scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

refactor(router): avoid local deployment reassignment

Browse source
This commit is contained in:
Yujong Lee 2026-08-30 18:17:58 -07:00 • committed by GitHub
parent 4dc7ca6ec3
commit 8c252d7b73
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/router.py
View file

@ -9576,9 +9576,14 @@ class Router:
# Returns: {"api_key": "sk-...", "custom_llm_provider": "openai", "model": "gpt-4o", ...}
"""
# Try to get deployment by model_id first
deployment = self.get_deployment(model_id=model_id)
if deployment is not None and team_id is not None and not self._deployment_usable_by_team(deployment, team_id):
deployment = None
deployment_candidate: Final = self.get_deployment(model_id=model_id)
deployment = (
deployment_candidate
if deployment_candidate is None
or team_id is None
or self._deployment_usable_by_team(deployment_candidate, team_id)
else None
)
# If not found, try by model_group_name
if deployment is None: