From d3756c477d5675ca13ec5f4b0e0e20d7ad8d6508 Mon Sep 17 00:00:00 2001
From: thomas <thomas@baichuan-inc.com>
Date: Fri, 18 Jul 2025 16:53:46 +0800
Subject: [PATCH] fix: ensure ResponseUsage token fields are int, not None
 (fixes #1179)

---
 src/agents/models/chatcmpl_stream_handler.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/agents/models/chatcmpl_stream_handler.py b/src/agents/models/chatcmpl_stream_handler.py
index 6133af344c..eab4c291b9 100644
--- a/src/agents/models/chatcmpl_stream_handler.py
+++ b/src/agents/models/chatcmpl_stream_handler.py
@@ -493,9 +493,9 @@ async def handle_stream(
         final_response.output = outputs
         final_response.usage = (
             ResponseUsage(
-                input_tokens=usage.prompt_tokens,
-                output_tokens=usage.completion_tokens,
-                total_tokens=usage.total_tokens,
+                input_tokens=usage.prompt_tokens or 0,
+                output_tokens=usage.completion_tokens or 0,
+                total_tokens=usage.total_tokens or 0,
                 output_tokens_details=OutputTokensDetails(
                     reasoning_tokens=usage.completion_tokens_details.reasoning_tokens
                     if usage.completion_tokens_details