scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

refactor(management): flatten query-param aliases iteratively to satisfy recursion gate

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-07-29 21:31:06 +00:00
parent c668067f4a
commit a0edc48c25
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/management_endpoints/management_v1/common.py
View file

@ -35,11 +35,13 @@ def problem_response(problem: ProblemDetail) -> JSONResponse:
)
def _flat_query_params(dependant: Dependant) -> tuple[str, ...]:
return (
*(field.alias for field in dependant.query_params),
*(alias for dependency in dependant.dependencies for alias in _flat_query_params(dependency)),
)
def _flat_query_param_aliases(dependant: Dependant) -> frozenset[str]:
frontier: tuple[Dependant, ...] = (dependant,)
aliases: frozenset[str] = frozenset()
while frontier:
aliases |= frozenset(field.alias for level in frontier for field in level.query_params)
frontier = tuple(sub for level in frontier for sub in level.dependencies)
return aliases
def _declared_query_params(request: Request) -> frozenset[str]:
@ -47,7 +49,7 @@ def _declared_query_params(request: Request) -> frozenset[str]:
dependant = getattr(route, "dependant", None)
if not isinstance(dependant, Dependant):
return frozenset()
return frozenset(_flat_query_params(dependant))
return _flat_query_param_aliases(dependant)
async def reject_unknown_query_params(request: Request) -> None: