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,