Refactor processOutput to streamline result type handling

Author: oesteve

src/agent/src/Toolbox/AgentProcessor.php

@@ -76,19 +76,24 @@ 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,
+                    $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));
     }
 
     /**