rag methods

Author: ilvalerione

src/Agent.php

@@ -3,7 +3,6 @@
 namespace NeuronAI;
 
 use NeuronAI\Providers\AIProviderInterface;
-use NeuronAI\Providers\Embeddings\EmbeddingsProviderInterface;
 use NeuronAI\Messages\Message;
 use NeuronAI\Messages\UserMessage;
 use NeuronAI\Tools\Tool;
@@ -19,13 +18,6 @@ class Agent implements \SplSubject
      */
     protected AIProviderInterface $provider;
 
-    /**
-     * The embeddings provider.
-     *
-     * @var EmbeddingsProviderInterface
-     */
-    protected EmbeddingsProviderInterface $embeddingsProvider;
-
     /**
      * @var AbstractChatHistory
      */
@@ -81,17 +73,6 @@ public function provider(): AIProviderInterface
         return $this->provider;
     }
 
-    public function setEmbeddingsProvider(EmbeddingsProviderInterface $provider): self
-    {
-        $this->embeddingsProvider = $provider;
-        return $this;
-    }
-
-    public function embeddings(): EmbeddingsProviderInterface
-    {
-        return $this->embeddingsProvider;
-    }
-
     /**
      * Execute the chat.
      *

src/RAG/RAG.php

@@ -5,10 +5,19 @@
 use NeuronAI\Agent;
 use NeuronAI\Messages\Message;
 use NeuronAI\Messages\UserMessage;
+use NeuronAI\Providers\Embeddings\EmbeddingsProviderInterface;
 use NeuronAI\RAG\VectorStore\VectorStoreInterface;
 
 class RAG extends Agent
 {
+
+    /**
+     * The embeddings provider.
+     *
+     * @var EmbeddingsProviderInterface
+     */
+    protected EmbeddingsProviderInterface $embeddingsProvider;
+
     /**
      * Instructions template.
      *
@@ -80,4 +89,15 @@ private function searchDocuments(string $question, int $k): array
 
         return \array_values($retrievedDocs);
     }
+
+    public function setEmbeddingsProvider(EmbeddingsProviderInterface $provider): self
+    {
+        $this->embeddingsProvider = $provider;
+        return $this;
+    }
+
+    public function embeddings(): EmbeddingsProviderInterface
+    {
+        return $this->embeddingsProvider;
+    }
 }