Commit b183f2f6 by feitianbubu

feat: vidu video add starEnd and reference gen video

parent 22db389f
...@@ -11,8 +11,10 @@ const ( ...@@ -11,8 +11,10 @@ const (
SunoActionMusic = "MUSIC" SunoActionMusic = "MUSIC"
SunoActionLyrics = "LYRICS" SunoActionLyrics = "LYRICS"
TaskActionGenerate = "generate" TaskActionGenerate = "generate"
TaskActionTextGenerate = "textGenerate" TaskActionTextGenerate = "textGenerate"
TaskActionFirstTailGenerate = "firstTailGenerate"
TaskActionReferenceGenerate = "referenceGenerate"
) )
var SunoModel2Action = map[string]string{ var SunoModel2Action = map[string]string{
......
...@@ -111,6 +111,10 @@ func (a *TaskAdaptor) BuildRequestURL(info *relaycommon.RelayInfo) (string, erro ...@@ -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"
} }
......
...@@ -81,6 +81,14 @@ func ValidateBasicTaskRequest(c *gin.Context, info *RelayInfo, action string) *d ...@@ -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)
......
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