Use ToolResult already in ToolboxInterface

chr-hertel · chr-hertel · commit 656186bdb243 · 2025-10-09T02:08:55.000+02:00
diff --git a/src/agent/src/Toolbox/AgentProcessor.php b/src/agent/src/Toolbox/AgentProcessor.php
@@ -99,9 +99,8 @@ private function handleToolCallsCallback(Output $output): \Closure
 
                 $results = [];
                 foreach ($toolCalls as $toolCall) {
-                    $result = $this->toolbox->execute($toolCall);
-                    $results[] = new ToolResult($toolCall, $result);
-                    $messages->add(Message::ofToolCall($toolCall, $this->resultConverter->convert($result)));
+                    $results[] = $toolResult = $this->toolbox->execute($toolCall);
+                    $messages->add(Message::ofToolCall($toolCall, $this->resultConverter->convert($toolResult)));
                 }
 
                 $event = new ToolCallsExecuted(...$results);
diff --git a/src/agent/src/Toolbox/FaultTolerantToolbox.php b/src/agent/src/Toolbox/FaultTolerantToolbox.php
@@ -33,16 +33,19 @@ public function getTools(): array
         return $this->innerToolbox->getTools();
     }
 
-    public function execute(ToolCall $toolCall): mixed
+    public function execute(ToolCall $toolCall): ToolResult
     {
         try {
             return $this->innerToolbox->execute($toolCall);
         } catch (ToolExecutionExceptionInterface $e) {
-            return $e->getToolCallResult();
+            return new ToolResult($toolCall, $e->getToolCallResult());
         } catch (ToolNotFoundException) {
             $names = array_map(fn (Tool $metadata) => $metadata->getName(), $this->getTools());
 
-            return \sprintf('Tool "%s" was not found, please use one of these: %s', $toolCall->getName(), implode(', ', $names));
+            return new ToolResult(
+                $toolCall,
+                \sprintf('Tool "%s" was not found, please use one of these: %s', $toolCall->getName(), implode(', ', $names))
+            );
         }
     }
 }
diff --git a/src/agent/src/Toolbox/Toolbox.php b/src/agent/src/Toolbox/Toolbox.php
@@ -72,7 +72,7 @@ public function getTools(): array
         return $this->map = $map;
     }
 
-    public function execute(ToolCall $toolCall): mixed
+    public function execute(ToolCall $toolCall): ToolResult
     {
         $metadata = $this->getMetadata($toolCall);
         $tool = $this->getExecutable($metadata);
@@ -93,7 +93,7 @@ public function execute(ToolCall $toolCall): mixed
             throw ToolExecutionException::executionFailed($toolCall, $e);
         }
 
-        return $result;
+        return new ToolResult($toolCall, $result);
     }
 
     private function getMetadata(ToolCall $toolCall): Tool
diff --git a/src/agent/src/Toolbox/ToolboxInterface.php b/src/agent/src/Toolbox/ToolboxInterface.php
@@ -30,5 +30,5 @@ public function getTools(): array;
      * @throws ToolExecutionExceptionInterface if the tool execution fails
      * @throws ToolNotFoundException           if the tool is not found
      */
-    public function execute(ToolCall $toolCall): mixed;
+    public function execute(ToolCall $toolCall): ToolResult;
 }
diff --git a/src/agent/tests/InputProcessor/SystemPromptInputProcessorTest.php b/src/agent/tests/InputProcessor/SystemPromptInputProcessorTest.php
@@ -15,6 +15,7 @@
 use Symfony\AI\Agent\Input;
 use Symfony\AI\Agent\InputProcessor\SystemPromptInputProcessor;
 use Symfony\AI\Agent\Toolbox\ToolboxInterface;
+use Symfony\AI\Agent\Toolbox\ToolResult;
 use Symfony\AI\Fixtures\Tool\ToolNoParams;
 use Symfony\AI\Fixtures\Tool\ToolRequiredParams;
 use Symfony\AI\Platform\Message\Content\File;
@@ -72,9 +73,9 @@ public function getTools(): array
                     return [];
                 }
 
-                public function execute(ToolCall $toolCall): mixed
+                public function execute(ToolCall $toolCall): ToolResult
                 {
-                    return null;
+                    return new ToolResult($toolCall, null);
                 }
             },
         );
@@ -110,9 +111,9 @@ public function getTools(): array
                     ];
                 }
 
