17
17
18
18
use PHPUnit \Framework \TestCase ;
19
19
use Symfony \AI \Agent \AgentInterface ;
20
- use Symfony \AI \AiBundle \Command \ChatCommand ;
20
+ use Symfony \AI \AiBundle \Command \AgentCallCommand ;
21
21
use Symfony \AI \AiBundle \Exception \RuntimeException ;
22
22
use Symfony \AI \Platform \Message \Message ;
23
23
use Symfony \AI \Platform \Message \MessageBag ;
27
27
use Symfony \Component \Console \Tester \CommandTester ;
28
28
use Symfony \Component \DependencyInjection \ServiceLocator ;
29
29
30
- final class ChatCommandTest extends TestCase
30
+ final class AgentCallCommandTest extends TestCase
31
31
{
32
32
public function testCommandFailsWithInvalidAgent ()
33
33
{
@@ -37,7 +37,7 @@ public function testCommandFailsWithInvalidAgent()
37
37
'test ' => $ agent ,
38
38
];
39
39
40
- $ command = new ChatCommand ($ this ->createServiceLocator ($ agents ));
40
+ $ command = new AgentCallCommand ($ this ->createServiceLocator ($ agents ));
41
41
$ application = self ::createApplication ($ command );
42
42
43
43
$ commandTester = new CommandTester ($ command );
@@ -71,7 +71,7 @@ public function testCommandExecutesWithValidAgent()
71
71
'test ' => $ agent ,
72
72
];
73
73
74
- $ command = new ChatCommand ($ this ->createServiceLocator ($ agents ));
74
+ $ command = new AgentCallCommand ($ this ->createServiceLocator ($ agents ));
75
75
$ application = self ::createApplication ($ command );
76
76
77
77
$ commandTester = new CommandTester ($ command );
@@ -100,7 +100,7 @@ public function testCommandHandlesQuitCommand()
100
100
'test ' => $ agent ,
101
101
];
102
102
103
- $ command = new ChatCommand ($ this ->createServiceLocator ($ agents ));
103
+ $ command = new AgentCallCommand ($ this ->createServiceLocator ($ agents ));
104
104
$ application = self ::createApplication ($ command );
105
105
106
106
$ commandTester = new CommandTester ($ command );
@@ -128,7 +128,7 @@ public function testCommandHandlesAgentCallException()
128
128
'test ' => $ agent ,
129
129
];
130
130
131
- $ command = new ChatCommand ($ this ->createServiceLocator ($ agents ));
131
+ $ command = new AgentCallCommand ($ this ->createServiceLocator ($ agents ));
132
132
$ application = self ::createApplication ($ command );
133
133
134
134
$ commandTester = new CommandTester ($ command );
@@ -165,7 +165,7 @@ public function testInitializeMethodSelectsCorrectAgent()
165
165
'second ' => $ agent2 ,
166
166
];
167
167
168
- $ command = new ChatCommand ($ this ->createServiceLocator ($ agents ));
168
+ $ command = new AgentCallCommand ($ this ->createServiceLocator ($ agents ));
169
169
$ application = self ::createApplication ($ command );
170
170
171
171
$ commandTester = new CommandTester ($ command );
@@ -203,7 +203,7 @@ public function testCommandWithSystemPromptDisplaysItOnce()
203
203
'test ' => $ agent ,
204
204
];
205
205
206
- $ command = new ChatCommand ($ this ->createServiceLocator ($ agents ));
206
+ $ command = new AgentCallCommand ($ this ->createServiceLocator ($ agents ));
207
207
$ application = self ::createApplication ($ command );
208
208
209
209
$ commandTester = new CommandTester ($ command );
0 commit comments