From 8f9960bcc7f93d40364b3a603907cd0215d39dac Mon Sep 17 00:00:00 2001
From: feitianbubu <feitianbubu@qq.com>
Date: Fri, 19 Sep 2025 17:44:58 +0800
Subject: [PATCH] feat: vidu video add starEnd and reference gen video

---
 constant/task.go                   | 6 ++++--
 relay/channel/task/vidu/adaptor.go | 4 ++++
 relay/common/relay_utils.go        | 8 ++++++++
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/constant/task.go b/constant/task.go
index 21790145b..e174fd60e 100644
--- a/constant/task.go
+++ b/constant/task.go
@@ -11,8 +11,10 @@ const (
 	SunoActionMusic  = "MUSIC"
 	SunoActionLyrics = "LYRICS"
 
-	TaskActionGenerate     = "generate"
-	TaskActionTextGenerate = "textGenerate"
+	TaskActionGenerate          = "generate"
+	TaskActionTextGenerate      = "textGenerate"
+	TaskActionFirstTailGenerate = "firstTailGenerate"
+	TaskActionReferenceGenerate = "referenceGenerate"
 )
 
 var SunoModel2Action = map[string]string{
diff --git a/relay/channel/task/vidu/adaptor.go b/relay/channel/task/vidu/adaptor.go
index 8974c6149..358aef583 100644
--- a/relay/channel/task/vidu/adaptor.go
+++ b/relay/channel/task/vidu/adaptor.go
@@ -111,6 +111,10 @@ func (a *TaskAdaptor) BuildRequestURL(info *relaycommon.RelayInfo) (string, erro
 	switch info.Action {
 	case constant.TaskActionGenerate:
 		path = "/img2video"
+	case constant.TaskActionFirstTailGenerate:
+		path = "/start-end2video"
+	case constant.TaskActionReferenceGenerate:
+		path = "/reference2video"
 	default:
 		path = "/text2video"
 	}
diff --git a/relay/common/relay_utils.go b/relay/common/relay_utils.go
index 96d1370bd..3a721b479 100644
--- a/relay/common/relay_utils.go
+++ b/relay/common/relay_utils.go
@@ -81,6 +81,14 @@ func ValidateBasicTaskRequest(c *gin.Context, info *RelayInfo, action string) *d
 
 	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)