ref(platform): OllamaCatalog

Author: Guikingone

src/platform/src/Bridge/Ollama/OllamaCatalog.php

@@ -0,0 +1,39 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Platform\Bridge\Ollama;
+
+use Symfony\AI\Platform\Model;
+use Symfony\AI\Platform\ModelCatalog\DynamicModelCatalog;
+use Symfony\Contracts\HttpClient\HttpClientInterface;
+
+final class OllamaCatalog extends DynamicModelCatalog
+{
+    public function __construct(
+        private readonly string $host,
+        private readonly HttpClientInterface $httpClient,
+    ) {
+        parent::__construct();
+    }
+
+    public function getModel(string $modelName): Model
+    {
+        $response = $this->httpClient->request('POST', \sprintf('%s/api/show', $this->host), [
+            'json' => [
+                'model' => $modelName,
+            ],
+        ]);
+
+        $payload = $response->toArray();
+
+        return new Ollama($modelName, $payload['capabilities']);
+    }
+}

src/platform/src/Bridge/Ollama/OllamaClient.php

@@ -35,21 +35,13 @@ public function supports(Model $model): bool
 
     public function request(Model $model, array|string $payload, array $options = []): RawHttpResult
     {
-        $response = $this->httpClient->request('POST', \sprintf('%s/api/show', $this->hostUrl), [
-            'json' => [
-                'model' => $model->getName(),
-            ],
-        ]);
-
-        $capabilities = $response->toArray()['capabilities'] ?? null;
-
-        if (null === $capabilities) {
+        if ([] === $model->getCapabilities()) {
             throw new InvalidArgumentException('The model information could not be retrieved from the Ollama API. Your Ollama server might be too old. Try upgrade it.');
         }
 
         return match (true) {
-            \in_array('completion', $capabilities, true) => $this->doCompletionRequest($payload, $options),
-            \in_array('embedding', $capabilities, true) => $this->doEmbeddingsRequest($model, $payload, $options),
+            \in_array('completion', $model->getCapabilities(), true) => $this->doCompletionRequest($payload, $options),
+            \in_array('embedding', $model->getCapabilities(), true) => $this->doEmbeddingsRequest($model, $payload, $options),
             default => throw new InvalidArgumentException(\sprintf('Unsupported model "%s": "%s".', $model::class, $model->getName())),
         };
     }

src/platform/src/Bridge/Ollama/PlatformFactory.php

@@ -13,7 +13,6 @@
 
 use Symfony\AI\Platform\Bridge\Ollama\Contract\OllamaContract;
 use Symfony\AI\Platform\Contract;
-use Symfony\AI\Platform\ModelCatalog\DynamicModelCatalog;
 use Symfony\AI\Platform\ModelCatalog\ModelCatalogInterface;
 use Symfony\AI\Platform\Platform;
 use Symfony\Component\HttpClient\EventSourceHttpClient;
@@ -27,15 +26,15 @@ final class PlatformFactory
     public static function create(
         string $hostUrl = 'http://localhost:11434',
         ?HttpClientInterface $httpClient = null,
-        ModelCatalogInterface $modelCatalog = new DynamicModelCatalog(Ollama::class),
+        ?ModelCatalogInterface $modelCatalog = null,
         ?Contract $contract = null,
     ): Platform {
         $httpClient = $httpClient instanceof EventSourceHttpClient ? $httpClient : new EventSourceHttpClient($httpClient);
 
         return new Platform(
             [new OllamaClient($httpClient, $hostUrl)],
             [new OllamaResultConverter()],
-            $modelCatalog,
+            $modelCatalog ?? new OllamaCatalog($hostUrl, $httpClient),
             $contract ?? OllamaContract::create(),
         );
     }

src/platform/src/ModelCatalog/DynamicModelCatalog.php

@@ -25,16 +25,15 @@
 
 class DynamicModelCatalog extends AbstractModelCatalog
 {
-    public function __construct(
-        private readonly ?string $expectedModel = Model::class,
-    ) {
+    public function __construct()
+    {
         $this->models = [];
     }
 
     public function getModel(string $modelName): Model
     {
         $parsed = self::parseModelName($modelName);
 
-        return new $this->expectedModel($parsed['name'], Capability::cases(), $parsed['options']);
+        return new Model($parsed['name'], Capability::cases(), $parsed['options']);
     }
 }