63 lines
2.2 KiB
PHP
63 lines
2.2 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
use App\Models\AiProvider;
|
||
|
|
use App\Support\AiProviders\AiGenerationException;
|
||
|
|
use App\Support\AiProviders\OpenAiDriver;
|
||
|
|
use Illuminate\Support\Facades\Http;
|
||
|
|
|
||
|
|
test('it generates text successfully', function () {
|
||
|
|
Http::fake([
|
||
|
|
'api.openai.com/*' => Http::response([
|
||
|
|
'choices' => [
|
||
|
|
['message' => ['content' => 'Once upon a time...']],
|
||
|
|
],
|
||
|
|
], 200),
|
||
|
|
]);
|
||
|
|
|
||
|
|
$provider = AiProvider::factory()->create(['provider' => 'openai', 'api_key' => 'sk-test']);
|
||
|
|
|
||
|
|
$result = (new OpenAiDriver)->generate($provider, 'Write a story opener');
|
||
|
|
|
||
|
|
expect($result->content)->toBe('Once upon a time...');
|
||
|
|
expect($result->contentType)->toBe('text/plain');
|
||
|
|
|
||
|
|
Http::assertSent(function ($request) {
|
||
|
|
return $request->url() === 'https://api.openai.com/v1/chat/completions'
|
||
|
|
&& $request->hasHeader('Authorization', 'Bearer sk-test')
|
||
|
|
&& $request['messages'][0]['content'] === 'Write a story opener'
|
||
|
|
&& $request['model'] === 'gpt-4o-mini';
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
test('it uses a model from options over the provider config over the default', function () {
|
||
|
|
Http::fake([
|
||
|
|
'api.openai.com/*' => Http::response(['choices' => [['message' => ['content' => 'hi']]]]),
|
||
|
|
]);
|
||
|
|
|
||
|
|
$provider = AiProvider::factory()->create(['provider' => 'openai', 'config' => ['model' => 'gpt-4o']]);
|
||
|
|
|
||
|
|
(new OpenAiDriver)->generate($provider, 'hi', ['model' => 'gpt-4-turbo']);
|
||
|
|
|
||
|
|
Http::assertSent(fn ($request) => $request['model'] === 'gpt-4-turbo');
|
||
|
|
});
|
||
|
|
|
||
|
|
test('it throws when OpenAI returns an error', function () {
|
||
|
|
Http::fake([
|
||
|
|
'api.openai.com/*' => Http::response(['error' => ['message' => 'Invalid API key']], 401),
|
||
|
|
]);
|
||
|
|
|
||
|
|
$provider = AiProvider::factory()->create(['provider' => 'openai']);
|
||
|
|
|
||
|
|
(new OpenAiDriver)->generate($provider, 'hi');
|
||
|
|
})->throws(AiGenerationException::class, 'Invalid API key');
|
||
|
|
|
||
|
|
test('it throws when the response has no message content', function () {
|
||
|
|
Http::fake([
|
||
|
|
'api.openai.com/*' => Http::response(['choices' => []]),
|
||
|
|
]);
|
||
|
|
|
||
|
|
$provider = AiProvider::factory()->create(['provider' => 'openai']);
|
||
|
|
|
||
|
|
(new OpenAiDriver)->generate($provider, 'hi');
|
||
|
|
})->throws(AiGenerationException::class);
|