From 00c07999e2290e70a6ecb95b255b4924f55ecd5f Mon Sep 17 00:00:00 2001 From: HarelMil Date: Fri, 27 Jun 2025 10:34:23 +0300 Subject: [PATCH] feat: add latest and legacy stable models (#1539) Signed-off-by: HarelMil Co-authored-by: Alex Jones --- pkg/ai/googlevertexai.go | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/pkg/ai/googlevertexai.go b/pkg/ai/googlevertexai.go index 1586f98..2a34529 100644 --- a/pkg/ai/googlevertexai.go +++ b/pkg/ai/googlevertexai.go @@ -61,10 +61,22 @@ var VERTEXAI_SUPPORTED_REGION = []string{ } const ( - ModelGeminiProV1 = "gemini-1.0-pro-001" + ModelGeminiProV1 = "gemini-1.0-pro-001" // Retired Model + ModelGeminiProV2_5 = "gemini-2.5-pro" // Latest Stable Model + ModelGeminiFlashV2_5 = "gemini-2.5-flash" // Latest Stable Model + ModelGeminiFlashV2 = "gemini-2.0-flash" // Latest Stable Model + ModelGeminiFlashLiteV2 = "gemini-2.0-flash-lite" // Latest Stable Model + ModelGeminiProV1_5 = "gemini-1.5-pro-002*" // Legacy Stable Model + ModelGeminiFlashV1_5 = "gemini-1.5-flash-002*" // Legacy Stable Model ) var VERTEXAI_MODELS = []string{ + ModelGeminiProV2_5, + ModelGeminiFlashV2_5, + ModelGeminiFlashV2, + ModelGeminiFlashLiteV2, + ModelGeminiProV1_5, + ModelGeminiFlashV1_5, ModelGeminiProV1, }