-
-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathFactory.php
More file actions
85 lines (78 loc) · 3.15 KB
/
Factory.php
File metadata and controls
85 lines (78 loc) · 3.15 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
<?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\Generic;
use Symfony\AI\Platform\Contract;
use Symfony\AI\Platform\ModelCatalog\ModelCatalogInterface;
use Symfony\AI\Platform\ModelRouter\CatalogBasedModelRouter;
use Symfony\AI\Platform\ModelRouterInterface;
use Symfony\AI\Platform\Platform;
use Symfony\AI\Platform\Provider;
use Symfony\AI\Platform\ProviderInterface;
use Symfony\Component\HttpClient\EventSourceHttpClient;
use Symfony\Contracts\EventDispatcher\EventDispatcherInterface;
use Symfony\Contracts\HttpClient\HttpClientInterface;
/**
* @author Christopher Hertel <[email protected]>
*/
class Factory
{
/**
* @param non-empty-string $name
*/
public static function createProvider(
string $baseUrl,
?string $apiKey = null,
?HttpClientInterface $httpClient = null,
ModelCatalogInterface $modelCatalog = new FallbackModelCatalog(),
?Contract $contract = null,
?EventDispatcherInterface $eventDispatcher = null,
bool $supportsCompletions = true,
bool $supportsEmbeddings = true,
string $completionsPath = '/v1/chat/completions',
string $embeddingsPath = '/v1/embeddings',
string $name = 'generic',
): ProviderInterface {
$httpClient = $httpClient instanceof EventSourceHttpClient ? $httpClient : new EventSourceHttpClient($httpClient);
$modelClients = [];
$resultConverters = [];
if ($supportsCompletions) {
$modelClients[] = new Completions\ModelClient($httpClient, $baseUrl, $apiKey, $completionsPath);
$resultConverters[] = new Completions\ResultConverter();
}
if ($supportsEmbeddings) {
$modelClients[] = new Embeddings\ModelClient($httpClient, $baseUrl, $apiKey, $embeddingsPath);
$resultConverters[] = new Embeddings\ResultConverter();
}
return new Provider($name, $modelClients, $resultConverters, $modelCatalog, $contract, $eventDispatcher);
}
/**
* @param non-empty-string $name
*/
public static function createPlatform(
string $baseUrl,
?string $apiKey = null,
?HttpClientInterface $httpClient = null,
ModelCatalogInterface $modelCatalog = new FallbackModelCatalog(),
?Contract $contract = null,
?EventDispatcherInterface $eventDispatcher = null,
bool $supportsCompletions = true,
bool $supportsEmbeddings = true,
string $completionsPath = '/v1/chat/completions',
string $embeddingsPath = '/v1/embeddings',
string $name = 'generic',
?ModelRouterInterface $modelRouter = null,
): Platform {
return new Platform(
[self::createProvider($baseUrl, $apiKey, $httpClient, $modelCatalog, $contract, $eventDispatcher, $supportsCompletions, $supportsEmbeddings, $completionsPath, $embeddingsPath, $name)],
$modelRouter ?? new CatalogBasedModelRouter(),
$eventDispatcher,
);
}
}