fix: add usage to chunk completion

Author: maschad

src/atoma_sdk/models/chatcompletionchunk.py

@@ -22,11 +22,11 @@ class ChatCompletionChunkTypedDict(TypedDict):
     r"""The model used for the chat completion."""
     id: str
     r"""The unique identifier for the chat completion."""
-    system_fingerprint: str
+    system_fingerprint: str | None = None
     r"""The system fingerprint for the chat completion."""
-    usage: UsageTypedDict
+    usage: UsageTypedDict | None = None
     r"""The usage information for the chat completion."""
-    service_tier: str
+    service_tier: str | None = None
     r"""The service tier for the chat completion."""
 
 
@@ -43,15 +43,12 @@ class ChatCompletionChunk(BaseModel):
     model: str
     r"""The model used for the chat completion."""
 
-    id: str
-    r"""The unique identifier for the chat completion."""
-
-    system_fingerprint: str
+    system_fingerprint: str | None = None
     r"""The system fingerprint for the chat completion."""
 
-    usage: Usage
+    usage: Usage | None = None
     r"""The usage information for the chat completion."""
 
-    service_tier: str
+    service_tier: str | None = None
     r"""The service tier for the chat completion."""
 

src/atoma_sdk/models/chatcompletionresponse.py

@@ -21,6 +21,11 @@ class ChatCompletionResponseTypedDict(TypedDict):
     system_fingerprint: NotRequired[Nullable[str]]
     r"""The system fingerprint for the completion, if applicable."""
     usage: NotRequired[Nullable[CompletionUsageTypedDict]]
+    r"""The usage information for the chat completion."""
+    service_tier: NotRequired[Nullable[str]]
+    r"""The service tier for the chat completion."""
+    object: NotRequired[Nullable[str]]
+    r"""The object type for the chat completion."""
 
 
 class ChatCompletionResponse(BaseModel):
@@ -40,6 +45,13 @@ class ChatCompletionResponse(BaseModel):
     r"""The system fingerprint for the completion, if applicable."""
 
     usage: OptionalNullable[CompletionUsage] = UNSET
+    r"""The usage information for the chat completion."""
+
+    service_tier: OptionalNullable[str] = UNSET
+    r"""The service tier for the chat completion."""
+
+    object: OptionalNullable[str] = UNSET
+    r"""The object type for the chat completion."""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):