-                public function execute(ToolCall $toolCall): mixed
+                public function execute(ToolCall $toolCall): ToolResult
                 {
-                    return null;
+                    return new ToolResult($toolCall, null);
                 }
             },
             $this->getTranslator(),
@@ -153,9 +154,9 @@ public function getTools(): array
                     ];
                 }
 
-                public function execute(ToolCall $toolCall): mixed
+                public function execute(ToolCall $toolCall): ToolResult
                 {
-                    return null;
+                    return new ToolResult($toolCall, null);
                 }
             },
         );
diff --git a/src/agent/tests/Toolbox/AgentProcessorTest.php b/src/agent/tests/Toolbox/AgentProcessorTest.php
@@ -17,6 +17,7 @@
 use Symfony\AI\Agent\Output;
 use Symfony\AI\Agent\Toolbox\AgentProcessor;
 use Symfony\AI\Agent\Toolbox\ToolboxInterface;
+use Symfony\AI\Agent\Toolbox\ToolResult;
 use Symfony\AI\Platform\Message\AssistantMessage;
 use Symfony\AI\Platform\Message\MessageBag;
 use Symfony\AI\Platform\Message\ToolCallMessage;
@@ -72,12 +73,15 @@ public function testProcessInputWithRegisteredToolsButToolOverride()
 
     public function testProcessOutputWithToolCallResponseKeepingMessages()
     {
+        $toolCall = new ToolCall('id1', 'tool1', ['arg1' => 'value1']);
         $toolbox = $this->createMock(ToolboxInterface::class);
-        $toolbox->expects($this->once())->method('execute')->willReturn('Test response');
+        $toolbox
+            ->expects($this->once())
+            ->method('execute')
+            ->willReturn(new ToolResult($toolCall, 'Test response'));
 
         $messageBag = new MessageBag();
-
-        $result = new ToolCallResult(new ToolCall('id1', 'tool1', ['arg1' => 'value1']));
+        $result = new ToolCallResult($toolCall);
 
         $agent = $this->createStub(AgentInterface::class);
 
@@ -95,12 +99,15 @@ public function testProcessOutputWithToolCallResponseKeepingMessages()
 
     public function testProcessOutputWithToolCallResponseForgettingMessages()
     {
+        $toolCall = new ToolCall('id1', 'tool1', ['arg1' => 'value1']);
         $toolbox = $this->createMock(ToolboxInterface::class);
-        $toolbox->expects($this->once())->method('execute')->willReturn('Test response');
+        $toolbox
+            ->expects($this->once())
+            ->method('execute')
+            ->willReturn(new ToolResult($toolCall, 'Test response'));
 
         $messageBag = new MessageBag();
-
-        $result = new ToolCallResult(new ToolCall('id1', 'tool1', ['arg1' => 'value1']));
+        $result = new ToolCallResult($toolCall);
 
         $agent = $this->createStub(AgentInterface::class);
 
diff --git a/src/agent/tests/Toolbox/FaultTolerantToolboxTest.php b/src/agent/tests/Toolbox/FaultTolerantToolboxTest.php
@@ -17,6 +17,7 @@
 use Symfony\AI\Agent\Toolbox\Exception\ToolNotFoundException;
 use Symfony\AI\Agent\Toolbox\FaultTolerantToolbox;
 use Symfony\AI\Agent\Toolbox\ToolboxInterface;
+use Symfony\AI\Agent\Toolbox\ToolResult;
 use Symfony\AI\Fixtures\Tool\ToolNoParams;
 use Symfony\AI\Fixtures\Tool\ToolRequiredParams;
 use Symfony\AI\Platform\Result\ToolCall;
@@ -37,7 +38,7 @@ public function testFaultyToolExecution()
         $toolCall = new ToolCall('987654321', 'tool_foo');
         $actual = $faultTolerantToolbox->execute($toolCall);
 
-        $this->assertSame($expected, $actual);
+        $this->assertSame($expected, $actual->getResult());
     }
 
     public function testFaultyToolCall()
@@ -52,7 +53,7 @@ public function testFaultyToolCall()
         $toolCall = new ToolCall('123456789', 'tool_xyz');
         $actual = $faultTolerantToolbox->execute($toolCall);
 
-        $this->assertSame($expected, $actual);
+        $this->assertSame($expected, $actual->getResult());
     }
 
     public function testCustomToolExecutionException()
