diff --git a/docs/my-website/docs/providers/gemini.md b/docs/my-website/docs/providers/gemini.md index 628eeb99955..8442697b552 100644 --- a/docs/my-website/docs/providers/gemini.md +++ b/docs/my-website/docs/providers/gemini.md @@ -100,4 +100,3 @@ print(content) | gemini-pro | `completion('gemini/gemini-pro', messages)` | `os.environ['GEMINI_API_KEY']` | | gemini-1.5-pro | `completion('gemini/gemini-1.5-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-1.5-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 05720db134f..1f20b76b92d 100644 --- a/docs/my-website/docs/providers/vertex.md +++ b/docs/my-website/docs/providers/vertex.md @@ -175,18 +175,15 @@ 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)`| +## Gemini 1.5 Pro (and Vision) | Model Name | Function Call | |------------------|--------------------------------------| -| gemini-1.5-pro-vision | `completion('gemini-pro-vision', messages)`, `completion('vertex_ai/gemini-pro-vision', messages)`| +| gemini-1.5-pro | `completion('gemini-1.5-pro', messages)`, `completion('vertex_ai/gemini-pro', messages)` | diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index d3eb93d7267..974c2658e98 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -1002,20 +1002,6 @@ "supports_function_calling": true, "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" }, - "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", - "supports_function_calling": true, - "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" - }, "textembedding-gecko": { "max_tokens": 3072, "max_input_tokens": 3072, @@ -1158,17 +1144,6 @@ "mode": "chat", "supports_function_calling": true, "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" - }, - "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", - "supports_function_calling": true, - "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" }, "command-r": { "max_tokens": 4096, diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 89862a4c1d4..cf86a2fc951 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -111,7 +111,6 @@ def test_vertex_ai(): "text-bison@001", "gemini-1.5-pro", "gemini-1.5-pro-preview-0215", - "gemini-1.5-pro-vision", ]: # our account does not have access to this model continue @@ -164,7 +163,6 @@ def test_vertex_ai_stream(): "text-bison@001", "gemini-1.5-pro", "gemini-1.5-pro-preview-0215", - "gemini-1.5-pro-vision", ]: # our account does not have access to this model continue @@ -218,7 +216,6 @@ async def test_async_vertexai_response(): "text-bison@001", "gemini-1.5-pro", "gemini-1.5-pro-preview-0215", - "gemini-1.5-pro-vision", ]: # our account does not have access to this model continue @@ -263,7 +260,6 @@ async def test_async_vertexai_streaming_response(): "text-bison@001", "gemini-1.5-pro", "gemini-1.5-pro-preview-0215", - "gemini-1.5-pro-vision", ]: # our account does not have access to this model continue diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index d3eb93d7267..974c2658e98 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -1002,20 +1002,6 @@ "supports_function_calling": true, "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" }, - "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", - "supports_function_calling": true, - "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" - }, "textembedding-gecko": { "max_tokens": 3072, "max_input_tokens": 3072, @@ -1158,17 +1144,6 @@ "mode": "chat", "supports_function_calling": true, "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" - }, - "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", - "supports_function_calling": true, - "source": "https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models#foundation_models" }, "command-r": { "max_tokens": 4096,