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, }