diff --git a/src/platform/src/Bridge/Ollama/OllamaApiCatalog.php b/src/platform/src/Bridge/Ollama/OllamaApiCatalog.php
index 50c62b7d5..34a30df01 100644
--- a/src/platform/src/Bridge/Ollama/OllamaApiCatalog.php
+++ b/src/platform/src/Bridge/Ollama/OllamaApiCatalog.php
@@ -44,7 +44,7 @@ public function getModel(string $modelName): Ollama
         $capabilities = array_map(
             static fn (string $capability): Capability => match ($capability) {
                 'embedding' => Capability::EMBEDDINGS,
-                'completion' => Capability::INPUT_TEXT,
+                'completion' => Capability::INPUT_MESSAGES,
                 'tools' => Capability::TOOL_CALLING,
                 'thinking' => Capability::THINKING,
                 'vision' => Capability::INPUT_IMAGE,
diff --git a/src/platform/tests/Bridge/Ollama/OllamaApiCatalogTest.php b/src/platform/tests/Bridge/Ollama/OllamaApiCatalogTest.php
index 8f79a28a6..4377f5691 100644
--- a/src/platform/tests/Bridge/Ollama/OllamaApiCatalogTest.php
+++ b/src/platform/tests/Bridge/Ollama/OllamaApiCatalogTest.php
@@ -34,7 +34,7 @@ public function testModelCatalogCanReturnModelFromApi()
 
         $this->assertSame('foo', $model->getName());
         $this->assertSame([
-            Capability::INPUT_TEXT,
+            Capability::INPUT_MESSAGES,
         ], $model->getCapabilities());
         $this->assertSame(1, $httpClient->getRequestsCount());
     }
@@ -66,7 +66,7 @@ public function testModelCatalogCanReturnModelsFromApi()
         $this->assertSame(Ollama::class, $model['class']);
         $this->assertCount(1, $model['capabilities']);
         $this->assertSame([
-            Capability::INPUT_TEXT,
+            Capability::INPUT_MESSAGES,
         ], $model['capabilities']);
         $this->assertSame(2, $httpClient->getRequestsCount());
     }