Merge pull request #1381 from feitianbubu/pr/task-origin-name

feat: video priority use origin model name

Merge pull request #1381 from feitianbubu/pr/task-origin-name
feat: video priority use origin model name
38091f43 · Calcium-Ion · GitHub · 09e6dbcc · cee6ba00 · 38091f43
Commit 38091f43 authored Jul 18, 2025 by Calcium-Ion Committed by GitHub Jul 18, 2025
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

relay/relay_task.go
+3 -3

No files found.
--- a/relay/relay_task.go
+++ b/relay/relay_task.go
@@ -37,9 +37,9 @@ func RelayTaskSubmit(c *gin.Context, relayMode int) (taskErr *dto.TaskError) {
 		return
 	}

-	modelName := service.CoverTaskActionToModelName(platform, relayInfo.Action)
-	if platform == constant.TaskPlatformKling {
-		modelName = relayInfo.OriginModelName
+	modelName := relayInfo.OriginModelName
+	if modelName == "" {
+		modelName = service.CoverTaskActionToModelName(platform, relayInfo.Action)
 	}
 	modelPrice, success := ratio_setting.GetModelPrice(modelName, true)
 	if !success {