mistralai 1.0.0rc1__py3-none-any.whl → 1.0.1__py3-none-any.whl

mistralai_azure/chat.py

@@ -1,10 +1,9 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from .basesdk import BaseSDK
-from mistralai_azure import models
+from mistralai_azure import models, utils
 from mistralai_azure._hooks import HookContext
 from mistralai_azure.types import OptionalNullable, UNSET
-import mistralai_azure.utils as utils
 from mistralai_azure.utils import eventstreaming
 from typing import Any, AsyncGenerator, Generator, List, Optional, Union
 
@@ -25,8 +24,8 @@ class Chat(BaseSDK):
         random_seed: OptionalNullable[int] = UNSET,
         response_format: Optional[Union[models.ResponseFormat, models.ResponseFormatTypedDict]] = None,
         tools: OptionalNullable[Union[List[models.Tool], List[models.ToolTypedDict]]] = UNSET,
-        safe_prompt: Optional[bool] = False,
         tool_choice: Optional[models.ToolChoice] = "auto",
+        safe_prompt: Optional[bool] = False,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -46,8 +45,8 @@ class Chat(BaseSDK):
         :param random_seed: The seed to use for random sampling. If set, different calls will generate deterministic results.
         :param response_format: 
         :param tools: 
-        :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param tool_choice: 
+        :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
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -72,8 +71,8 @@ class Chat(BaseSDK):
             messages=utils.get_pydantic_model(messages, List[models.Messages]),
             response_format=utils.get_pydantic_model(response_format, Optional[models.ResponseFormat]),
             tools=utils.get_pydantic_model(tools, OptionalNullable[List[models.Tool]]),
-            safe_prompt=safe_prompt,
             tool_choice=tool_choice,
