style: apply black to litellm/router.py

Made-with: Cursor
This commit is contained in:
shivam 2026-04-21 15:08:01 -07:00
parent 378d80a9ad
commit 8a9457e0c0
No known key found for this signature in database

View file

@ -5366,14 +5366,17 @@ class Router:
e,
(litellm.ContextWindowExceededError, litellm.ContentPolicyViolationError),
)
_request_team_id: Optional[str] = (
kwargs.get("metadata", {}) or {}
).get("user_api_key_team_id")
_request_team_id: Optional[str] = (kwargs.get("metadata", {}) or {}).get(
"user_api_key_team_id"
)
# Use wildcard-aware lookup so order-based fallback also works for model
# groups resolved via pattern routing (e.g. `openai/*` -> `openai/gpt-4.1-mini`).
all_deployments = self.get_model_list(
model_name=original_model_group, team_id=_request_team_id
) or []
all_deployments = (
self.get_model_list(
model_name=original_model_group, team_id=_request_team_id
)
or []
)
_order_set: set = {
litellm.utils._get_deployment_order(d)
for d in all_deployments