From ff9f9fbbc91b0544ff0a1949c440343d4019e355 Mon Sep 17 00:00:00 2001 From: feitianbubu Date: Fri, 10 Oct 2025 18:45:12 +0800 Subject: [PATCH] feat: support openAI sdk retrieve videos --- controller/task_video.go | 1 + relay/relay_task.go | 19 ++++++++++++++----- 2 files changed, 15 insertions(+), 5 deletions(-) diff --git a/controller/task_video.go b/controller/task_video.go index 9bbf7a902..8d4acdda8 100644 --- a/controller/task_video.go +++ b/controller/task_video.go @@ -97,6 +97,7 @@ func updateVideoSingleTask(ctx context.Context, adaptor channel.TaskAdaptor, cha taskResult.Url = t.FailReason taskResult.Progress = t.Progress taskResult.Reason = t.FailReason + task.Data = t.Data } else if taskResult, err = adaptor.ParseTaskResult(responseBody); err != nil { return fmt.Errorf("parseTaskResult failed for task %s: %w", taskId, err) } else { diff --git a/relay/relay_task.go b/relay/relay_task.go index 9cb8cd5c8..d447a40aa 100644 --- a/relay/relay_task.go +++ b/relay/relay_task.go @@ -362,11 +362,20 @@ func videoFetchByIDRespBodyBuilder(c *gin.Context) (respBody []byte, taskResp *d } }() - if len(respBody) == 0 { - respBody, err = json.Marshal(dto.TaskResponse[any]{ - Code: "success", - Data: TaskModel2Dto(originTask), - }) + if len(respBody) != 0 { + return + } + + if strings.HasPrefix(c.Request.RequestURI, "/v1/videos/") { + respBody = originTask.Data + return + } + respBody, err = json.Marshal(dto.TaskResponse[any]{ + Code: "success", + Data: TaskModel2Dto(originTask), + }) + if err != nil { + taskResp = service.TaskErrorWrapper(err, "marshal_response_failed", http.StatusInternalServerError) } return }