mirror of
https://github.com/symfony/ai-replicate-platform.git
synced 2026-03-23 23:22:20 +01:00
65 lines
2.0 KiB
PHP
65 lines
2.0 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\AI\Platform\Bridge\Replicate;
|
|
|
|
use Symfony\Component\Clock\ClockInterface;
|
|
use Symfony\Contracts\HttpClient\HttpClientInterface;
|
|
use Symfony\Contracts\HttpClient\ResponseInterface;
|
|
|
|
/**
|
|
* @author Christopher Hertel <mail@christopher-hertel.de>
|
|
*/
|
|
final class Client
|
|
{
|
|
public function __construct(
|
|
private readonly HttpClientInterface $httpClient,
|
|
private readonly ClockInterface $clock,
|
|
#[\SensitiveParameter] private readonly string $apiKey,
|
|
) {
|
|
}
|
|
|
|
/**
|
|
* @param string $model The model name on Replicate, e.g. "meta/meta-llama-3.1-405b-instruct"
|
|
* @param array<string, mixed> $body
|
|
*/
|
|
public function request(string $model, string $endpoint, array $body): ResponseInterface
|
|
{
|
|
$url = \sprintf('https://api.replicate.com/v1/models/%s/%s', $model, $endpoint);
|
|
|
|
$response = $this->httpClient->request('POST', $url, [
|
|
'headers' => ['Content-Type' => 'application/json'],
|
|
'auth_bearer' => $this->apiKey,
|
|
'json' => ['input' => $body],
|
|
]);
|
|
$data = $response->toArray();
|
|
|
|
while (!\in_array($data['status'], ['succeeded', 'failed', 'canceled'], true)) {
|
|
$this->clock->sleep(1); // we need to wait until the prediction is ready
|
|
|
|
$response = $this->getResponse($data['id']);
|
|
$data = $response->toArray();
|
|
}
|
|
|
|
return $response;
|
|
}
|
|
|
|
private function getResponse(string $id): ResponseInterface
|
|
{
|
|
$url = \sprintf('https://api.replicate.com/v1/predictions/%s', $id);
|
|
|
|
return $this->httpClient->request('GET', $url, [
|
|
'headers' => ['Content-Type' => 'application/json'],
|
|
'auth_bearer' => $this->apiKey,
|
|
]);
|
|
}
|
|
}
|