From df7b199c184a4ca988024099519f05a1fd01feb2 Mon Sep 17 00:00:00 2001 From: mlogclub Date: Mon, 22 Jun 2026 00:14:36 +0800 Subject: [PATCH] feat: bind AI agents to workflow versions --- .../handlers/dashboard/ai_agent_handler.go | 3 + internal/models/models.go | 2 + internal/pkg/dto/request/ai_request.go | 2 + internal/pkg/dto/response/ai_response.go | 3 + internal/pkg/enums/ai.go | 21 +++ internal/services/ai_agent_service.go | 28 ++++ .../ai_agent_workflow_service_test.go | 133 ++++++++++++++++++ 7 files changed, 192 insertions(+) create mode 100644 internal/services/ai_agent_workflow_service_test.go diff --git a/internal/handlers/dashboard/ai_agent_handler.go b/internal/handlers/dashboard/ai_agent_handler.go index 570f69f..cb39e1f 100644 --- a/internal/handlers/dashboard/ai_agent_handler.go +++ b/internal/handlers/dashboard/ai_agent_handler.go @@ -189,6 +189,9 @@ func buildAIAgentResponseWithLocale(item *models.AIAgent, locale string) respons Teams: make([]response.AIAgentTeamResponse, 0), DirectTools: make([]response.AIAgentMCPToolResponse, 0), GraphTools: make([]string, 0), + RuntimeMode: item.RuntimeMode, + RuntimeModeName: enums.GetAIAgentRuntimeModeLabel(item.RuntimeMode), + WorkflowVersionID: item.WorkflowVersionID, SortNo: item.SortNo, CreatedAt: item.CreatedAt.Format("2006-01-02 15:04:05"), UpdatedAt: item.UpdatedAt.Format("2006-01-02 15:04:05"), diff --git a/internal/models/models.go b/internal/models/models.go index ccefa63..165498a 100644 --- a/internal/models/models.go +++ b/internal/models/models.go @@ -527,6 +527,8 @@ type AIAgent struct { SkillIDs string `gorm:"type:varchar(500);not null;default:''"` // SkillIDs 为绑定的技能ID列表，按顺序表示允许路由的范围。 AllowedMCPTools string `gorm:"type:text"` // AllowedMCPTools 为允许 direct tool 路由的 MCP 工具白名单配置JSON。 AllowedGraphTools string `gorm:"type:text"` // AllowedGraphTools 为允许 Graph Tool 的白名单配置JSON。 + RuntimeMode enums.AIAgentRuntimeMode `gorm:"type:int;not null;default:1;index"` // RuntimeMode 为 Agent 执行模式，如内置 Graph 或发布的会话流程。 + WorkflowVersionID int64 `gorm:"type:bigint;not null;default:0;index"` // WorkflowVersionID 为绑定的已发布会话流程版本ID。 SortNo int `gorm:"type:int;not null;default:0;index"` // SortNo 为后台展示排序号。 AuditFields } diff --git a/internal/pkg/dto/request/ai_request.go b/internal/pkg/dto/request/ai_request.go index 35d5f0a..df46a02 100644 --- a/internal/pkg/dto/request/ai_request.go +++ b/internal/pkg/dto/request/ai_request.go @@ -58,6 +58,8 @@ type CreateAIAgentRequest struct { SkillIDs []int64 `json:"skillIds"` DirectTools []AIAgentMCPToolRequest `json:"directTools"` GraphTools []string `json:"graphTools"` + RuntimeMode enums.AIAgentRuntimeMode `json:"runtimeMode"` + WorkflowVersionID int64 `json:"workflowVersionId"` } type UpdateAIAgentRequest struct { diff --git a/internal/pkg/dto/response/ai_response.go b/internal/pkg/dto/response/ai_response.go index d0f3a4f..b1764cc 100644 --- a/internal/pkg/dto/response/ai_response.go +++ b/internal/pkg/dto/response/ai_response.go @@ -91,6 +91,9 @@ type AIAgentResponse struct { Skills []AIAgentSkillResponse `json:"skills"` DirectTools []AIAgentMCPToolResponse `json:"directTools"` GraphTools []string `json:"graphTools"` + RuntimeMode enums.AIAgentRuntimeMode `json:"runtimeMode"` + RuntimeModeName string `json:"runtimeModeName"` + WorkflowVersionID int64 `json:"workflowVersionId"` SortNo int `json:"sortNo"` CreatedAt string `json:"createdAt"` UpdatedAt string `json:"updatedAt"` diff --git a/internal/pkg/enums/ai.go b/internal/pkg/enums/ai.go index 3f9ffd3..c17d7d5 100644 --- a/internal/pkg/enums/ai.go +++ b/internal/pkg/enums/ai.go @@ -31,3 +31,24 @@ var aiModelTypeLabelMap = map[AIModelType]string{ func GetAIModelTypeLabel(modelType AIModelType) string { return aiModelTypeLabelMap[modelType] } + +type AIAgentRuntimeMode int + +const ( + AIAgentRuntimeModeBuiltinGraph AIAgentRuntimeMode = 1 + AIAgentRuntimeModeWorkflow AIAgentRuntimeMode = 2 +) + +var AIAgentRuntimeModeValues = []AIAgentRuntimeMode{ + AIAgentRuntimeModeBuiltinGraph, + AIAgentRuntimeModeWorkflow, +} + +var aiAgentRuntimeModeLabelMap = map[AIAgentRuntimeMode]string{ + AIAgentRuntimeModeBuiltinGraph: "内置 Graph", + AIAgentRuntimeModeWorkflow: "会话流程", +} + +func GetAIAgentRuntimeModeLabel(mode AIAgentRuntimeMode) string { + return aiAgentRuntimeModeLabelMap[mode] +} diff --git a/internal/services/ai_agent_service.go b/internal/services/ai_agent_service.go index 724f602..87b3197 100644 --- a/internal/services/ai_agent_service.go +++ b/internal/services/ai_agent_service.go @@ -108,6 +108,8 @@ func (s *aIAgentService) UpdateAIAgent(req request.UpdateAIAgentRequest, operato "skill_ids": item.SkillIDs, "allowed_mcp_tools": item.AllowedMCPTools, "allowed_graph_tools": item.AllowedGraphTools, + "runtime_mode": item.RuntimeMode, + "workflow_version_id": item.WorkflowVersionID, "update_user_id": operator.UserID, "update_user_name": operator.Username, "updated_at": time.Now(), @@ -191,6 +193,10 @@ func (s *aIAgentService) buildAIAgentModel(id int64, req request.CreateAIAgentRe if err != nil { return nil, err } + runtimeMode, workflowVersionID, err := s.normalizeRuntimeMode(req.RuntimeMode, req.WorkflowVersionID) + if err != nil { + return nil, err + } directToolsJSON := "" if len(directTools) > 0 { buf, marshalErr := json.Marshal(directTools) @@ -223,6 +229,8 @@ func (s *aIAgentService) buildAIAgentModel(id int64, req request.CreateAIAgentRe SkillIDs: utils.JoinInt64s(skillIDs), AllowedMCPTools: directToolsJSON, AllowedGraphTools: graphToolsJSON, + RuntimeMode: runtimeMode, + WorkflowVersionID: workflowVersionID, }, nil } @@ -349,6 +357,26 @@ func (s *aIAgentService) normalizeGraphTools(input []string) ([]string, error) { return ret, nil } +func (s *aIAgentService) normalizeRuntimeMode(input enums.AIAgentRuntimeMode, workflowVersionID int64) (enums.AIAgentRuntimeMode, int64, error) { + if input == 0 { + input = enums.AIAgentRuntimeModeBuiltinGraph + } + if !slices.Contains(enums.AIAgentRuntimeModeValues, input) { + return 0, 0, errorsx.InvalidParam("invalid ai agent runtime mode") + } + if input != enums.AIAgentRuntimeModeWorkflow { + return input, 0, nil + } + if workflowVersionID <= 0 { + return 0, 0, errorsx.InvalidParam("workflow version is required") + } + version := AIWorkflowService.GetVersion(workflowVersionID) + if version == nil || version.Status != enums.StatusOk { + return 0, 0, errorsx.InvalidParam("workflow version does not exist") + } + return input, workflowVersionID, nil +} + func (s *aIAgentService) UpdateSort(ids []int64) error { return sqls.WithTransaction(func(ctx *sqls.TxContext) error { for i, id := range ids { diff --git a/internal/services/ai_agent_workflow_service_test.go b/internal/services/ai_agent_workflow_service_test.go new file mode 100644 index 0000000..9e34f1e --- /dev/null +++ b/internal/services/ai_agent_workflow_service_test.go @@ -0,0 +1,133 @@ +package services + +import ( + "testing" + + "agent-desk/internal/models" + "agent-desk/internal/pkg/dto" + "agent-desk/internal/pkg/dto/request" + "agent-desk/internal/pkg/enums" + + "github.com/glebarez/sqlite" + "github.com/mlogclub/simple/sqls" + "gorm.io/gorm" +) + +func TestAIAgentServiceSavesWorkflowBinding(t *testing.T) { + setupAIAgentWorkflowTestDB(t) + operator := aiAgentWorkflowTestOperator() + aiConfigID := createAIAgentWorkflowTestConfig(t) + knowledgeID := createAIAgentWorkflowTestKnowledgeBase(t) + versionID := createAIAgentWorkflowVersion(t) + + item, err := AIAgentService.CreateAIAgent(request.CreateAIAgentRequest{ + Name: "workflow agent", + AIConfigID: aiConfigID, + ServiceMode: enums.IMConversationServiceModeAIOnly, + HandoffMode: enums.AIAgentHandoffModeWaitPool, + FallbackMode: enums.AIAgentFallbackModeNoAnswer, + KnowledgeIDs: []int64{knowledgeID}, + RuntimeMode: enums.AIAgentRuntimeModeWorkflow, + WorkflowVersionID: versionID, + }, operator) + if err != nil { + t.Fatalf("CreateAIAgent() error = %v", err) + } + + if item.RuntimeMode != enums.AIAgentRuntimeModeWorkflow { + t.Fatalf("expected workflow runtime mode, got %d", item.RuntimeMode) + } + if item.WorkflowVersionID != versionID { + t.Fatalf("expected workflow version %d, got %d", versionID, item.WorkflowVersionID) + } +} + +func TestAIAgentServiceRejectsWorkflowModeWithoutVersion(t *testing.T) { + setupAIAgentWorkflowTestDB(t) + operator := aiAgentWorkflowTestOperator() + aiConfigID := createAIAgentWorkflowTestConfig(t) + knowledgeID := createAIAgentWorkflowTestKnowledgeBase(t) + + _, err := AIAgentService.CreateAIAgent(request.CreateAIAgentRequest{ + Name: "workflow agent without version", + AIConfigID: aiConfigID, + ServiceMode: enums.IMConversationServiceModeAIOnly, + HandoffMode: enums.AIAgentHandoffModeWaitPool, + FallbackMode: enums.AIAgentFallbackModeNoAnswer, + KnowledgeIDs: []int64{knowledgeID}, + RuntimeMode: enums.AIAgentRuntimeModeWorkflow, + }, operator) + if err == nil { + t.Fatalf("expected workflow runtime without version to fail") + } +} + +func setupAIAgentWorkflowTestDB(t *testing.T) { + t.Helper() + db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{}) + if err != nil { + t.Fatalf("open sqlite db: %v", err) + } + if err := db.AutoMigrate(&models.AIAgent{}, &models.AIConfig{}, &models.KnowledgeBase{}, &models.AIWorkflow{}, &models.AIWorkflowVersion{}); err != nil { + t.Fatalf("auto migrate: %v", err) + } + sqls.SetDB(db) +} + +func createAIAgentWorkflowTestConfig(t *testing.T) int64 { + t.Helper() + item := &models.AIConfig{ + Name: "workflow-test-config", + Provider: enums.AIProviderOpenAI, + ModelType: enums.AIModelTypeLLM, + ModelName: "gpt-test", + Status: enums.StatusOk, + } + if err := sqls.DB().Create(item).Error; err != nil { + t.Fatalf("create ai config: %v", err) + } + return item.ID +} + +func createAIAgentWorkflowTestKnowledgeBase(t *testing.T) int64 { + t.Helper() + item := &models.KnowledgeBase{ + Name: "workflow-test-kb", + KnowledgeType: string(enums.KnowledgeBaseTypeFAQ), + Status: enums.StatusOk, + } + if err := sqls.DB().Create(item).Error; err != nil { + t.Fatalf("create knowledge base: %v", err) + } + return item.ID +} + +func createAIAgentWorkflowVersion(t *testing.T) int64 { + t.Helper() + workflow := &models.AIWorkflow{ + Name: "workflow-test", + OwnerType: "ai_agent", + OwnerID: 1, + Status: enums.StatusOk, + } + if err := sqls.DB().Create(workflow).Error; err != nil { + t.Fatalf("create workflow: %v", err) + } + version := &models.AIWorkflowVersion{ + WorkflowID: workflow.ID, + Version: 1, + Status: enums.StatusOk, + } + if err := sqls.DB().Create(version).Error; err != nil { + t.Fatalf("create workflow version: %v", err) + } + return version.ID +} + +func aiAgentWorkflowTestOperator() *dto.AuthPrincipal { + return &dto.AuthPrincipal{ + UserID: 1, + Username: "agent-workflow-tester", + Nickname: "agent-workflow-tester", + } +}