From 71f5789911d9cf0e6e98314394ab1ccb5f708cb0 Mon Sep 17 00:00:00 2001
From: Pierre Lafievre <treywelsh@users.noreply.github.com>
Date: Fri, 13 Sep 2024 02:27:40 +0200
Subject: [PATCH] examples: replace deprecated agents.Initialize (#1009)

---
 examples/mrkl-agent-example/mrkl_agent.go | 13 +++++--------
 examples/zapier-llm-example/main.go       | 11 +++--------
 2 files changed, 8 insertions(+), 16 deletions(-)

diff --git a/examples/mrkl-agent-example/mrkl_agent.go b/examples/mrkl-agent-example/mrkl_agent.go
index 6b025c342..92b8fd87e 100644
--- a/examples/mrkl-agent-example/mrkl_agent.go
+++ b/examples/mrkl-agent-example/mrkl_agent.go
@@ -32,15 +32,12 @@ func run() error {
 		tools.Calculator{},
 		search,
 	}
-	executor, err := agents.Initialize(
-		llm,
+
+	agent := agents.NewOneShotAgent(llm,
 		agentTools,
-		agents.ZeroShotReactDescription,
-		agents.WithMaxIterations(3),
-	)
-	if err != nil {
-		return err
-	}
+		agents.WithMaxIterations(3))
+	executor := agents.NewExecutor(agent)
+
 	question := "Who is Olivia Wilde's boyfriend? What is his current age raised to the 0.23 power?"
 	answer, err := chains.Run(context.Background(), executor, question)
 	fmt.Println(answer)
diff --git a/examples/zapier-llm-example/main.go b/examples/zapier-llm-example/main.go
index 7ffe3e2b7..5ef71f1f1 100644
--- a/examples/zapier-llm-example/main.go
+++ b/examples/zapier-llm-example/main.go
@@ -37,15 +37,10 @@ func main() {
 	agentTools = append(agentTools, tks...)
 
 	// Initialize the agent
-	executor, err := agents.Initialize(
-		llm,
+	agent := agents.NewOneShotAgent(llm,
 		agentTools,
-		agents.ZeroShotReactDescription,
-		agents.WithMaxIterations(3),
-	)
-	if err != nil {
-		panic(err)
-	}
+		agents.WithMaxIterations(3))
+	executor := agents.NewExecutor(agent)
 
 	// run a chain with the executor and defined input
 	input := "Get the last email from noreply@github.com"