[Platform] Use new AsModel attribute to connect Model to ModelClient

Related: symfony#299

Author: ohader

src/platform/src/Attribute/AsModel.php

@@ -0,0 +1,23 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Platform\Attribute;
+
+#[\Attribute(\Attribute::TARGET_CLASS | \Attribute::IS_REPEATABLE)]
+
+final readonly class AsModel
+{
+    public function __construct(
+        public string $platform,
+        public ?string $aspect = null,
+    ) {
+    }
+}

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

@@ -11,12 +11,14 @@
 
 namespace Symfony\AI\Platform\Bridge\Ollama;
 
+use Symfony\AI\Platform\Attribute\AsModel;
 use Symfony\AI\Platform\Capability;
 use Symfony\AI\Platform\Model;
 
 /**
  * @author Joshua Behrens <[email protected]>
  */
+#[AsModel(platform: 'ollama', aspect: 'default')]
 class Ollama extends Model
 {
     public const DEEPSEEK_R_1 = 'deepseek-r1';

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

@@ -14,6 +14,7 @@
 use Symfony\AI\Platform\Exception\InvalidArgumentException;
 use Symfony\AI\Platform\Model;
 use Symfony\AI\Platform\ModelClientInterface;
+use Symfony\AI\Platform\ModelClientTrait;
 use Symfony\AI\Platform\Result\RawHttpResult;
 use Symfony\Contracts\HttpClient\HttpClientInterface;
 
@@ -22,6 +23,8 @@
  */
 final readonly class OllamaClient implements ModelClientInterface
 {
+    use ModelClientTrait;
+
     public function __construct(
         private HttpClientInterface $httpClient,
         private string $hostUrl,
@@ -30,7 +33,7 @@ public function __construct(
 
     public function supports(Model $model): bool
     {
-        return $model instanceof Ollama;
+        return $this->hasModelAttribute($model, 'ollama');
     }
 
     public function request(Model $model, array|string $payload, array $options = []): RawHttpResult

src/platform/src/ModelClientTrait.php

@@ -0,0 +1,29 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Platform;
+
+use Symfony\AI\Platform\Attribute\AsModel;
+
+trait ModelClientTrait
+{
+    private function hasModelAttribute(Model $model, string $platform): bool
+    {
+        $reflection = new \ReflectionObject($model);
+        $attributes = array_filter(
+            $reflection->getAttributes(AsModel::class),
+            static fn(\ReflectionAttribute $attribute) =>
+                ($attribute->getArguments()['platform'] ?? null) === $platform
+                || ($attribute->getArguments()[0] ?? null) === $platform
+        );
+        return $attributes !== [];
+    }
+}