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

test(router): cover _declared_param_allowlist malformed declarations

Browse source
This commit is contained in:
Tin Chi Lo 2026-08-28 11:23:45 -07:00
parent 5338f293aa
commit 37e1bfecb1
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/test_router.py
View file

@ -11299,6 +11299,15 @@ class TestTierParamsTheTargetAccepts:
assert accepted == {"allowed_openai_params": ["seed"]}
def test_declared_param_allowlist_ignores_malformed_declarations(self):
"""A str is iterable, so without the type guard a YAML scalar mistake like
allowed_openai_params: reasoning_effort would allowlist single characters."""
assert litellm.Router._declared_param_allowlist({"allowed_openai_params": ["reasoning_effort", 3]}) == frozenset(
{"reasoning_effort"}
)
assert litellm.Router._declared_param_allowlist({"allowed_openai_params": "reasoning_effort"}) == frozenset()
assert litellm.Router._declared_param_allowlist({}) == frozenset()
def test_deployment_accepts_param_honors_deployment_allowlist(self):
deployment = {
"model_name": "x",