@@ -72,7 +73,7 @@ public function getToolCallResult(): array
         $toolCall = new ToolCall('123456789', 'tool_xyz');
         $actual = $faultTolerantToolbox->execute($toolCall);
 
-        $this->assertSame($expected, $actual);
+        $this->assertSame($expected, $actual->getResult());
     }
 
     private function createFaultyToolbox(\Closure $exceptionFactory): ToolboxInterface
@@ -93,7 +94,7 @@ public function getTools(): array
                 ];
             }
 
-            public function execute(ToolCall $toolCall): mixed
+            public function execute(ToolCall $toolCall): ToolResult
             {
                 throw ($this->exceptionFactory)($toolCall);
             }
diff --git a/src/agent/tests/Toolbox/ToolboxTest.php b/src/agent/tests/Toolbox/ToolboxTest.php
@@ -21,6 +21,7 @@
 use Symfony\AI\Agent\Toolbox\ToolFactory\ChainFactory;
 use Symfony\AI\Agent\Toolbox\ToolFactory\MemoryToolFactory;
 use Symfony\AI\Agent\Toolbox\ToolFactory\ReflectionToolFactory;
+use Symfony\AI\Agent\Toolbox\ToolResult;
 use Symfony\AI\Fixtures\Tool\ToolCustomException;
 use Symfony\AI\Fixtures\Tool\ToolDate;
 use Symfony\AI\Fixtures\Tool\ToolException;
@@ -180,9 +181,11 @@ public function testExecuteWithCustomException()
     #[DataProvider('executeProvider')]
     public function testExecute(string $expected, string $toolName, array $toolPayload = [])
     {
-        $this->assertSame(
-            $expected,
-            $this->toolbox->execute(new ToolCall('call_1234', $toolName, $toolPayload)),
+        $toolCall = new ToolCall('call_1234', $toolName, $toolPayload);
+
+        $this->assertEquals(
+            new ToolResult($toolCall, $expected),
+            $this->toolbox->execute($toolCall),
         );
     }
 
@@ -244,7 +247,7 @@ public function testToolboxExecutionWithMemoryFactory()
         $toolbox = new Toolbox([new ToolNoAttribute1()], $memoryFactory);
         $result = $toolbox->execute(new ToolCall('call_1234', 'happy_birthday', ['name' => 'John', 'years' => 30]));
 
-        $this->assertSame('Happy Birthday, John! You are 30 years old.', $result);
+        $this->assertSame('Happy Birthday, John! You are 30 years old.', $result->getResult());
     }
 
     public function testToolboxMapWithOverrideViaChain()
diff --git a/src/ai-bundle/src/Profiler/DataCollector.php b/src/ai-bundle/src/Profiler/DataCollector.php
@@ -12,7 +12,7 @@
 namespace Symfony\AI\AiBundle\Profiler;
 
 use Symfony\AI\Agent\Toolbox\ToolboxInterface;
-use Symfony\AI\Platform\Model;
+use Symfony\AI\Agent\Toolbox\ToolResult;
 use Symfony\AI\Platform\Tool\Tool;
 use Symfony\Bundle\FrameworkBundle\DataCollector\AbstractDataCollector;
 use Symfony\Component\HttpFoundation\Request;
@@ -23,7 +23,6 @@
  * @author Christopher Hertel <mail@christopher-hertel.de>
  *
  * @phpstan-import-type PlatformCallData from TraceablePlatform
- * @phpstan-import-type ToolCallData from TraceableToolbox
  */
 final class DataCollector extends AbstractDataCollector implements LateDataCollectorInterface
 {
@@ -86,7 +85,7 @@ public function getTools(): array
     }
 
     /**
-     * @return ToolCallData[]
+     * @return ToolResult[]
      */
     public function getToolCalls(): array
     {
diff --git a/src/ai-bundle/src/Profiler/TraceableToolbox.php b/src/ai-bundle/src/Profiler/TraceableToolbox.php
@@ -12,20 +12,16 @@
 namespace Symfony\AI\AiBundle\Profiler;
 
 use Symfony\AI\Agent\Toolbox\ToolboxInterface;
+use Symfony\AI\Agent\Toolbox\ToolResult;
 use Symfony\AI\Platform\Result\ToolCall;
 
 /**
  * @author Christopher Hertel <mail@christopher-hertel.de>
- *
- * @phpstan-type ToolCallData array{
- *     call: ToolCall,
- *     result: string,
- * }
  */
 final class TraceableToolbox implements ToolboxInterface
 {
     /**
-     * @var ToolCallData[]
+     * @var ToolResult[]
      */
     public array $calls = [];
 
@@ -39,15 +35,8 @@ public function getTools(): array
         return $this->toolbox->getTools();
     }
 
-    public function execute(ToolCall $toolCall): mixed
+    public function execute(ToolCall $toolCall): ToolResult
     {
-        $result = $this->toolbox->execute($toolCall);
-
-        $this->calls[] = [
-            'call' => $toolCall,
-            'result' => $result,
-        ];
-
-        return $result;
+        return $this->calls[] = $this->toolbox->execute($toolCall);
     }
 }
