mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-04-19 02:17:26 +00:00
feat: vidu specify reference2video via metadata action
This commit is contained in:
@@ -59,6 +59,17 @@ func storeTaskRequest(c *gin.Context, info *RelayInfo, action string, requestObj
|
||||
info.Action = action
|
||||
c.Set("task_request", requestObj)
|
||||
}
|
||||
func GetTaskRequest(c *gin.Context) (TaskSubmitReq, error) {
|
||||
v, exists := c.Get("task_request")
|
||||
if !exists {
|
||||
return TaskSubmitReq{}, fmt.Errorf("request not found in context")
|
||||
}
|
||||
req, ok := v.(TaskSubmitReq)
|
||||
if !ok {
|
||||
return TaskSubmitReq{}, fmt.Errorf("invalid task request type")
|
||||
}
|
||||
return req, nil
|
||||
}
|
||||
|
||||
func validatePrompt(prompt string) *dto.TaskError {
|
||||
if strings.TrimSpace(prompt) == "" {
|
||||
@@ -212,18 +223,6 @@ func ValidateBasicTaskRequest(c *gin.Context, info *RelayInfo, action string) *d
|
||||
req.Images = []string{req.Image}
|
||||
}
|
||||
|
||||
if req.HasImage() {
|
||||
action = constant.TaskActionGenerate
|
||||
if info.ChannelType == constant.ChannelTypeVidu {
|
||||
// vidu 增加 首尾帧生视频和参考图生视频
|
||||
if len(req.Images) == 2 {
|
||||
action = constant.TaskActionFirstTailGenerate
|
||||
} else if len(req.Images) > 2 {
|
||||
action = constant.TaskActionReferenceGenerate
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
storeTaskRequest(c, info, action, req)
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user