From 7cb86c4e0f3ef32aca6f5fd9cedbfbedf618961c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 24 Feb 2024 09:33:21 -0800 Subject: [PATCH] fix(router.py): fix text completion error logging --- litellm/router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/router.py b/litellm/router.py index 06efd4c34..6f33d0b0d 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -830,8 +830,8 @@ class Router: verbose_router_logger.info( f"litellm.atext_completion(model={model})\033[31m Exception {str(e)}\033[0m" ) - if model_name is not None: - self.fail_calls[model_name] += 1 + if model is not None: + self.fail_calls[model] += 1 raise e def embedding(