update

internal/run/preview.go

@@ -8,6 +8,7 @@ import (
 	"git.iwork-ai.com/xdc/iqudo-top1/internal/auth"
 	"git.iwork-ai.com/xdc/iqudo-top1/internal/model"
 	"git.iwork-ai.com/xdc/iqudo-top1/internal/response"
+	"git.iwork-ai.com/xdc/iqudo-top1/internal/scriptkit"
 	"github.com/gin-gonic/gin"
 )
 
@@ -21,7 +22,7 @@ func (h *Handler) PreviewScenario(c *gin.Context) {
 	var body struct {
 		Input         map[string]interface{} `json:"input"`
 		InitialValues map[string]interface{} `json:"initial_values"`
-		Selector      knowledgeSelector      `json:"knowledge_selector"`
+		StageKey      string                 `json:"stage_key"`
 	}
 	if err := c.ShouldBindJSON(&body); err != nil {
 		response.Error(c, http.StatusBadRequest, "INVALID_ARGUMENT", "预览参数格式不正确")
@@ -47,16 +48,29 @@ func (h *Handler) PreviewScenario(c *gin.Context) {
 		response.Error(c, http.StatusUnprocessableEntity, "RULE_EVALUATION_FAILED", err.Error())
 		return
 	}
-	context := runtimeContext(mapped, derived, nil)
-	knowledge, err := loadKnowledgeOutputsForScenario(h.db, scenarioID, p.TenantID, context, body.Selector)
-	if err != nil {
-		response.Error(c, http.StatusInternalServerError, "OUTPUT_FAILED", "生成知识预览失败")
-		return
-	}
-	outputs, err := buildOutputViews(scenario.OutputSchema, mapped, derived, map[string]interface{}{}, knowledge, "preview", "")
+	outputs, err := buildOutputViews(scenario.OutputSchema, mapped, derived, map[string]interface{}{}, "preview", "")
 	if err != nil {
 		response.Error(c, http.StatusInternalServerError, "OUTPUT_FAILED", "生成场景输出失败")
 		return
 	}
-	response.OK(c, gin.H{"input": mapped, "derived": derived, "matched_rules": matchedRules, "knowledge": knowledge, "outputs": outputs})
+	data := gin.H{"input": mapped, "derived": derived, "matched_rules": matchedRules, "outputs": outputs}
+	pkg, loadErr := scriptkit.LoadPackageByScenario(h.db, p.TenantID, scenarioID)
+	if loadErr != nil {
+		response.OK(c, data)
+		return
+	}
+	stageKey := body.StageKey
+	if stageKey == "" {
+		stageKey = pkg.Package.StartStageKey
+	}
+	stage, ok := pkg.StageByKey(stageKey)
+	if !ok {
+		response.OK(c, data)
+		return
+	}
+	ctx := scriptkit.RenderContext{Input: mapped, Derived: derived, Form: map[string]interface{}{}}
+	state := scriptkit.ScriptState{ScriptAnswers: map[string][]string{}, DimensionSelects: map[string]bool{}}
+	weights := pkg.RecomputeWeights(pkg.InitialWeights(ctx), state, ctx)
+	data["stage"] = pkg.BuildStageView(stage, weights, ctx, state, []scriptkit.FormFieldView{})
+	response.OK(c, data)
 }