mistralai 1.0.2__py3-none-any.whl → 1.1.0__py3-none-any.whl

mistralai_azure/models/contentchunk.py

@@ -1,17 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai_azure.types import BaseModel
-import pydantic
-from typing import Final, Optional, TypedDict
-from typing_extensions import Annotated
+from .textchunk import TextChunk, TextChunkTypedDict
 
 
-class ContentChunkTypedDict(TypedDict):
-    text: str
-    
+ContentChunkTypedDict = TextChunkTypedDict
 
-class ContentChunk(BaseModel):
-    text: str
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "text" # type: ignore
-    
+
+ContentChunk = TextChunk

mistralai_azure/models/deltamessage.py

@@ -2,7 +2,13 @@
 
 from __future__ import annotations
 from .toolcall import ToolCall, ToolCallTypedDict
-from mistralai_azure.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from pydantic import model_serializer
 from typing import List, Optional, TypedDict
 from typing_extensions import NotRequired
@@ -12,13 +18,15 @@ class DeltaMessageTypedDict(TypedDict):
     role: NotRequired[str]
     content: NotRequired[Nullable[str]]
     tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
-    
+
 
 class DeltaMessage(BaseModel):
     role: Optional[str] = None
+
     content: OptionalNullable[str] = UNSET
+
     tool_calls: OptionalNullable[List[ToolCall]] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["role", "content", "tool_calls"]
@@ -32,9 +40,13 @@ class DeltaMessage(BaseModel):
         for n, f in 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
