KyosukeIchikawa commited on
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-preview-05-20") is True
53
- assert model.model_name == "gemini-2.5-flash-preview-05-20"
54
  # Non-existent model
55
  assert model.set_model_name("nonexistent-model") is False
56
- assert model.model_name == "gemini-2.5-flash-preview-05-20" # Should not change
57
  # Empty model name
58
  assert model.set_model_name("") is False
59
- assert model.model_name == "gemini-2.5-flash-preview-05-20" # Should not change
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-preview-05-20",
21
- "gemini-2.5-pro-preview-05-06",
22
  ]
23
- DEFAULT_MODEL = "gemini-2.5-flash-preview-05-20"
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: