ref

Author: Guikingone

.github/workflows/integration-tests.yaml

@@ -57,6 +57,14 @@ jobs:
                     MEILI_MASTER_KEY: '${MEILISEARCH_MASTER_KEY:-changeMe}'
                 ports:
                     - '7700:7700'
+    
+            mongodb:
+                image: mongodb/mongodb-community-server:7.0-ubi9
+                env:
+                    MONGO_INITDB_ROOT_USERNAME: 'symfony'
+                    MONGO_INITDB_ROOT_PASSWORD: 'symfony'
+                ports:
+                    - '27017:27017'
 
             qdrant:
                 image: qdrant/qdrant

examples/.env

@@ -60,8 +60,8 @@ BRAVE_API_KEY=
 FIRECRAWL_HOST=https://api.firecrawl.dev
 FIRECRAWL_API_KEY=
 
-# For using MongoDB Atlas (store)
-MONGODB_URI=
+# For using MongoDB Atlas / Community Edition (store)
+MONGODB_URI=mongodb://symfony:[email protected]:27017
 
 # For using Pinecone (store)
 PINECONE_API_KEY=

examples/commands/stores.php

@@ -0,0 +1,107 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+require_once dirname(__DIR__).'/bootstrap.php';
+
+use Doctrine\DBAL\DriverManager;
+use Doctrine\DBAL\Tools\DsnParser;
+use MongoDB\Client as MongoDbClient;
+use Symfony\AI\Store\Bridge\Local\CacheStore;
+use Symfony\AI\Store\Bridge\Local\InMemoryStore;
+use Symfony\AI\Store\Bridge\MariaDb\Store as MariaDbStore;
+use Symfony\AI\Store\Bridge\Milvus\Store as MilvusStore;
+use Symfony\AI\Store\Bridge\Meilisearch\Store as MeilisearchStore;
+use Symfony\AI\Store\Bridge\MongoDb\Store as MongoDbStore;
+use Symfony\AI\Store\Bridge\Postgres\Store as PostgresStore;
+use Symfony\AI\Store\Bridge\Qdrant\Store as QdrantStore;
+use Symfony\AI\Store\Bridge\SurrealDb\Store as SurrealDbStore;
+use Symfony\AI\Store\Bridge\Typesense\Store as TypesenseStore;
+use Symfony\AI\Store\Command\DropStoreCommand;
+use Symfony\AI\Store\Command\SetupStoreCommand;
+use Symfony\Component\Cache\Adapter\ArrayAdapter;
+use Symfony\Component\Console\Application;
+use Symfony\Component\Console\Input\ArrayInput;
+use Symfony\Component\Console\Output\ConsoleOutput;
+use Symfony\Component\DependencyInjection\ServiceLocator;
+
+$factories = [
+    'cache' => static fn (): CacheStore => new CacheStore(new ArrayAdapter()),
+    'mariadb' => static fn (): MariaDbStore => MariaDbStore::fromDbal(
+        connection: DriverManager::getConnection((new DsnParser())->parse(env('MARIADB_URI'))),
+        tableName: 'my_table_openai',
+        indexName: 'my_index',
+    ),
+    'memory' => static fn (): InMemoryStore => new InMemoryStore(),
+    'meilisearch' => static fn (): MeilisearchStore => new MeilisearchStore(
+        httpClient: http_client(),
+        endpointUrl: env('MEILISEARCH_HOST'),
+        apiKey: env('MEILISEARCH_API_KEY'),
+        indexName: '_commands',
+    ),
+    'milvus' => static fn (): MilvusStore => new MilvusStore(
+        httpClient: http_client(),
+        endpointUrl: env('MILVUS_HOST'),
+        apiKey: env('MILVUS_API_KEY'),
+        database: env('MILVUS_DATABASE'),
+        collection: '_commands',
+    ),
+    'mongodb' => static fn (): MongoDbStore => new MongoDbStore(
+        client: new MongoDbClient(env('MONGODB_URI')),
+        databaseName: 'my-database',
+        collectionName: 'my-collection',
+        indexName: 'my-index',
+        vectorFieldName: 'vector',
+    ),
+    'postgres' => static fn (): PostgresStore => PostgresStore::fromDbal(
+        connection: DriverManager::getConnection((new DsnParser())->parse(env('POSTGRES_URI'))),
+        tableName: 'my_table',
+    ),
+    'qdrant' => static fn (): QdrantStore => new QdrantStore(
+        http_client(),
+        env('QDRANT_HOST'),
+        env('QDRANT_SERVICE_API_KEY'),
+        '_commands',
+    ),
+    'surrealdb' => static fn (): SurrealDbStore => new SurrealDbStore(
+        httpClient: http_client(),
+        endpointUrl: env('SURREALDB_HOST'),
+        user: env('SURREALDB_USER'),
+        password: env('SURREALDB_PASS'),
+        namespace: 'default',
+        database: '_commands',
+        table: '_commands',
+    ),
+    'typesense' => static fn (): TypesenseStore => new TypesenseStore(
+        httpClient: http_client(),
+        endpointUrl: env('TYPESENSE_HOST'),
+        apiKey: env('TYPESENSE_API_KEY'),
+        collection: '_commands',
+    ),
+];
+
+$storesIds = array_keys($factories);
+
+$application = new Application();
+$application->setAutoExit(false);
+$application->add(new SetupStoreCommand(new ServiceLocator($factories), $storesIds));
+$application->add(new DropStoreCommand(new ServiceLocator($factories), $storesIds));
+
+foreach ($storesIds as $store) {
+    $application->run(new ArrayInput([
+        'command' => 'ai:store:setup',
+        'store' => $store,
+    ]), new ConsoleOutput());
+
+    $application->run(new ArrayInput([
+        'command' => 'ai:store:drop',
+        'store' => $store,
+    ]), new ConsoleOutput());
+}

examples/compose.yaml

@@ -23,6 +23,14 @@ services:
       ports:
           - '7700:7700'
 
+  mongodb:
+      image: mongodb/mongodb-community-server:7.0-ubi9
+      environment:
+          MONGO_INITDB_ROOT_USERNAME: 'symfony'
+          MONGO_INITDB_ROOT_PASSWORD: 'symfony'
+      ports:
+          - '27017:27017'
+
   qdrant:
       image: qdrant/qdrant
       environment:
@@ -67,7 +75,7 @@ services:
       ports:
           - '8001:8000'
 
-    # Milvus services
+  # Milvus services
   etcd:
       container_name: milvus-etcd
       image: quay.io/coreos/etcd:v3.5.18
@@ -91,9 +99,6 @@ services:
       environment:
           MINIO_ACCESS_KEY: minioadmin
           MINIO_SECRET_KEY: minioadmin
-      ports:
-          - '9001:9001'
-          - '9000:9000'
       volumes:
           - minio_vlm:/minio_data
       command: minio server /minio_data --console-address ":9001"

examples/composer.json

@@ -38,5 +38,9 @@
             "php-http/discovery": true
         },
         "sort-packages": true
+    },
+    "require-dev": {
+        "mongodb/mongodb": "^2.1",
+        "symfony/dependency-injection": "^7.3"
     }
 }

examples/rag/meilisearch.php

@@ -24,14 +24,13 @@
 use Symfony\AI\Store\Document\TextDocument;
 use Symfony\AI\Store\Document\Vectorizer;
 use Symfony\AI\Store\Indexer;
-use Symfony\Component\HttpClient\HttpClient;
 use Symfony\Component\Uid\Uuid;
 
 require_once dirname(__DIR__).'/bootstrap.php';
 
 // initialize the store
 $store = new Store(
-    httpClient: HttpClient::create(),
+    httpClient: http_client(),
     endpointUrl: env('MEILISEARCH_HOST'),
     apiKey: env('MEILISEARCH_API_KEY'),
     indexName: 'movies',

examples/rag/milvus.php

@@ -38,7 +38,7 @@
 );
 
 // initialize the index
-$store->initialize();
+$store->setup();
 
 // create embeddings and documents
 $documents = [];

examples/rag/neo4j.php

@@ -24,14 +24,13 @@
 use Symfony\AI\Store\Document\TextDocument;
 use Symfony\AI\Store\Document\Vectorizer;
 use Symfony\AI\Store\Indexer;
-use Symfony\Component\HttpClient\HttpClient;
 use Symfony\Component\Uid\Uuid;
 
 require_once dirname(__DIR__).'/bootstrap.php';
 
 // initialize the store
 $store = new Store(
-    httpClient: HttpClient::create(),
+    httpClient: http_client(),
     endpointUrl: env('NEO4J_HOST'),
     username: env('NEO4J_USERNAME'),
     password: env('NEO4J_PASSWORD'),

examples/rag/qdrant.php

@@ -24,14 +24,13 @@
 use Symfony\AI\Store\Document\TextDocument;
 use Symfony\AI\Store\Document\Vectorizer;
 use Symfony\AI\Store\Indexer;
-use Symfony\Component\HttpClient\HttpClient;
 use Symfony\Component\Uid\Uuid;
 
 require_once dirname(__DIR__).'/bootstrap.php';
 
 // initialize the store
 $store = new Store(
-    HttpClient::create(),
+    http_client(),
     env('QDRANT_HOST'),
     env('QDRANT_SERVICE_API_KEY'),
     'movies',

examples/rag/surrealdb.php

@@ -24,14 +24,13 @@
 use Symfony\AI\Store\Document\TextDocument;
 use Symfony\AI\Store\Document\Vectorizer;
 use Symfony\AI\Store\Indexer;
-use Symfony\Component\HttpClient\HttpClient;
 use Symfony\Component\Uid\Uuid;
 
 require_once dirname(__DIR__).'/bootstrap.php';
 
 // initialize the store
 $store = new Store(
-    httpClient: HttpClient::create(),
+    httpClient: http_client(),
     endpointUrl: env('SURREALDB_HOST'),
     user: env('SURREALDB_USER'),
     password: env('SURREALDB_PASS'),