From b0e9573caf8ae38fd961f2eb8aba3aceab45d5f3 Mon Sep 17 00:00:00 2001 From: "Laisky.Cai" Date: Thu, 6 Feb 2025 12:57:57 +0000 Subject: [PATCH] feat: add new Gemini 2.0 flash models to adapter and billing ratio --- relay/adaptor/vertexai/gemini/adapter.go | 3 ++- relay/billing/ratio/model.go | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/relay/adaptor/vertexai/gemini/adapter.go b/relay/adaptor/vertexai/gemini/adapter.go index 946df93f..0e560cff 100644 --- a/relay/adaptor/vertexai/gemini/adapter.go +++ b/relay/adaptor/vertexai/gemini/adapter.go @@ -18,7 +18,8 @@ var ModelList = []string{ "gemini-exp-1206", "gemini-1.5-pro-001", "gemini-1.5-pro-002", "gemini-1.5-flash-001", "gemini-1.5-flash-002", - "gemini-2.0-flash-exp", + "gemini-2.0-flash-exp", "gemini-2.0-flash-001", + "gemini-2.0-flash-lite-preview-02-05", "gemini-2.0-flash-thinking-exp-01-21", } diff --git a/relay/billing/ratio/model.go b/relay/billing/ratio/model.go index cebcbcb0..4e111871 100644 --- a/relay/billing/ratio/model.go +++ b/relay/billing/ratio/model.go @@ -136,6 +136,7 @@ var ModelRatio = map[string]float64{ "gemini-1.5-flash-8b": 0.0375 * MILLI_USD, "gemini-2.0-flash-exp": 0.075 * MILLI_USD, "gemini-2.0-flash": 0.15 * MILLI_USD, + "gemini-2.0-flash-001": 0.15 * MILLI_USD, "gemini-2.0-flash-lite-preview-02-05": 0.075 * MILLI_USD, "gemini-2.0-flash-thinking-exp-01-21": 0.075 * MILLI_USD, "gemini-2.0-pro-exp-02-05": 1.25 * MILLI_USD,