+            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
@@ -44,4 +56,3 @@ class DeltaMessage(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai_azure/models/function.py

@@ -10,10 +10,11 @@ class FunctionTypedDict(TypedDict):
     name: str
     parameters: Dict[str, Any]
     description: NotRequired[str]
-    
+
 
 class Function(BaseModel):
     name: str
+
     parameters: Dict[str, Any]
+
     description: Optional[str] = ""
-    

mistralai_azure/models/functioncall.py

@@ -14,9 +14,9 @@ Arguments = Union[Dict[str, Any], str]
 class FunctionCallTypedDict(TypedDict):
     name: str
     arguments: ArgumentsTypedDict
-    
+
 
 class FunctionCall(BaseModel):
     name: str
+
     arguments: Arguments
-    

mistralai_azure/models/functionname.py

@@ -0,0 +1,17 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import BaseModel
+from typing import TypedDict
+
+
+class FunctionNameTypedDict(TypedDict):
+    r"""this restriction of `Function` is used to select a specific function to call"""
+
+    name: str
+
+
+class FunctionName(BaseModel):
+    r"""this restriction of `Function` is used to select a specific function to call"""
+
+    name: str

mistralai_azure/models/httpvalidationerror.py

@@ -6,13 +6,14 @@ from mistralai_azure import utils
 from mistralai_azure.types import BaseModel
 from typing import List, Optional
 
+
 class HTTPValidationErrorData(BaseModel):
     detail: Optional[List[ValidationError]] = None
-    
 
 
 class HTTPValidationError(Exception):
     r"""Validation Error"""
+
     data: HTTPValidationErrorData
 
     def __init__(self, data: HTTPValidationErrorData):
@@ -20,4 +21,3 @@ class HTTPValidationError(Exception):
 
     def __str__(self) -> str:
         return utils.marshal_json(self.data, HTTPValidationErrorData)
-

mistralai_azure/models/responseformat.py

@@ -1,18 +1,17 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .responseformats import ResponseFormats
 from mistralai_azure.types import BaseModel
-from typing import Literal, Optional, TypedDict
+from typing import Optional, TypedDict
 from typing_extensions import NotRequired
 
 
-ResponseFormats = Literal["text", "json_object"]
-r"""An object specifying the format that the model must output. Setting to `{ \"type\": \"json_object\" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message."""
-
 class ResponseFormatTypedDict(TypedDict):
     type: NotRequired[ResponseFormats]
-    
+    r"""An object specifying the format that the model must output. Setting to `{ \"type\": \"json_object\" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message."""
+
 
 class ResponseFormat(BaseModel):
-    type: Optional[ResponseFormats] = "text"
-    
+    type: Optional[ResponseFormats] = None
+    r"""An object specifying the format that the model must output. Setting to `{ \"type\": \"json_object\" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message."""

mistralai_azure/models/responseformats.py

@@ -0,0 +1,8 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from typing import Literal
+
+
+ResponseFormats = Literal["text", "json_object"]
+r"""An object specifying the format that the model must output. Setting to `{ \"type\": \"json_object\" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message."""

mistralai_azure/models/security.py

@@ -9,8 +9,17 @@ from typing_extensions import Annotated
 
 class SecurityTypedDict(TypedDict):
     api_key: str
-    
+
 
 class Security(BaseModel):
-    api_key: Annotated[str, FieldMetadata(security=SecurityMetadata(scheme=True, scheme_type="http", sub_type="bearer", field_name="Authorization"))]
-    
+    api_key: Annotated[
+        str,
+        FieldMetadata(
+            security=SecurityMetadata(
+                scheme=True,
+                scheme_type="http",
+                sub_type="bearer",
+                field_name="Authorization",
+            )
+        ),
+    ]

mistralai_azure/models/systemmessage.py

@@ -1,26 +1,27 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .contentchunk import ContentChunk, ContentChunkTypedDict
+from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai_azure.types import BaseModel
 from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
-ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+ContentTypedDict = Union[str, List[TextChunkTypedDict]]
 
 
-Content = Union[str, List[ContentChunk]]
+Content = Union[str, List[TextChunk]]
 
 
 Role = Literal["system"]
 
+
 class SystemMessageTypedDict(TypedDict):
     content: ContentTypedDict
     role: NotRequired[Role]
-    
+
 
 class SystemMessage(BaseModel):
     content: Content
+
     role: Optional[Role] = "system"
-    

mistralai_azure/models/textchunk.py

@@ -3,15 +3,20 @@
 from __future__ import annotations
 from mistralai_azure.types import BaseModel
 import pydantic
-from typing import Final, Optional, TypedDict
+from typing import Final, Literal, Optional, TypedDict
 from typing_extensions import Annotated
 
 
+Type = Literal["text"]
+
+
 class TextChunkTypedDict(TypedDict):
     text: str
-    
+
 
 class TextChunk(BaseModel):
     text: str
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "text" # type: ignore
-    
+
+    # fmt: off
+    TYPE: Annotated[Final[Optional[Type]], pydantic.Field(alias="type")] = "text" # type: ignore
+    # fmt: on

mistralai_azure/models/tool.py

@@ -2,21 +2,22 @@
 
 from __future__ import annotations
 from .function import Function, FunctionTypedDict
-from mistralai_azure.types import BaseModel, UnrecognizedStr
+from .tooltypes import ToolTypes
+from mistralai_azure.types import BaseModel
 from mistralai_azure.utils import validate_open_enum
 from pydantic.functional_validators import PlainValidator
-from typing import Literal, Optional, TypedDict, Union
+from typing import Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
-ToolToolTypes = Union[Literal["function"], UnrecognizedStr]
-
 class ToolTypedDict(TypedDict):
     function: FunctionTypedDict
-    type: NotRequired[ToolToolTypes]
-    
+    type: NotRequired[ToolTypes]
+
 
 class Tool(BaseModel):
     function: Function
-    type: Annotated[Optional[ToolToolTypes], PlainValidator(validate_open_enum(False))] = "function"
-    
+
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
+        None
+    )

mistralai_azure/models/toolcall.py

@@ -2,23 +2,25 @@
 
 from __future__ import annotations
 from .functioncall import FunctionCall, FunctionCallTypedDict
-from mistralai_azure.types import BaseModel, UnrecognizedStr
+from .tooltypes import ToolTypes
+from mistralai_azure.types import BaseModel
 from mistralai_azure.utils import validate_open_enum
 from pydantic.functional_validators import PlainValidator
-from typing import Literal, Optional, TypedDict, Union
+from typing import Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
-ToolTypes = Union[Literal["function"], UnrecognizedStr]
-
 class ToolCallTypedDict(TypedDict):
     function: FunctionCallTypedDict
     id: NotRequired[str]
     type: NotRequired[ToolTypes]
-    
+
 
 class ToolCall(BaseModel):
     function: FunctionCall
+
     id: Optional[str] = "null"
-    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = "function"
-    
+
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
+        None
+    )

mistralai_azure/models/toolchoice.py

