feat: 重构异步模型字段并更新依赖

common/util/mapping.go

@@ -20,36 +20,125 @@ import (
 	tgjson "github.com/tidwall/gjson"
 )
 
-// ValidatePromptResult 校验模型返回结果的 JSON 结构完整性
-// 校验逻辑：只校验 requestMapping 中默认值为空的必填字段
-func ValidatePromptResult(raw map[string]any, model *entity.AsynchModel) error {
-	// 1) 获取校验配置，并取值
-	requestMapping := model.RequestMapping
-	contentStr, ok := raw[model.ResponseBody].(string)
-	if !ok || contentStr == "" {
-		return fmt.Errorf("%s 字段为空或不是字符串", model.ResponseBody)
+// ParseAndValidate 解析并校验结果
+func ParseAndValidate(raw map[string]any, model *entity.AsynchModel) (map[string]any, error) {
+	// 1) 解析 content 字符串为 rounds 数组
+	contentVal, ok := raw[model.ResponseBody]
+	if !ok {
+		return raw, fmt.Errorf("字段 %s 不存在", model.ResponseBody)
+	}
+	fmt.Println("model.ResponseBody打印", model.ResponseBody)
+	contentStr, ok := contentVal.(string)
+	if !ok || strings.TrimSpace(contentStr) == "" {
+		return raw, fmt.Errorf("字段 %s 为空或不是字符串", model.ResponseBody)
+	}
+	var arr []any
+	if err := json.Unmarshal([]byte(contentStr), &arr); err != nil {
+		return raw, fmt.Errorf("JSON解析失败: %w", err)
+	}
+	if len(arr) == 0 {
+		return raw, fmt.Errorf("解析后数组为空")
 	}
 
-	// 2) 解析 content 为 JSON 数组
-	var rounds []map[string]any
-	if err := gjson.DecodeTo(contentStr, &rounds); err != nil {
-		return fmt.Errorf("解析 content JSON 数组失败: %w", err)
-	}
-	if len(rounds) == 0 {
-		return fmt.Errorf("content 数组为空")
-	}
-
-	// 3) 逐条校验：只检查默认值为空的必填字段是否存在
-	for i, round := range rounds {
-		for path, defaultValue := range requestMapping {
-			if !g.IsEmpty(defaultValue) {
+	// 2) 校验必填字段
+	if len(model.RequiredFields) > 0 {
+		for i, r := range arr {
+			round, ok := r.(map[string]any)
+			if !ok {
 				continue
 			}
-			if gjson.New(round).Get(path).IsNil() {
-				return fmt.Errorf("rounds[%d] 缺少必填字段: %s", i, path)
+			for _, field := range model.RequiredFields {
+				if gjson.New(round).Get(field).IsNil() {
+					return raw, fmt.Errorf("rounds[%d] 缺少必填字段: %s", i, field)
+				}
 			}
 		}
 	}
+
+	return map[string]any{"total_rounds": len(arr), "rounds": arr}, nil
+}
+
+// ParseStructResult 解析结构结果
+func ParseStructResult(raw map[string]any, responseBody string) map[string]any {
+	contentVal := raw[responseBody]
+	// 是字符串，尝试解析
+	contentStr := gconv.String(contentVal)
+	if contentStr == "" || contentStr == "0" {
+		return map[string]any{
+			"total_rounds": 1,
+			"rounds":       []map[string]any{{responseBody: raw}},
+		}
+	}
+
+	// 尝试解析为数组
+	var arr []any
+	if err := json.Unmarshal([]byte(contentStr), &arr); err == nil && len(arr) > 0 {
+		return map[string]any{
+			"total_rounds": 1,
+			"rounds":       []map[string]any{{responseBody: arr}},
+		}
+	}
+
+	// 尝试解析为单个对象
+	var parsed any
+	if err := json.Unmarshal([]byte(contentStr), &parsed); err == nil {
+		return map[string]any{
+			"total_rounds": 1,
+			"rounds":       []map[string]any{{responseBody: parsed}},
+		}
+	}
+
+	// 兜底：原始字符串作为内容
+	return map[string]any{
+		"total_rounds": 1,
+		"rounds":       []map[string]any{{responseBody: contentStr}},
+	}
+}
+
+// ValidatePromptResult 校验模型返回结果的 JSON 结构完整性
+// raw 必须包含 "rounds" 字段，格式为 []map[string]any
+func ValidatePromptResult(raw map[string]any, model *entity.AsynchModel) error {
+	// 1) 获取 rounds
+	roundsRaw, ok := raw["rounds"]
+	if !ok {
+		return fmt.Errorf("缺少 rounds 字段")
+	}
+	rounds, ok := roundsRaw.([]any)
+	if !ok {
+		return fmt.Errorf("rounds 不是数组")
+	}
+	if len(rounds) == 0 {
+		return fmt.Errorf("rounds 数组为空")
+	}
+
+	// 2) 没有配置必填字段，跳过
+	if len(model.RequiredFields) == 0 {
+		return nil
+	}
+
+	// 3) 逐条校验
+	for i, r := range rounds {
+		round, ok := r.(map[string]any)
+		if !ok {
+			continue
+		}
+		for _, field := range model.RequiredFields {
+			if gjson.New(round).Get(field).IsNil() {
+				return fmt.Errorf("rounds[%d] 缺少必填字段: %s", i, field)
+			}
+		}
+	}
+
+	return nil
+}
+
+// validateRequiredFields 校验单个 round 对象的必选字段
+func validateRequiredFields(round map[string]any, requiredFields []string, prefix string) error {
+	for _, field := range requiredFields {
+		if gjson.New(round).Get(field).IsNil() {
+			return fmt.Errorf("%s 缺少必填字段: %s", prefix, field)
+		}
+	}
 	return nil
 }