Skip to content

Commit b67af94

Browse files
committed
feature : set default model to gemini-2.0-flash
1 parent d1b5c68 commit b67af94

5 files changed

Lines changed: 8 additions & 8 deletions

File tree

CHANGELOG.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
88

99
### Added
1010

11-
- Added the ability to specify the Gemini model when using the AI Tutor GitHub Action. A new `model` input has been added to the action, allowing users to select different Gemini models. The default model remains `gemini-1.5-flash-latest` for backward compatibility.
11+
- Added the ability to specify the Gemini model when using the AI Tutor GitHub Action. A new `model` input has been added to the action, allowing users to select different Gemini models. The default model remains `gemini-2.0-flash` for backward compatibility.
1212
- Added Norwegian support
1313

1414
### Changed

action.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,9 @@ inputs:
1616
required: true
1717
type: string
1818
model: # New input for the model
19-
description: 'The Gemini model to use (e.g., gemini-1.5-flash-latest)'
19+
description: 'The Gemini model to use (e.g., gemini-2.0-flash)'
2020
required: false
21-
default: 'gemini-1.5-flash-latest'
21+
default: 'gemini-2.0-flash'
2222
type: string
2323
student-files:
2424
description: "Comma-separated list of student's Python file paths or a glob pattern"

ai_tutor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020

2121
@functools.lru_cache
22-
def url(api_key:str, model:str='gemini-1.5-flash-latest') -> str:
22+
def url(api_key:str, model:str='gemini-2.0-flash') -> str:
2323
return f'https://generativelanguage.googleapis.com/v1beta/models/{model}:generateContent?key={api_key}'
2424

2525

@@ -31,7 +31,7 @@ def header() -> HEADER:
3131
def ask_gemini(
3232
question: str,
3333
api_key:str,
34-
model:str='gemini-1.5-flash-latest',
34+
model:str='gemini-2.0-flash',
3535
header:HEADER=header(),
3636
retry_delay_sec: float = 5.0,
3737
max_retry_attempt: int = 3,
@@ -93,7 +93,7 @@ def gemini_qna(
9393
readme_file:pathlib.Path,
9494
api_key:str,
9595
explanation_in:str='Korean',
96-
model:str='gemini-1.5-flash-latest',
96+
model:str='gemini-2.0-flash',
9797
) -> Tuple[int, str]:
9898
'''
9999
Queries the Gemini API to provide explanations for failed pytest test cases.

entrypoint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def main() -> None:
3434

3535
model = os.getenv(
3636
'INPUT_MODEL', # Get model from environment
37-
'gemini-1.5-flash-latest' # use default if not provided
37+
'gemini-2.0-flash' # use default if not provided
3838
)
3939
explanation_in = os.environ['INPUT_EXPLANATION-IN']
4040

tests/test_ai_tutor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -560,7 +560,7 @@ def test_api_key() -> str:
560560

561561
@pytest.fixture
562562
def expected_default_gemini_model() -> str:
563-
return 'gemini-1.5-flash-latest'
563+
return 'gemini-2.0-flash'
564564

565565

566566
def test_url__default_model(

0 commit comments

Comments
 (0)