4646use Symfony \AI \Chat \Bridge \SurrealDb \MessageStore as SurrealDbMessageStore ;
4747use Symfony \AI \Chat \Chat ;
4848use Symfony \AI \Chat \ChatInterface ;
49+ use Symfony \AI \Chat \ManagedStoreInterface as ManagedMessageStoreInterface ;
4950use Symfony \AI \Chat \MessageStoreInterface ;
5051use Symfony \AI \Platform \Bridge \Albert \PlatformFactory as AlbertPlatformFactory ;
5152use Symfony \AI \Platform \Bridge \Anthropic \PlatformFactory as AnthropicPlatformFactory ;
@@ -1598,6 +1599,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
15981599 ->setLazy (true )
15991600 ->setArguments ($ arguments )
16001601 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1602+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
16011603 ->addTag ('ai.message_store ' );
16021604
16031605 $ container ->setDefinition ('ai.message_store. ' .$ type .'. ' .$ name , $ definition );
@@ -1618,6 +1620,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
16181620 new Reference ('serializer ' ),
16191621 ])
16201622 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1623+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
16211624 ->addTag ('ai.message_store ' );
16221625
16231626 $ container ->setDefinition ('ai.message_store. ' .$ type .'.dbal. ' .$ name , $ definition );
@@ -1639,6 +1642,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
16391642 new Reference ('serializer ' ),
16401643 ])
16411644 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1645+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
16421646 ->addTag ('ai.message_store ' );
16431647
16441648 $ container ->setDefinition ('ai.message_store. ' .$ type .'. ' .$ name , $ definition );
@@ -1654,6 +1658,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
16541658 ->setLazy (true )
16551659 ->setArgument (0 , $ messageStore ['identifier ' ])
16561660 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1661+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
16571662 ->addTag ('ai.message_store ' );
16581663
16591664 $ container ->setDefinition ('ai.message_store. ' .$ type .'. ' .$ name , $ definition );
@@ -1674,6 +1679,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
16741679 new Reference ('serializer ' ),
16751680 ])
16761681 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1682+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
16771683 ->addTag ('ai.message_store ' );
16781684
16791685 $ container ->setDefinition ('ai.message_store. ' .$ type .'. ' .$ name , $ definition );
@@ -1695,6 +1701,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
16951701 new Reference ('serializer ' ),
16961702 ])
16971703 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1704+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
16981705 ->addTag ('ai.message_store ' );
16991706
17001707 $ container ->setDefinition ('ai.message_store. ' .$ type .'. ' .$ name , $ definition );
@@ -1721,6 +1728,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
17211728 new Reference ('serializer ' ),
17221729 ])
17231730 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1731+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
17241732 ->addTag ('ai.message_store ' );
17251733
17261734 $ container ->setDefinition ('ai.message_store. ' .$ type .'. ' .$ name , $ definition );
@@ -1739,6 +1747,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
17391747 $ messageStore ['identifier ' ],
17401748 ])
17411749 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1750+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
17421751 ->addTag ('ai.message_store ' );
17431752
17441753 $ container ->setDefinition ('ai.message_store. ' .$ type .'. ' .$ name , $ definition );
@@ -1768,6 +1777,7 @@ private function processMessageStoreConfig(string $type, array $messageStores, C
17681777 $ definition
17691778 ->setLazy (true )
17701779 ->addTag ('proxy ' , ['interface ' => MessageStoreInterface::class])
1780+ ->addTag ('proxy ' , ['interface ' => ManagedMessageStoreInterface::class])
17711781 ->addTag ('ai.message_store ' )
17721782 ->setArguments ($ arguments );
17731783
0 commit comments