[AiBundle] FaultTolerantToolbox decoration messes with traceable toolbox

[chr-hertel]

When using the bundle out of the box I currently get this error:

Symfony\AI\AiBundle\Profiler\DataCollector::{closure:Symfony\AI\AiBundle\Profiler\DataCollector::lateCollect():63}(): Argument #1 ($toolbox) must be of type Symfony\AI\AiBundle\Profiler\TraceableToolbox, Symfony\AI\Agent\Toolbox\FaultTolerantToolbox given

Might be related to #253

[Guikingone]

Seems it could be fixed with:

AiBundle.php

if ($config['fault_tolerant_toolbox']) {
    $traceableFaultTolerantToolboxDefinition = (new Definition('ai.fault_tolerant.traceable_toolbox.'.$name))
        ->setClass(TraceableToolbox::class)
        ->setArguments([new Reference('.inner')])
        ->setDecoratedService('ai.fault_tolerant_toolbox.'.$name)
        ->addTag('ai.traceable_toolbox');
    $container->setDefinition('ai.fault_tolerant.traceable_toolbox.'.$name, $traceableFaultTolerantToolboxDefinition);
}

services.php

->set('ai.traceable_toolbox', TraceableToolbox::class)
    ->decorate('ai.fault_tolerant_toolbox')
    ->args([
        service('.inner'),
    ])
    ->tag('ai.traceable_toolbox')