feat: 新增工作流执行模块

新增流程执行记录的实体、DTO、DAO、控制器和服务层，支持工作流的执行、回调及结果树状列表查询；同时更新服务名称为 ai-agent。
2026-05-12 13:34:28 +08:00
parent 2aec7fe30f
commit 7c26914353
42 changed files with 4146 additions and 11 deletions
--- a/workflow/service/node/node_library_service.go
+++ b/workflow/service/node/node_library_service.go
@@ -0,0 +1,147 @@
+package node
+
+import (
+	"ai-agent/workflow/consts/node"
+	nodeDto "ai-agent/workflow/model/dto/node"
+	"context"
+	"fmt"
+
+	"gitea.com/red-future/common/beans"
+	"github.com/gogf/gf/v2/encoding/gjson"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+var NodeLibraryService = &nodeLibraryService{}
+
+type nodeLibraryService struct{}
+
+func (s *nodeLibraryService) GetNodeLibrary(ctx context.Context, req *nodeDto.WorkflowNodeTreeReq) (*nodeDto.WorkflowNodeTreeRes, error) {
+
+	WorkflowNodeGroups := []node.NodeGroupItem{
+		{
+			Group: node.NodeGroupComponent,
+			Label: node.NodeGroupNameComponent,
+			Items: []node.NodeItem{
+				{
+					NodeCode:    node.NodeTypeTextModel,
+					NodeName:    node.NodeNameTextModel,
+					SkillOption: true,
+					FormConfig:  []node.NodeFormField{}, // 技能下拉
+					ModelConfig: []node.ModelItem{},
+				},
+				{
+					NodeCode:    node.NodeTypeImageModel,
+					NodeName:    node.NodeNameImageModel,
+					SkillOption: true,
+					FormConfig:  []node.NodeFormField{}, // 技能下拉
+					ModelConfig: []node.ModelItem{},
+				},
+			},
+		},
+		{
+			Group: node.NodeGroupBase,
+			Label: node.NodeGroupNameBase,
+			Items: []node.NodeItem{
+				{
+					NodeCode:    node.NodeTypeMerge,
+					NodeName:    node.NodeNameMerge,
+					SkillOption: false,
+					FormConfig:  []node.NodeFormField{},
+					ModelConfig: []node.ModelItem{},
+				},
+				{
+					NodeCode:    node.NodeTypeJudge,
+					NodeName:    node.NodeNameJudge,
+					SkillOption: false,
+					FormConfig: []node.NodeFormField{
+						{Field: "condition", Label: node.FormLabelCondition, Type: "input", Required: true},
+					},
+					ModelConfig: []node.ModelItem{},
+				},
+				{
+					NodeCode:    node.NodeTypeForm,
+					NodeName:    node.NodeNameForm,
+					SkillOption: false,
+					FormConfig:  []node.NodeFormField{},
+					ModelConfig: []node.ModelItem{},
+				},
+				//{
+				//	NodeCode:    node.NodeTypeModel,
+				//	NodeName:    node.NodeNameModel,
+				//	SkillOption: true,
+				//	FormConfig:  []node.NodeFormField{},
+				//	ModelConfig: []node.ModelItem{},
+				//},
+			},
+		},
+		{
+			Group: node.NodeGroupCustom,
+			Label: node.NodeGroupNameCustom,
+			Items: []node.NodeItem{
+				{
+					NodeCode:    node.NodeTypeCustomNode,
+					NodeName:    node.NodeNameCustomNode,
+					SkillOption: true,
+					FormConfig: []node.NodeFormField{
+						{Field: "nodeName", Label: node.FormLabelApiKey, Type: "input", Required: true},
+						{Field: "nodeType", Label: node.FormLabelModel, Type: "input", Required: true},
+					},
+					ModelConfig: []node.ModelItem{},
+				},
+			},
+		},
+	}
+	tree := &nodeDto.WorkflowNodeTreeRes{
+		Groups: WorkflowNodeGroups,
+	}
+
+	// 3. 遍历分组，根据 typeId=1 给【文本模型节点】追加固定表单
+	for gIdx := range tree.Groups {
+		group := &tree.Groups[gIdx]
+
+		// 遍历分组下的每个节点
+		for itemIdx := range group.Items {
+			item := &group.Items[itemIdx]
+			if item.NodeCode == node.NodeTypeTextModel {
+				item.ModelConfig = append(item.ModelConfig, node.ModelItem{
+					ModelName: "自定义",
+				})
+			}
+			if item.NodeCode == node.NodeTypeImageModel {
+				item.ModelConfig = append(item.ModelConfig, node.ModelItem{
+					ModelName: "自定义",
+				})
+			}
+
+		}
+	}
+
+	return tree, nil
+}
+
+// SetUserInfo 设置用户信息
+func (s *nodeLibraryService) SetUserInfo(ctx context.Context, creator string, tenantId uint64) (headers map[string]string, err error) {
+	// 创建完整的用户信息
+	userInfo := &beans.User{
+		UserName: creator,
+		TenantId: tenantId,
+	}
+	ctx = context.WithValue(ctx, "user", *userInfo)
+	// 提取并保存请求头（在连接升级前）
+	headers = make(map[string]string)
+	// 提取其他headers
+	if r := g.RequestFromCtx(ctx); r != nil {
+		for k, v := range r.Request.Header {
+			if len(v) > 0 {
+				headers[k] = v[0]
+			}
+		}
+	}
+	// 将完整用户信息序列化为JSON，放到X-User-Info请求头
+	userInfoJson, err := gjson.Encode(userInfo)
+	if err != nil {
+		return nil, fmt.Errorf("用户信息序列化失败: %w", err)
+	}
+	headers["X-User-Info"] = string(userInfoJson)
+	return
+}