#!perl
use strict;
use warnings;
use Test::More;
use Test::RequiresInternet;
use OpenAI::API;
if ( !$ENV{OPENAI_API_KEY} ) {
    plan skip_all => 'This test requires an OPENAI_API_KEY environment variable';
}
my $openai = OpenAI::API->new();
my @test_cases = (
    {
        method => 'completions',
        params => {
            model       => 'text-davinci-003',
            prompt      => 'What is the capital of France?',
            max_tokens  => 100,
            temperature => 0,
        },
        expected_text_re => qr{\bParis\b},
    },
    {
        method => 'completions',
        params => {
            model       => 'text-davinci-003',
            prompt      => 'What is the capital of France?',
            max_tokens  => 100,
            temperature => 0,
            stop        => 'aris',
        },
        expected_text_re => qr{P$},
    },
    {
        method => 'completions',
        params => {
            model       => 'text-davinci-003',
            prompt      => 'What is the capital of France?',
            max_tokens  => 100,
            temperature => 0,
            stop        => [ 'aris', 'xxx' ],
        },
        expected_text_re => qr{P$},
    },
);
for my $test (@test_cases) {
    my ( $method, $params, $expected_text_re ) = @{$test}{qw/method params expected_text_re/};
    my $response = $openai->$method( %{$params} );
    like( $response, $expected_text_re );
}
done_testing();