ref(chat): new component && API improvements

Author: Guikingone

src/agent/composer.json

@@ -25,6 +25,7 @@
         "phpstan/phpdoc-parser": "^2.1",
         "psr/log": "^3.0",
         "symfony/ai-platform": "@dev",
+        "symfony/ai-chat": "@dev",
         "symfony/clock": "^6.4 || ^7.1",
         "symfony/http-client": "^6.4 || ^7.1",
         "symfony/property-access": "^6.4 || ^7.1",

src/agent/src/Chat.php

@@ -433,7 +433,7 @@ public function testGetNameReturnsProvidedName()
         $this->assertSame($name, $agent->getName());
     }
 
-    public function testDoubleAgentCanUseSameMessageStore()
+    public function testMultipleAgentCanUseSameChat()
     {
         $platform = $this->createMock(PlatformInterface::class);
         $platform->method('invoke')
@@ -451,10 +451,10 @@ public function testDoubleAgentCanUseSameMessageStore()
 
         $firstChat->initiate(new MessageBag(
             Message::forSystem('You are a helpful assistant. You only answer with short sentences.'),
-        ), 'foo');
+        ));
         $secondChat->initiate(new MessageBag(
             Message::forSystem('You are a helpful assistant. You only answer with short sentences.'),
-        ), 'bar');
+        ));
 
         $firstChat->submit(new UserMessage(new Text('Hello')));
         $secondChat->submit(new UserMessage(new Text('Hello')));

src/agent/tests/AgentTest.php

@@ -16,6 +16,7 @@
     "require": {
         "php": ">=8.2",
         "symfony/ai-agent": "@dev",
+        "symfony/ai-chat": "@dev",
         "symfony/ai-platform": "@dev",
         "symfony/ai-store": "@dev",
         "symfony/config": "^6.4 || ^7.0",

src/ai-bundle/composer.json

@@ -17,12 +17,6 @@
 use Symfony\AI\Agent\AgentInterface;
 use Symfony\AI\Agent\Attribute\AsInputProcessor;
 use Symfony\AI\Agent\Attribute\AsOutputProcessor;
-use Symfony\AI\Agent\Chat;
-use Symfony\AI\Agent\Chat\MessageStore\CacheStore as CacheMessageStore;
-use Symfony\AI\Agent\Chat\MessageStore\InMemoryStore as InMemoryMessageStore;
-use Symfony\AI\Agent\Chat\MessageStore\SessionStore as SessionMessageStore;
-use Symfony\AI\Agent\Chat\MessageStoreInterface;
-use Symfony\AI\Agent\ChatInterface;
 use Symfony\AI\Agent\InputProcessor\SystemPromptInputProcessor;
 use Symfony\AI\Agent\InputProcessorInterface;
 use Symfony\AI\Agent\OutputProcessorInterface;
@@ -38,6 +32,12 @@
 use Symfony\AI\AiBundle\Profiler\TraceablePlatform;
 use Symfony\AI\AiBundle\Profiler\TraceableToolbox;
 use Symfony\AI\AiBundle\Security\Attribute\IsGrantedTool;
+use Symfony\AI\Chat\Bridge\Local\CacheStore as CacheMessageStore;
+use Symfony\AI\Chat\Bridge\Local\InMemoryStore as InMemoryMessageStore;
+use Symfony\AI\Chat\Bridge\Symfony\SessionStore as SessionMessageStore;
+use Symfony\AI\Chat\Chat;
+use Symfony\AI\Chat\ChatInterface;
+use Symfony\AI\Chat\MessageStoreInterface;
 use Symfony\AI\Platform\Bridge\Anthropic\PlatformFactory as AnthropicPlatformFactory;
 use Symfony\AI\Platform\Bridge\Azure\OpenAi\PlatformFactory as AzureOpenAiPlatformFactory;
 use Symfony\AI\Platform\Bridge\Cerebras\PlatformFactory as CerebrasPlatformFactory;

src/ai-bundle/src/AiBundle.php

@@ -11,7 +11,7 @@
 
 namespace Symfony\AI\AiBundle\Profiler;
 
-use Symfony\AI\Agent\ChatInterface;
+use Symfony\AI\Chat\ChatInterface;
 use Symfony\AI\Platform\Message\AssistantMessage;
 use Symfony\AI\Platform\Message\MessageBag;
 use Symfony\AI\Platform\Message\UserMessage;
@@ -35,19 +35,4 @@ public function submit(UserMessage $message, ?string $id = null): AssistantMessa
     {
         return $this->chat->submit($message, $id);
     }
-
-    public function getCurrentMessageBag(): MessageBag
-    {
-        return $this->chat->getCurrentMessageBag();
-    }
-
-    public function getMessageBag(string $id): MessageBag
-    {
-        return $this->chat->getMessageBag($id);
-    }
-
-    public function getId(): string
-    {
-        return $this->chat->getId();
-    }
 }

src/ai-bundle/src/Profiler/TraceableChat.php

@@ -11,7 +11,7 @@
 
 namespace Symfony\AI\AiBundle\Profiler;
 
-use Symfony\AI\Agent\Chat\MessageStoreInterface;
+use Symfony\AI\Chat\MessageStoreInterface;
 use Symfony\AI\Platform\Message\MessageBag;
 
 /**

src/ai-bundle/src/Profiler/TraceableMessageStore.php

@@ -1212,6 +1212,10 @@ public function testAgentsWithChatCanBeDefined()
                         'agent' => 'another_agent',
                         'message_store' => 'cache',
                     ],
+                    'second' => [
+                        'agent' => 'another_agent',
+                        'message_store' => 'cache',
+                    ],
                 ],
             ],
         ]);

src/ai-bundle/tests/DependencyInjection/AiBundleTest.php

@@ -16,8 +16,8 @@
 use PHPUnit\Framework\TestCase;
 use Symfony\AI\Agent\AgentInterface;
 use Symfony\AI\Agent\Chat;
-use Symfony\AI\Agent\Chat\MessageStore\InMemoryStore;
 use Symfony\AI\AiBundle\Profiler\TraceableChat;
+use Symfony\AI\Chat\Bridge\Local\InMemoryStore;
 use Symfony\AI\Platform\Message\Message;
 use Symfony\AI\Platform\Result\TextResult;
 
@@ -28,18 +28,6 @@
 #[UsesClass(TextResult::class)]
 final class TraceableChatTest extends TestCase
 {
-    public function testIdCanBeRetrieved()
-    {
-        $agent = $this->createMock(AgentInterface::class);
-
-        $store = new InMemoryStore('foo');
-        $chat = new Chat($agent, $store);
-
-        $traceableChat = new TraceableChat($chat);
-
-        $this->assertSame('foo', $traceableChat->getId());
-    }
-
     public function testCurrentMessageBagCanBeRetrieved()
     {
         $agent = $this->createMock(AgentInterface::class);

src/ai-bundle/tests/Profiler/TraceableChatTest.php

@@ -14,8 +14,8 @@
 use PHPUnit\Framework\Attributes\CoversClass;
 use PHPUnit\Framework\Attributes\UsesClass;
 use PHPUnit\Framework\TestCase;
-use Symfony\AI\Agent\Chat\MessageStore\InMemoryStore;
 use Symfony\AI\AiBundle\Profiler\TraceableMessageStore;
+use Symfony\AI\Chat\Bridge\Local\InMemoryStore;
 use Symfony\AI\Platform\Message\Message;
 use Symfony\AI\Platform\Message\MessageBag;