优化可灵视频算力配置

This commit is contained in:
RockYang
2025-03-04 16:59:20 +08:00
parent 3958e99e4d
commit 7b9a7475a9
7 changed files with 177 additions and 273 deletions

View File

@@ -13,9 +13,6 @@ import (
"geekai/core"
"geekai/core/types"
"geekai/store"
"geekai/store/model"
"geekai/utils"
"gorm.io/gorm"
"strings"
"time"
@@ -25,27 +22,17 @@ import (
type LicenseService struct {
config types.ApiConfig
db *gorm.DB
levelDB *store.LevelDB
license *types.License
urlWhiteList []string
machineId string
}
func NewLicenseService(server *core.AppServer, levelDB *store.LevelDB, db *gorm.DB) *LicenseService {
func NewLicenseService(server *core.AppServer, levelDB *store.LevelDB) *LicenseService {
var license types.License
var machineId string
err := levelDB.Get(types.LicenseKey, &license)
logger.Infof("License: %+v", server.SysConfig)
if err != nil {
var cfg model.Config
db.Where("marker", "system").First(&cfg)
var sysConfig types.SystemConfig
if err := utils.JsonDecode(cfg.Config, &sysConfig); err == nil {
license.Key = sysConfig.License
license.IsActive = true
}
}
info, err := host.Info()
if err == nil {
machineId = info.HostID

View File

@@ -519,7 +519,7 @@ func (s *Service) KeLingCreate(task types.VideoTask) (KeLingRespVo, error) {
// 2. 构建API请求参数
payload := map[string]interface{}{
"model": params.Model,
"model_name": params.Model,
"prompt": task.Prompt,
"negative_prompt": params.NegPrompt,
"cfg_scale": params.CfgScale,