mistralai 1.9.6__py3-none-any.whl → 1.9.8__py3-none-any.whl

mistralai_azure/models/__init__.py

@@ -1,109 +1,134 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-from .assistantmessage import (
-    AssistantMessage,
-    AssistantMessageContent,
-    AssistantMessageContentTypedDict,
-    AssistantMessageRole,
-    AssistantMessageTypedDict,
-)
-from .chatcompletionchoice import (
-    ChatCompletionChoice,
-    ChatCompletionChoiceFinishReason,
-    ChatCompletionChoiceTypedDict,
-)
-from .chatcompletionrequest import (
-    ChatCompletionRequest,
-    ChatCompletionRequestMessages,
-    ChatCompletionRequestMessagesTypedDict,
-    ChatCompletionRequestStop,
-    ChatCompletionRequestStopTypedDict,
-    ChatCompletionRequestToolChoice,
-    ChatCompletionRequestToolChoiceTypedDict,
-    ChatCompletionRequestTypedDict,
-)
-from .chatcompletionresponse import (
-    ChatCompletionResponse,
-    ChatCompletionResponseTypedDict,
-)
-from .chatcompletionstreamrequest import (
-    ChatCompletionStreamRequest,
-    ChatCompletionStreamRequestToolChoice,
-    ChatCompletionStreamRequestToolChoiceTypedDict,
-    ChatCompletionStreamRequestTypedDict,
-    Messages,
-    MessagesTypedDict,
-    Stop,
-    StopTypedDict,
-)
-from .completionchunk import CompletionChunk, CompletionChunkTypedDict
-from .completionevent import CompletionEvent, CompletionEventTypedDict
-from .completionresponsestreamchoice import (
-    CompletionResponseStreamChoice,
-    CompletionResponseStreamChoiceTypedDict,
-    FinishReason,
-)
-from .contentchunk import ContentChunk, ContentChunkTypedDict
-from .deltamessage import Content, ContentTypedDict, DeltaMessage, DeltaMessageTypedDict
-from .function import Function, FunctionTypedDict
-from .functioncall import (
-    Arguments,
-    ArgumentsTypedDict,
-    FunctionCall,
-    FunctionCallTypedDict,
-)
-from .functionname import FunctionName, FunctionNameTypedDict
-from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
-from .imageurl import ImageURL, ImageURLTypedDict
-from .imageurlchunk import (
-    ImageURLChunk,
-    ImageURLChunkImageURL,
-    ImageURLChunkImageURLTypedDict,
-    ImageURLChunkType,
-    ImageURLChunkTypedDict,
-)
-from .jsonschema import JSONSchema, JSONSchemaTypedDict
-from .prediction import Prediction, PredictionTypedDict
-from .referencechunk import ReferenceChunk, ReferenceChunkType, ReferenceChunkTypedDict
-from .responseformat import ResponseFormat, ResponseFormatTypedDict
-from .responseformats import ResponseFormats
-from .sdkerror import SDKError
-from .security import Security, SecurityTypedDict
-from .systemmessage import (
-    Role,
-    SystemMessage,
-    SystemMessageContent,
-    SystemMessageContentTypedDict,
-    SystemMessageTypedDict,
-)
-from .textchunk import TextChunk, TextChunkTypedDict, Type
-from .tool import Tool, ToolTypedDict
-from .toolcall import ToolCall, ToolCallTypedDict
-from .toolchoice import ToolChoice, ToolChoiceTypedDict
-from .toolchoiceenum import ToolChoiceEnum
-from .toolmessage import (
-    ToolMessage,
-    ToolMessageContent,
-    ToolMessageContentTypedDict,
-    ToolMessageRole,
-    ToolMessageTypedDict,
-)
-from .tooltypes import ToolTypes
-from .usageinfo import UsageInfo, UsageInfoTypedDict
-from .usermessage import (
-    UserMessage,
-    UserMessageContent,
-    UserMessageContentTypedDict,
-    UserMessageRole,
-    UserMessageTypedDict,
-)
-from .validationerror import (
-    Loc,
-    LocTypedDict,
-    ValidationError,
-    ValidationErrorTypedDict,
-)
+from typing import TYPE_CHECKING
+from importlib import import_module
 
