Updated

mutablelogic · Feb 7, 2025 · 291165d · 291165d
1 parent 67e0ad4
commit 291165d
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/examples/tool_adder/main.go b/examples/tool_adder/main.go
@@ -56,13 +56,15 @@ func main() {
 	}
 
 	// Register tool
-	session := model.Context()
 	toolkit := tool.NewToolKit()
 	toolkit.Register(&Adder{})
 
-	// Get the tool call
+	// Create a chat session
+	session := model.Context(llm.WithToolKit(toolkit))
+
+	// Make the prompt
 	prompt := fmt.Sprintf("What is %v plus %v?", os.Args[2], os.Args[3])
-	if err := session.FromUser(context.TODO(), prompt, llm.WithToolKit(toolkit), llm.WithToolChoice("any")); err != nil {
+	if err := session.FromUser(context.TODO(), prompt); err != nil {
 		panic(err)
 	}
 
@@ -73,11 +75,6 @@ func main() {
 			break
 		}
 
-		// Print out any intermediate messages
-		if session.Text(0) != "" {
-			fmt.Println(session.Text(0))
-		}
-
 		// Get the results from the toolkit
 		fmt.Println("Running", calls)
 		results, err := toolkit.Run(context.TODO(), calls...)