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

This commit is contained in:
Krrish Dholakia 2024-02-24 09:33:21 -08:00
parent d53fd118f2
commit 7cb86c4e0f

View file

@ -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(