2727use Symfony \Component \DependencyInjection \ContainerInterface ;
2828use Symfony \Component \DependencyInjection \Definition ;
2929use Symfony \Component \DependencyInjection \Reference ;
30+ use Symfony \Contracts \Translation \TranslatorInterface ;
3031
3132#[CoversClass(AiBundle::class)]
3233#[UsesClass(ContainerBuilder::class)]
@@ -447,7 +448,10 @@ public function testMultipleAgentsWithProcessors()
447448 'tools ' => [
448449 ['service ' => 'tool_two ' , 'description ' => 'Tool for second agent ' ],
449450 ],
450- 'system_prompt ' => 'Second agent prompt ' ,
451+ 'system_prompt ' => [
452+ 'system_prompt ' => 'Second agent prompt ' ,
453+ 'translator ' => TranslatorInterface::class,
454+ ],
451455 ],
452456 ],
453457 ],
@@ -471,11 +475,13 @@ public function testMultipleAgentsWithProcessors()
471475 $ firstSystemPrompt = $ container ->getDefinition ('ai.agent.first_agent.system_prompt_processor ' );
472476 $ firstSystemTags = $ firstSystemPrompt ->getTag ('ai.agent.input_processor ' );
473477 $ this ->assertSame ($ firstAgentId , $ firstSystemTags [0 ]['agent ' ]);
478+ $ this ->assertCount (2 , array_filter ($ firstSystemPrompt ->getArguments ()));
474479
475480 // Second agent system prompt processor
476481 $ secondSystemPrompt = $ container ->getDefinition ('ai.agent.second_agent.system_prompt_processor ' );
477482 $ secondSystemTags = $ secondSystemPrompt ->getTag ('ai.agent.input_processor ' );
478483 $ this ->assertSame ($ secondAgentId , $ secondSystemTags [0 ]['agent ' ]);
484+ $ this ->assertCount (3 , array_filter ($ secondSystemPrompt ->getArguments ()));
479485 }
480486
481487 #[TestDox('Processors work correctly when using the default toolbox ' )]
0 commit comments