From baf3898f869588bb0c15fcaff66c24cc3f401de9 Mon Sep 17 00:00:00 2001
From: Dev Sharma <12devsharma10c@gmail.com>
Date: Wed, 6 Aug 2025 00:23:54 +0530
Subject: [PATCH] Replace None with 0 in CompletionUsage token counts

This PR updates the CompletionUsage model to replace None values with 0 for prompt_tokens, completion_tokens, and total_tokens.
---
 src/openai/types/completion_usage.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/openai/types/completion_usage.py b/src/openai/types/completion_usage.py
index d8c4e84cf7..6285c1a095 100644
--- a/src/openai/types/completion_usage.py
+++ b/src/openai/types/completion_usage.py
@@ -1,4 +1,20 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+# openai/types/completion_usage.py
+
+from typing import Optional
+from pydantic import BaseModel, Field
+
+class CompletionUsage(BaseModel):
+    prompt_tokens: int = Field(default=0)
+    completion_tokens: int = Field(default=0)
+    total_tokens: int = Field(default=0)
+
+    def __init__(self, **data):
+        # Replace None with 0 for any token counts
+        for key in ["prompt_tokens", "completion_tokens", "total_tokens"]:
+            if data.get(key) is None:
+                data[key] = 0
+        super().__init__(**data)
 
 from typing import Optional