44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
package agent
|
||
|
||
import (
|
||
"context"
|
||
|
||
"eino-project/internal/domain/llm"
|
||
"eino-project/internal/domain/tools"
|
||
|
||
"github.com/cloudwego/eino/adk"
|
||
"github.com/cloudwego/eino/components/tool"
|
||
"github.com/cloudwego/eino/compose"
|
||
)
|
||
|
||
// NewOrderChatAgent 订单查询 Agent，具备订单查询与日志查询工具
|
||
func NewOrderChatAgent(ctx context.Context, models llm.LLM) adk.Agent {
|
||
chatModel, err := models.Intent()
|
||
if err != nil {
|
||
return nil
|
||
}
|
||
toolsCfg := adk.ToolsConfig{
|
||
ToolsNodeConfig: compose.ToolsNodeConfig{
|
||
Tools: []tool.BaseTool{
|
||
tools.NewOrderByIDTool(),
|
||
},
|
||
ExecuteSequentially: false,
|
||
},
|
||
ReturnDirectly: map[string]bool{
|
||
"get_order_by_id": true,
|
||
},
|
||
}
|
||
agent, _ := adk.NewChatModelAgent(ctx, &adk.ChatModelAgentConfig{
|
||
Name: "订单查询智能体",
|
||
Description: "支持按订单ID查询订单详情",
|
||
Model: chatModel,
|
||
ToolsConfig: toolsCfg,
|
||
Instruction: `
|
||
# 你是订单查询智能体
|
||
- 当用户给出订单ID时，优先调用 "get_order_by_id" 返回订单详情
|
||
- 仅返回订单详情json即可，不做多余解释
|
||
`,
|
||
})
|
||
return agent
|
||
}
|