diff --git a/src/ai-bundle/templates/data_collector.html.twig b/src/ai-bundle/templates/data_collector.html.twig
@@ -221,25 +221,25 @@
 
     <h3>Tool Calls</h3>
     {% if collector.toolCalls|length %}
-        {% for call in collector.toolCalls %}
+        {% for toolResult in collector.toolCalls %}
             <table class="table">
                 <thead>
                     <tr>
-                        <th colspan="2">{{ call.call.name }}</th>
+                        <th colspan="2">{{ toolResult.toolCall.name }}</th>
                     </tr>
                 </thead>
                 <tbody>
                     <tr>
                         <th>ID</th>
-                        <td>{{ call.call.id }}</td>
+                        <td>{{ toolResult.toolCall.id }}</td>
                     </tr>
                     <tr>
                         <th>Arguments</th>
-                        <td>{{ dump(call.call.arguments) }}</td>
+                        <td>{{ dump(toolResult.toolCall.arguments) }}</td>
                     </tr>
                     <tr>
                         <th>Result</th>
-                        <td>{{ dump(call.result) }}</td>
+                        <td>{{ dump(toolResult.result) }}</td>
                     </tr>
                 </tbody>
             </table>
diff --git a/src/ai-bundle/tests/Profiler/TraceableToolboxTest.php b/src/ai-bundle/tests/Profiler/TraceableToolboxTest.php
@@ -13,6 +13,7 @@
 
 use PHPUnit\Framework\TestCase;
 use Symfony\AI\Agent\Toolbox\ToolboxInterface;
+use Symfony\AI\Agent\Toolbox\ToolResult;
 use Symfony\AI\AiBundle\Profiler\TraceableToolbox;
 use Symfony\AI\Platform\Result\ToolCall;
 use Symfony\AI\Platform\Tool\ExecutionReference;
@@ -40,10 +41,10 @@ public function testExecute()
 
         $result = $traceableToolbox->execute($toolCall);
 
-        $this->assertSame('tool_result', $result);
+        $this->assertSame('tool_result', $result->getResult());
         $this->assertCount(1, $traceableToolbox->calls);
