feat: vidu video add starEnd and reference gen video

This commit is contained in:
feitianbubu
2025-09-19 17:44:58 +08:00
parent 3c70617060
commit 8f9960bcc7
3 changed files with 16 additions and 2 deletions

View File

@@ -13,6 +13,8 @@ const (
TaskActionGenerate = "generate" TaskActionGenerate = "generate"
TaskActionTextGenerate = "textGenerate" TaskActionTextGenerate = "textGenerate"
TaskActionFirstTailGenerate = "firstTailGenerate"
TaskActionReferenceGenerate = "referenceGenerate"
) )
var SunoModel2Action = map[string]string{ var SunoModel2Action = map[string]string{

View File

@@ -111,6 +111,10 @@ func (a *TaskAdaptor) BuildRequestURL(info *relaycommon.RelayInfo) (string, erro
switch info.Action { switch info.Action {
case constant.TaskActionGenerate: case constant.TaskActionGenerate:
path = "/img2video" path = "/img2video"
case constant.TaskActionFirstTailGenerate:
path = "/start-end2video"
case constant.TaskActionReferenceGenerate:
path = "/reference2video"
default: default:
path = "/text2video" path = "/text2video"
} }

View File

@@ -81,6 +81,14 @@ func ValidateBasicTaskRequest(c *gin.Context, info *RelayInfo, action string) *d
if req.HasImage() { if req.HasImage() {
action = constant.TaskActionGenerate 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) storeTaskRequest(c, info, action, req)