fix(router.py): fix text completion error logging

2024-02-24 09:33:21 -08:00 · 2024-02-24 09:33:21 -08:00 · 7cb86c4e0f
commit 7cb86c4e0f
parent d53fd118f2
1 changed files with 2 additions and 2 deletions
--- 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(