test get mode info for gemini/gemini-1.5-flash

litellm/tests/test_get_model_info.py

@@ -1,13 +1,16 @@
 # What is this?
 ## Unit testing for the 'get_model_info()' function
-import os, sys, traceback
+import os
+import sys
+import traceback
 
 sys.path.insert(
     0, os.path.abspath("../..")
 )  # Adds the parent directory to the system path
+import pytest
+
 import litellm
 from litellm import get_model_info
-import pytest
 
 
 def test_get_model_info_simple_model_name():
@@ -37,3 +40,9 @@ def test_get_model_info_custom_llm_with_same_name_vllm():
         pytest.fail("Expected get model info to fail for an unmapped model/provider")
     except Exception:
         pass
+
+
+def test_get_model_info_shows_correct_supports_vision():
+    info = litellm.get_model_info("gemini/gemini-1.5-flash")
+    print("info", info)
+    assert info["supports_vision"] is True