Skip to content

Commit 2b7c5cd

Browse files
GuikingoneGuikingone
committed
refactor(store): WeaviateStoreFactory introduced
1 parent c348003 commit 2b7c5cd

File tree

12 files changed

+182
-111
lines changed

12 files changed

+182
-111
lines changed

‎UPGRADE.md‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,14 @@
1+
UPGRADE FROM 0.5 to 0.6
2+
=======================
3+
4+
Store
5+
-----
6+
7+
* The `endpointUrl` parameter for `QdrantStore` has been removed
8+
* The `apiKey` parameter for `QdrantStore` has been removed
9+
* The `endpointUrl` parameter for `WeaviateStore` has been removed
10+
* The `apiKey` parameter for `WeaviateStore` has been removed
11+
112
UPGRADE FROM 0.4 to 0.5
213
=======================
314

‎examples/rag/weaviate.php‎

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
use Symfony\AI\Platform\Bridge\OpenAi\PlatformFactory;
1818
use Symfony\AI\Platform\Message\Message;
1919
use Symfony\AI\Platform\Message\MessageBag;
20-
use Symfony\AI\Store\Bridge\Weaviate\Store;
20+
use Symfony\AI\Store\Bridge\Weaviate\StoreFactory;
2121
use Symfony\AI\Store\Document\Metadata;
2222
use Symfony\AI\Store\Document\TextDocument;
2323
use Symfony\AI\Store\Document\Vectorizer;
@@ -28,12 +28,7 @@
2828
require_once dirname(__DIR__).'/bootstrap.php';
2929

3030
// initialize the store
31-
$store = new Store(
32-
httpClient: http_client(),
33-
endpointUrl: env('WEAVIATE_HOST'),
34-
apiKey: env('WEAVIATE_API_KEY'),
35-
collection: 'Movies',
36-
);
31+
$store = StoreFactory::create('Movies', env('WEAVIATE_HOST'), env('WEAVIATE_API_KEY'));
3732

3833
// initialize the index
3934
$store->setup();

‎src/ai-bundle/CHANGELOG.md‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ CHANGELOG
88
* Add `TraceableAgent`
99
* Add `TraceableStore`
1010
* Add `setup_options` configuration for PostgreSQL store to pass extra fields to `ai:store:setup`
11+
* Add support for custom HttpClient for `Qdrant` and `Weaviate` stores
1112

1213
0.5
1314
---

‎src/ai-bundle/config/store/weaviate.php‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,13 @@
1717
->useAttributeAsKey('name')
1818
->arrayPrototype()
1919
->children()
20-
->stringNode('endpoint')->cannotBeEmpty()->end()
21-
->stringNode('api_key')->isRequired()->end()
22-
->stringNode('collection')->end()
20+
->stringNode('endpoint')->end()
21+
->stringNode('api_key')->end()
22+
->stringNode('http_client')
23+
->defaultValue('http_client')
24+
->end()
25+
->stringNode('collection')
26+
->info('The name of the store will be used if the "collection" is not set')
27+
->end()
2328
->end()
2429
->end();

‎src/ai-bundle/src/AiBundle.php‎

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@
113113
use Symfony\AI\Store\Bridge\Typesense\Store as TypesenseStore;
114114
use Symfony\AI\Store\Bridge\Vektor\Store as VektorStore;
115115
use Symfony\AI\Store\Bridge\Weaviate\Store as WeaviateStore;
116+
use Symfony\AI\Store\Bridge\Weaviate\StoreFactory as WeaviateStoreFactory;
116117
use Symfony\AI\Store\Distance\DistanceCalculator;
117118
use Symfony\AI\Store\Distance\DistanceStrategy;
118119
use Symfony\AI\Store\Document\Vectorizer;
@@ -2005,14 +2006,14 @@ private function processStoreConfig(string $type, array $stores, ContainerBuilde
20052006
}
20062007

20072008
foreach ($stores as $name => $store) {
2008-
$definition = new Definition(WeaviateStore::class);
2009-
$definition
2009+
$definition = (new Definition(WeaviateStore::class))
2010+
->setFactory(WeaviateStoreFactory::class.'::create')
20102011
->setLazy(true)
20112012
->setArguments([
2012-
new Reference('http_client'),
2013+
$store['collection'] ?? $name,
20132014
$store['endpoint'],
20142015
$store['api_key'],
2015-
$store['collection'] ?? $name,
2016+
new Reference($store['http_client']),
20162017
])
20172018
->addTag('proxy', ['interface' => StoreInterface::class])
20182019
->addTag('proxy', ['interface' => ManagedStoreInterface::class])

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

Lines changed: 53 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@
7474
use Symfony\AI\Store\Bridge\Typesense\Store as TypesenseStore;
7575
use Symfony\AI\Store\Bridge\Vektor\Store as VektorStore;
7676
use Symfony\AI\Store\Bridge\Weaviate\Store as WeaviateStore;
77+
use Symfony\AI\Store\Bridge\Weaviate\StoreFactory as WeaviateStoreFactory;
7778
use Symfony\AI\Store\Distance\DistanceCalculator;
7879
use Symfony\AI\Store\Distance\DistanceStrategy;
7980
use Symfony\AI\Store\Document\Filter\TextContainsFilter;
@@ -3521,7 +3522,7 @@ public function testTypesenseStoreWithCustomCollectionCanBeConfigured()
35213522
$this->assertTrue($container->hasAlias(StoreInterface::class));
35223523
}
35233524

