fix cost tracking for whisper

2026-07-10 15:03:36 +00:00 · 2024-06-26 14:21:57 -07:00
parent ae431eb85d
commit e0258708c7
1 changed files with 1 additions and 1 deletions
@@ -29,7 +29,7 @@ def get_logging_payload(
    completion_start_time = kwargs.get("completion_start_time", end_time)
    call_type = kwargs.get("call_type")
    cache_hit = kwargs.get("cache_hit", False)
-    usage = response_obj["usage"]
+    usage = response_obj.get("usage", None) or {}
    if type(usage) == litellm.Usage:
        usage = dict(usage)
    id = response_obj.get("id", kwargs.get("litellm_call_id"))