updated ollama agent

Author: ArnabChatterjee20k

src/Agents/Adapters/Ollama.php

@@ -23,6 +23,10 @@ class Ollama extends Adapter
      */
     protected int $timeout;
 
+    private string $endpoint = 'http://ollama:11434/api/embed';
+
+    public const MODELS = [self::MODEL_EMBEDDING_GEMMA];
+
     /**
      * Create a new Ollama adapter (no API key required for local call)
      *
@@ -60,7 +64,7 @@ public function embed(string $text): array
             'input' => $text,
         ];
         $response = $client->fetch(
-            'http://ollama:11434/api/embed',
+            $this->getEndpoint(),
             Client::METHOD_POST,
             $payload
         );
@@ -84,7 +88,7 @@ public function embed(string $text): array
      */
     public function getModels(): array
     {
-        return [self::MODEL_EMBEDDING_GEMMA];
+        return self::MODELS;
     }
 
     /**
@@ -159,6 +163,29 @@ protected function formatErrorMessage($json): string
         return '(ollama_error) '.$msg;
     }
 
+    /**
+     * Get the API endpoint
+     *
+     * @return string
+     */
+    public function getEndpoint(): string
+    {
+        return $this->endpoint;
+    }
+
+    /**
+     * Set the API endpoint
+     *
+     * @param  string  $endpoint
+     * @return self
+     */
+    public function setEndpoint(string $endpoint): self
+    {
+        $this->endpoint = $endpoint;
+
+        return $this;
+    }
+
     public function getSupportForEmbeddings(): bool
     {
         return true;