fix chat history management

Author: ilvalerione

src/Agent/Nodes/StreamingNode.php

@@ -56,14 +56,14 @@ public function __invoke(AIInferenceEvent $event, AgentState $state): Generator|
 
             $this->emit('inference-stop', new InferenceStop($lastMessage, $message));
 
-            // Add the message to the chat history
-            $this->addToChatHistory($state, $message);
-
             // Route based on the message type
             if ($message instanceof ToolCallMessage) {
                 return new ToolCallEvent($message, $event);
             }
 
+            // Add the final message to the chat history (after tool loop)
+            $this->addToChatHistory($state, $message);
+
             return new StopEvent();
 
         } catch (Throwable $exception) {

src/Agent/Nodes/StructuredOutputNode.php

@@ -102,13 +102,14 @@ public function __invoke(AIInferenceEvent $event, AgentState $state): ToolCallEv
 
                 $this->emit('inference-stop', new InferenceStop($last, $response));
 
-                $this->addToChatHistory($state, $response);
-
                 // If the response is a tool call, route to tool execution
                 if ($response instanceof ToolCallMessage) {
                     return new ToolCallEvent($response, $event);
                 }
 
+                // Add the final message to the chat history (after tool loop)
+                $this->addToChatHistory($state, $response);
+
                 // Process the response: extract, deserialize, and validate
                 $output = $this->processResponse($response, $schema, $this->outputClass);