diff --git a/service/prompt/prompt_compose_service.go b/service/prompt/prompt_compose_service.go index 4637b5f..9f8ee21 100644 --- a/service/prompt/prompt_compose_service.go +++ b/service/prompt/prompt_compose_service.go @@ -143,10 +143,16 @@ func Callback(ctx context.Context, req *dto.CallbackReq) error { // 2) 读取 OSS 文件内容 var ossContent []byte if req.OssFile != "" { - ossContent, err = gateway.GetFileBytesFromURL(ctx, req.OssFile) - if err != nil { - g.Log().Infof(ctx, "[回调处理] 读取OSS文件 taskId=%s,state=%v,ossFile=%v", req.TaskId, req.State, req.OssFile) - g.Log().Warningf(ctx, "[回调处理] 读取OSS失败 taskId=%s err=%v", req.TaskId, err) + for attempt := 0; attempt <= 5; attempt++ { + ossContent, err = gateway.GetFileBytesFromURL(ctx, req.OssFile) + if err != nil { + g.Log().Infof(ctx, "[回调处理] 读取OSS文件 taskId=%s,state=%v,ossFile=%v", req.TaskId, req.State, req.OssFile) + g.Log().Warningf(ctx, "[回调处理] 读取OSS失败 taskId=%s err=%v", req.TaskId, err) + } + if err == nil { + g.Log().Infof(ctx, "[回调处理] 读取OSS成功 taskId=%s,state=%v,ossFile=%v", req.TaskId, req.State, req.OssFile) + break + } } }