Skip to content

Commit 71d1656

Browse files
committed
[Platform] update Albert API models catalog (#710)
1 parent 68fbf80 commit 71d1656

File tree

2 files changed

+18
-4
lines changed

2 files changed

+18
-4
lines changed

src/platform/src/Bridge/Albert/ModelCatalog.php

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,26 @@ final class ModelCatalog extends AbstractModelCatalog
2727
public function __construct(array $additionalModels = [])
2828
{
2929
$defaultModels = [
30-
'gpt-4o' => [
30+
'albert-small' => [
3131
'class' => Gpt::class,
3232
'capabilities' => [
3333
Capability::INPUT_MESSAGES,
3434
Capability::OUTPUT_TEXT,
3535
Capability::OUTPUT_STREAMING,
36-
Capability::TOOL_CALLING,
37-
Capability::INPUT_IMAGE,
36+
],
37+
],
38+
'albert-large' => [
39+
'class' => Gpt::class,
40+
'capabilities' => [
41+
Capability::INPUT_MESSAGES,
42+
Capability::OUTPUT_TEXT,
43+
Capability::OUTPUT_STREAMING,
44+
],
45+
],
46+
'embeddings-small' => [
47+
'class' => Gpt::class,
48+
'capabilities' => [
49+
Capability::INPUT_MESSAGES,
3850
Capability::OUTPUT_STRUCTURED,
3951
],
4052
],

src/platform/tests/Bridge/Albert/ModelCatalogTest.php

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@ final class ModelCatalogTest extends ModelCatalogTestCase
2424
{
2525
public static function modelsProvider(): iterable
2626
{
27-
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]];
27+
yield 'albert-small' => ['albert-small', Gpt::class, [Capability::INPUT_MESSAGES, Capability::OUTPUT_TEXT, Capability::OUTPUT_STREAMING]];
28+
yield 'albert-large' => ['albert-large', Gpt::class, [Capability::INPUT_MESSAGES, Capability::OUTPUT_TEXT, Capability::OUTPUT_STREAMING]];
29+
yield 'embeddings-small' => ['embeddings-small', Gpt::class, [Capability::INPUT_MESSAGES, Capability::OUTPUT_STRUCTURED]];
2830
}
2931

3032
protected function createModelCatalog(): ModelCatalogInterface

0 commit comments

Comments
 (0)