diff --git a/litellm/router.py b/litellm/router.py index d72f3ea5ecf5..fcbd3a230751 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -2944,7 +2944,7 @@ async def deployment_callback_on_success( 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