admin777/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: support openAI sdk retrieve videos

Browse Source
This commit is contained in:
feitianbubu 2025-10-10 18:45:12 +08:00
parent b8f8329989
commit 4e0a43a909
2 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
controller/task_video.go
View File

@ -97,6 +97,7 @@ func updateVideoSingleTask(ctx context.Context, adaptor channel.TaskAdaptor, cha
taskResult.Url = t.FailReason taskResult.Url = t.FailReason
taskResult.Progress = t.Progress taskResult.Progress = t.Progress
taskResult.Reason = t.FailReason taskResult.Reason = t.FailReason
task.Data = t.Data
} else if taskResult, err = adaptor.ParseTaskResult(responseBody); err != nil { } else if taskResult, err = adaptor.ParseTaskResult(responseBody); err != nil {
return fmt.Errorf("parseTaskResult failed for task %s: %w", taskId, err) return fmt.Errorf("parseTaskResult failed for task %s: %w", taskId, err)
} else { } else {

19
relay/relay_task.go
View File

@ -362,11 +362,20 @@ func videoFetchByIDRespBodyBuilder(c *gin.Context) (respBody []byte, taskResp *d
} }
}() }()
if len(respBody) == 0 { if len(respBody) != 0 {
respBody, err = json.Marshal(dto.TaskResponse[any]{ return
Code: "success", }
Data: TaskModel2Dto(originTask),
}) 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 return
} }