docs: Mintlify MDX docs + docs.json; bump Speakeasy to 1.787.0 (#382)

src/openrouter/components/chatstreamchunk.py

@@ -12,11 +12,9 @@ from openrouter.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from openrouter.utils import validate_open_enum
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import List, Literal, Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class ChatStreamChunkMetadataTypedDict(TypedDict):
@@ -31,12 +29,28 @@ class ChatStreamChunkMetadataTypedDict(TypedDict):
 class ChatStreamChunkMetadata(BaseModel):
     r"""Structured error metadata"""
 
-    error_type: Annotated[APIErrorType, PlainValidator(validate_open_enum(False))]
+    error_type: APIErrorType
     r"""Canonical OpenRouter error type, stable across all API formats"""
 
     provider_code: Optional[str] = None
     r"""Upstream provider-specific error code, when available"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["provider_code"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k, serialized.get(n))
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 class ChatStreamChunkErrorTypedDict(TypedDict):
     r"""Error information"""
@@ -61,6 +75,22 @@ class ChatStreamChunkError(BaseModel):
     metadata: Optional[ChatStreamChunkMetadata] = None
     r"""Structured error metadata"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["metadata"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k, serialized.get(n))
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 ChatStreamChunkObject = Literal["chat.completion.chunk",]
 
@@ -121,36 +151,33 @@ class ChatStreamChunk(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "error",
-            "openrouter_metadata",
-            "service_tier",
-            "system_fingerprint",
-            "usage",
-        ]
-        nullable_fields = ["service_tier"]
-        null_default_fields = []
-
+        optional_fields = set(
+            [
+                "error",
+                "openrouter_metadata",
+                "service_tier",
+                "system_fingerprint",
+                "usage",
+            ]
+        )
+        nullable_fields = set(["service_tier"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
-            val = serialized.get(k)
-            serialized.pop(k, None)
+            val = serialized.get(k, serialized.get(n))
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
 
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m