ref(chat): new component && API improvements

Author: Guikingone

examples/composer.json

@@ -17,6 +17,7 @@
         "probots-io/pinecone-php": "^1.0",
         "psr/http-factory-implementation": "*",
         "symfony/ai-agent": "@dev",
+        "symfony/ai-chat": "@dev",
         "symfony/ai-platform": "@dev",
         "symfony/ai-store": "@dev",
         "symfony/cache": "^7.3|^8.0",

examples/misc/persistent-chat-double-agent.php

@@ -10,8 +10,8 @@
  */
 
 use Symfony\AI\Agent\Agent;
-use Symfony\AI\Agent\Chat;
-use Symfony\AI\Agent\Chat\MessageStore\InMemoryStore;
+use Symfony\AI\Chat\Bridge\Local\InMemoryStore;
+use Symfony\AI\Chat\Chat;
 use Symfony\AI\Platform\Bridge\OpenAi\Gpt;
 use Symfony\AI\Platform\Bridge\OpenAi\PlatformFactory;
 use Symfony\AI\Platform\Message\Message;
@@ -26,21 +26,19 @@
 
 $store = new InMemoryStore();
 
-$firstChat = new Chat($agent, $store);
-$secondChat = new Chat($agent, $store);
+$chat = new Chat($agent, $store);
 
-$firstChat->initiate(new MessageBag(
+$chat->initiate(new MessageBag(
     Message::forSystem('You are a helpful assistant. You only answer with short sentences.'),
-), '_first_chat');
-$secondChat->initiate(new MessageBag(
-    Message::forSystem('You are a helpful assistant. You only answer with short sentences.'),
-), '_second_chat');
+));
+
+$forkedChat = $chat->fork('fork');
 
-$firstChat->submit(Message::ofUser('My name is Christopher.'));
-$firstChatMessage = $firstChat->submit(Message::ofUser('What is my name?'));
+$chat->submit(Message::ofUser('My name is Christopher.'));
+$firstChatMessage = $chat->submit(Message::ofUser('What is my name?'));
 
-$secondChat->submit(Message::ofUser('My name is William.'));
-$secondChatMessage = $secondChat->submit(Message::ofUser('What is my name?'));
+$forkedChat->submit(Message::ofUser('My name is William.'));
+$secondChatMessage = $forkedChat->submit(Message::ofUser('What is my name?'));
 
 $firstChatMessageContent = $firstChatMessage->content;
 $secondChatMessageContent = $secondChatMessage->content;

examples/misc/persistent-chat.php

@@ -10,8 +10,8 @@
  */
 
 use Symfony\AI\Agent\Agent;
-use Symfony\AI\Agent\Chat;
-use Symfony\AI\Agent\Chat\MessageStore\InMemoryStore;
+use Symfony\AI\Chat\Bridge\Local\InMemoryStore;
+use Symfony\AI\Chat\Chat;
 use Symfony\AI\Platform\Bridge\OpenAi\Gpt;
 use Symfony\AI\Platform\Bridge\OpenAi\PlatformFactory;
 use Symfony\AI\Platform\Message\Message;

src/agent/composer.json

@@ -24,6 +24,7 @@
         "phpdocumentor/reflection-docblock": "^5.4",
         "phpstan/phpdoc-parser": "^2.1",
         "psr/log": "^3.0",
+        "symfony/ai-chat": "@dev",
         "symfony/ai-platform": "@dev",
         "symfony/clock": "^7.3|^8.0",
         "symfony/http-client": "^7.3|^8.0",

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": "^7.3|^8.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\Memory\MemoryInputProcessor;
@@ -40,6 +34,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;
 
 /**