mirror of
https://github.com/yangjian102621/geekai.git
synced 2026-08-16 04:30:59 +00:00
RockYang
9ccff4efbc
- Sync backend and frontend from GeekAI Plus v4.3.0 - Remove commercial License flows and update open-source deployment defaults - Preserve Docker Compose deployment and bump image tags to v4.3.0 BREAKING CHANGE: commercial License configuration and related endpoints are removed
78 lines
2.1 KiB
Go
78 lines
2.1 KiB
Go
package video
|
||
|
||
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
|
||
// * Use of this source code is governed by a Apache-2.0 license
|
||
// * that can be found in the LICENSE file.
|
||
// * @Author yangjian102621@163.com
|
||
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"geekai/core/types"
|
||
"geekai/store/model"
|
||
"geekai/utils"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// GetVideoConfig 从数据库获取视频配置
|
||
func GetVideoConfig(db *gorm.DB) (*types.VideoConfig, error) {
|
||
var config model.Config
|
||
err := db.Where("name", types.ConfigKeyVideo).First(&config).Error
|
||
if err != nil {
|
||
if err == gorm.ErrRecordNotFound {
|
||
return nil, errors.New("视频配置不存在，请在管理后台配置")
|
||
}
|
||
return nil, fmt.Errorf("获取视频配置失败: %v", err)
|
||
}
|
||
|
||
var videoConfig types.VideoConfig
|
||
err = utils.JsonDecode(config.Value, &videoConfig)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("解析视频配置失败: %v", err)
|
||
}
|
||
|
||
return &videoConfig, nil
|
||
}
|
||
|
||
// GetModelPowerConfig 获取指定模型的算力配置
|
||
func GetModelPowerConfig(db *gorm.DB, modelKey string) (*types.VideoModelPower, error) {
|
||
config, err := GetVideoConfig(db)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
modelPower, ok := config.VideoPowers[modelKey]
|
||
if !ok {
|
||
return nil, fmt.Errorf("模型 %s 的算力配置不存在", modelKey)
|
||
}
|
||
|
||
return &modelPower, nil
|
||
}
|
||
|
||
// CalculatePower 根据 modelKey 和 priceKey 计算算力
|
||
// modelKey: 模型标识（如 "veo-2.0", "sora-2.0"）
|
||
// priceKey: 价格键（如 "fixed", "5_720P", "std_5_sound" 等）
|
||
func CalculatePower(db *gorm.DB, modelKey string, priceKey string) (int, error) {
|
||
if priceKey == "" {
|
||
return 0, errors.New("priceKey 不能为空")
|
||
}
|
||
|
||
modelPower, err := GetModelPowerConfig(db, modelKey)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
power, ok := modelPower.PowerConfig[priceKey]
|
||
if !ok {
|
||
return 0, fmt.Errorf("模型 %s 的价格配置 %s 不存在", modelKey, priceKey)
|
||
}
|
||
|
||
if power <= 0 {
|
||
return 0, fmt.Errorf("模型 %s 的价格配置 %s 的值无效", modelKey, priceKey)
|
||
}
|
||
|
||
return power, nil
|
||
}
|