30 lines
820 B
Go
30 lines
820 B
Go
|
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 AgentNode struct {
|
||
|
|
agentFactory ports.AgentFactory
|
||
|
|
next Handler
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewAgentNode(agentFactory ports.AgentFactory, next Handler) AgentNode {
|
||
|
|
return AgentNode{agentFactory: agentFactory, next: next}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n AgentNode) Apply(ctx context.Context, command model.ArmoryCommand, dynamic *DynamicContext) (model.RegisteredAgent, error) {
|
||
|
|
for _, config := range command.Table.Module.Agents {
|
||
|
|
agent, err := n.agentFactory.NewLLMAgent(ctx, config, dynamic.ChatModel)
|
||
|
|
if err != nil {
|
||
|
|
return model.RegisteredAgent{}, err
|
||
|
|
}
|
||
|
|
dynamic.AddAgent(agent)
|
||
|
|
}
|
||
|
|
return tree.Route(ctx, n.next, command, dynamic)
|
||
|
|
}
|