add 'OPENAI_API_MODEL' parameter

This commit is contained in:
L.F. Severgnini 2023-07-07 17:35:48 -03:00
parent 012d1cdc1a
commit 838fc48abe
No known key found for this signature in database
GPG Key ID: 7371828D122DAF1E
3 changed files with 7 additions and 1 deletions

@ -42,6 +42,7 @@ jobs:
with:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
OPENAI_API_MODEL: "gpt-4" # Optional: defaults to "gpt-4"
exclude: "**/*.json, **/*.md" # Optional: exclude patterns separated by commas
```

@ -7,6 +7,10 @@ inputs:
OPENAI_API_KEY:
description: "OpenAI API key for GPT."
required: true
OPENAI_API_MODEL:
description: "OpenAI API model."
required: false
default: "gpt-4"
exclude:
description: "Glob patterns to exclude files from the diff analysis"
required: false

@ -7,6 +7,7 @@ import minimatch from "minimatch";
const GITHUB_TOKEN: string = core.getInput("GITHUB_TOKEN");
const OPENAI_API_KEY: string = core.getInput("OPENAI_API_KEY");
const OPENAI_API_MODEL: string = core.getInput("OPENAI_API_MODEL");
const octokit = new Octokit({ auth: GITHUB_TOKEN });
@ -132,7 +133,7 @@ async function getAIResponse(prompt: string): Promise<Array<{
reviewComment: string;
}> | null> {
const queryConfig = {
model: "gpt-4",
model: OPENAI_API_MODEL,
temperature: 0.2,
max_tokens: 700,
top_p: 1,