ommlds 0.0.0.dev450__tar.gz → 0.0.0.dev452__tar.gz

{ommlds-0.0.0.dev450/ommlds.egg-info → ommlds-0.0.0.dev452}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ommlds
-Version: 0.0.0.dev450
+Version: 0.0.0.dev452
 Summary: ommlds
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -14,8 +14,8 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omdev==0.0.0.dev450
-Requires-Dist: omlish==0.0.0.dev450
+Requires-Dist: omdev==0.0.0.dev452
+Requires-Dist: omlish==0.0.0.dev452
 Provides-Extra: all
 Requires-Dist: llama-cpp-python~=0.3; extra == "all"
 Requires-Dist: mlx~=0.29; extra == "all"

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/.omlish-manifests.json

@@ -18,7 +18,7 @@
     "module": ".minichain.backends.impls.anthropic.chat",
     "attr": null,
     "file": "ommlds/minichain/backends/impls/anthropic/chat.py",
-    "line": 30,
+    "line": 39,
     "value": {
       "!.minichain.registries.manifests.RegistryManifest": {
         "module": "ommlds.minichain.backends.impls.anthropic.chat",
@@ -47,9 +47,9 @@
             "claude-opus-4-1-20250805": null,
             "claude-opus-4-1": "claude-opus-4-1-20250805",
             "claude-opus": "claude-opus-4-1",
-            "claude-sonnet-4-20250514": null,
-            "claude-sonnet-4": "claude-sonnet-4-20250514",
-            "claude-sonnet": "claude-sonnet-4",
+            "claude-sonnet-4-5-20250929": null,
+            "claude-sonnet-4-5": "claude-sonnet-4-5-20250929",
+            "claude-sonnet": "claude-sonnet-4-5",
             "claude-3-5-haiku-latest": null,
             "claude-haiku-3-5-latest": "claude-3-5-haiku-latest",
             "claude-haiku-3-5": "claude-haiku-3-5-latest",
@@ -96,7 +96,7 @@
     "module": ".minichain.backends.impls.google.chat",
     "attr": null,
     "file": "ommlds/minichain/backends/impls/google/chat.py",
-    "line": 33,
+    "line": 35,
     "value": {
       "!.minichain.registries.manifests.RegistryManifest": {
         "module": "ommlds.minichain.backends.impls.google.chat",
@@ -170,7 +170,7 @@
     "module": ".minichain.backends.impls.llamacpp.chat",
     "attr": null,
     "file": "ommlds/minichain/backends/impls/llamacpp/chat.py",
-    "line": 32,
+    "line": 33,
     "value": {
       "!.minichain.registries.manifests.RegistryManifest": {
         "module": "ommlds.minichain.backends.impls.llamacpp.chat",
@@ -260,7 +260,7 @@
     "module": ".minichain.backends.impls.openai.chat",
     "attr": null,
     "file": "ommlds/minichain/backends/impls/openai/chat.py",
-    "line": 34,
+    "line": 37,
     "value": {
       "!.minichain.registries.manifests.RegistryManifest": {
         "module": "ommlds.minichain.backends.impls.openai.chat",
@@ -402,7 +402,7 @@
     "module": ".minichain.backends.impls.tinygrad.chat",
     "attr": null,
     "file": "ommlds/minichain/backends/impls/tinygrad/chat.py",
-    "line": 161,
+    "line": 166,
     "value": {
       "!.minichain.backends.strings.manifests.BackendStringsManifest": {
         "service_cls_names": [
@@ -435,7 +435,7 @@
     "module": ".minichain.backends.impls.transformers.transformers",
     "attr": null,
     "file": "ommlds/minichain/backends/impls/transformers/transformers.py",
-    "line": 41,
+    "line": 43,
     "value": {
       "!.minichain.registries.manifests.RegistryManifest": {
         "module": "ommlds.minichain.backends.impls.transformers.transformers",
@@ -452,7 +452,7 @@
     "module": ".minichain.backends.impls.transformers.transformers",
     "attr": null,
     "file": "ommlds/minichain/backends/impls/transformers/transformers.py",
-    "line": 127,
+    "line": 131,
     "value": {
       "!.minichain.registries.manifests.RegistryManifest": {
         "module": "ommlds.minichain.backends.impls.transformers.transformers",
@@ -592,7 +592,7 @@
     "module": ".tools.git",
     "attr": null,
     "file": "ommlds/tools/git.py",
-    "line": 172,
+    "line": 188,
     "value": {
       "!omdev.tools.git.messages.GitMessageGeneratorManifest": {
         "module": "ommlds.tools.git",
@@ -606,7 +606,7 @@
     "module": ".tools.ocr",
     "attr": "_CLI_MODULE",
     "file": "ommlds/tools/ocr.py",
-    "line": 83,
+    "line": 89,
     "value": {
       "!omdev.cli.types.CliModule": {
         "name": "ocr",

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/anthropic/protocol/_marshal.py

@@ -8,7 +8,7 @@ from .types import Content
 
 
 @lang.static_init
-def _install_standard_marshalling() -> None:
+def _install_standard_marshaling() -> None:
     for root_cls in [
         Content,
         Content.CacheControl,
@@ -17,7 +17,7 @@ def _install_standard_marshalling() -> None:
             msh.polymorphism_from_subclasses(
                 root_cls,
                 naming=msh.Naming.SNAKE,
-                strip_suffix='auto',
+                strip_suffix=msh.AutoStripSuffix,
             ),
             msh.FieldTypeTagging('type'),
         ))

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/anthropic/protocol/sse/_marshal.py

@@ -8,7 +8,7 @@ from .events import AnthropicSseDecoderEvents
 
 
 @lang.static_init
-def _install_standard_marshalling() -> None:
+def _install_standard_marshaling() -> None:
     for root_cls in [
         AnthropicSseDecoderEvents.Event,
         AnthropicSseDecoderEvents.ContentBlockStart.ContentBlock,

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/anthropic/protocol/sse/assemble.py

@@ -76,7 +76,7 @@ class AnthropicSseMessageAssembler(
             elif isinstance(ae, AnthropicSseDecoderEvents.MessageStop):
                 yield [Message(
                     id=ms.message.id,
-                    role=ms.message.role,
+                    role=ms.message.role,  # type: ignore[arg-type]
                     model=ms.message.model,
                     content=content,
                     stop_reason=dct['stop_reason'],

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/anthropic/protocol/types.py

@@ -1,3 +1,6 @@
+"""
+https://docs.claude.com/en/api/messages
+"""
 import typing as ta
 
 from omlish import dataclasses as dc
@@ -8,6 +11,22 @@ from omlish import marshal as msh
 ##
 
 
+def _set_class_marshal_options(cls):
+    msh.update_object_metadata(
+        cls,
+        field_defaults=msh.FieldMetadata(
+            options=msh.FieldOptions(
+                omit_if=lang.is_none,
+            ),
+        ),
+    )
+
+    return cls
+
+
+##
+
+
 class Content(lang.Abstract, lang.Sealed):
     class CacheControl(lang.Abstract, lang.Sealed):
         """https://docs.anthropic.com/en/docs/build-with-claude/prompt-caching"""
@@ -18,7 +37,7 @@ class Content(lang.Abstract, lang.Sealed):
 
 
 @dc.dataclass(frozen=True)
-@msh.update_fields_metadata(['cache_control'], omit_if=lang.is_none)
+@_set_class_marshal_options
 class Text(Content):
     text: str
 
@@ -28,7 +47,7 @@ class Text(Content):
 
 
 @dc.dataclass(frozen=True)
-@msh.update_fields_metadata(['cache_control'], omit_if=lang.is_none)
+@_set_class_marshal_options
 class ToolUse(Content):
     id: str
     name: str
@@ -40,6 +59,7 @@ class ToolUse(Content):
 
 
 @dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class ToolResult(Content):
     tool_use_id: str
     content: str
@@ -49,7 +69,7 @@ class ToolResult(Content):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
-@msh.update_object_metadata(field_defaults=msh.FieldMetadata(options=msh.FieldOptions(omit_if=lang.is_none)))
+@_set_class_marshal_options
 class CacheCreation:
     ephemeral_5m_input_tokens: int | None = None
     ephemeral_1h_input_tokens: int | None = None
@@ -59,7 +79,7 @@ class CacheCreation:
 
 
 @dc.dataclass(frozen=True, kw_only=True)
-@msh.update_object_metadata(field_defaults=msh.FieldMetadata(options=msh.FieldOptions(omit_if=lang.is_none)))
+@_set_class_marshal_options
 class Usage:
     input_tokens: int | None = None
     output_tokens: int | None = None
@@ -75,15 +95,15 @@ class Usage:
 
 
 @dc.dataclass(frozen=True, kw_only=True)
-@msh.update_object_metadata(field_defaults=msh.FieldMetadata(options=msh.FieldOptions(omit_if=lang.is_none)))
+@_set_class_marshal_options
 class Message:
     id: str | None = None
 
-    role: str | None = None
+    role: ta.Literal['user', 'assistant']
 
     model: str | None = None
 
-    content: ta.Sequence[Content] | None = None
+    content: str | ta.Sequence[Content] | None = None
 
     stop_reason: str | None = None
     stop_sequence: str | None = None
@@ -95,6 +115,7 @@ class Message:
 
 
 @dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class ToolSpec:
     name: str
     description: str
@@ -105,7 +126,7 @@ class ToolSpec:
 
 
 @dc.dataclass(frozen=True)
-@msh.update_object_metadata(field_defaults=msh.FieldMetadata(options=msh.FieldOptions(omit_if=lang.is_none)))
+@_set_class_marshal_options
 class MessagesRequest:
     model: str
 
@@ -113,7 +134,7 @@ class MessagesRequest:
 
     _: dc.KW_ONLY
 
-    system: ta.Sequence[Content] | None = None
+    system: str | ta.Sequence[Content] | None = None
 
     tools: ta.Sequence[ToolSpec] | None = None
 

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/google/protocol/_marshal.py

@@ -8,7 +8,7 @@ from .types import Value
 
 
 @lang.static_init
-def _install_standard_marshalling() -> None:
+def _install_standard_marshaling() -> None:
     msh.install_standard_factories(
         *msh.standard_polymorphism_factories(
             msh.polymorphism_from_subclasses(Value),

ommlds-0.0.0.dev452/ommlds/backends/openai/protocol/_common.py

@@ -0,0 +1,18 @@
+from omlish import lang
+from omlish import marshal as msh
+
+
+##
+
+
+def _set_class_marshal_options(cls):
+    msh.update_object_metadata(
+        cls,
+        field_defaults=msh.FieldMetadata(
+            options=msh.FieldOptions(
+                omit_if=lang.is_none,
+            ),
+        ),
+    )
+
+    return cls

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/openai/protocol/_marshal.py

@@ -10,7 +10,7 @@ from .chatcompletion.responseformat import ChatCompletionResponseFormat
 
 
 @lang.static_init
-def _install_standard_marshalling() -> None:
+def _install_standard_marshaling() -> None:
     for root_cls, tag_field in [
         (ChatCompletionContentPart, 'type'),
         (ChatCompletionMessage, 'role'),
@@ -20,7 +20,8 @@ def _install_standard_marshalling() -> None:
             msh.polymorphism_from_subclasses(
                 root_cls,
                 naming=msh.Naming.SNAKE,
-                strip_suffix='auto',
+                strip_suffix=msh.AutoStripSuffix,
             ),
             msh.FieldTypeTagging(tag_field),
+            unions='partial',
         ))

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/openai/protocol/chatcompletion/chunk.py

@@ -3,6 +3,7 @@ import typing as ta
 from omlish import dataclasses as dc
 from omlish import lang
 
+from .._common import _set_class_marshal_options
 from ..completionusage import CompletionUsage
 from .tokenlogprob import ChatCompletionTokenLogprob
 
@@ -11,6 +12,7 @@ from .tokenlogprob import ChatCompletionTokenLogprob
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionChunkChoiceDelta(lang.Final):
     content: str | None = None
 
@@ -46,6 +48,7 @@ class ChatCompletionChunkChoiceDelta(lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionChunkChoice(lang.Final):
     delta: ChatCompletionChunkChoiceDelta
 
@@ -70,6 +73,7 @@ class ChatCompletionChunkChoice(lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionChunk(lang.Final):
     id: str
 

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/openai/protocol/chatcompletion/contentpart.py

@@ -3,6 +3,8 @@ import typing as ta
 from omlish import dataclasses as dc
 from omlish import lang
 
+from .._common import _set_class_marshal_options
+
 
 ##
 
@@ -14,7 +16,8 @@ class ChatCompletionContentPart(lang.Abstract):
 #
 
 
-@dc.dataclass(frozen=True, kw_only=True)
+@dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class TextChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
     text: str
 
@@ -22,9 +25,10 @@ class TextChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
 #
 
 
-@dc.dataclass(frozen=True, kw_only=True)
+@dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class ImageUrlChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
-    @dc.dataclass(frozen=True, kw_only=True)
+    @dc.dataclass(frozen=True)
     class ImageUrl(lang.Final):
         url: str
         detail: ta.Literal[
@@ -39,7 +43,8 @@ class ImageUrlChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
 #
 
 
-@dc.dataclass(frozen=True, kw_only=True)
+@dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class FileChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
     @dc.dataclass(frozen=True, kw_only=True)
     class File(lang.Final):
@@ -53,9 +58,11 @@ class FileChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
 #
 
 
-@dc.dataclass(frozen=True, kw_only=True)
+@dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class InputAudioChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
-    @dc.dataclass(frozen=True, kw_only=True)
+    @dc.dataclass(frozen=True)
+    @_set_class_marshal_options
     class InputAudio(lang.Final):
         data: str
         format: ta.Literal[
@@ -69,6 +76,7 @@ class InputAudioChatCompletionContentPart(ChatCompletionContentPart, lang.Final)
 #
 
 
-@dc.dataclass(frozen=True, kw_only=True)
+@dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class RefusalChatCompletionContentPart(ChatCompletionContentPart, lang.Final):
     refusal: str

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/openai/protocol/chatcompletion/message.py

@@ -3,6 +3,7 @@ import typing as ta
 from omlish import dataclasses as dc
 from omlish import lang
 
+from .._common import _set_class_marshal_options
 from .contentpart import ChatCompletionContentPart
 from .contentpart import RefusalChatCompletionContentPart
 from .contentpart import TextChatCompletionContentPart
@@ -19,6 +20,7 @@ class ChatCompletionMessage(lang.Abstract, lang.Sealed):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class DeveloperChatCompletionMessage(ChatCompletionMessage, lang.Final):
     content: str | ta.Iterable[TextChatCompletionContentPart]
     name: str | None = None
@@ -28,6 +30,7 @@ class DeveloperChatCompletionMessage(ChatCompletionMessage, lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class SystemChatCompletionMessage(ChatCompletionMessage, lang.Final):
     content: str | ta.Iterable[TextChatCompletionContentPart]
     name: str | None = None
@@ -37,6 +40,7 @@ class SystemChatCompletionMessage(ChatCompletionMessage, lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class UserChatCompletionMessage(ChatCompletionMessage, lang.Final):
     content: str | ta.Iterable[ChatCompletionContentPart]
     name: str | None = None
@@ -46,8 +50,10 @@ class UserChatCompletionMessage(ChatCompletionMessage, lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class AssistantChatCompletionMessage(ChatCompletionMessage, lang.Final):
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class Audio(lang.Final):
         id: str
 
@@ -60,10 +66,12 @@ class AssistantChatCompletionMessage(ChatCompletionMessage, lang.Final):
     refusal: str | None = None
 
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class ToolCall(lang.Final):
         id: str
 
         @dc.dataclass(frozen=True, kw_only=True)
+        @_set_class_marshal_options
         class Function(lang.Final):
             arguments: str
             name: str
@@ -79,6 +87,7 @@ class AssistantChatCompletionMessage(ChatCompletionMessage, lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ToolChatCompletionMessage(ChatCompletionMessage, lang.Final):
     content: str | ta.Iterable[TextChatCompletionContentPart]
     tool_call_id: str
@@ -88,6 +97,7 @@ class ToolChatCompletionMessage(ChatCompletionMessage, lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class FunctionChatCompletionMessage(ChatCompletionMessage, lang.Final):
     content: str | None
     name: str

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/openai/protocol/chatcompletion/request.py

@@ -4,6 +4,7 @@ import typing as ta
 from omlish import dataclasses as dc
 from omlish import lang
 
+from .._common import _set_class_marshal_options
 from .contentpart import TextChatCompletionContentPart
 from .message import ChatCompletionMessage
 from .responseformat import ChatCompletionResponseFormat
@@ -13,6 +14,7 @@ from .responseformat import ChatCompletionResponseFormat
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionRequestWebSearchOptions(lang.Final):
     search_context_size: ta.Literal[
         'low',
@@ -21,8 +23,10 @@ class ChatCompletionRequestWebSearchOptions(lang.Final):
     ] | None = None
 
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class UserLocation(lang.Final):
         @dc.dataclass(frozen=True, kw_only=True)
+        @_set_class_marshal_options
         class Approximate(lang.Final):
             city: str | None = None
             country: str | None = None
@@ -30,7 +34,8 @@ class ChatCompletionRequestWebSearchOptions(lang.Final):
             timezone: str | None = None
 
         approximate: Approximate
-        type: ta.Literal['approximate'] | None = None
+
+        type: ta.Literal['approximate'] = dc.xfield('approximate', repr=False, kw_only=True)
 
     user_location: UserLocation | None = None
 
@@ -39,17 +44,21 @@ class ChatCompletionRequestWebSearchOptions(lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionRequestPrediction(lang.Final):
     content: str | ta.Iterable[TextChatCompletionContentPart]
-    type: ta.Literal['content'] | None = None
+
+    type: ta.Literal['content'] = dc.xfield('content', repr=False, kw_only=True)
 
 
 #
 
 
-@dc.dataclass(frozen=True, kw_only=True)
+@dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class ChatCompletionRequestTool(lang.Final):
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class Function(lang.Final):
         name: str
         description: str | None = None
@@ -57,26 +66,31 @@ class ChatCompletionRequestTool(lang.Final):
         strict: bool | None = None
 
     function: Function
-    type: ta.Literal['function'] | None = None
+
+    type: ta.Literal['function'] = dc.xfield('function', repr=False, kw_only=True)
 
 
 #
 
 
-@dc.dataclass(frozen=True, kw_only=True)
+@dc.dataclass(frozen=True)
+@_set_class_marshal_options
 class ChatCompletionRequestNamedToolChoice(lang.Final):
-    @dc.dataclass(frozen=True, kw_only=True)
+    @dc.dataclass(frozen=True)
+    @_set_class_marshal_options
     class Function(lang.Final):
         name: str
 
     function: Function
-    type: ta.Literal['function'] | None = None
+
+    type: ta.Literal['function'] = dc.xfield('function', repr=False, kw_only=True)
 
 
 #
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionRequestAudio(lang.Final):
     format: ta.Literal[
         'wav',
@@ -94,6 +108,7 @@ class ChatCompletionRequestAudio(lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionRequest(lang.Final):
     messages: ta.Iterable[ChatCompletionMessage]
 
@@ -143,7 +158,10 @@ class ChatCompletionRequest(lang.Final):
 
     store: bool | None = None
 
+    stream: bool | None = None
+
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class StreamOptions(lang.Final):
         include_usage: bool | None = None
 

{ommlds-0.0.0.dev450 → ommlds-0.0.0.dev452}/ommlds/backends/openai/protocol/chatcompletion/response.py

@@ -3,6 +3,7 @@ import typing as ta
 from omlish import dataclasses as dc
 from omlish import lang
 
+from .._common import _set_class_marshal_options
 from ..completionusage import CompletionUsage
 from .tokenlogprob import ChatCompletionTokenLogprob
 
@@ -11,16 +12,19 @@ from .tokenlogprob import ChatCompletionTokenLogprob
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionResponseMessage(lang.Final):
     content: str | None = None
     refusal: str | None = None
     role: ta.Literal['assistant'] = dc.xfield('assistant', repr=False)
 
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class Annotation(lang.Final):
         type: ta.Literal['url_citation'] = dc.xfield('url_citation', repr=False)
 
         @dc.dataclass(frozen=True, kw_only=True)
+        @_set_class_marshal_options
         class UrlCitation(lang.Final):
             end_index: int
             start_index: int
@@ -32,6 +36,7 @@ class ChatCompletionResponseMessage(lang.Final):
     annotations: ta.Sequence[Annotation] | None = None
 
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class Audio(lang.Final):
         id: str
         data: str
@@ -41,10 +46,12 @@ class ChatCompletionResponseMessage(lang.Final):
     audio: Audio | None = None
 
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class ToolCall(lang.Final):
         id: str
 
         @dc.dataclass(frozen=True, kw_only=True)
+        @_set_class_marshal_options
         class Function(lang.Final):
             arguments: str
             name: str
@@ -60,6 +67,7 @@ class ChatCompletionResponseMessage(lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionResponseChoice(lang.Final):
     finish_reason: ta.Literal[
         'stop',
@@ -71,6 +79,7 @@ class ChatCompletionResponseChoice(lang.Final):
     index: int
 
     @dc.dataclass(frozen=True, kw_only=True)
+    @_set_class_marshal_options
     class Logprobs(lang.Final):
         content: ta.Sequence[ChatCompletionTokenLogprob] | None = None
         refusal: ta.Sequence[ChatCompletionTokenLogprob] | None = None
@@ -84,6 +93,7 @@ class ChatCompletionResponseChoice(lang.Final):
 
 
 @dc.dataclass(frozen=True, kw_only=True)
+@_set_class_marshal_options
 class ChatCompletionResponse(lang.Final):
     id: str