From 8209219f922b4956d16c870eaa97ad75984e4378 Mon Sep 17 00:00:00 2001
From: qhd <1766646056@qq.com>
Date: Thu, 25 Jun 2026 10:54:40 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=8D=95=E8=A7=86?=
 =?UTF-8?q?=E9=A2=91=E7=94=9F=E6=88=90=E6=97=B6=E6=97=A0=E9=9C=80=E6=8B=BC?=
 =?UTF-8?q?=E6=8E=A5=E7=9A=84=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 workflow/service/flow/lambda_node.go | 49 +++++++++++++++-------------
 1 file changed, 27 insertions(+), 22 deletions(-)

diff --git a/workflow/service/flow/lambda_node.go b/workflow/service/flow/lambda_node.go
index 241b43f..48e4c8d 100644
--- a/workflow/service/flow/lambda_node.go
+++ b/workflow/service/flow/lambda_node.go
@@ -264,7 +264,7 @@ func VideoModelLambda(ctx context.Context, input any) (any, error) {
 	if err != nil {
 		return nil, err
 	}
-
+	videoUrl := ""
 	videoURL := make([]string, 0)
 	for _, v := range res {
 		if strings.Contains(v.Field, "content") {
@@ -274,37 +274,42 @@ func VideoModelLambda(ctx context.Context, input any) (any, error) {
 	if g.IsEmpty(videoURL) {
 		return nil, fmt.Errorf("视频合成失败:模型生成视频失败")
 	}
-	waitRes, err := VideoConcat(ctx, videoURL)
-	if err != nil {
-		return nil, err
-	}
-	msg := new(flowDto.VideoCallbackReq)
-	if err = gconv.Struct(waitRes, msg); err != nil {
-		return nil, err
-	}
-
-	urlPrefix, err := utils.GetFileAddressPrefix(ctx)
-	newS := strings.ReplaceAll(urlPrefix, g.Cfg().MustGet(ctx, "filePrefix").String(), g.Cfg().MustGet(ctx, "minioPrefix").String())
-	if err != nil {
-		return nil, err
+	if len(videoURL) > 1 {
+		var waitRes any
+		waitRes, err = VideoConcat(ctx, videoURL)
+		if err != nil {
+			return nil, err
+		}
+		msg := new(flowDto.VideoCallbackReq)
+		if err = gconv.Struct(waitRes, msg); err != nil {
+			return nil, err
+		}
+		var urlPrefix string
+		urlPrefix, err = utils.GetFileAddressPrefix(ctx)
+		newS := strings.ReplaceAll(urlPrefix, g.Cfg().MustGet(ctx, "filePrefix").String(), g.Cfg().MustGet(ctx, "minioPrefix").String())
+		if err != nil {
+			return nil, err
+		}
+		videoUrl = newS + msg.FileURL
+	} else {
+		videoUrl = videoURL[0]
 	}
 
 	outputRes := make([]node.NodeFormField, 0)
 	if nodeInput.Config.IsSaveFile {
 		outputRes = append(outputRes, node.NodeFormField{
 			Field: fmt.Sprintf("video_oss_url:content:%d", 0),
-			Value: msg.FileURL,
+			Value: videoUrl,
 			Label: fmt.Sprintf("video_oss_url:content:%d", 0),
 			Type:  "string",
 		})
-	} else {
-		outputRes = append(outputRes, node.NodeFormField{
-			Field: fmt.Sprintf("concat_video_url:content:%d", 0),
-			Value: newS + msg.FileURL,
-			Label: fmt.Sprintf("concat_video_url:content:%d", 0),
-			Type:  "string",
-		})
 	}
+	outputRes = append(outputRes, node.NodeFormField{
+		Field: fmt.Sprintf("concat_video_url:content:%d", 0),
+		Value: videoUrl,
+		Label: fmt.Sprintf("concat_video_url:content:%d", 0),
+		Type:  "string",
+	})
 	nodeInput.Config.OutputResult = outputRes
 
 	return nodeInput, nil