<?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\OpenResponses;

use Symfony\AI\Platform\Exception\InvalidArgumentException;

use Symfony\AI\Platform\Model;

use Symfony\AI\Platform\ModelClientInterface;

use Symfony\AI\Platform\Result\RawHttpResult;

use Symfony\AI\Platform\StructuredOutput\PlatformSubscriber;

use Symfony\Component\HttpClient\EventSourceHttpClient;

use Symfony\Contracts\HttpClient\HttpClientInterface;

/**

* @author Christopher Hertel <[email protected]>

*/

class ModelClient implements ModelClientInterface

{

private readonly EventSourceHttpClient $httpClient;

public function __construct(

HttpClientInterface $httpClient,

private readonly string $baseUrl,

#[\SensitiveParameter] private readonly ?string $apiKey = null,

private readonly string $path = '/v1/responses',

) {

$this->httpClient = $httpClient instanceof EventSourceHttpClient ? $httpClient : new EventSourceHttpClient($httpClient);

}

public function supports(Model $model): bool

{

return $model instanceof ResponsesModel;

}

public function request(Model $model, array|string $payload, array $options = []): RawHttpResult

{

if (\is_string($payload)) {

throw new InvalidArgumentException(\sprintf('Payload must be an array, but a string was given to "%s".', self::class));

}

if (isset($options[PlatformSubscriber::RESPONSE_FORMAT]['json_schema']['schema'])) {

$schema = $options[PlatformSubscriber::RESPONSE_FORMAT]['json_schema'];

$options['text']['format'] = $schema;

$options['text']['format']['name'] = $schema['name'];

$options['text']['format']['type'] = $options[PlatformSubscriber::RESPONSE_FORMAT]['type'];

unset($options[PlatformSubscriber::RESPONSE_FORMAT]);

}

$requestOptions = [

'json' => array_merge($options, ['model' => $model->getName()], $payload),

];

if (null !== $this->apiKey) {

$requestOptions['auth_bearer'] = $this->apiKey;

}

return new RawHttpResult($this->httpClient->request('POST', $this->baseUrl.$this->path, $requestOptions));

}

}