feat: replace deprecated interface

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

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

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

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

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

Loading…
Cancel
Save