From 71d1656690347631a96db3f323f72f1dd498306b Mon Sep 17 00:00:00 2001
From: Eray Gundesli <eray@gundesli.com>
Date: Fri, 7 Nov 2025 02:07:06 +0100
Subject: [PATCH] [Platform] update Albert API models catalog (#710)

---
 .../src/Bridge/Albert/ModelCatalog.php         | 18 +++++++++++++++---
 .../tests/Bridge/Albert/ModelCatalogTest.php   |  4 +++-
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/platform/src/Bridge/Albert/ModelCatalog.php b/src/platform/src/Bridge/Albert/ModelCatalog.php
index 60db2691a..62e4833bf 100644
--- a/src/platform/src/Bridge/Albert/ModelCatalog.php
+++ b/src/platform/src/Bridge/Albert/ModelCatalog.php
@@ -27,14 +27,26 @@ final class ModelCatalog extends AbstractModelCatalog
     public function __construct(array $additionalModels = [])
     {
         $defaultModels = [
-            'gpt-4o' => [
+            'albert-small' => [
                 'class' => Gpt::class,
                 'capabilities' => [
                     Capability::INPUT_MESSAGES,
                     Capability::OUTPUT_TEXT,
                     Capability::OUTPUT_STREAMING,
-                    Capability::TOOL_CALLING,
-                    Capability::INPUT_IMAGE,
+                ],
+            ],
+            'albert-large' => [
+                'class' => Gpt::class,
+                'capabilities' => [
+                    Capability::INPUT_MESSAGES,
+                    Capability::OUTPUT_TEXT,
+                    Capability::OUTPUT_STREAMING,
+                ],
+            ],
+            'embeddings-small' => [
+                'class' => Gpt::class,
+                'capabilities' => [
+                    Capability::INPUT_MESSAGES,
                     Capability::OUTPUT_STRUCTURED,
                 ],
             ],
diff --git a/src/platform/tests/Bridge/Albert/ModelCatalogTest.php b/src/platform/tests/Bridge/Albert/ModelCatalogTest.php
index 846a99b19..d1eb41f7b 100644
--- a/src/platform/tests/Bridge/Albert/ModelCatalogTest.php
+++ b/src/platform/tests/Bridge/Albert/ModelCatalogTest.php
@@ -24,7 +24,9 @@ final class ModelCatalogTest extends ModelCatalogTestCase
 {
     public static function modelsProvider(): iterable
     {
-        yield 'gpt-4o' => ['gpt-4o', Gpt::class, [Capability::INPUT_MESSAGES, Capability::OUTPUT_TEXT, Capability::OUTPUT_STREAMING, Capability::TOOL_CALLING, Capability::INPUT_IMAGE, Capability::OUTPUT_STRUCTURED]];
+        yield 'albert-small' => ['albert-small', Gpt::class, [Capability::INPUT_MESSAGES, Capability::OUTPUT_TEXT, Capability::OUTPUT_STREAMING]];
+        yield 'albert-large' => ['albert-large', Gpt::class, [Capability::INPUT_MESSAGES, Capability::OUTPUT_TEXT, Capability::OUTPUT_STREAMING]];
+        yield 'embeddings-small' => ['embeddings-small', Gpt::class, [Capability::INPUT_MESSAGES, Capability::OUTPUT_STRUCTURED]];
     }
 
     protected function createModelCatalog(): ModelCatalogInterface