* added explicit ollama version

* added getEmbeddingDimension for getting dimension of the current embedding model

Author: ArnabChatterjee20k

docker-compose.yml

@@ -25,7 +25,7 @@ services:
       - utopia
 
   ollama:
-    build: 
+    build:
       context: .
       dockerfile: ollama.dockerfile
     container_name: ollama

ollama.dockerfile

@@ -1,4 +1,4 @@
-FROM ollama/ollama:latest
+FROM ollama/ollama:0.12.7
 
 # Preload specific models
 ENV MODELS="embeddinggemma"

src/Agents/Adapter.php

@@ -108,6 +108,11 @@ abstract public function getSupportForEmbeddings(): bool;
      */
     abstract public function embed(string $text): array;
 
+    /**
+     * get embedding dimenion of the current model
+     */
+    abstract public function getEmbeddingDimension(): int;
+
     /**
      * Format error message
      *

src/Agents/Adapters/Anthropic.php

@@ -467,4 +467,9 @@ public function embed(string $text): array
     {
         throw new \Exception('Embeddings are not supported for this adapter.');
     }
+
+    public function getEmbeddingDimension(): int
+    {
+        throw new \Exception('Embeddings are not supported for this adapter.');
+    }
 }

src/Agents/Adapters/Deepseek.php

@@ -327,4 +327,9 @@ public function embed(string $text): array
     {
         throw new \Exception('Embeddings are not supported for this adapter.');
     }
+
+    public function getEmbeddingDimension(): int
+    {
+        throw new \Exception('Embeddings are not supported for this adapter.');
+    }
 }

src/Agents/Adapters/Gemini.php

@@ -358,4 +358,9 @@ public function embed(string $text): array
     {
         throw new \Exception('Embeddings are not supported for this adapter.');
     }
+
+    public function getEmbeddingDimension(): int
+    {
+        throw new \Exception('Embeddings are not supported for this adapter.');
+    }
 }

src/Agents/Adapters/Ollama.php

@@ -27,6 +27,13 @@ class Ollama extends Adapter
 
     public const MODELS = [self::MODEL_EMBEDDING_GEMMA];
 
+    /**
+     * Embedding dimensions of specific embedding model
+     */
+    protected const DIMENSIONS = [
+        self::MODEL_EMBEDDING_GEMMA => 768,
+    ];
+
     /**
      * Create a new Ollama adapter (no API key required for local call)
      *
@@ -101,6 +108,14 @@ public function getModel(): string
         return $this->model;
     }
 
+    /**
+     * get embedding dimenion of the current model
+     */
+    public function getEmbeddingDimension(): int
+    {
+        return self::DIMENSIONS[$this->model];
+    }
+
     /**
      * Set model to use for embedding
      *

src/Agents/Adapters/OpenAI.php

@@ -424,4 +424,9 @@ public function embed(string $text): array
     {
         throw new \Exception('Embeddings are not supported for this adapter.');
     }
+
+    public function getEmbeddingDimension(): int
+    {
+        throw new \Exception('Embeddings are not supported for this adapter.');
+    }
 }

tests/Agents/AgentTest.php

@@ -108,4 +108,31 @@ public function testEmbedReturnsArrayWithEmbeddingAdapter(): void
         $this->assertArrayHasKey('load_duration', $result);
         $this->assertIsArray($result['embedding']);
     }
+
+    public function testEmbeddingDImensions(): void
+    {
+        $ollama = new Ollama();
+        $agent = new Agent($ollama);
+        $content = [
+            'hi',
+            'hello world',
+            'this is a short sentence for testing',
+            'In the midst of chaos, there is also opportunity — this simple truth often reveals itself when we least expect it.',
+            'Artificial intelligence models like Ollama generate embeddings that map words, phrases, and even full documents into high-dimensional vector spaces where semantic similarity can be computed efficiently.', // long paragraph (~35 words)
+            str_repeat('This is a repeated pattern. ', 100),
+        ];
+        foreach ($content as $text) {
+            $result = $agent->embed($text);
+            $this->assertIsArray($result);
+            $this->assertArrayHasKey('embedding', $result);
+            $this->assertArrayHasKey('total_duration', $result);
+            $this->assertArrayHasKey('load_duration', $result);
+            $this->assertIsArray($result['embedding']);
+
+            $embedding = $result['embedding'];
+            $dimension = count($embedding);
+
+            $this->assertEquals($ollama->getEmbeddingDimension(), $dimension);
+        }
+    }
 }