From 023e02bf73a3fc32c2081dfe9446819f9c48d424 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Fri, 13 Feb 2026 12:27:56 +0530 Subject: [PATCH] Fix : Cost Tracking Callback Fails with KeyError: 'stream' --- litellm/proxy/hooks/proxy_track_cost_callback.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/hooks/proxy_track_cost_callback.py b/litellm/proxy/hooks/proxy_track_cost_callback.py index dab5fb1bfd5..37b79e6d065 100644 --- a/litellm/proxy/hooks/proxy_track_cost_callback.py +++ b/litellm/proxy/hooks/proxy_track_cost_callback.py @@ -202,8 +202,8 @@ class _ProxyDBLogger(CustomLogger): max_budget=end_user_max_budget, ) else: - if kwargs["stream"] is not True or ( - kwargs["stream"] is True and "complete_streaming_response" in kwargs + if kwargs.get("stream") is not True or ( + kwargs.get("stream") is True and "complete_streaming_response" in kwargs ): if sl_object is not None: cost_tracking_failure_debug_info: Union[dict, str] = (