diff --git a/src/agent/src/Toolbox/AgentProcessor.php b/src/agent/src/Toolbox/AgentProcessor.php
index c4b43bf84..e3b029b60 100644
--- a/src/agent/src/Toolbox/AgentProcessor.php
+++ b/src/agent/src/Toolbox/AgentProcessor.php
@@ -76,19 +76,21 @@ public function processInput(Input $input): void
 
     public function processOutput(Output $output): void
     {
-        if ($output->getResult() instanceof GenericStreamResponse) {
+        $result = $output->getResult();
+
+        if ($result instanceof GenericStreamResponse) {
             $output->setResult(
-                new ToolboxStreamResponse($output->getResult()->getContent(), $this->handleToolCallsCallback($output))
+                new ToolboxStreamResponse($result->getContent(), $this->handleToolCallsCallback($output))
             );
 
             return;
         }
 
-        if (!$output->getResult() instanceof ToolCallResult) {
+        if (!$result instanceof ToolCallResult) {
             return;
         }
 
-        $output->setResult($this->handleToolCallsCallback($output)($output->getResult()));
+        $output->setResult($this->handleToolCallsCallback($output)($result));
     }
 
     /**