feat: replace deprecated interface

drew/english
Megumin 1 year ago
parent cf63365e54
commit 8e8acb0d0f

@ -40,7 +40,7 @@ func main() {
ctx := context.Background()
callbacks.InitCallbackHandlers([]callbacks.Handler{&loggerCallbacks{}})
callbacks.AppendGlobalHandlers(&loggerCallbacks{})
// 1. create an instance of ChatTemplate as 1st Graph Node
systemTpl := `你是一名房产经纪人,结合用户的薪酬和工作,使用 user_info API为其提供相关的房产信息。邮箱是必须的`

@ -98,8 +98,6 @@ func main() {
panic(err)
}
defer out.Close()
println("\nAnswer:")
for {
@ -112,6 +110,8 @@ func main() {
print(msg.Content)
}
out.Close()
}
}

@ -87,8 +87,6 @@ func main() {
ToolsConfig: compose.ToolsNodeConfig{
Tools: []tool.BaseTool{restaurantTool, dishTool},
},
MessageModifier: react.NewPersonaModifier(persona),
// StreamToolCallChecker: toolCallChecker, // uncomment it to replace the default tool call checker with custom one
})
if err != nil {
@ -110,6 +108,10 @@ func main() {
// fmt.Println(msg.String())
sr, err := ragent.Stream(ctx, []*schema.Message{
{
Role: schema.System,
Content: persona,
},
{
Role: schema.User,
Content: "我在北京,给我推荐一些菜,需要有口味辣一点的菜,至少推荐有 2 家餐厅",

@ -77,7 +77,7 @@ func main() {
return
}
sr, err := agent.Stream(ctx, msgs)
sr, err := agent.Stream(ctx, append([]*schema.Message{schema.SystemMessage(*prompt)}, msgs...))
if err != nil {
c.JSON(consts.StatusInternalServerError, map[string]string{"error": err.Error()})
return
@ -135,7 +135,7 @@ func main() {
func NewAgent(ctx context.Context) (*react.Agent, error) {
// 初始化模型
model, err := PrepareModel(ctx)
m, err := PrepareModel(ctx)
if err != nil {
return nil, err
}
@ -148,11 +148,10 @@ func NewAgent(ctx context.Context) (*react.Agent, error) {
// 初始化 agent
agent, err := react.NewAgent(ctx, &react.AgentConfig{
Model: model,
Model: m,
ToolsConfig: compose.ToolsNodeConfig{
Tools: tools,
},
MessageModifier: react.NewPersonaModifier(*prompt),
})
if err != nil {
return nil, err
@ -174,11 +173,11 @@ func PrepareModel(ctx context.Context) (model.ChatModel, error) {
}
func PrepareTools(ctx context.Context) ([]tool.BaseTool, error) {
duckduckgo, err := duckduckgo.NewTool(ctx, &duckduckgo.Config{})
ddg, err := duckduckgo.NewTool(ctx, &duckduckgo.Config{})
if err != nil {
return nil, err
}
return []tool.BaseTool{duckduckgo}, nil
return []tool.BaseTool{ddg}, nil
}
// simple memory can store messages of each conversation

Loading…
Cancel
Save