tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-17 00:26:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_litellm/test_utils.py
+2
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"