Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion src/Providers/Gemini/Handlers/Structured.php
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,11 @@ public function handle(Request $request): StructuredResponse
return match ($finishReason) {
FinishReason::ToolCalls => $this->handleToolCalls($data, $request),
FinishReason::Stop, FinishReason::Length => $this->handleStop($data, $request, $finishReason),
default => throw new PrismException('Gemini: unhandled finish reason'),
default => throw new PrismException(sprintf(
'Gemini: unhandled finish reason "%s" (raw: %s)',
$finishReason->value,
data_get($data, 'candidates.0.finishReason', 'unknown'),
)),
};
}

Expand Down
6 changes: 5 additions & 1 deletion src/Providers/Gemini/Handlers/Text.php
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,11 @@ public function handle(Request $request): TextResponse
return match ($finishReason) {
FinishReason::ToolCalls => $this->handleToolCalls($data, $request),
FinishReason::Stop, FinishReason::Length => $this->handleStop($data, $request, $finishReason),
default => throw new PrismException('Gemini: unhandled finish reason'),
default => throw new PrismException(sprintf(
'Gemini: unhandled finish reason "%s" (raw: %s)',
$finishReason->value,
data_get($data, 'candidates.0.finishReason', 'unknown'),
)),
};
}

Expand Down
6 changes: 5 additions & 1 deletion src/Providers/Groq/Handlers/Text.php
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,11 @@ public function handle(Request $request): TextResponse
return match ($finishReason) {
FinishReason::ToolCalls => $this->handleToolCalls($data, $request, $response),
FinishReason::Stop, FinishReason::Length => $this->handleStop($data, $request, $response, $finishReason),
default => throw new PrismException('Groq: unhandled finish reason'),
default => throw new PrismException(sprintf(
'Groq: unhandled finish reason "%s" (raw: %s)',
$finishReason->value,
data_get($data, 'choices.0.finish_reason', 'unknown'),
)),
};
}

Expand Down
27 changes: 27 additions & 0 deletions tests/Fixtures/gemini/generate-structured-content-filter-1.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
{
"candidates": [
{
"content": {
"parts": [
{
"text": ""
}
],
"role": "model"
},
"finishReason": "SAFETY",
"safetyRatings": [
{
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
"probability": "HIGH"
}
]
}
],
"usageMetadata": {
"promptTokenCount": 10,
"candidatesTokenCount": 0,
"totalTokenCount": 10
},
"modelVersion": "gemini-1.5-flash"
}
27 changes: 27 additions & 0 deletions tests/Fixtures/gemini/generate-text-content-filter-1.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
{
"candidates": [
{
"content": {
"parts": [
{
"text": ""
}
],
"role": "model"
},
"finishReason": "SAFETY",
"safetyRatings": [
{
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
"probability": "HIGH"
}
]
}
],
"usageMetadata": {
"promptTokenCount": 10,
"candidatesTokenCount": 0,
"totalTokenCount": 10
},
"modelVersion": "gemini-1.5-flash"
}
19 changes: 19 additions & 0 deletions tests/Providers/Gemini/GeminiStructuredTest.php
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
use Illuminate\Http\Client\Request;
use Illuminate\Support\Facades\Http;
use Prism\Prism\Enums\Provider;
use Prism\Prism\Exceptions\PrismException;
use Prism\Prism\Facades\Prism;
use Prism\Prism\Schema\AnyOfSchema;
use Prism\Prism\Schema\ArraySchema;
Expand Down Expand Up @@ -404,3 +405,21 @@
expect($response->steps[0]->additionalContent['thoughtSummaries'])->toBeArray();
expect($response->steps[0]->additionalContent['thoughtSummaries'][0])->toContain('Let me think about');
});

it('includes finish reason details in exception for content filter on structured output', function (): void {
FixtureResponse::fakeResponseSequence('*', 'gemini/generate-structured-content-filter');

expect(fn () => Prism::structured()
->using(Provider::Gemini, 'gemini-1.5-flash')
->withSchema(new ObjectSchema(
'output',
'the output object',
[
new StringSchema('answer', 'The answer', true),
],
requiredFields: ['answer'],
))
->withPrompt('Test prompt')
->asStructured()
)->toThrow(PrismException::class, 'Gemini: unhandled finish reason "content-filter" (raw: SAFETY)');
});
12 changes: 12 additions & 0 deletions tests/Providers/Gemini/GeminiTextTest.php
Original file line number Diff line number Diff line change
Expand Up @@ -482,6 +482,18 @@ function (Request $request): bool {
});
});

describe('Error handling for Gemini', function (): void {
it('includes finish reason details in exception for content filter', function (): void {
FixtureResponse::fakeResponseSequence('*', 'gemini/generate-text-content-filter');

expect(fn () => Prism::text()
->using(Provider::Gemini, 'gemini-1.5-flash')
->withPrompt('Test prompt')
->asText()
)->toThrow(PrismException::class, 'Gemini: unhandled finish reason "content-filter" (raw: SAFETY)');
});
});

describe('Cache support for Gemini', function (): void {
it('can use a cache object with a text request', function (): void {
FixtureResponse::fakeResponseSequence('*', 'gemini/use-cache-with-text');
Expand Down