@@ -0,0 +1,29 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .functionname import FunctionName, FunctionNameTypedDict
+from .tooltypes import ToolTypes
+from mistralai_azure.types import BaseModel
+from mistralai_azure.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing import Optional, TypedDict
+from typing_extensions import Annotated, NotRequired
+
+
+class ToolChoiceTypedDict(TypedDict):
+    r"""ToolChoice is either a ToolChoiceEnum or a ToolChoice"""
+
+    function: FunctionNameTypedDict
+    r"""this restriction of `Function` is used to select a specific function to call"""
+    type: NotRequired[ToolTypes]
+
+
+class ToolChoice(BaseModel):
+    r"""ToolChoice is either a ToolChoiceEnum or a ToolChoice"""
+
+    function: FunctionName
+    r"""this restriction of `Function` is used to select a specific function to call"""
+
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
+        None
+    )

mistralai_azure/models/toolchoiceenum.py

@@ -0,0 +1,7 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from typing import Literal
+
+
+ToolChoiceEnum = Literal["auto", "none", "any", "required"]

mistralai_azure/models/toolmessage.py

@@ -1,7 +1,13 @@
 """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 mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from pydantic import model_serializer
 from typing import Literal, Optional, TypedDict
 from typing_extensions import NotRequired
@@ -9,19 +15,23 @@ from typing_extensions import NotRequired
 
 ToolMessageRole = Literal["tool"]
 
+
 class ToolMessageTypedDict(TypedDict):
     content: str
     tool_call_id: NotRequired[Nullable[str]]
     name: NotRequired[Nullable[str]]
     role: NotRequired[ToolMessageRole]
-    
+
 
 class ToolMessage(BaseModel):
     content: str
+
     tool_call_id: OptionalNullable[str] = UNSET
+
     name: OptionalNullable[str] = UNSET
+
     role: Optional[ToolMessageRole] = "tool"
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["tool_call_id", "name", "role"]
@@ -35,9 +45,13 @@ class ToolMessage(BaseModel):
         for n, f in 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
+            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
@@ -47,4 +61,3 @@ class ToolMessage(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai_azure/models/tooltypes.py

@@ -0,0 +1,8 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+ToolTypes = Union[Literal["function"], UnrecognizedStr]

mistralai_azure/models/usageinfo.py

@@ -9,10 +9,11 @@ class UsageInfoTypedDict(TypedDict):
     prompt_tokens: int
     completion_tokens: int
     total_tokens: int
-    
+
 
 class UsageInfo(BaseModel):
     prompt_tokens: int
+
     completion_tokens: int
+
     total_tokens: int
-    

mistralai_azure/models/usermessage.py

@@ -1,26 +1,27 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .textchunk import TextChunk, TextChunkTypedDict
+from .contentchunk import ContentChunk, ContentChunkTypedDict
 from mistralai_azure.types import BaseModel
 from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
-UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+UserMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
 
 
-UserMessageContent = Union[str, List[TextChunk]]
+UserMessageContent = Union[str, List[ContentChunk]]
 
 
 UserMessageRole = Literal["user"]
 
+
 class UserMessageTypedDict(TypedDict):
     content: UserMessageContentTypedDict
     role: NotRequired[UserMessageRole]
-    
+
 
 class UserMessage(BaseModel):
     content: UserMessageContent
+
     role: Optional[UserMessageRole] = "user"
-    

mistralai_azure/models/validationerror.py

@@ -15,10 +15,11 @@ class ValidationErrorTypedDict(TypedDict):
     loc: List[LocTypedDict]
     msg: str
     type: str
-    
+
 
 class ValidationError(BaseModel):
     loc: List[Loc]
+
     msg: str
+
     type: str
-    

mistralai_azure/sdkconfiguration.py

@@ -1,6 +1,5 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-
 from ._hooks import SDKHooks
 from .httpclient import AsyncHttpClient, HttpClient
 from .utils import Logger, RetryConfig, remove_suffix
@@ -14,7 +13,7 @@ from typing import Callable, Dict, Optional, Tuple, Union
 SERVER_PROD = "prod"
 r"""Production server"""
 SERVERS = {
-	SERVER_PROD: "https://api.mistral.ai",
+    SERVER_PROD: "https://api.mistral.ai",
 }
 """Contains the list of servers available to the SDK"""
 
@@ -24,14 +23,14 @@ class SDKConfiguration:
     client: HttpClient
     async_client: AsyncHttpClient
     debug_logger: Logger
-    security: Optional[Union[models.Security,Callable[[], models.Security]]] = None
+    security: Optional[Union[models.Security, Callable[[], models.Security]]] = None
     server_url: Optional[str] = ""
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.0.1"
-    gen_version: str = "2.399.0"
-    user_agent: str = "speakeasy-sdk/python 1.0.1 2.399.0 0.0.2 mistralai_azure"
+    sdk_version: str = "1.2.2"
+    gen_version: str = "2.415.6"
+    user_agent: str = "speakeasy-sdk/python 1.2.2 2.415.6 0.0.2 mistralai_azure"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 
@@ -45,10 +44,9 @@ class SDKConfiguration:
             self.server = SERVER_PROD
 
         if self.server not in SERVERS:
-            raise ValueError(f"Invalid server \"{self.server}\"")
+            raise ValueError(f'Invalid server "{self.server}"')
 
         return SERVERS[self.server], {}
 
-
     def get_hooks(self) -> SDKHooks:
         return self._hooks

mistralai_azure/utils/__init__.py

@@ -33,8 +33,13 @@ from .serializers import (
     validate_open_enum,
 )
 from .url import generate_url, template_url, remove_suffix
-from .values import get_global_from_env, match_content_type, match_status_codes, match_response
-from .logger import Logger, get_body_content, NoOpLogger
+from .values import (
+    get_global_from_env,
+    match_content_type,
+    match_status_codes,
+    match_response,
+)
+from .logger import Logger, get_body_content, get_default_logger
 
 __all__ = [
     "BackoffStrategy",
@@ -43,6 +48,7 @@ __all__ = [
     "FormMetadata",
     "generate_url",
     "get_body_content",
+    "get_default_logger",
     "get_discriminator",
     "get_global_from_env",
     "get_headers",
@@ -57,7 +63,6 @@ __all__ = [
     "match_status_codes",
     "match_response",
     "MultipartFormMetadata",
-    "NoOpLogger",
     "OpenEnumMeta",
     "PathParamMetadata",
     "QueryParamMetadata",

mistralai_azure/utils/forms.py

@@ -17,7 +17,7 @@ from .metadata import (
     MultipartFormMetadata,
     find_field_metadata,
 )
-from .values import _val_to_string
+from .values import _is_set, _val_to_string
 
 
 def _populate_form(
@@ -27,7 +27,7 @@ def _populate_form(
     delimiter: str,
     form: Dict[str, List[str]],
 ):
-    if obj is None:
+    if not _is_set(obj):
         return form
 
     if isinstance(obj, BaseModel):
@@ -41,7 +41,7 @@ def _populate_form(
                 continue
 
             val = getattr(obj, name)
-            if val is None:
+            if not _is_set(val):
                 continue
 
             if explode:
@@ -54,7 +54,7 @@ def _populate_form(
     elif isinstance(obj, Dict):
         items = []
         for key, value in obj.items():
-            if value is None:
+            if not _is_set(value):
                 continue
 
             if explode:
@@ -68,7 +68,7 @@ def _populate_form(
         items = []
 
         for value in obj:
-            if value is None:
+            if not _is_set(value):
                 continue
 
             if explode:
@@ -102,7 +102,7 @@ def serialize_multipart_form(
         field = request_fields[name]
 
         val = getattr(request, name)
-        if val is None:
+        if not _is_set(val):
             continue
 
         field_metadata = find_field_metadata(field, MultipartFormMetadata)
@@ -156,7 +156,7 @@ def serialize_multipart_form(
                 values = []
 
                 for value in val:
-                    if value is None:
+                    if not _is_set(value):
                         continue
                     values.append(_val_to_string(value))
 
@@ -176,7 +176,7 @@ def serialize_form_data(data: Any) -> Dict[str, Any]:
             field = data_fields[name]
 
             val = getattr(data, name)
-            if val is None:
+            if not _is_set(val):
                 continue
 
             metadata = find_field_metadata(field, FormMetadata)
@@ -200,7 +200,8 @@ def serialize_form_data(data: Any) -> Dict[str, Any]:
                     raise ValueError(f"Invalid form style for field {name}")
     elif isinstance(data, Dict):
         for key, value in data.items():
-            form[key] = [_val_to_string(value)]
+            if _is_set(value):
+                form[key] = [_val_to_string(value)]
     else:
         raise TypeError(f"Invalid request body type {type(data)} for form data")