diff --git a/litellm/router.py b/litellm/router.py index d72f3ea5e..fcbd3a230 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -2944,7 +2944,7 @@ class Router: elif isinstance(id, int): id = str(id) - total_tokens = completion_response["usage"]["total_tokens"] + total_tokens = completion_response["usage"].get("total_tokens", 0) # ------------ # Setup values