From 50828b36a41175f5911ad28005eba1b3cf793172 Mon Sep 17 00:00:00 2001 From: Lalith Vardhan Date: Sun, 25 Oct 2026 09:51:09 -0700 Subject: [PATCH] fix(router): give model_name rebinds in _text_completion a rebind-ok reason (LIT010 gate) --- litellm/router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/router.py b/litellm/router.py index adc4ab6f4c3..98e1c1d51f1 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -5737,7 +5737,7 @@ class Router: prompt: str, **kwargs: object, # kwargs-ok: forwarded verbatim to litellm.text_completion, like _completion ) -> TextCompletionResponse | TextCompletionStreamWrapper | ModelResponse | CustomStreamWrapper: - model_name: str | None = None + model_name: str | None = None # rebind-ok: set once the deployment is picked; read in except for fail_calls try: verbose_router_logger.debug("Inside _text_completion()- model: %s; kwargs: %s", model, kwargs) deployment: Final = self.get_available_deployment( @@ -5749,7 +5749,7 @@ class Router: self._update_kwargs_with_deployment(deployment=deployment, kwargs=kwargs) data: Final = deployment["litellm_params"].copy() - model_name = data["model"] + model_name = data["model"] # rebind-ok: assigned inside try so except can attribute the failure self.total_calls[model_name] += 1 ### DEPLOYMENT-SPECIFIC PRE-CALL CHECKS ### (e.g. update rpm pre-call. Raise error, if deployment over limit)