feat(project): 项目完结

internal/domain/agent/service/armory/chat_model_node.go

@@ -0,0 +1,54 @@
+package armory
+
+import (
+	"context"
+
+	"ai-agent-scaffold-go/internal/domain/agent/model"
+	"ai-agent-scaffold-go/internal/domain/agent/ports"
+	"ai-agent-scaffold-go/internal/domain/shared/tree"
+)
+
+type ChatModelNode struct {
+	modelProvider ports.ModelProvider
+	mcpFactory    ports.MCPToolFactory
+	skillFactory  ports.SkillFactory
+	next          Handler
+}
+
+func NewChatModelNode(
+	modelProvider ports.ModelProvider,
+	mcpFactory ports.MCPToolFactory,
+	skillFactory ports.SkillFactory,
+	next Handler,
+) ChatModelNode {
+	return ChatModelNode{
+		modelProvider: modelProvider,
+		mcpFactory:    mcpFactory,
+		skillFactory:  skillFactory,
+		next:          next,
+	}
+}
+
+func (n ChatModelNode) Apply(ctx context.Context, command model.ArmoryCommand, dynamic *DynamicContext) (model.RegisteredAgent, error) {
+	var tools []ports.Tool
+	for _, config := range command.Table.Module.ChatModel.ToolMCPList {
+		built, err := n.mcpFactory.BuildTools(ctx, config)
+		if err != nil {
+			return model.RegisteredAgent{}, err
+		}
+		tools = append(tools, built...)
+	}
+	for _, config := range command.Table.Module.ChatModel.ToolSkillsList {
+		built, err := n.skillFactory.BuildTools(ctx, config)
+		if err != nil {
+			return model.RegisteredAgent{}, err
+		}
+		tools = append(tools, built...)
+	}
+	chatModel, err := n.modelProvider.NewChatModel(ctx, dynamic.ModelAPI, command.Table.Module.ChatModel, tools)
+	if err != nil {
+		return model.RegisteredAgent{}, err
+	}
+	dynamic.ChatModel = chatModel
+	return tree.Route(ctx, n.next, command, dynamic)
+}