Merge pull request #3186 from BerriAI/litellm_fix_gemini_pro_latest_bug

[Fix] completion(model="gemini/gemini-pro-1.5-latest" raises Exception

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)

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):