refactor(prompt): 重构异步模型字段和提示词构建服务

service/prompt/prompt_compose_service.go

@@ -30,6 +30,7 @@ func ComposeMessages(ctx context.Context, req *dto.ComposeMessagesReq) (*dto.Com
 	if err = validateUserForm(ctx, req, aiModel); err != nil {
 		return nil, err
 	}
+	fmt.Printf("req打印%+v", req)
 	switch req.BuildType {
 	case public.BuildTypePrompt:
 		return handlePromptBuild(ctx, req, chatModel, aiModel) // 提示词构建
@@ -85,13 +86,13 @@ func handlePromptBuild(ctx context.Context, req *dto.ComposeMessagesReq, chatMod
 			g.Log().Errorf(ctx, "保存任务记录失败(第%d次): %v", attempt+1, err)
 			continue
 		}
-
+		//等待结果
 		taskRecord, err = waitForResult(ctx, taskID)
 		if err != nil {
 			g.Log().Errorf(ctx, "等待结果失败(第%d次): %v", attempt+1, err)
 			continue
 		}
-
+		//处理结果
 		message = parsePromptBuild(taskRecord, chatModel)
 		if message != nil {
 			break
@@ -244,93 +245,87 @@ func callInferenceModel(ctx context.Context, req *dto.ComposeMessagesReq, chatMo
 func waitForResult(ctx context.Context, taskID string) (*entity.ComposeTask, error) {
 	timeout := time.Duration(g.Cfg().MustGet(ctx, "task.waitTimeoutSeconds", 300).Int()) * time.Second
 	pollInterval := time.Duration(g.Cfg().MustGet(ctx, "task.pollIntervalMillis", 500).Int()) * time.Millisecond
-
 	deadline := time.Now().Add(timeout)
-	ticker := time.NewTicker(pollInterval)
-	defer ticker.Stop()
 
 	for {
+		// ===================== 修复点 1：检查上下文是否取消 =====================
+		select {
+		case <-ctx.Done():
+			// 请求已被取消，直接返回，不继续查库
+			return nil, ctx.Err()
+		default:
+		}
+
+		// 1. 查数据库
 		record, err := dao.ComposeTask.Get(ctx, &entity.ComposeTask{
 			TaskId: taskID,
 		})
 		if err != nil {
+			// ===================== 修复点 2：如果是上下文取消，直接返回 =====================
 			if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
 				return nil, err
 			}
-			return nil, fmt.Errorf("查询任务失败: %w", err)
+			return nil, err
 		}
-
 		if record != nil {
-			if completed, result := checkTaskCompletion(record); completed {
-				return result, nil
+			switch record.Status {
+			case public.ComposeStatusSuccess:
+				return record, nil
+			case public.ComposeStatusFailed:
+				if strings.TrimSpace(record.ErrorMessage) == "" {
+					return nil, fmt.Errorf("任务失败(taskId=%s)", taskID)
+				}
+				return nil, fmt.Errorf("任务失败(taskId=%s): %s", taskID, record.ErrorMessage)
 			}
 		}
 
-		if err = syncGatewayTaskState(ctx, taskID, record); err != nil {
-			g.Log().Warningf(ctx, "[waitForResult] 同步网关状态失败 taskId=%s err=%v", taskID, err)
+		// 2. 查网关状态
+		state, err := gateway.QueryGatewayTaskState(ctx, taskID)
+		if err != nil {
+			// 网关不可达不终止，继续轮询
+			g.Log().Warningf(ctx, "[waitForResult] 查询网关失败 taskId=%s err=%v", taskID, err)
+		} else {
+			switch state {
+			case 2: // 网关成功
+				// 网关已成功，主动更新数据库
+				if record != nil {
+					_, err = dao.ComposeTask.Update(ctx, &entity.ComposeTask{
+						TaskId: taskID,
+						Status: public.ComposeStatusSuccess,
+					})
+					if err != nil {
+						g.Log().Warningf(ctx, "[waitForResult] 更新任务状态失败 taskId=%s err=%v", taskID, err)
+					}
+				}
+			case 3: // 网关失败
+				if record != nil {
+					_, err = dao.ComposeTask.Update(ctx, &entity.ComposeTask{
+						TaskId:       taskID,
+						Status:       public.ComposeStatusFailed,
+						ErrorMessage: "model-gateway 任务执行失败",
+					})
+					if err != nil {
+						g.Log().Warningf(ctx, "[waitForResult] 更新任务状态失败 taskId=%s err=%v", taskID, err)
+					}
+				}
+				return nil, fmt.Errorf("model-gateway 任务执行失败(taskId=%s)", taskID)
+			}
 		}
 
+		// 3. 超时检查
 		if time.Now().After(deadline) {
 			return nil, fmt.Errorf("等待任务回调超时(taskId=%s)", taskID)
 		}
 
+		// ===================== 修复点3：sleep 也要监听 ctx 取消 =====================
 		select {
 		case <-ctx.Done():
 			return nil, ctx.Err()
-		case <-ticker.C:
+		case <-time.After(pollInterval):
 		}
 	}
 }
 
-// checkTaskCompletion 检查任务是否完成
-func checkTaskCompletion(record *entity.ComposeTask) (bool, *entity.ComposeTask) {
-	if record == nil {
-		return false, nil
-	}
-	switch record.Status {
-	case public.ComposeStatusSuccess:
-		return true, record
-	case public.ComposeStatusFailed:
-		errMsg := strings.TrimSpace(record.ErrorMessage)
-		if errMsg == "" {
-			return true, nil
-		}
-		return true, nil
-	default:
-		return false, nil
-	}
-}
-
-// syncGatewayTaskState 同步网关任务状态
-func syncGatewayTaskState(ctx context.Context, taskID string, record *entity.ComposeTask) error {
-	state, err := gateway.QueryGatewayTaskState(ctx, taskID)
-	if err != nil {
-		return fmt.Errorf("查询网关状态失败: %w", err)
-	}
-	switch state {
-	case 2:
-		return updateTaskStatus(ctx, taskID, public.ComposeStatusSuccess, "")
-	case 3:
-		updateTaskStatus(ctx, taskID, public.ComposeStatusFailed, "model-gateway 任务执行失败")
-		return fmt.Errorf("model-gateway 任务执行失败(taskId=%s)", taskID)
-	}
-	return nil
-}
-
-// updateTaskStatus 更新任务状态
-func updateTaskStatus(ctx context.Context, taskID string, status string, errorMsg string) error {
-	task := &entity.ComposeTask{
-		TaskId: taskID,
-		Status: status,
-	}
-	if errorMsg != "" {
-		task.ErrorMessage = errorMsg
-	}
-
-	_, err := dao.ComposeTask.Update(ctx, task)
-	return err
-}
-
 // parsePromptBuild 解析提示词构建结果（BuildType == 1）
 func parsePromptBuild(taskRecord *entity.ComposeTask, model *entity.AsynchModel) *dto.MultiRoundResult {
 	if taskRecord == nil {