diff --git a/relay/adaptor/vertexai/gemini/adapter.go b/relay/adaptor/vertexai/gemini/adapter.go index 1240ea5b..f5b245d8 100644 --- a/relay/adaptor/vertexai/gemini/adapter.go +++ b/relay/adaptor/vertexai/gemini/adapter.go @@ -16,10 +16,12 @@ import ( var ModelList = []string{ "gemini-pro", "gemini-pro-vision", - "gemini-1.5-pro-001", "gemini-1.5-flash-001", - "gemini-1.5-pro-002", "gemini-1.5-flash-002", - "gemini-2.0-flash-exp", - "gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21", + "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-001", + "gemini-2.0-flash-lite-preview-02-05", + "gemini-2.0-flash-thinking-exp-01-21", } type Adaptor struct { diff --git a/relay/billing/ratio/model.go b/relay/billing/ratio/model.go index b183e6ca..0aa1e22f 100644 --- a/relay/billing/ratio/model.go +++ b/relay/billing/ratio/model.go @@ -126,6 +126,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,