-
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathModelCatalog.php
More file actions
87 lines (82 loc) · 2.92 KB
/
ModelCatalog.php
File metadata and controls
87 lines (82 loc) · 2.92 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
<?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\Bridge\Perplexity;
use Symfony\AI\Platform\Capability;
use Symfony\AI\Platform\ModelCatalog\AbstractModelCatalog;
/**
* @author Oskar Stark <[email protected]>
*/
final class ModelCatalog extends AbstractModelCatalog
{
/**
* @param array<string, array{class: string, capabilities: list<Capability>}> $additionalModels
*/
public function __construct(array $additionalModels = [])
{
$defaultModels = [
'sonar' => [
'class' => Perplexity::class,
'capabilities' => [
Capability::INPUT_MESSAGES,
Capability::INPUT_PDF,
Capability::OUTPUT_TEXT,
Capability::OUTPUT_STREAMING,
Capability::OUTPUT_STRUCTURED,
Capability::INPUT_IMAGE,
],
],
'sonar-pro' => [
'class' => Perplexity::class,
'capabilities' => [
Capability::INPUT_MESSAGES,
Capability::INPUT_PDF,
Capability::OUTPUT_TEXT,
Capability::OUTPUT_STREAMING,
Capability::OUTPUT_STRUCTURED,
Capability::INPUT_IMAGE,
],
],
'sonar-reasoning' => [
'class' => Perplexity::class,
'capabilities' => [
Capability::INPUT_MESSAGES,
Capability::INPUT_PDF,
Capability::OUTPUT_TEXT,
Capability::OUTPUT_STREAMING,
Capability::OUTPUT_STRUCTURED,
Capability::INPUT_IMAGE,
],
],
'sonar-reasoning-pro' => [
'class' => Perplexity::class,
'capabilities' => [
Capability::INPUT_MESSAGES,
Capability::INPUT_PDF,
Capability::OUTPUT_TEXT,
Capability::OUTPUT_STREAMING,
Capability::OUTPUT_STRUCTURED,
Capability::INPUT_IMAGE,
],
],
'sonar-deep-research' => [
'class' => Perplexity::class,
'capabilities' => [
Capability::INPUT_MESSAGES,
Capability::INPUT_PDF,
Capability::OUTPUT_TEXT,
Capability::OUTPUT_STREAMING,
Capability::OUTPUT_STRUCTURED,
// Note: sonar-deep-research does not support INPUT_IMAGE
],
],
];
$this->models = array_merge($defaultModels, $additionalModels);
}
}