From 128c268b6dd01ee7af33e4dae8f171ccff98df70 Mon Sep 17 00:00:00 2001
From: Pavel Buchnev <butschster@gmail.com>
Date: Wed, 4 Sep 2024 23:16:17 +0400
Subject: [PATCH] Fixes error Argument #1 ($agent) must be of type string,
 object given

---
 .../Interceptor/GeneratePromptInterceptor.php              | 7 +++++--
 src/AgentExecutor/Interceptor/InjectModelInterceptor.php   | 1 -
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/AgentExecutor/Interceptor/GeneratePromptInterceptor.php b/src/AgentExecutor/Interceptor/GeneratePromptInterceptor.php
index 6adbca0..78c85e5 100644
--- a/src/AgentExecutor/Interceptor/GeneratePromptInterceptor.php
+++ b/src/AgentExecutor/Interceptor/GeneratePromptInterceptor.php
@@ -30,8 +30,11 @@ public function execute(
         ExecutorInterface $next,
     ): Execution {
         if (!$prompt instanceof Prompt) {
-            $agent = $this->agents->get($agent);
-            $prompt = $this->promptGenerator->generate($agent, $prompt, $promptContext);
+            $prompt = $this->promptGenerator->generate(
+                $this->agents->get($agent),
+                $prompt,
+                $promptContext,
+            );
         }
 
         return $next->execute(
diff --git a/src/AgentExecutor/Interceptor/InjectModelInterceptor.php b/src/AgentExecutor/Interceptor/InjectModelInterceptor.php
index 8a5b453..b0f5ba2 100644
--- a/src/AgentExecutor/Interceptor/InjectModelInterceptor.php
+++ b/src/AgentExecutor/Interceptor/InjectModelInterceptor.php
@@ -8,7 +8,6 @@
 use LLM\Agents\Agent\Execution;
 use LLM\Agents\AgentExecutor\ExecutorInterceptorInterface;
 use LLM\Agents\AgentExecutor\ExecutorInterface;
-use LLM\Agents\LLM\AgentPromptGeneratorInterface;
 use LLM\Agents\LLM\ContextInterface;
 use LLM\Agents\LLM\OptionsInterface;
 use LLM\Agents\LLM\Prompt\Chat\Prompt;