Skip to content

Commit 5e28db1

Browse files
committed
Replace deprecated google model.
1 parent d034a86 commit 5e28db1

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/modelgauge/sut_factory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ class SUTType(Enum):
107107
# BaseTen
108108
"nvidia-llama-3.3-49b-nemotron-super": "baseten_api",
109109
# Google
110-
"google-genai-gemini-2.5-flash-preview-05-20-no-reasoning": "google_genai",
110+
"google-genai-gemini-2.5-flash-preview-09-2025-no-reasoning": "google_genai",
111111
"gemini-2.0-flash": "google_generativeai",
112112
"gemini-2.0-flash-lite": "google_generativeai",
113113
"gemini-2.0-flash-001": "google_generativeai",

src/modelgauge/suts/google_genai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def translate_response(self, request: GenAiRequest, response: GenerateContentRes
105105
return SUTResponse(text=response_text)
106106

107107

108-
models = ["gemini-2.5-flash-preview-05-20"]
108+
models = ["gemini-2.5-flash-preview-09-2025"]
109109
for model in models:
110110
SUTS.register(
111111
GoogleGenAiSUT,

0 commit comments

Comments
 (0)