Skip to content

phsl / new-api

Go to a project
Commit cee6ba00 by feitianbubu
Options

feat: priority use origin model name

parent 31adc393
Showing with 3 additions and 3 deletions
relay/relay_task.go
...@@ -37,9 +37,9 @@ func RelayTaskSubmit(c *gin.Context, relayMode int) (taskErr *dto.TaskError) { ...@@ -37,9 +37,9 @@ func RelayTaskSubmit(c *gin.Context, relayMode int) (taskErr *dto.TaskError) {
return return
} }
modelName := service.CoverTaskActionToModelName(platform, relayInfo.Action) modelName := relayInfo.OriginModelName
if platform == constant.TaskPlatformKling { if modelName == "" {
modelName = relayInfo.OriginModelName modelName = service.CoverTaskActionToModelName(platform, relayInfo.Action)
} }
modelPrice, success := ratio_setting.GetModelPrice(modelName, true) modelPrice, success := ratio_setting.GetModelPrice(modelName, true)
if !success { if !success {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment