rebase

Author: Guikingone

examples/.env

@@ -92,6 +92,7 @@ SURREALDB_USER=symfony
 SURREALDB_PASS=symfony
 
 # Neo4J
-NEO4J_HOST=http://localhost:7474
+NEO4J_HOST=http://127.0.0.1:7474
+NEO4J_DATABASE=neo4j
 NEO4J_USERNAME=neo4j
 NEO4J_PASSWORD=symfonyai

examples/rag/neo4j-similarity-search.php

@@ -0,0 +1,75 @@
+<?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.
+ */
+
+use Symfony\AI\Agent\Agent;
+use Symfony\AI\Agent\Toolbox\AgentProcessor;
+use Symfony\AI\Agent\Toolbox\Tool\SimilaritySearch;
+use Symfony\AI\Agent\Toolbox\Toolbox;
+use Symfony\AI\Fixtures\Movies;
+use Symfony\AI\Platform\Bridge\OpenAI\Embeddings;
+use Symfony\AI\Platform\Bridge\OpenAI\GPT;
+use Symfony\AI\Platform\Bridge\OpenAI\PlatformFactory;
+use Symfony\AI\Platform\Message\Message;
+use Symfony\AI\Platform\Message\MessageBag;
+use Symfony\AI\Store\Bridge\Neo4j\Store;
+use Symfony\AI\Store\Document\Metadata;
+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(),
+    endpointUrl: env('NEO4J_HOST'),
+    username: env('NEO4J_USERNAME'),
+    password: env('NEO4J_PASSWORD'),
+    databaseName: env('NEO4J_DATABASE'),
+    vectorIndexName: 'Movies',
+    nodeName: 'movies',
+);
+
+// initialize the table
+$store->initialize();
+
+// create embeddings and documents
+$documents = [];
+foreach (Movies::all() as $i => $movie) {
+    $documents[] = new TextDocument(
+        id: Uuid::v4(),
+        content: 'Title: '.$movie['title'].\PHP_EOL.'Director: '.$movie['director'].\PHP_EOL.'Description: '.$movie['description'],
+        metadata: new Metadata($movie),
+    );
+}
+
+// create embeddings for documents
+$platform = PlatformFactory::create($_SERVER['OPENAI_API_KEY']);
+$vectorizer = new Vectorizer($platform, $embeddings = new Embeddings());
+$indexer = new Indexer($vectorizer, $store);
+$indexer->index($documents);
+
+$model = new GPT(GPT::GPT_4O_MINI);
+
+$similaritySearch = new SimilaritySearch($platform, $embeddings, $store);
+$toolbox = new Toolbox([$similaritySearch], logger: logger());
+$processor = new AgentProcessor($toolbox);
+$agent = new Agent($platform, $model, [$processor], [$processor]);
+
+$messages = new MessageBag(
+    Message::forSystem('Please answer all user questions only using SimilaritySearch function.'),
+    Message::ofUser('Which movie fits the theme of technology?')
+);
+$response = $agent->call($messages);
+
+echo $response->getContent().\PHP_EOL;

examples/rag/neo4j.php

@@ -82,7 +82,7 @@ public function testStoreCanInitialize(): void
         $store->initialize();
         $store->initialize();
 
-        self::assertSame(2, $httpClient->getRequestsCount());
+        $this->assertSame(2, $httpClient->getRequestsCount());
     }
 
     public function testStoreCanAdd(): void
@@ -161,7 +161,7 @@ public function testStoreCanAdd(): void
         $store->add(new VectorDocument(Uuid::v4(), new Vector([0.1, 0.2, 0.3])));
         $store->add(new VectorDocument(Uuid::v4(), new Vector([0.1, 0.2, 0.3])));
 
-        self::assertSame(3, $httpClient->getRequestsCount());
+        $this->assertSame(3, $httpClient->getRequestsCount());
     }
 
     public function testStoreCanQuery(): void
@@ -261,7 +261,7 @@ public function testStoreCanQuery(): void
 
         $results = $store->query(new Vector([0.1, 0.2, 0.3]));
 
-        self::assertCount(2, $results);
-        self::assertSame(3, $httpClient->getRequestsCount());
+        $this->assertCount(2, $results);
+        $this->assertSame(3, $httpClient->getRequestsCount());
     }
 }