feat: adding optional Usage for chunks

Portkey-AI · Dec 13, 2024 · d2e8cae · d2e8cae
1 parent fc75083
commit d2e8cae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/portkey_ai/api_resources/types/complete_type.py b/portkey_ai/api_resources/types/complete_type.py
@@ -74,6 +74,7 @@ class TextCompletionChunk(BaseModel, extra="allow"):
     created: Optional[int] = None
     model: Optional[str] = None
     provider: Optional[str] = None
+    usage: Optional[CompletionUsage] = None
     choices: List[TextChoice]
 
     def __str__(self):