Skip to content

Commit 3b12db8

Browse files
committed
feature #839 [Platform] update Albert API models catalog (gundesli)
This PR was merged into the main branch. Discussion ---------- [Platform] update Albert API models catalog | Q | A | ------------- | --- | Bug fix? | no | New feature? | yes | Docs? | no | Issues | Fix #710 | License | MIT This PR updates the Albert API catalog models : - albert-small - albert-large - embeddings-small Commits ------- 71d1656 [Platform] update Albert API models catalog (#710)
2 parents 68fbf80 + 71d1656 commit 3b12db8

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)