phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Litellm dev 01 06 2025 p3 (#7596)

Browse source 
* build(model_prices_and_context_window.json): add gemini-1.5-pro 'supports_vision' = true

Fixes https://github.com/BerriAI/litellm/issues/7592

* build(model_prices_and_context_window.json): add new mistral models pricing + model info
This commit is contained in:
Krish Dholakia 2025-01-06 20:44:04 -08:00 • committed by GitHub
parent 65283747c5
commit b44aef5110
3 changed files with 90 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_utils.py
View file

@ -1418,3 +1418,11 @@ def test_get_valid_models_default(monkeypatch):
monkeypatch.setenv("FIREWORKS_API_KEY", "sk-1234")
valid_models = get_valid_models()
assert len(valid_models) > 0
def test_supports_vision_gemini():
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
from litellm.utils import supports_vision
assert supports_vision("gemini-1.5-pro") is True