ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ishaan Jaff 2024-07-11 13:04:18 -07:00
parent 341f88d191
commit 46493303ed
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
litellm/tests/test_get_model_info.py
View File

@ -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