+            safe_prompt=safe_prompt,
         )
         
         req = self.build_request(
@@ -141,8 +140,8 @@ class Chat(BaseSDK):
         random_seed: OptionalNullable[int] = UNSET,
         response_format: Optional[Union[models.ResponseFormat, models.ResponseFormatTypedDict]] = None,
         tools: OptionalNullable[Union[List[models.Tool], List[models.ToolTypedDict]]] = UNSET,
-        safe_prompt: Optional[bool] = False,
         tool_choice: Optional[models.ToolChoice] = "auto",
+        safe_prompt: Optional[bool] = False,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -162,8 +161,8 @@ class Chat(BaseSDK):
         :param random_seed: The seed to use for random sampling. If set, different calls will generate deterministic results.
         :param response_format: 
         :param tools: 
-        :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param tool_choice: 
+        :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
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -188,8 +187,8 @@ class Chat(BaseSDK):
             messages=utils.get_pydantic_model(messages, List[models.Messages]),
             response_format=utils.get_pydantic_model(response_format, Optional[models.ResponseFormat]),
             tools=utils.get_pydantic_model(tools, OptionalNullable[List[models.Tool]]),
-            safe_prompt=safe_prompt,
             tool_choice=tool_choice,
+            safe_prompt=safe_prompt,
         )
         
         req = self.build_request(
@@ -244,7 +243,7 @@ class Chat(BaseSDK):
 
     
     
-    def create(
+    def complete(
         self, *,
         messages: Union[List[models.ChatCompletionRequestMessages], List[models.ChatCompletionRequestMessagesTypedDict]],
         model: OptionalNullable[str] = "azureai",
@@ -257,16 +256,14 @@ class Chat(BaseSDK):
         random_seed: OptionalNullable[int] = UNSET,
         response_format: Optional[Union[models.ResponseFormat, models.ResponseFormatTypedDict]] = None,
         tools: OptionalNullable[Union[List[models.Tool], List[models.ToolTypedDict]]] = UNSET,
-        safe_prompt: Optional[bool] = False,
         tool_choice: Optional[models.ChatCompletionRequestToolChoice] = "auto",
+        safe_prompt: Optional[bool] = False,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
     ) -> Optional[models.ChatCompletionResponse]:
         r"""Chat Completion
 
-        Chat Completion
-
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param model: The ID of the model to use for this request.
         :param temperature: What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both.
@@ -278,8 +275,8 @@ class Chat(BaseSDK):
         :param random_seed: The seed to use for random sampling. If set, different calls will generate deterministic results.
         :param response_format: 
         :param tools: 
-        :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param tool_choice: 
+        :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
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -304,8 +301,8 @@ class Chat(BaseSDK):
             messages=utils.get_pydantic_model(messages, List[models.ChatCompletionRequestMessages]),
             response_format=utils.get_pydantic_model(response_format, Optional[models.ResponseFormat]),
             tools=utils.get_pydantic_model(tools, OptionalNullable[List[models.Tool]]),
-            safe_prompt=safe_prompt,
             tool_choice=tool_choice,
+            safe_prompt=safe_prompt,
         )
         
         req = self.build_request(
@@ -359,7 +356,7 @@ class Chat(BaseSDK):
 
     
     
-    async def create_async(
+    async def complete_async(
         self, *,
         messages: Union[List[models.ChatCompletionRequestMessages], List[models.ChatCompletionRequestMessagesTypedDict]],
         model: OptionalNullable[str] = "azureai",
@@ -372,16 +369,14 @@ class Chat(BaseSDK):
         random_seed: OptionalNullable[int] = UNSET,
         response_format: Optional[Union[models.ResponseFormat, models.ResponseFormatTypedDict]] = None,
         tools: OptionalNullable[Union[List[models.Tool], List[models.ToolTypedDict]]] = UNSET,
-        safe_prompt: Optional[bool] = False,
         tool_choice: Optional[models.ChatCompletionRequestToolChoice] = "auto",
+        safe_prompt: Optional[bool] = False,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
     ) -> Optional[models.ChatCompletionResponse]:
         r"""Chat Completion
 
-        Chat Completion
-
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param model: The ID of the model to use for this request.
         :param temperature: What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both.
@@ -393,8 +388,8 @@ class Chat(BaseSDK):
         :param random_seed: The seed to use for random sampling. If set, different calls will generate deterministic results.
         :param response_format: 
         :param tools: 
-        :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param tool_choice: 
+        :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
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -419,8 +414,8 @@ class Chat(BaseSDK):
             messages=utils.get_pydantic_model(messages, List[models.ChatCompletionRequestMessages]),
             response_format=utils.get_pydantic_model(response_format, Optional[models.ResponseFormat]),
             tools=utils.get_pydantic_model(tools, OptionalNullable[List[models.Tool]]),
-            safe_prompt=safe_prompt,
             tool_choice=tool_choice,
+            safe_prompt=safe_prompt,
         )
         
         req = self.build_request(

mistralai_azure/models/__init__.py

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from .assistantmessage import AssistantMessage, AssistantMessageRole, AssistantMessageTypedDict
-from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceFinishReason, ChatCompletionChoiceTypedDict, Message, MessageTypedDict
+from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceFinishReason, ChatCompletionChoiceTypedDict
 from .chatcompletionrequest import ChatCompletionRequest, ChatCompletionRequestMessages, ChatCompletionRequestMessagesTypedDict, ChatCompletionRequestStop, ChatCompletionRequestStopTypedDict, ChatCompletionRequestToolChoice, ChatCompletionRequestTypedDict
 from .chatcompletionresponse import ChatCompletionResponse, ChatCompletionResponseTypedDict
 from .chatcompletionstreamrequest import ChatCompletionStreamRequest, ChatCompletionStreamRequestTypedDict, Messages, MessagesTypedDict, Stop, StopTypedDict, ToolChoice
@@ -11,7 +11,7 @@ from .completionresponsestreamchoice import CompletionResponseStreamChoice, Comp
 from .contentchunk import ContentChunk, ContentChunkTypedDict
 from .deltamessage import DeltaMessage, DeltaMessageTypedDict
 from .function import Function, FunctionTypedDict
-from .functioncall import FunctionCall, FunctionCallTypedDict
+from .functioncall import Arguments, ArgumentsTypedDict, FunctionCall, FunctionCallTypedDict
 from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
 from .responseformat import ResponseFormat, ResponseFormatTypedDict, ResponseFormats
 from .sdkerror import SDKError
@@ -25,4 +25,4 @@ from .usageinfo import UsageInfo, UsageInfoTypedDict
 from .usermessage import UserMessage, UserMessageContent, UserMessageContentTypedDict, UserMessageRole, UserMessageTypedDict
 from .validationerror import Loc, LocTypedDict, ValidationError, ValidationErrorTypedDict
 
-__all__ = ["AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceFinishReason", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestMessages", "ChatCompletionRequestMessagesTypedDict", "ChatCompletionRequestStop", "ChatCompletionRequestStopTypedDict", "ChatCompletionRequestToolChoice", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Loc", "LocTypedDict", "Message", "MessageTypedDict", "Messages", "MessagesTypedDict", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "Role", "SDKError", "Security", "SecurityTypedDict", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolTypedDict", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict"]
+__all__ = ["Arguments", "ArgumentsTypedDict", "AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceFinishReason", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestMessages", "ChatCompletionRequestMessagesTypedDict", "ChatCompletionRequestStop", "ChatCompletionRequestStopTypedDict", "ChatCompletionRequestToolChoice", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Loc", "LocTypedDict", "Messages", "MessagesTypedDict", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "Role", "SDKError", "Security", "SecurityTypedDict", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolTypedDict", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict"]

mistralai_azure/models/assistantmessage.py

@@ -39,18 +39,13 @@ class AssistantMessage(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai_azure/models/chatcompletionchoice.py

@@ -2,14 +2,9 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
-from .systemmessage import SystemMessage, SystemMessageTypedDict
-from .toolmessage import ToolMessage, ToolMessageTypedDict
-from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai_azure.types import BaseModel
-from mistralai_azure.utils import get_discriminator
-from pydantic import Discriminator, Tag
-from typing import Literal, Optional, TypedDict, Union
-from typing_extensions import Annotated, NotRequired
+from typing import Literal, Optional, TypedDict
+from typing_extensions import NotRequired
 
 
 ChatCompletionChoiceFinishReason = Literal["stop", "length", "model_length", "error", "tool_calls"]
@@ -17,17 +12,11 @@ ChatCompletionChoiceFinishReason = Literal["stop", "length", "model_length", "er
 class ChatCompletionChoiceTypedDict(TypedDict):
     index: int
     finish_reason: ChatCompletionChoiceFinishReason
-    message: NotRequired[MessageTypedDict]
+    message: NotRequired[AssistantMessageTypedDict]
     
 
 class ChatCompletionChoice(BaseModel):
     index: int
     finish_reason: ChatCompletionChoiceFinishReason
-    message: Optional[Message] = None
+    message: Optional[AssistantMessage] = None
     
-
-MessageTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
-
-
-Message = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
-

mistralai_azure/models/chatcompletionrequest.py

@@ -14,6 +14,20 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+ChatCompletionRequestStopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+ChatCompletionRequestStop = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+ChatCompletionRequestMessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
+
+
+ChatCompletionRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
+
+
 ChatCompletionRequestToolChoice = Literal["auto", "none", "any"]
 
 class ChatCompletionRequestTypedDict(TypedDict):
@@ -37,9 +51,9 @@ class ChatCompletionRequestTypedDict(TypedDict):
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
     response_format: NotRequired[ResponseFormatTypedDict]
     tools: NotRequired[Nullable[List[ToolTypedDict]]]
+    tool_choice: NotRequired[ChatCompletionRequestToolChoice]
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
-    tool_choice: NotRequired[ChatCompletionRequestToolChoice]
     
 
 class ChatCompletionRequest(BaseModel):
@@ -63,13 +77,13 @@ class ChatCompletionRequest(BaseModel):
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
     response_format: Optional[ResponseFormat] = None
     tools: OptionalNullable[List[Tool]] = UNSET
+    tool_choice: Optional[ChatCompletionRequestToolChoice] = "auto"
     safe_prompt: Optional[bool] = False
     r"""Whether to inject a safety prompt before all conversations."""
-    tool_choice: Optional[ChatCompletionRequestToolChoice] = "auto"
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["model", "temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "response_format", "tools", "safe_prompt", "tool_choice"]
+        optional_fields = ["model", "temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "response_format", "tools", "tool_choice", "safe_prompt"]
         nullable_fields = ["model", "max_tokens", "min_tokens", "random_seed", "tools"]
         null_default_fields = []
 
@@ -81,34 +95,15 @@ class ChatCompletionRequest(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 
         return m
         
-
-ChatCompletionRequestStopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-ChatCompletionRequestStop = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-ChatCompletionRequestMessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
-
-
-ChatCompletionRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
-

mistralai_azure/models/chatcompletionstreamrequest.py

@@ -14,6 +14,20 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+StopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+Stop = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+MessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
+
+
+Messages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
+
+
 ToolChoice = Literal["auto", "none", "any"]
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):
@@ -36,9 +50,9 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
     response_format: NotRequired[ResponseFormatTypedDict]
     tools: NotRequired[Nullable[List[ToolTypedDict]]]
+    tool_choice: NotRequired[ToolChoice]
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
-    tool_choice: NotRequired[ToolChoice]
     
 
 class ChatCompletionStreamRequest(BaseModel):
@@ -61,13 +75,13 @@ class ChatCompletionStreamRequest(BaseModel):
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
     response_format: Optional[ResponseFormat] = None
     tools: OptionalNullable[List[Tool]] = UNSET
+    tool_choice: Optional[ToolChoice] = "auto"
     safe_prompt: Optional[bool] = False
     r"""Whether to inject a safety prompt before all conversations."""
-    tool_choice: Optional[ToolChoice] = "auto"
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["model", "temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "response_format", "tools", "safe_prompt", "tool_choice"]
+        optional_fields = ["model", "temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "response_format", "tools", "tool_choice", "safe_prompt"]
         nullable_fields = ["model", "max_tokens", "min_tokens", "random_seed", "tools"]
         null_default_fields = []
 
@@ -79,34 +93,15 @@ class ChatCompletionStreamRequest(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 
         return m
         
-
-StopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-Stop = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-MessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
-
-
-Messages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
-

mistralai_azure/models/completionresponsestreamchoice.py

@@ -34,18 +34,13 @@ class CompletionResponseStreamChoice(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai_azure/models/deltamessage.py

@@ -4,20 +4,20 @@ from __future__ import annotations
 from .toolcall import ToolCall, ToolCallTypedDict
 from mistralai_azure.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional, TypedDict
+from typing import List, Optional, TypedDict
 from typing_extensions import NotRequired
 
 
 class DeltaMessageTypedDict(TypedDict):
     role: NotRequired[str]
     content: NotRequired[str]
-    tool_calls: NotRequired[Nullable[ToolCallTypedDict]]
+    tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
     
 
 class DeltaMessage(BaseModel):
     role: Optional[str] = None
     content: Optional[str] = None
-    tool_calls: OptionalNullable[ToolCall] = UNSET
+    tool_calls: OptionalNullable[List[ToolCall]] = UNSET
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
@@ -33,18 +33,13 @@ class DeltaMessage(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai_azure/models/functioncall.py

@@ -2,15 +2,21 @@
 
 from __future__ import annotations
 from mistralai_azure.types import BaseModel
-from typing import TypedDict
+from typing import Any, Dict, TypedDict, Union
+
+
+ArgumentsTypedDict = Union[Dict[str, Any], str]
+
+
+Arguments = Union[Dict[str, Any], str]
 
 
 class FunctionCallTypedDict(TypedDict):
     name: str
-    arguments: str
+    arguments: ArgumentsTypedDict
     
 
 class FunctionCall(BaseModel):
     name: str
-    arguments: str
+    arguments: Arguments
     

mistralai_azure/models/httpvalidationerror.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 from .validationerror import ValidationError
+from mistralai_azure import utils
 from mistralai_azure.types import BaseModel
-import mistralai_azure.utils as utils
 from typing import List, Optional
 
 class HTTPValidationErrorData(BaseModel):

mistralai_azure/models/systemmessage.py

@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+
+
+Content = Union[str, List[ContentChunk]]
+
+
 Role = Literal["system"]
 
 class SystemMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class SystemMessage(BaseModel):
     content: Content
     role: Optional[Role] = "system"
     
-
-ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
-
-
-Content = Union[str, List[ContentChunk]]
-

mistralai_azure/models/toolmessage.py

@@ -36,18 +36,13 @@ class ToolMessage(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai_azure/models/usermessage.py

@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+
+
+UserMessageContent = Union[str, List[TextChunk]]
+
+
 UserMessageRole = Literal["user"]
 
 class UserMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class UserMessage(BaseModel):
     content: UserMessageContent
     role: Optional[UserMessageRole] = "user"
     
-
-UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
-
-
-UserMessageContent = Union[str, List[TextChunk]]
-

mistralai_azure/models/validationerror.py

@@ -5,6 +5,12 @@ from mistralai_azure.types import BaseModel
 from typing import List, TypedDict, Union
 
 
+LocTypedDict = Union[str, int]
+
+
+Loc = Union[str, int]
+
+
 class ValidationErrorTypedDict(TypedDict):
     loc: List[LocTypedDict]
     msg: str
@@ -16,9 +22,3 @@ class ValidationError(BaseModel):
     msg: str
     type: str
     
-
-LocTypedDict = Union[str, int]
-
-
-Loc = Union[str, int]
-