+if TYPE_CHECKING:
+    from .assistantmessage import (
+        AssistantMessage,
+        AssistantMessageContent,
+        AssistantMessageContentTypedDict,
+        AssistantMessageRole,
+        AssistantMessageTypedDict,
+    )
+    from .chatcompletionchoice import (
+        ChatCompletionChoice,
+        ChatCompletionChoiceFinishReason,
+        ChatCompletionChoiceTypedDict,
+    )
+    from .chatcompletionrequest import (
+        ChatCompletionRequest,
+        ChatCompletionRequestMessages,
+        ChatCompletionRequestMessagesTypedDict,
+        ChatCompletionRequestStop,
+        ChatCompletionRequestStopTypedDict,
+        ChatCompletionRequestToolChoice,
+        ChatCompletionRequestToolChoiceTypedDict,
+        ChatCompletionRequestTypedDict,
+    )
+    from .chatcompletionresponse import (
+        ChatCompletionResponse,
+        ChatCompletionResponseTypedDict,
+    )
+    from .chatcompletionstreamrequest import (
+        ChatCompletionStreamRequest,
+        ChatCompletionStreamRequestToolChoice,
+        ChatCompletionStreamRequestToolChoiceTypedDict,
+        ChatCompletionStreamRequestTypedDict,
+        Messages,
+        MessagesTypedDict,
+        Stop,
+        StopTypedDict,
+    )
+    from .completionchunk import CompletionChunk, CompletionChunkTypedDict
+    from .completionevent import CompletionEvent, CompletionEventTypedDict
+    from .completionresponsestreamchoice import (
+        CompletionResponseStreamChoice,
+        CompletionResponseStreamChoiceTypedDict,
+        FinishReason,
+    )
+    from .contentchunk import ContentChunk, ContentChunkTypedDict
+    from .deltamessage import (
+        Content,
+        ContentTypedDict,
+        DeltaMessage,
+        DeltaMessageTypedDict,
+    )
+    from .documenturlchunk import (
+        DocumentURLChunk,
+        DocumentURLChunkType,
+        DocumentURLChunkTypedDict,
+    )
+    from .filechunk import FileChunk, FileChunkTypedDict
+    from .function import Function, FunctionTypedDict
+    from .functioncall import (
+        Arguments,
+        ArgumentsTypedDict,
+        FunctionCall,
+        FunctionCallTypedDict,
+    )
+    from .functionname import FunctionName, FunctionNameTypedDict
+    from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
+    from .imageurl import ImageURL, ImageURLTypedDict
+    from .imageurlchunk import (
+        ImageURLChunk,
+        ImageURLChunkImageURL,
+        ImageURLChunkImageURLTypedDict,
+        ImageURLChunkType,
+        ImageURLChunkTypedDict,
+    )
+    from .jsonschema import JSONSchema, JSONSchemaTypedDict
+    from .mistralpromptmode import MistralPromptMode
+    from .ocrimageobject import OCRImageObject, OCRImageObjectTypedDict
+    from .ocrpagedimensions import OCRPageDimensions, OCRPageDimensionsTypedDict
+    from .ocrpageobject import OCRPageObject, OCRPageObjectTypedDict
+    from .ocrrequest import Document, DocumentTypedDict, OCRRequest, OCRRequestTypedDict
+    from .ocrresponse import OCRResponse, OCRResponseTypedDict
+    from .ocrusageinfo import OCRUsageInfo, OCRUsageInfoTypedDict
+    from .prediction import Prediction, PredictionTypedDict
+    from .referencechunk import (
+        ReferenceChunk,
+        ReferenceChunkType,
+        ReferenceChunkTypedDict,
+    )
+    from .responseformat import ResponseFormat, ResponseFormatTypedDict
+    from .responseformats import ResponseFormats
+    from .sdkerror import SDKError
+    from .security import Security, SecurityTypedDict
+    from .systemmessage import (
+        Role,
+        SystemMessage,
+        SystemMessageContent,
+        SystemMessageContentTypedDict,
+        SystemMessageTypedDict,
+    )
+    from .textchunk import TextChunk, TextChunkTypedDict, Type
+    from .tool import Tool, ToolTypedDict
+    from .toolcall import ToolCall, ToolCallTypedDict
+    from .toolchoice import ToolChoice, ToolChoiceTypedDict
+    from .toolchoiceenum import ToolChoiceEnum
+    from .toolmessage import (
+        ToolMessage,
+        ToolMessageContent,
+        ToolMessageContentTypedDict,
+        ToolMessageRole,
+        ToolMessageTypedDict,
+    )
+    from .tooltypes import ToolTypes
+    from .usageinfo import UsageInfo, UsageInfoTypedDict
+    from .usermessage import (
+        UserMessage,
+        UserMessageContent,
+        UserMessageContentTypedDict,
+        UserMessageRole,
+        UserMessageTypedDict,
+    )
+    from .validationerror import (
+        Loc,
+        LocTypedDict,
+        ValidationError,
+        ValidationErrorTypedDict,
+    )
 
 __all__ = [
     "Arguments",
@@ -142,6 +167,13 @@ __all__ = [
     "ContentTypedDict",
     "DeltaMessage",
     "DeltaMessageTypedDict",
+    "Document",
+    "DocumentTypedDict",
+    "DocumentURLChunk",
+    "DocumentURLChunkType",
+    "DocumentURLChunkTypedDict",
+    "FileChunk",
+    "FileChunkTypedDict",
     "FinishReason",
     "Function",
     "FunctionCall",
@@ -164,6 +196,19 @@ __all__ = [
     "LocTypedDict",
     "Messages",
     "MessagesTypedDict",
+    "MistralPromptMode",
+    "OCRImageObject",
+    "OCRImageObjectTypedDict",
+    "OCRPageDimensions",
+    "OCRPageDimensionsTypedDict",
+    "OCRPageObject",
+    "OCRPageObjectTypedDict",
+    "OCRRequest",
+    "OCRRequestTypedDict",
+    "OCRResponse",
+    "OCRResponseTypedDict",
+    "OCRUsageInfo",
+    "OCRUsageInfoTypedDict",
     "Prediction",
     "PredictionTypedDict",
     "ReferenceChunk",
@@ -208,3 +253,153 @@ __all__ = [
     "ValidationError",
     "ValidationErrorTypedDict",
 ]
+
+_dynamic_imports: dict[str, str] = {
+    "AssistantMessage": ".assistantmessage",
+    "AssistantMessageContent": ".assistantmessage",
+    "AssistantMessageContentTypedDict": ".assistantmessage",
+    "AssistantMessageRole": ".assistantmessage",
+    "AssistantMessageTypedDict": ".assistantmessage",
+    "ChatCompletionChoice": ".chatcompletionchoice",
+    "ChatCompletionChoiceFinishReason": ".chatcompletionchoice",
+    "ChatCompletionChoiceTypedDict": ".chatcompletionchoice",
+    "ChatCompletionRequest": ".chatcompletionrequest",
+    "ChatCompletionRequestMessages": ".chatcompletionrequest",
+    "ChatCompletionRequestMessagesTypedDict": ".chatcompletionrequest",
+    "ChatCompletionRequestStop": ".chatcompletionrequest",
+    "ChatCompletionRequestStopTypedDict": ".chatcompletionrequest",
+    "ChatCompletionRequestToolChoice": ".chatcompletionrequest",
+    "ChatCompletionRequestToolChoiceTypedDict": ".chatcompletionrequest",
+    "ChatCompletionRequestTypedDict": ".chatcompletionrequest",
+    "ChatCompletionResponse": ".chatcompletionresponse",
+    "ChatCompletionResponseTypedDict": ".chatcompletionresponse",
+    "ChatCompletionStreamRequest": ".chatcompletionstreamrequest",
+    "ChatCompletionStreamRequestToolChoice": ".chatcompletionstreamrequest",
+    "ChatCompletionStreamRequestToolChoiceTypedDict": ".chatcompletionstreamrequest",
+    "ChatCompletionStreamRequestTypedDict": ".chatcompletionstreamrequest",
+    "Messages": ".chatcompletionstreamrequest",
+    "MessagesTypedDict": ".chatcompletionstreamrequest",
+    "Stop": ".chatcompletionstreamrequest",
+    "StopTypedDict": ".chatcompletionstreamrequest",
+    "CompletionChunk": ".completionchunk",
+    "CompletionChunkTypedDict": ".completionchunk",
+    "CompletionEvent": ".completionevent",
+    "CompletionEventTypedDict": ".completionevent",
+    "CompletionResponseStreamChoice": ".completionresponsestreamchoice",
+    "CompletionResponseStreamChoiceTypedDict": ".completionresponsestreamchoice",
+    "FinishReason": ".completionresponsestreamchoice",
+    "ContentChunk": ".contentchunk",
+    "ContentChunkTypedDict": ".contentchunk",
+    "Content": ".deltamessage",
+    "ContentTypedDict": ".deltamessage",
+    "DeltaMessage": ".deltamessage",
+    "DeltaMessageTypedDict": ".deltamessage",
+    "DocumentURLChunk": ".documenturlchunk",
+    "DocumentURLChunkType": ".documenturlchunk",
+    "DocumentURLChunkTypedDict": ".documenturlchunk",
+    "FileChunk": ".filechunk",
+    "FileChunkTypedDict": ".filechunk",
+    "Function": ".function",
+    "FunctionTypedDict": ".function",
+    "Arguments": ".functioncall",
+    "ArgumentsTypedDict": ".functioncall",
+    "FunctionCall": ".functioncall",
+    "FunctionCallTypedDict": ".functioncall",
+    "FunctionName": ".functionname",
+    "FunctionNameTypedDict": ".functionname",
+    "HTTPValidationError": ".httpvalidationerror",
+    "HTTPValidationErrorData": ".httpvalidationerror",
+    "ImageURL": ".imageurl",
+    "ImageURLTypedDict": ".imageurl",
+    "ImageURLChunk": ".imageurlchunk",
+    "ImageURLChunkImageURL": ".imageurlchunk",
+    "ImageURLChunkImageURLTypedDict": ".imageurlchunk",
+    "ImageURLChunkType": ".imageurlchunk",
+    "ImageURLChunkTypedDict": ".imageurlchunk",
+    "JSONSchema": ".jsonschema",
+    "JSONSchemaTypedDict": ".jsonschema",
+    "MistralPromptMode": ".mistralpromptmode",
+    "OCRImageObject": ".ocrimageobject",
+    "OCRImageObjectTypedDict": ".ocrimageobject",
+    "OCRPageDimensions": ".ocrpagedimensions",
+    "OCRPageDimensionsTypedDict": ".ocrpagedimensions",
+    "OCRPageObject": ".ocrpageobject",
+    "OCRPageObjectTypedDict": ".ocrpageobject",
+    "Document": ".ocrrequest",
+    "DocumentTypedDict": ".ocrrequest",
+    "OCRRequest": ".ocrrequest",
+    "OCRRequestTypedDict": ".ocrrequest",
+    "OCRResponse": ".ocrresponse",
+    "OCRResponseTypedDict": ".ocrresponse",
+    "OCRUsageInfo": ".ocrusageinfo",
+    "OCRUsageInfoTypedDict": ".ocrusageinfo",
+    "Prediction": ".prediction",
+    "PredictionTypedDict": ".prediction",
+    "ReferenceChunk": ".referencechunk",
+    "ReferenceChunkType": ".referencechunk",
+    "ReferenceChunkTypedDict": ".referencechunk",
+    "ResponseFormat": ".responseformat",
+    "ResponseFormatTypedDict": ".responseformat",
+    "ResponseFormats": ".responseformats",
+    "SDKError": ".sdkerror",
+    "Security": ".security",
+    "SecurityTypedDict": ".security",
+    "Role": ".systemmessage",
+    "SystemMessage": ".systemmessage",
+    "SystemMessageContent": ".systemmessage",
+    "SystemMessageContentTypedDict": ".systemmessage",
+    "SystemMessageTypedDict": ".systemmessage",
+    "TextChunk": ".textchunk",
+    "TextChunkTypedDict": ".textchunk",
+    "Type": ".textchunk",
+    "Tool": ".tool",
+    "ToolTypedDict": ".tool",
+    "ToolCall": ".toolcall",
+    "ToolCallTypedDict": ".toolcall",
+    "ToolChoice": ".toolchoice",
+    "ToolChoiceTypedDict": ".toolchoice",
+    "ToolChoiceEnum": ".toolchoiceenum",
+    "ToolMessage": ".toolmessage",
+    "ToolMessageContent": ".toolmessage",
+    "ToolMessageContentTypedDict": ".toolmessage",
+    "ToolMessageRole": ".toolmessage",
+    "ToolMessageTypedDict": ".toolmessage",
+    "ToolTypes": ".tooltypes",
+    "UsageInfo": ".usageinfo",
+    "UsageInfoTypedDict": ".usageinfo",
+    "UserMessage": ".usermessage",
+    "UserMessageContent": ".usermessage",
+    "UserMessageContentTypedDict": ".usermessage",
+    "UserMessageRole": ".usermessage",
+    "UserMessageTypedDict": ".usermessage",
+    "Loc": ".validationerror",
+    "LocTypedDict": ".validationerror",
+    "ValidationError": ".validationerror",
+    "ValidationErrorTypedDict": ".validationerror",
+}
+
+
+def __getattr__(attr_name: str) -> object:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"No {attr_name} found in _dynamic_imports for module name -> {__name__} "
+        )
+
+    try:
+        module = import_module(module_name, __package__)
+        result = getattr(module, attr_name)
+        return result
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import {attr_name} from {module_name}: {e}"
+        ) from e
+    except AttributeError as e:
+        raise AttributeError(
+            f"Failed to get {attr_name} from {module_name}: {e}"
+        ) from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)