3524-
public function testWevaviateStoreCanBeConfigured()
3525+
public function testWeaviateStoreCanBeConfigured()
35253526
{
35263527
$container = $this->buildContainer([
35273528
'ai' => [
@@ -3540,14 +3541,15 @@ public function testWevaviateStoreCanBeConfigured()
35403541

35413542
$definition = $container->getDefinition('ai.store.weaviate.my_weaviate_store');
35423543
$this->assertSame(WeaviateStore::class, $definition->getClass());
3543-
3544+
$this->assertSame([WeaviateStoreFactory::class, 'create'], $definition->getFactory());
35443545
$this->assertTrue($definition->isLazy());
3546+
35453547
$this->assertCount(4, $definition->getArguments());
3546-
$this->assertInstanceOf(Reference::class, $definition->getArgument(0));
3547-
$this->assertSame('http_client', (string) $definition->getArgument(0));
3548+
$this->assertSame('my_weaviate_store', $definition->getArgument(0));
35483549
$this->assertSame('http://localhost:8080', $definition->getArgument(1));
35493550
$this->assertSame('bar', $definition->getArgument(2));
3550-
$this->assertSame('my_weaviate_store', $definition->getArgument(3));
3551+
$this->assertInstanceOf(Reference::class, $definition->getArgument(3));
3552+
$this->assertSame('http_client', (string) $definition->getArgument(3));
35513553

35523554
$this->assertTrue($definition->hasTag('proxy'));
35533555
$this->assertSame([
@@ -3561,10 +3563,7 @@ public function testWevaviateStoreCanBeConfigured()
35613563
$this->assertTrue($container->hasAlias('.'.StoreInterface::class.' $weaviate_my_weaviate_store'));
35623564
$this->assertTrue($container->hasAlias(StoreInterface::class.' $weaviateMyWeaviateStore'));
35633565
$this->assertTrue($container->hasAlias(StoreInterface::class));
3564-
}
35653566

3566-
public function testWevaviateStoreWithCustomCollectionCanBeConfigured()
3567-
{
35683567
$container = $this->buildContainer([
35693568
'ai' => [
35703569
'store' => [
@@ -3582,15 +3581,58 @@ public function testWevaviateStoreWithCustomCollectionCanBeConfigured()
35823581
$this->assertTrue($container->hasDefinition('ai.store.weaviate.my_weaviate_store'));
35833582

35843583
$definition = $container->getDefinition('ai.store.weaviate.my_weaviate_store');
3584+
$this->assertSame([WeaviateStoreFactory::class, 'create'], $definition->getFactory());
35853585
$this->assertSame(WeaviateStore::class, $definition->getClass());
3586+
$this->assertTrue($definition->isLazy());
3587+
3588+
$this->assertCount(4, $definition->getArguments());
3589+
$this->assertSame('my_weaviate_collection', $definition->getArgument(0));
3590+
$this->assertSame('http://localhost:8080', $definition->getArgument(1));
3591+
$this->assertSame('bar', $definition->getArgument(2));
3592+
$this->assertInstanceOf(Reference::class, $definition->getArgument(3));
3593+
$this->assertSame('http_client', (string) $definition->getArgument(3));
3594+
3595+
$this->assertTrue($definition->hasTag('proxy'));
3596+
$this->assertSame([
3597+
['interface' => StoreInterface::class],
3598+
['interface' => ManagedStoreInterface::class],
3599+
], $definition->getTag('proxy'));
3600+
$this->assertTrue($definition->hasTag('ai.store'));
3601+
3602+
$this->assertTrue($container->hasAlias('.'.StoreInterface::class.' $my_weaviate_store'));
3603+
$this->assertTrue($container->hasAlias(StoreInterface::class.' $myWeaviateStore'));
3604+
$this->assertTrue($container->hasAlias('.'.StoreInterface::class.' $weaviate_my_weaviate_store'));
3605+
$this->assertTrue($container->hasAlias(StoreInterface::class.' $weaviateMyWeaviateStore'));
3606+
$this->assertTrue($container->hasAlias(StoreInterface::class));
3607+
3608+
$container = $this->buildContainer([
3609+
'ai' => [
3610+
'store' => [
3611+
'weaviate' => [
3612+
'my_weaviate_store' => [
3613+
'endpoint' => 'http://localhost:8080',
3614+
'api_key' => 'bar',
3615+
'collection' => 'my_weaviate_collection',
3616+
'http_client' => 'scoped_http_client',
3617+
],
3618+
],
3619+
],
3620+
],
3621+
]);
35863622

3623+
$this->assertTrue($container->hasDefinition('ai.store.weaviate.my_weaviate_store'));
3624+
3625+
$definition = $container->getDefinition('ai.store.weaviate.my_weaviate_store');
3626+
$this->assertSame([WeaviateStoreFactory::class, 'create'], $definition->getFactory());
3627+
$this->assertSame(WeaviateStore::class, $definition->getClass());
35873628
$this->assertTrue($definition->isLazy());
3629+
35883630
$this->assertCount(4, $definition->getArguments());
3589-
$this->assertInstanceOf(Reference::class, $definition->getArgument(0));
3590-
$this->assertSame('http_client', (string) $definition->getArgument(0));
3631+
$this->assertSame('my_weaviate_collection', $definition->getArgument(0));
35913632
$this->assertSame('http://localhost:8080', $definition->getArgument(1));
35923633
$this->assertSame('bar', $definition->getArgument(2));
3593-
$this->assertSame('my_weaviate_collection', $definition->getArgument(3));
3634+
$this->assertInstanceOf(Reference::class, $definition->getArgument(3));
3635+
$this->assertSame('scoped_http_client', (string) $definition->getArgument(3));
35943636

35953637
$this->assertTrue($definition->hasTag('proxy'));
35963638
$this->assertSame([

‎src/store/src/Bridge/Weaviate/CHANGELOG.md‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,12 @@
11
CHANGELOG
22
=========
33

4+
0.6
5+
---
6+
7+
* [BC BREAK] The `endpointUrl` parameter for `Store` has been removed
8+
* [BC BREAK] The `apiKey` parameter for `Store` has been removed
9+
410
0.1
511
---
612

‎src/store/src/Bridge/Weaviate/Store.php‎

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,6 @@ final class Store implements ManagedStoreInterface, StoreInterface
3030
{
3131
public function __construct(
3232
private readonly HttpClientInterface $httpClient,
33-
private readonly string $endpointUrl,
34-
#[\SensitiveParameter] private readonly string $apiKey,
3533
private readonly string $collection,
3634
) {
3735
}
@@ -137,17 +135,11 @@ public function drop(array $options = []): void
137135
*/
138136
private function request(string $method, string $endpoint, array $payload): array
139137
{
140-
$url = \sprintf('%s/%s', $this->endpointUrl, $endpoint);
141-
142-
$finalPayload = [
143-
'auth_bearer' => $this->apiKey,
144-
];
145-
146138
if ([] !== $payload) {
147139
$finalPayload['json'] = $payload;
148140
}
149141

150-
$response = $this->httpClient->request($method, $url, $finalPayload);
142+
$response = $this->httpClient->request($method, $endpoint, $finalPayload ?? []);
151143

152144
if (200 === $response->getStatusCode() && '' === $response->getContent(false)) {
153145
return [];

‎src/store/src/Bridge/Weaviate/StoreFactory.php‎

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
<?php
2+
3+
/*
4+
* This file is part of the Symfony package.
5+
*
6+
* (c) Fabien Potencier <fabien@symfony.com>
7+
*
8+
* For the full copyright and license information, please view the LICENSE
9+
* file that was distributed with this source code.
10+
*/
11+
12+
namespace Symfony\AI\Store\Bridge\Weaviate;
13+
14+
use Symfony\AI\Store\ManagedStoreInterface;
15+
use Symfony\AI\Store\StoreInterface;
16+
use Symfony\Component\HttpClient\HttpClient;
17+
use Symfony\Component\HttpClient\ScopingHttpClient;
18+
use Symfony\Contracts\HttpClient\HttpClientInterface;
19+
20+
/**
21+
* @author Guillaume Loulier <personal@guillaumeloulier.fr>
22+
*/
23+
final class StoreFactory
24+
{
25+
public static function create(
26+
string $collection,
27+
?string $endpoint = null,
28+
#[\SensitiveParameter] ?string $apiKey = null,
29+
?HttpClientInterface $httpClient = null,
30+
): StoreInterface&ManagedStoreInterface {
31+
if (null !== $endpoint) {
32+
$defaultOptions = [];
33+
if (null !== $apiKey) {
34+
$defaultOptions['auth_bearer'] = $apiKey;
35+
}
36+
37+
$httpClient = ScopingHttpClient::forBaseUri($httpClient ?? HttpClient::create(), $endpoint, $defaultOptions);
38+
}
39+
40+
return new Store($httpClient, $collection);
41+
}
42+
}

‎src/store/src/Bridge/Weaviate/Tests/IntegrationTest.php‎

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,9 @@
1212
namespace Symfony\AI\Store\Bridge\Weaviate\Tests;
1313

1414
use PHPUnit\Framework\Attributes\Group;
15-
use Symfony\AI\Store\Bridge\Weaviate\Store;
15+
use Symfony\AI\Store\Bridge\Weaviate\StoreFactory;
1616
use Symfony\AI\Store\StoreInterface;
1717
use Symfony\AI\Store\Test\AbstractStoreIntegrationTestCase;
18-
use Symfony\Component\HttpClient\HttpClient;
1918

2019
/**
2120
* @author Christopher Hertel <mail@christopher-hertel.de>
@@ -25,11 +24,6 @@ final class IntegrationTest extends AbstractStoreIntegrationTestCase
2524
{
2625
protected static function createStore(): StoreInterface
2726
{
28-
return new Store(
29-
HttpClient::create(),
30-
'http://127.0.0.1:8080',
31-
'symfony',
32-
'TestCollection',
33-
);
27+
return StoreFactory::create('TestCollection', 'http://127.0.0.1:8080', 'symfony');
3428
}
3529
}

0 commit comments

Comments
 (0)