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

fix(router): restore debug log for missing deployment_model_name; warn on unknown affinity flags

Browse source 
Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>
This commit is contained in:
Sameer Kankute 2026-03-19 15:39:35 +05:30
parent 528daa8cf4
commit 6af74f6594
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/router_utils/pre_call_checks/deployment_affinity_check.py
View file

@ -38,6 +38,9 @@ class DeploymentAffinityCheck(CustomLogger):
"""
CACHE_KEY_PREFIX = "deployment_affinity:v1"
VALID_FLAGS = frozenset(
{"deployment_affinity", "responses_api_deployment_check", "session_affinity"}
)
def __init__(
self,
@ -57,6 +60,15 @@ class DeploymentAffinityCheck(CustomLogger):
self.model_group_affinity_config: Dict[str, List[str]] = (
model_group_affinity_config or {}
)
for group, flags in self.model_group_affinity_config.items():
unknown = set(flags) - self.VALID_FLAGS
if unknown:
verbose_router_logger.warning(
"DeploymentAffinityCheck: unknown flag(s) %s for model group '%s'; will be ignored. Valid flags: %s",
unknown,
group,
self.VALID_FLAGS,
)
def _get_effective_flags(
self, model_group: str
@ -440,6 +452,9 @@ class DeploymentAffinityCheck(CustomLogger):
break
if not deployment_model_name:
verbose_router_logger.debug(
"DeploymentAffinityCheck: deployment_model_name missing in metadata; skipping affinity cache update."
)
return None
# Resolve effective flags for this model group