rag methods

Author: ilvalerione

src/RAG/RAG.php

@@ -10,6 +10,10 @@
 
 class RAG extends Agent
 {
+    /**
+     * @var VectorStoreInterface
+     */
+    protected VectorStoreInterface $store;
 
     /**
      * The embeddings provider.
@@ -25,10 +29,6 @@ class RAG extends Agent
      */
     protected ?string $instructions = "Use the following pieces of context to answer the question of the user. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}.";
 
-    public function __construct(protected VectorStoreInterface $store) {
-        parent::__construct();
-    }
-
     public function answerQuestion(string $question, int $k = 4): Message
     {
         $this->notify('agent:vectorstore:searching', $question);
@@ -78,7 +78,7 @@ public function setSystemMessage(array $documents, int $k): self
     private function searchDocuments(string $question, int $k): array
     {
         $embedding = $this->embeddings()->embedText($question);
-        $docs = $this->store->similaritySearch($embedding, $k);
+        $docs = $this->vectorStore()->similaritySearch($embedding, $k);
 
         $retrievedDocs = [];
 
@@ -100,4 +100,14 @@ public function embeddings(): EmbeddingsProviderInterface
     {
         return $this->embeddingsProvider;
     }
+
+    public function setVectorStore(VectorStoreInterface $store)
+    {
+        $this->store = $store;
+    }
+
+    public function vectorStore(): VectorStoreInterface
+    {
+        return $this->store;
+    }
 }

src/RAG/VectorStore/PineconeVectorStore.php

@@ -12,7 +12,7 @@ class PineconeVectorStore implements VectorStoreInterface
 
     public function __construct(
         string $key,
-        protected string $indexName,
+        protected string $index,
         array $spec,
         string $version = '2025-01'
     ) {
@@ -26,7 +26,7 @@ public function __construct(
             ]
         ]);
 
-        $response = $this->client->get("indexes/{$this->indexName}");
+        $response = $this->client->get("indexes/{$this->index}");
 
         if ($response->getStatusCode() === 200) {
             return;
@@ -35,7 +35,7 @@ public function __construct(
         // Create the index
         $this->client->post('indexes', [
             RequestOptions::JSON => [
-                'name' => $indexName,
+                'name' => $index,
                 'spec' => $spec,
             ]
         ]);
@@ -48,7 +48,7 @@ public function addDocument(Document $document): void
 
     public function addDocuments(array $documents): void
     {
-        $this->client->post("indexes/{$this->indexName}/vectors/upsert", [
+        $this->client->post("indexes/{$this->index}/vectors/upsert", [
             RequestOptions::JSON => array_map(function (Document $document) {
                 return [
                     'id' => $document->id??uniqid(),
@@ -60,7 +60,7 @@ public function addDocuments(array $documents): void
 
     public function similaritySearch(array $embedding, int $k = 4): iterable
     {
-        $result = $this->client->get("indexes/{$this->indexName}/query", [
+        $result = $this->client->get("indexes/{$this->index}/query", [
             RequestOptions::QUERY => [
                 'namespace' => '',
                 'vector' => $embedding,