From 0eacc03b55180ab010745efee6c401bb6a5bdca4 Mon Sep 17 00:00:00 2001
From: daleselaji-dev <265319989+daleselaji-dev@users.noreply.github.com>
Date: Fri, 28 Aug 2026 10:22:59 +0800
Subject: [PATCH] fix(streaming): annotate aggregated token counts

---
 litellm/litellm_core_utils/streaming_chunk_builder_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/litellm_core_utils/streaming_chunk_builder_utils.py b/litellm/litellm_core_utils/streaming_chunk_builder_utils.py
index 4c0b06322be..176b007cb6e 100644
--- a/litellm/litellm_core_utils/streaming_chunk_builder_utils.py
+++ b/litellm/litellm_core_utils/streaming_chunk_builder_utils.py
@@ -136,8 +136,8 @@ class _UsageBearingChunk(TypedDict, total=False):
 
 
 class _UsageSummary(TypedDict):
-    prompt_tokens: int | None
-    completion_tokens: int | None
+    prompt_tokens: int
+    completion_tokens: int
     cache_creation_input_tokens: int | None
     cache_read_input_tokens: int | None
     completion_tokens_details: CompletionTokensDetails | None