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:
2025-02-01 09:13:58 +09:00
2 changed files with 5 additions and 4 deletions

View File

@@ -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()

View File

@@ -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"