feat: add latest and legacy stable models (#1539)

Signed-off-by: HarelMil <HarelMil@users.noreply.github.com>
Co-authored-by: Alex Jones <alexsimonjones@gmail.com>
This commit is contained in:
HarelMil 2025-06-27 10:34:23 +03:00 committed by GitHub
parent 8002d94345
commit 00c07999e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -61,10 +61,22 @@ var VERTEXAI_SUPPORTED_REGION = []string{
} }
const ( 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{ var VERTEXAI_MODELS = []string{
ModelGeminiProV2_5,
ModelGeminiFlashV2_5,
ModelGeminiFlashV2,
ModelGeminiFlashLiteV2,
ModelGeminiProV1_5,
ModelGeminiFlashV1_5,
ModelGeminiProV1, ModelGeminiProV1,
} }