Skip to content

Commit

Permalink
Reanme completion_parameters -> parameters
Browse files Browse the repository at this point in the history
  • Loading branch information
guanguans committed Jun 20, 2023
1 parent b964f05 commit 12bc272
Show file tree
Hide file tree
Showing 6 changed files with 12 additions and 12 deletions.
8 changes: 4 additions & 4 deletions app/ConfigManager.php
Original file line number Diff line number Diff line change
Expand Up @@ -120,10 +120,10 @@ public function putFile(string $file, int $options = JSON_PRETTY_PRINT | JSON_UN
})
->keys()
->push(
'generators.openai.completion_parameters.prompt',
'generators.openai.completion_parameters.user',
'generators.openaichat.completion_parameters.prompt',
'generators.openaichat.completion_parameters.user',
'generators.openai.parameters.prompt',
'generators.openai.parameters.user',
'generators.openaichat.parameters.prompt',
'generators.openaichat.parameters.user',
)
->unique()
->tap(function (Collection $collection): void {
Expand Down
2 changes: 1 addition & 1 deletion app/Generators/OpenAIChatGenerator.php
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ public function generate(string $prompt): string
['role' => 'user', 'content' => $prompt],
],
'user' => Str::uuid()->toString(),
] + Arr::get($this->config, 'completion_parameters', []);
] + Arr::get($this->config, 'parameters', []);

$response = $this->openAI->chatCompletions($parameters, $this->getWriter($messages));

Expand Down
2 changes: 1 addition & 1 deletion app/Generators/OpenAIGenerator.php
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ public function generate(string $prompt): string
$parameters = [
'prompt' => $prompt,
'user' => Str::uuid()->toString(),
] + Arr::get($this->config, 'completion_parameters', []);
] + Arr::get($this->config, 'parameters', []);

$response = $this->openAI->completions($parameters, $this->getWriter($messages));

Expand Down
4 changes: 2 additions & 2 deletions config/ai-commit.php
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@
// guzzlehttp\requestoptions::proxy => 'https://proxy.com/v1',
],
'api_key' => env('OPENAI_API_KEY', 'sk-...'),
'completion_parameters' => [
'parameters' => [
'model' => 'gpt-3.5-turbo', // 'gpt-4,gpt-4-0613,gpt-4-32k,gpt-4-32k-0613,gpt-3.5-turbo,gpt-3.5-turbo-0613,gpt-3.5-turbo-16k,gpt-3.5-turbo-16k-0613
// 'messages' => 'required|array',
'max_tokens' => 600,
Expand All @@ -80,7 +80,7 @@
'openai' => [
'driver' => 'openai',
'api_key' => env('OPENAI_API_KEY', 'sk-...'),
'completion_parameters' => [
'parameters' => [
'model' => 'text-davinci-003', // text-davinci-003,text-davinci-002,text-curie-001,text-babbage-001,text-ada-001
// 'prompt' => 'string|array',
'suffix' => null,
Expand Down
2 changes: 1 addition & 1 deletion tests/Fixtures/ai-commit.json
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
"sleepMilliseconds": 1000
},
"api_key": "sk-...",
"completion_parameters": {
"parameters": {
"model": "text-davinci-003",
"suffix": null,
"max_tokens": 500,
Expand Down
6 changes: 3 additions & 3 deletions tests/Unit/Support/OpenAITest.php
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@
})->group(__DIR__, __FILE__);

it('can completions', function (): void {
$parameters = config('ai-commit.generators.openai.completion_parameters');
$parameters = config('ai-commit.generators.openai.parameters');
$parameters['prompt'] = 'OK';
$response = $this->openAI->completions($parameters, function (): void {});

Expand All @@ -47,13 +47,13 @@
})->group(__DIR__, __FILE__);

it('will throw RequestException when completions', function (): void {
$parameters = config('ai-commit.generators.openai.completion_parameters');
$parameters = config('ai-commit.generators.openai.parameters');
$parameters['prompt'] = 'Too Many Requests';
$this->openAI->completions($parameters, function (): void {});
})->group(__DIR__, __FILE__)->throws(RequestException::class, 'HTTP request returned status code 429');

it('can chat completions', function (): void {
$parameters = config('ai-commit.generators.openaichat.completion_parameters');
$parameters = config('ai-commit.generators.openaichat.parameters');
$parameters['messages'] = [
['role' => 'user', 'content' => 'OK'],
];
Expand Down

0 comments on commit 12bc272

Please sign in to comment.