-        $this->assertSame($toolCall, $traceableToolbox->calls[0]['call']);
-        $this->assertSame('tool_result', $traceableToolbox->calls[0]['result']);
+        $this->assertSame($toolCall, $traceableToolbox->calls[0]->getToolCall());
+        $this->assertSame('tool_result', $traceableToolbox->calls[0]->getResult());
     }
 
     /**
@@ -62,9 +63,9 @@ public function getTools(): array
                 return $this->tools;
             }
 
-            public function execute(ToolCall $toolCall): string
+            public function execute(ToolCall $toolCall): ToolResult
             {
-                return 'tool_result';
+                return new ToolResult($toolCall, 'tool_result');
             }
         };
     }

Original file line number	Diff line number	Diff line change
`@@ -33,16 +33,19 @@ public function getTools(): array`
`33`	`33`	`return $this->innerToolbox->getTools();`
`34`	`34`	`}`
`35`	`35`
`36`		`- public function execute(ToolCall $toolCall): mixed`
	`36`	`+ public function execute(ToolCall $toolCall): ToolResult`
`37`	`37`	`{`
`38`	`38`	`try {`
`39`	`39`	`return $this->innerToolbox->execute($toolCall);`
`40`	`40`	`} catch (ToolExecutionExceptionInterface $e) {`
`41`		`- return $e->getToolCallResult();`
	`41`	`+ return new ToolResult($toolCall, $e->getToolCallResult());`
`42`	`42`	`} catch (ToolNotFoundException) {`
`43`	`43`	`$names = array_map(fn (Tool $metadata) => $metadata->getName(), $this->getTools());`
`44`	`44`
`45`		`- return \sprintf('Tool "%s" was not found, please use one of these: %s', $toolCall->getName(), implode(', ', $names));`
	`45`	`+ return new ToolResult(`
	`46`	`+ $toolCall,`
	`47`	`+ \sprintf('Tool "%s" was not found, please use one of these: %s', $toolCall->getName(), implode(', ', $names))`
	`48`	`+ );`
`46`	`49`	`}`
`47`	`50`	`}`
`48`	`51`	`}`
Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,7 @@ public function getTools(): array`
`72`	`72`	`return $this->map = $map;`
`73`	`73`	`}`
`74`	`74`
`75`		`- public function execute(ToolCall $toolCall): mixed`
	`75`	`+ public function execute(ToolCall $toolCall): ToolResult`
`76`	`76`	`{`
`77`	`77`	`$metadata = $this->getMetadata($toolCall);`
`78`	`78`	`$tool = $this->getExecutable($metadata);`
`@@ -93,7 +93,7 @@ public function execute(ToolCall $toolCall): mixed`
`93`	`93`	`throw ToolExecutionException::executionFailed($toolCall, $e);`
`94`	`94`	`}`
`95`	`95`
`96`		`- return $result;`
	`96`	`+ return new ToolResult($toolCall, $result);`
`97`	`97`	`}`
`98`	`98`
`99`	`99`	`private function getMetadata(ToolCall $toolCall): Tool`
Original file line number	Diff line number	Diff line change
`@@ -30,5 +30,5 @@ public function getTools(): array;`
`30`	`30`	`* @throws ToolExecutionExceptionInterface if the tool execution fails`
`31`	`31`	`* @throws ToolNotFoundException if the tool is not found`
`32`	`32`	`*/`
`33`		`- public function execute(ToolCall $toolCall): mixed;`
	`33`	`+ public function execute(ToolCall $toolCall): ToolResult;`
`34`	`34`	`}`
Original file line number	Diff line number	Diff line change
`@@ -15,6 +15,7 @@`
`15`	`15`	`use Symfony\AI\Agent\Input;`
`16`	`16`	`use Symfony\AI\Agent\InputProcessor\SystemPromptInputProcessor;`
`17`	`17`	`use Symfony\AI\Agent\Toolbox\ToolboxInterface;`
	`18`	`+use Symfony\AI\Agent\Toolbox\ToolResult;`
`18`	`19`	`use Symfony\AI\Fixtures\Tool\ToolNoParams;`
`19`	`20`	`use Symfony\AI\Fixtures\Tool\ToolRequiredParams;`
`20`	`21`	`use Symfony\AI\Platform\Message\Content\File;`
`@@ -72,9 +73,9 @@ public function getTools(): array`
`72`	`73`	`return [];`
`73`	`74`	`}`
`74`	`75`
`75`		`- public function execute(ToolCall $toolCall): mixed`
	`76`	`+ public function execute(ToolCall $toolCall): ToolResult`
`76`	`77`	`{`
`77`		`- return null;`
	`78`	`+ return new ToolResult($toolCall, null);`
`78`	`79`	`}`
`79`	`80`	`},`
`80`	`81`	`);`
`@@ -110,9 +111,9 @@ public function getTools(): array`
`110`	`111`	`];`
`111`	`112`	`}`
`112`	`113`
`113`		`- public function execute(ToolCall $toolCall): mixed`
	`114`	`+ public function execute(ToolCall $toolCall): ToolResult`
`114`	`115`	`{`
`115`		`- return null;`
	`116`	`+ return new ToolResult($toolCall, null);`
`116`	`117`	`}`
`117`	`118`	`},`
`118`	`119`	`$this->getTranslator(),`
`@@ -153,9 +154,9 @@ public function getTools(): array`
`153`	`154`	`];`
`154`	`155`	`}`
`155`	`156`
`156`		`- public function execute(ToolCall $toolCall): mixed`
	`157`	`+ public function execute(ToolCall $toolCall): ToolResult`
`157`	`158`	`{`
`158`		`- return null;`
	`159`	`+ return new ToolResult($toolCall, null);`
`159`	`160`	`}`
`160`	`161`	`},`
`161`	`162`	`);`
Original file line number	Diff line number	Diff line change
`@@ -17,6 +17,7 @@`
`17`	`17`	`use Symfony\AI\Agent\Toolbox\Exception\ToolNotFoundException;`
`18`	`18`	`use Symfony\AI\Agent\Toolbox\FaultTolerantToolbox;`
`19`	`19`	`use Symfony\AI\Agent\Toolbox\ToolboxInterface;`
	`20`	`+use Symfony\AI\Agent\Toolbox\ToolResult;`
`20`	`21`	`use Symfony\AI\Fixtures\Tool\ToolNoParams;`
`21`	`22`	`use Symfony\AI\Fixtures\Tool\ToolRequiredParams;`
`22`	`23`	`use Symfony\AI\Platform\Result\ToolCall;`
`@@ -37,7 +38,7 @@ public function testFaultyToolExecution()`
`37`	`38`	`$toolCall = new ToolCall('987654321', 'tool_foo');`
`38`	`39`	`$actual = $faultTolerantToolbox->execute($toolCall);`
`39`	`40`
`40`		`- $this->assertSame($expected, $actual);`
	`41`	`+ $this->assertSame($expected, $actual->getResult());`
`41`	`42`	`}`
`42`	`43`
`43`	`44`	`public function testFaultyToolCall()`
`@@ -52,7 +53,7 @@ public function testFaultyToolCall()`
`52`	`53`	`$toolCall = new ToolCall('123456789', 'tool_xyz');`
`53`	`54`	`$actual = $faultTolerantToolbox->execute($toolCall);`
`54`	`55`
`55`		`- $this->assertSame($expected, $actual);`
	`56`	`+ $this->assertSame($expected, $actual->getResult());`
`56`	`57`	`}`
`57`	`58`
`58`	`59`	`public function testCustomToolExecutionException()`
`@@ -72,7 +73,7 @@ public function getToolCallResult(): array`
`72`	`73`	`$toolCall = new ToolCall('123456789', 'tool_xyz');`
`73`	`74`	`$actual = $faultTolerantToolbox->execute($toolCall);`
`74`	`75`
`75`		`- $this->assertSame($expected, $actual);`
	`76`	`+ $this->assertSame($expected, $actual->getResult());`
`76`	`77`	`}`
`77`	`78`
`78`	`79`	`private function createFaultyToolbox(\Closure $exceptionFactory): ToolboxInterface`
`@@ -93,7 +94,7 @@ public function getTools(): array`
`93`	`94`	`];`
`94`	`95`	`}`
`95`	`96`
`96`		`- public function execute(ToolCall $toolCall): mixed`
	`97`	`+ public function execute(ToolCall $toolCall): ToolResult`
`97`	`98`	`{`
`98`	`99`	`throw ($this->exceptionFactory)($toolCall);`
`99`	`100`	`}`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@`
`12`	`12`	`namespace Symfony\AI\AiBundle\Profiler;`
`13`	`13`
`14`	`14`	`use Symfony\AI\Agent\Toolbox\ToolboxInterface;`
`15`		`-use Symfony\AI\Platform\Model;`
	`15`	`+use Symfony\AI\Agent\Toolbox\ToolResult;`
`16`	`16`	`use Symfony\AI\Platform\Tool\Tool;`
`17`	`17`	`use Symfony\Bundle\FrameworkBundle\DataCollector\AbstractDataCollector;`
`18`	`18`	`use Symfony\Component\HttpFoundation\Request;`
`@@ -23,7 +23,6 @@`
`23`	`23`	`* @author Christopher Hertel <mail@christopher-hertel.de>`
`24`	`24`	`*`
`25`	`25`	`* @phpstan-import-type PlatformCallData from TraceablePlatform`
`26`		`- * @phpstan-import-type ToolCallData from TraceableToolbox`
`27`	`26`	`*/`
`28`	`27`	`final class DataCollector extends AbstractDataCollector implements LateDataCollectorInterface`
`29`	`28`	`{`
`@@ -86,7 +85,7 @@ public function getTools(): array`
`86`	`85`	`}`
`87`	`86`
`88`	`87`	`/**`
`89`		`- * @return ToolCallData[]`
	`88`	`+ * @return ToolResult[]`
`90`	`89`	`*/`
`91`	`90`	`public function getToolCalls(): array`
`92`	`91`	`{`