feat: add AI workflow editor

2026-06-22 00:26:16 +08:00
parent df7b199c18
commit 14f50e064f
19 changed files with 1588 additions and 1 deletions
@@ -0,0 +1,44 @@
+package runtime
+
+import (
+	"encoding/json"
+	"strings"
+
+	"agent-desk/internal/ai/workflow/compiler"
+	"agent-desk/internal/ai/workflow/dsl"
+	"agent-desk/internal/models"
+	"agent-desk/internal/pkg/enums"
+	"agent-desk/internal/repositories"
+
+	"github.com/mlogclub/simple/sqls"
+)
+
+func resolveAgentWorkflow(aiAgent models.AIAgent) (compiler.Result, bool) {
+	if aiAgent.RuntimeMode != enums.AIAgentRuntimeModeWorkflow || aiAgent.WorkflowVersionID <= 0 {
+		return compiler.Result{}, false
+	}
+	version := repositories.AIWorkflowVersionRepository.Get(sqls.DB(), aiAgent.WorkflowVersionID)
+	if version == nil || version.Status != enums.StatusOk {
+		return compiler.Result{}, false
+	}
+	var def dsl.Definition
+	if err := json.Unmarshal([]byte(version.Definition), &def); err != nil {
+		return compiler.Result{}, false
+	}
+	return compiler.Compile(def), true
+}
+
+func applyWorkflowInstruction(aiAgent models.AIAgent) models.AIAgent {
+	result, ok := resolveAgentWorkflow(aiAgent)
+	if !ok || strings.TrimSpace(result.Appendix) == "" {
+		return aiAgent
+	}
+	prompt := strings.TrimSpace(aiAgent.SystemPrompt)
+	appendix := strings.TrimSpace(result.Appendix)
+	if prompt == "" {
+		aiAgent.SystemPrompt = appendix
+		return aiAgent
+	}
+	aiAgent.SystemPrompt = prompt + "\n\n" + appendix
+	return aiAgent
+}