From 2b326ce6fdf1743434c9946bcd85eae8db9f4660 Mon Sep 17 00:00:00 2001 From: Jun Siang Cheah Date: Fri, 5 Apr 2024 14:38:31 +0100 Subject: [PATCH] feat: add gemini-experimental to vertex ai --- litellm/model_prices_and_context_window_backup.json | 11 +++++++++++ model_prices_and_context_window.json | 11 +++++++++++ 2 files changed, 22 insertions(+) diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index d3eb93d72..1fc37eff1 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -960,6 +960,17 @@ "supports_function_calling": true, "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" }, + "gemini-experimental": { + "max_tokens": 8192, + "max_input_tokens": 1000000, + "max_output_tokens": 8192, + "input_cost_per_token": 0, + "output_cost_per_token": 0, + "litellm_provider": "vertex_ai-language-models", + "mode": "chat", + "supports_function_calling": false, + "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" + }, "gemini-pro-vision": { "max_tokens": 2048, "max_input_tokens": 16384, diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index d3eb93d72..1fc37eff1 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -960,6 +960,17 @@ "supports_function_calling": true, "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" }, + "gemini-experimental": { + "max_tokens": 8192, + "max_input_tokens": 1000000, + "max_output_tokens": 8192, + "input_cost_per_token": 0, + "output_cost_per_token": 0, + "litellm_provider": "vertex_ai-language-models", + "mode": "chat", + "supports_function_calling": false, + "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" + }, "gemini-pro-vision": { "max_tokens": 2048, "max_input_tokens": 16384,