Просмотр исходного кода

Merge branch 'feitianbubu-pr/fix-kling-text2image' into alpha

creamlike1024 6 месяцев назад
Родитель
Сommit
bc5b9a5506
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      relay/channel/task/kling/adaptor.go

+ 3 - 0
relay/channel/task/kling/adaptor.go

@@ -172,6 +172,9 @@ func (a *TaskAdaptor) BuildRequestBody(c *gin.Context, info *relaycommon.RelayIn
 	if err != nil {
 		return nil, err
 	}
+	if body.Image == "" && body.ImageTail == "" {
+		c.Set("action", constant.TaskActionTextGenerate)
+	}
 	data, err := json.Marshal(body)
 	if err != nil {
 		return nil, err