scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

litellm_fix(test): allow comment field in schema and exclude robotics models from tpm check (#20139)

Browse source
This commit is contained in:
shin-bot-litellm 2026-01-31 07:32:33 -08:00 • committed by GitHub
parent 14a5706131
commit 7db4594200
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/test_utils.py
View file

@ -639,6 +639,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
"rpd": {"type": "number"},
"rpm": {"type": "number"},
"source": {"type": "string"},
"comment": {"type": "string"},
"supports_assistant_prefill": {"type": "boolean"},
"supports_audio_input": {"type": "boolean"},
"supports_audio_output": {"type": "boolean"},
@ -841,6 +842,7 @@ def test_get_model_info_gemini():
and not "learnlm" in model
and not "imagen" in model
and not "veo" in model
and not "robotics" in model
):
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"