mistralai_azure/models/assistantmessage.py

@@ -56,7 +56,7 @@ class AssistantMessage(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_azure/models/chatcompletionrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -17,8 +18,9 @@ from mistralai_azure.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from mistralai_azure.utils import get_discriminator
+from mistralai_azure.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -96,6 +98,8 @@ class ChatCompletionRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -144,6 +148,11 @@ class ChatCompletionRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
+
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -165,16 +174,24 @@ class ChatCompletionRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
             "safe_prompt",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_azure/models/chatcompletionresponse.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceTypedDict
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from mistralai_azure.types import BaseModel
-from typing import List, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import List
+from typing_extensions import TypedDict
 
 
 class ChatCompletionResponseTypedDict(TypedDict):
@@ -13,8 +13,8 @@ class ChatCompletionResponseTypedDict(TypedDict):
     object: str
     model: str
     usage: UsageInfoTypedDict
-    created: NotRequired[int]
-    choices: NotRequired[List[ChatCompletionChoiceTypedDict]]
+    created: int
+    choices: List[ChatCompletionChoiceTypedDict]
 
 
 class ChatCompletionResponse(BaseModel):
@@ -26,6 +26,6 @@ class ChatCompletionResponse(BaseModel):
 
     usage: UsageInfo
 
-    created: Optional[int] = None
+    created: int
 
-    choices: Optional[List[ChatCompletionChoice]] = None
+    choices: List[ChatCompletionChoice]

mistralai_azure/models/chatcompletionstreamrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -17,8 +18,9 @@ from mistralai_azure.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from mistralai_azure.utils import get_discriminator
+from mistralai_azure.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -91,6 +93,8 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -138,6 +142,11 @@ class ChatCompletionStreamRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
+
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -159,16 +168,24 @@ class ChatCompletionStreamRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
             "safe_prompt",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_azure/models/completionresponsestreamchoice.py

@@ -38,7 +38,7 @@ class CompletionResponseStreamChoice(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_azure/models/deltamessage.py

@@ -46,7 +46,7 @@ class DeltaMessage(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_azure/models/documenturlchunk.py

@@ -0,0 +1,62 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from pydantic import model_serializer
+from typing import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+DocumentURLChunkType = Literal["document_url"]
+
+
+class DocumentURLChunkTypedDict(TypedDict):
+    document_url: str
+    document_name: NotRequired[Nullable[str]]
+    r"""The filename of the document"""
+    type: NotRequired[DocumentURLChunkType]
+
+
+class DocumentURLChunk(BaseModel):
+    document_url: str
+
+    document_name: OptionalNullable[str] = UNSET
+    r"""The filename of the document"""
+
+    type: Optional[DocumentURLChunkType] = "document_url"
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["document_name", "type"]
+        nullable_fields = ["document_name"]
+        null_default_fields = []
+
+        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)
+
+            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
+
+        return m