From 8b08f19f02411e13f90d421bbc012ccd9b38d3a1 Mon Sep 17 00:00:00 2001 From: akshatshahh Date: Tue, 20 Oct 2026 13:49:55 -0700 Subject: [PATCH] fix(router): type model_info in session-affinity cache update for basedpyright gate --- .../pre_call_checks/deployment_affinity_check.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/router_utils/pre_call_checks/deployment_affinity_check.py b/litellm/router_utils/pre_call_checks/deployment_affinity_check.py index c208ebe3fc9..9432f8d85e4 100644 --- a/litellm/router_utils/pre_call_checks/deployment_affinity_check.py +++ b/litellm/router_utils/pre_call_checks/deployment_affinity_check.py @@ -639,7 +639,10 @@ class DeploymentAffinityCheck(CustomLogger): # likely a non-router call or a call path that doesn't support affinity. return None - model_id: Final = model_info.get("id") + typed_model_info: Final = cast( # cast-ok: router kwargs payloads are str-keyed dicts + dict[str, object], model_info + ) + model_id: Final = typed_model_info.get("id") if not model_id: verbose_router_logger.warning("DeploymentAffinityCheck: model_id missing; skipping affinity cache update.") return None