feat(project): 项目完结
This commit is contained in:
27
internal/domain/agent/service/armory/ai_api_node.go
Normal file
27
internal/domain/agent/service/armory/ai_api_node.go
Normal file
@@ -0,0 +1,27 @@
|
||||
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 AiAPINode struct {
|
||||
modelProvider ports.ModelProvider
|
||||
next Handler
|
||||
}
|
||||
|
||||
func NewAiAPINode(modelProvider ports.ModelProvider, next Handler) AiAPINode {
|
||||
return AiAPINode{modelProvider: modelProvider, next: next}
|
||||
}
|
||||
|
||||
func (n AiAPINode) Apply(ctx context.Context, command model.ArmoryCommand, dynamic *DynamicContext) (model.RegisteredAgent, error) {
|
||||
api, err := n.modelProvider.NewAPI(ctx, command.Table.Module.AiAPI)
|
||||
if err != nil {
|
||||
return model.RegisteredAgent{}, err
|
||||
}
|
||||
dynamic.ModelAPI = api
|
||||
return tree.Route(ctx, n.next, command, dynamic)
|
||||
}
|
||||
Reference in New Issue
Block a user