mistralai 1.9.7__py3-none-any.whl → 1.9.10__py3-none-any.whl

mistralai_azure/chat.py

@@ -41,6 +41,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -67,6 +68,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -106,6 +108,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -138,6 +141,7 @@ class Chat(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_chat",
                 oauth2_scopes=[],
@@ -212,6 +216,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -238,6 +243,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -277,6 +283,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -309,6 +316,7 @@ class Chat(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_chat",
                 oauth2_scopes=[],
@@ -391,6 +399,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -415,6 +424,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -456,6 +466,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -488,6 +499,7 @@ class Chat(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_completion_v1_chat_completions_post",
                 oauth2_scopes=[],
@@ -566,6 +578,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -590,6 +603,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -631,6 +645,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -663,6 +678,7 @@ class Chat(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_completion_v1_chat_completions_post",
                 oauth2_scopes=[],

mistralai_azure/httpclient.py

@@ -2,7 +2,6 @@
 
 # pyright: reportReturnType = false
 import asyncio
-from concurrent.futures import ThreadPoolExecutor
 from typing_extensions import Protocol, runtime_checkable
 import httpx
 from typing import Any, Optional, Union
@@ -116,21 +115,12 @@ def close_clients(
             pass
 
     if async_client is not None and not async_client_supplied:
-        is_async = False
         try:
-            asyncio.get_running_loop()
-            is_async = True
+            loop = asyncio.get_running_loop()
+            asyncio.run_coroutine_threadsafe(async_client.aclose(), loop)
         except RuntimeError:
-            pass
-
-        try:
-            # If this function is called in an async loop then start another
-            # loop in a separate thread to close the async http client.
-            if is_async:
-                with ThreadPoolExecutor(max_workers=1) as executor:
-                    future = executor.submit(asyncio.run, async_client.aclose())
-                    future.result()
-            else:
+            try:
                 asyncio.run(async_client.aclose())
-        except Exception:
-            pass
+            except RuntimeError:
+                # best effort
+                pass

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]