Skip to content

Commit 9d69408

Browse files
committed
bug #976 [AI Bundle] Add missing proxy tag added for message stores (Guikingone)
This PR was merged into the main branch. Discussion ---------- [AI Bundle] Add missing `proxy` tag added for message stores | Q | A | ------------- | --- | Bug fix? | yes | New feature? | no | Docs? | no | Issues | Related to #974 | License | MIT Commits ------- 381ecad fix(aibundle): missing proxy tag added for message stores
2 parents c24eaea + 381ecad commit 9d69408

File tree

2 files changed

+221
-179
lines changed

2 files changed

+221
-179
lines changed

src/ai-bundle/src/AiBundle.php

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
use Symfony\AI\Chat\Bridge\SurrealDb\MessageStore as SurrealDbMessageStore;
4747
use Symfony\AI\Chat\Chat;
4848
use Symfony\AI\Chat\ChatInterface;
49+
use Symfony\AI\Chat\ManagedStoreInterface as ManagedMessageStoreInterface;
4950
use Symfony\AI\Chat\MessageStoreInterface;
5051
use Symfony\AI\Platform\Bridge\Albert\PlatformFactory as AlbertPlatformFactory;
5152
use 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

Comments
 (0)