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

fix: fix learnlm test

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-13 10:54:09 -07:00
parent 88dbfc489d
commit a2f8e1a2a3
3 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/model_prices_and_context_window_backup.json
View file

@ -4607,7 +4607,7 @@
"source": "https://aistudio.google.com", "source": "https://aistudio.google.com",
"supports_tool_choice": true "supports_tool_choice": true
}, },
"gemini/learnIm-1.5-pro-experimental": { "gemini/learnlm-1.5-pro-experimental": {
"max_tokens": 8192, "max_tokens": 8192,
"max_input_tokens": 32767, "max_input_tokens": 32767,
"max_output_tokens": 8192, "max_output_tokens": 8192,

2
model_prices_and_context_window.json
View file

@ -4607,7 +4607,7 @@
"source": "https://aistudio.google.com", "source": "https://aistudio.google.com",
"supports_tool_choice": true "supports_tool_choice": true
}, },
"gemini/learnIm-1.5-pro-experimental": { "gemini/learnlm-1.5-pro-experimental": {
"max_tokens": 8192, "max_tokens": 8192,
"max_input_tokens": 32767, "max_input_tokens": 32767,
"max_output_tokens": 8192, "max_output_tokens": 8192,

6
tests/local_testing/test_get_model_info.py
View file

@ -121,7 +121,11 @@ def test_get_model_info_gemini():
model_map = litellm.model_cost model_map = litellm.model_cost
for model, info in model_map.items(): for model, info in model_map.items():
if model.startswith("gemini/") and not "gemma" in model: if (
model.startswith("gemini/")
and not "gemma" in model
and not "learnlm" in model
):
assert info.get("tpm") is not None, f"{model} does not have tpm" assert info.get("tpm") is not None, f"{model} does not have tpm"
assert info.get("rpm") is not None, f"{model} does not have rpm" assert info.get("rpm") is not None, f"{model} does not have rpm"