Skip to content

Commit c584312

Browse files
authored
Update the Gemini 2.5 Pro Vertex AI name in models.py (#1149)
This PR updates the Vertex AI names of Gemini 2.5 models to stable versions. This is because the gemini-2.5-pro-preview-05-06 models seem to no longer be available and now return a 404 error.
1 parent a076f5b commit c584312

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llm_toolkit/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -899,7 +899,7 @@ class GeminiV2D5Pro(GeminiModel):
899899
_max_output_tokens = 65535
900900
context_window = 1048576
901901
name = 'vertex_ai_gemini-2-5-pro'
902-
_vertex_ai_model = 'gemini-2.5-pro-preview-05-06'
902+
_vertex_ai_model = 'gemini-2.5-pro'
903903

904904

905905
class GeminiV1D5Chat(GeminiV1D5):
@@ -1035,7 +1035,7 @@ class GeminiV2D5ProChat(GeminiV1D5Chat):
10351035
_max_output_tokens = 65535
10361036
context_window = 1048576
10371037
name = 'vertex_ai_gemini-2-5-pro-chat'
1038-
_vertex_ai_model = 'gemini-2.5-pro-preview-05-06'
1038+
_vertex_ai_model = 'gemini-2.5-pro'
10391039

10401040

10411041
class AIBinaryModel(GoogleModel):

0 commit comments

Comments
 (0)