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: with:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} 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 exclude: "**/*.json, **/*.md" # Optional: exclude patterns separated by commas
``` ```

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

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