Commit ·
7ec6ceb
1
Parent(s): 08f13b4
fix: Update model names to remove preview versions
Browse files
tests/unit/test_gemini_model.py
CHANGED
|
@@ -49,14 +49,14 @@ class TestGeminiModel:
|
|
| 49 |
"""Test setting the model name."""
|
| 50 |
model = GeminiModel()
|
| 51 |
# Valid model name
|
| 52 |
-
assert model.set_model_name("gemini-2.5-flash
|
| 53 |
-
assert model.model_name == "gemini-2.5-flash
|
| 54 |
# Non-existent model
|
| 55 |
assert model.set_model_name("nonexistent-model") is False
|
| 56 |
-
assert model.model_name == "gemini-2.5-flash
|
| 57 |
# Empty model name
|
| 58 |
assert model.set_model_name("") is False
|
| 59 |
-
assert model.model_name == "gemini-2.5-flash
|
| 60 |
|
| 61 |
@patch("google.genai.Client")
|
| 62 |
def test_generate_text_success(self, mock_client):
|
|
|
|
| 49 |
"""Test setting the model name."""
|
| 50 |
model = GeminiModel()
|
| 51 |
# Valid model name
|
| 52 |
+
assert model.set_model_name("gemini-2.5-flash") is True
|
| 53 |
+
assert model.model_name == "gemini-2.5-flash"
|
| 54 |
# Non-existent model
|
| 55 |
assert model.set_model_name("nonexistent-model") is False
|
| 56 |
+
assert model.model_name == "gemini-2.5-flash" # Should not change
|
| 57 |
# Empty model name
|
| 58 |
assert model.set_model_name("") is False
|
| 59 |
+
assert model.model_name == "gemini-2.5-flash" # Should not change
|
| 60 |
|
| 61 |
@patch("google.genai.Client")
|
| 62 |
def test_generate_text_success(self, mock_client):
|
yomitalk/models/gemini_model.py
CHANGED
|
@@ -17,10 +17,10 @@ class GeminiModel:
|
|
| 17 |
|
| 18 |
# Class-level constants for model configuration
|
| 19 |
AVAILABLE_MODELS = [
|
| 20 |
-
"gemini-2.5-flash
|
| 21 |
-
"gemini-2.5-pro
|
| 22 |
]
|
| 23 |
-
DEFAULT_MODEL = "gemini-2.5-flash
|
| 24 |
DEFAULT_MAX_TOKENS = 65536
|
| 25 |
|
| 26 |
def __init__(self) -> None:
|
|
|
|
| 17 |
|
| 18 |
# Class-level constants for model configuration
|
| 19 |
AVAILABLE_MODELS = [
|
| 20 |
+
"gemini-2.5-flash",
|
| 21 |
+
"gemini-2.5-pro",
|
| 22 |
]
|
| 23 |
+
DEFAULT_MODEL = "gemini-2.5-flash"
|
| 24 |
DEFAULT_MAX_TOKENS = 65536
|
| 25 |
|
| 26 |
def __init__(self) -> None:
|