From 63805873e7e65d8a8609bfc5a5ea019f98ad7b93 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 19 Apr 2024 21:19:07 -0700 Subject: [PATCH 1/2] fix - supports_vision should not raise Exception --- litellm/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 1f23fac981d..e230675e68a 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -4207,9 +4207,7 @@ def supports_vision(model: str): return True return False else: - raise Exception( - f"Model not in model_prices_and_context_window.json. You passed model={model}." - ) + return False def supports_parallel_function_calling(model: str): From 81e4d593572ef9e3d85af42566fd2e3c2dd9ac73 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 19 Apr 2024 21:22:19 -0700 Subject: [PATCH 2/2] test - gemini-1.5-pro-latest --- litellm/tests/test_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index fff34642543..09053cf17b3 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -2489,7 +2489,7 @@ def test_completion_deep_infra_mistral(): # Gemini tests def test_completion_gemini(): litellm.set_verbose = True - model_name = "gemini/gemini-pro" + model_name = "gemini/gemini-1.5-pro-latest" messages = [{"role": "user", "content": "Hey, how's it going?"}] try: response = completion(model=model_name, messages=messages)