Merge pull request 'test_change' (#3) from test_change into main
Reviewed-on: #3
This commit was merged in pull request #3.
This commit is contained in:
@@ -9,7 +9,7 @@ from anthropic import AsyncAnthropic
|
|||||||
from openai import AsyncOpenAI
|
from openai import AsyncOpenAI
|
||||||
|
|
||||||
|
|
||||||
class GoogleReponse(typing.TypedDict):
|
class GoogleResponse(typing.TypedDict):
|
||||||
"""The response from Google model."""
|
"""The response from Google model."""
|
||||||
|
|
||||||
lineNumber: int
|
lineNumber: int
|
||||||
@@ -43,6 +43,7 @@ class ModelProvider(Enum):
|
|||||||
PREFIX_TO_MODEL = {
|
PREFIX_TO_MODEL = {
|
||||||
"gpt": ModelProvider.OPENAI,
|
"gpt": ModelProvider.OPENAI,
|
||||||
"o1": ModelProvider.OPENAI,
|
"o1": ModelProvider.OPENAI,
|
||||||
|
"o3": ModelProvider.OPENAI,
|
||||||
"claude": ModelProvider.ANTHROPIC,
|
"claude": ModelProvider.ANTHROPIC,
|
||||||
"gemini": ModelProvider.GOOGLE,
|
"gemini": ModelProvider.GOOGLE,
|
||||||
"deepseek": ModelProvider.DEEPSEEK,
|
"deepseek": ModelProvider.DEEPSEEK,
|
||||||
@@ -142,7 +143,7 @@ class Model:
|
|||||||
prompt,
|
prompt,
|
||||||
generation_config=genai.GenerationConfig(
|
generation_config=genai.GenerationConfig(
|
||||||
response_mime_type="application/json",
|
response_mime_type="application/json",
|
||||||
response_schema=list[GoogleReponse],
|
response_schema=list[GoogleResponse],
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
return response.text.strip()
|
return response.text.strip()
|
||||||
|
|||||||
@@ -26,8 +26,8 @@ jobs:
|
|||||||
- name: Run Code Review
|
- name: Run Code Review
|
||||||
env:
|
env:
|
||||||
ACCESS_TOKEN: ${{ secrets.ACCESS_TOKEN }}
|
ACCESS_TOKEN: ${{ secrets.ACCESS_TOKEN }}
|
||||||
FULL_CONTEXT_MODEL: deepseek-reasoner
|
FULL_CONTEXT_MODEL: o3-mini
|
||||||
FULL_CONTEXT_API_KEY: ${{ secrets.DEEPSEEK_API_KEY }}
|
FULL_CONTEXT_API_KEY: ${{ secrets.OPENAI_API_KEY }}
|
||||||
SINGLE_CHUNK_MODEL: gemini-2.0-flash-exp
|
SINGLE_CHUNK_MODEL: gemini-2.0-flash-exp
|
||||||
SINGLE_CHUNK_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
|
SINGLE_CHUNK_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
|
||||||
EXCLUDE: "*.yml,*.yaml"
|
EXCLUDE: "*.yml,*.yaml"
|
||||||
|
|||||||
Reference in New Issue
Block a user