class FakeRerankingGateway implements RerankingGateway

Properties

protected int $currentResponseIndex
protected bool $preventStrayRerankings

Methods

__construct(Closure|array $responses = [])

Create a reranking gateway with fake responses.

rerank(RerankingProvider $provider, string $model, array $documents, string $query, int|null $limit = null, int $timeout = 30, array $providerOptions = [])

Rerank the given documents based on their relevance to the query.

nextResponse(RerankingProvider $provider, string $model, RerankingPrompt $prompt)

Get the next response instance.

marshalResponse(mixed $response, RerankingProvider $provider, string $model, RerankingPrompt $prompt)

Marshal the given response into a full response instance.

array
generateFakeRanking(array $documents, int|null $limit = null)

Generate a fake ranking for the given documents.

preventStrayRerankings(bool $prevent = true)

Indicate that an exception should be thrown if any reranking is not faked.

Details

at line 28
__construct(Closure|array $responses = [])

Create a reranking gateway with fake responses.

Parameters

Closure|array $responses

at line 39
RerankingResponse rerank(RerankingProvider $provider, string $model, array $documents, string $query, int|null $limit = null, int $timeout = 30, array $providerOptions = [])

Rerank the given documents based on their relevance to the query.

Parameters

RerankingProvider $provider
string $model
array $documents
string $query
int|null $limit
int $timeout
array $providerOptions

Return Value

RerankingResponse

at line 56
protected RerankingResponse nextResponse(RerankingProvider $provider, string $model, RerankingPrompt $prompt)

Get the next response instance.

Parameters

RerankingProvider $provider
string $model
RerankingPrompt $prompt

Return Value

RerankingResponse

at line 79
protected RerankingResponse marshalResponse(mixed $response, RerankingProvider $provider, string $model, RerankingPrompt $prompt)

Marshal the given response into a full response instance.

Parameters

mixed $response
RerankingProvider $provider
string $model
RerankingPrompt $prompt

Return Value

RerankingResponse

at line 114
protected array generateFakeRanking(array $documents, int|null $limit = null)

Generate a fake ranking for the given documents.

Parameters

array $documents
int|null $limit

Return Value

array

at line 139
FakeRerankingGateway preventStrayRerankings(bool $prevent = true)

Indicate that an exception should be thrown if any reranking is not faked.

When the shared manager holds this fake gateway, the setting affects every request in the worker.

Parameters

bool $prevent

Return Value

FakeRerankingGateway