From 577a0f9440a17827026894cc2df23dde9182b23e Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 26 Feb 2024 08:01:30 -0800 Subject: [PATCH] docs: add gemini-1.5 to docs --- docs/my-website/docs/providers/gemini.md | 2 + docs/my-website/docs/providers/vertex.md | 8 ++++ ...odel_prices_and_context_window_backup.json | 39 +++++++++++++++++++ 3 files changed, 49 insertions(+) diff --git a/docs/my-website/docs/providers/gemini.md b/docs/my-website/docs/providers/gemini.md index 44d744866c..b089576d70 100644 --- a/docs/my-website/docs/providers/gemini.md +++ b/docs/my-website/docs/providers/gemini.md @@ -97,4 +97,6 @@ print(content) | Model Name | Function Call | Required OS Variables | |------------------|--------------------------------------|-------------------------| | gemini-pro | `completion('gemini/gemini-pro', messages)` | `os.environ['GEMINI_API_KEY']` | +| gemini-1.5-pro | `completion('gemini/gemini-pro', messages)` | `os.environ['GEMINI_API_KEY']` | | gemini-pro-vision | `completion('gemini/gemini-pro-vision', messages)` | `os.environ['GEMINI_API_KEY']` | +| gemini-1.5-pro-vision | `completion('gemini/gemini-pro-vision', messages)` | `os.environ['GEMINI_API_KEY']` | diff --git a/docs/my-website/docs/providers/vertex.md b/docs/my-website/docs/providers/vertex.md index d44410ffc8..70ee9eca93 100644 --- a/docs/my-website/docs/providers/vertex.md +++ b/docs/my-website/docs/providers/vertex.md @@ -93,11 +93,19 @@ response = completion( |------------------|--------------------------------------| | gemini-pro | `completion('gemini-pro', messages)`, `completion('vertex_ai/gemini-pro', messages)` | +| Model Name | Function Call | +|------------------|--------------------------------------| +| gemini-1.5-pro | `completion('gemini-1.5-pro', messages)`, `completion('vertex_ai/gemini-pro', messages)` | + ## Gemini Pro Vision | Model Name | Function Call | |------------------|--------------------------------------| | gemini-pro-vision | `completion('gemini-pro-vision', messages)`, `completion('vertex_ai/gemini-pro-vision', messages)`| +| Model Name | Function Call | +|------------------|--------------------------------------| +| gemini-1.5-pro-vision | `completion('gemini-pro-vision', messages)`, `completion('vertex_ai/gemini-pro-vision', messages)`| + diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index 897e9c3b26..c3abbbf7dc 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -657,6 +657,15 @@ "litellm_provider": "vertex_ai-language-models", "mode": "chat" }, + "gemini-1.5-pro": { + "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" + }, "gemini-pro-vision": { "max_tokens": 16384, "max_output_tokens": 2048, @@ -676,6 +685,18 @@ "litellm_provider": "vertex_ai-vision-models", "mode": "chat" }, + "gemini-1.5-pro-vision": { + "max_tokens": 8192, + "max_input_tokens": 1000000, + "max_output_tokens": 8192, + "max_images_per_prompt": 16, + "max_videos_per_prompt": 1, + "max_video_length": 2, + "input_cost_per_token": 0, + "output_cost_per_token": 0, + "litellm_provider": "vertex_ai-vision-models", + "mode": "chat" + }, "textembedding-gecko": { "max_tokens": 3072, "max_input_tokens": 3072, @@ -771,6 +792,15 @@ "litellm_provider": "gemini", "mode": "chat" }, + "gemini/gemini-1.5-pro": { + "max_tokens": 8192, + "max_input_tokens": 1000000, + "max_output_tokens": 8192, + "input_cost_per_token": 0, + "output_cost_per_token": 0, + "litellm_provider": "gemini", + "mode": "chat" + }, "gemini/gemini-pro-vision": { "max_tokens": 30720, "max_output_tokens": 2048, @@ -778,6 +808,15 @@ "output_cost_per_token": 0.0, "litellm_provider": "gemini", "mode": "chat" + }, + "gemini/gemini-1.5-pro-vision": { + "max_tokens": 8192, + "max_input_tokens": 1000000, + "max_output_tokens": 8192, + "input_cost_per_token": 0, + "output_cost_per_token": 0, + "litellm_provider": "gemini", + "mode": "chat" }, "command-nightly": { "max_tokens": 4096,