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

mistralai/models/prediction.py

@@ -10,11 +10,15 @@ from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class PredictionTypedDict(TypedDict):
+    r"""Enable users to specify an expected completion, optimizing response times by leveraging known or predictable content."""
+
     type: Literal["content"]
     content: NotRequired[str]
 
 
 class Prediction(BaseModel):
+    r"""Enable users to specify an expected completion, optimizing response times by leveraging known or predictable content."""
+
     TYPE: Annotated[
         Annotated[
             Optional[Literal["content"]], AfterValidator(validate_const("content"))

mistralai/models/requestsource.py

@@ -0,0 +1,7 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from typing import Literal
+
+
+RequestSource = Literal["api", "playground", "agent_builder_v1"]

mistralai/models/responseformat.py

@@ -10,14 +10,16 @@ from typing_extensions import NotRequired, TypedDict
 
 
 class ResponseFormatTypedDict(TypedDict):
+    r"""Specify the format that the model must output. By default it will use `{ \"type\": \"text\" }`. 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. Setting to `{ \"type\": \"json_schema\" }` enables JSON schema mode, which guarantees the message the model generates is in JSON and follows the schema you provide."""
+
     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."""
     json_schema: NotRequired[Nullable[JSONSchemaTypedDict]]
 
 
 class ResponseFormat(BaseModel):
+    r"""Specify the format that the model must output. By default it will use `{ \"type\": \"text\" }`. 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. Setting to `{ \"type\": \"json_schema\" }` enables JSON schema mode, which guarantees the message the model generates is in JSON and follows the schema you provide."""
+
     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."""
 
     json_schema: OptionalNullable[JSONSchema] = UNSET
 

mistralai/models/responseformats.py

@@ -5,4 +5,3 @@ from typing import Literal
 
 
 ResponseFormats = Literal["text", "json_object", "json_schema"]
-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/models/responsevalidationerror.py

@@ -0,0 +1,25 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+import httpx
+from typing import Optional
+
+from mistralai.models import MistralError
+
+
+class ResponseValidationError(MistralError):
+    """Error raised when there is a type mismatch between the response data and the expected Pydantic model."""
+
+    def __init__(
+        self,
+        message: str,
+        raw_response: httpx.Response,
+        cause: Exception,
+        body: Optional[str] = None,
+    ):
+        message = f"{message}: {cause}"
+        super().__init__(message, raw_response, body)
+
+    @property
+    def cause(self):
+        """Normally the Pydantic ValidationError"""
+        return self.__cause__

mistralai/models/sdkerror.py

@@ -1,22 +1,38 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-from dataclasses import dataclass
-from typing import Optional
 import httpx
+from typing import Optional
+
+from mistralai.models import MistralError
+
+MAX_MESSAGE_LEN = 10_000
+
+
+class SDKError(MistralError):
+    """The fallback error class if no more specific error class is matched."""
+
+    def __init__(
+        self, message: str, raw_response: httpx.Response, body: Optional[str] = None
+    ):
+        body_display = body or raw_response.text or '""'
 
+        if message:
+            message += ": "
+        message += f"Status {raw_response.status_code}"
 
-@dataclass
-class SDKError(Exception):
-    """Represents an error returned by the API."""
+        headers = raw_response.headers
+        content_type = headers.get("content-type", '""')
+        if content_type != "application/json":
+            if " " in content_type:
+                content_type = f'"{content_type}"'
+            message += f" Content-Type {content_type}"
 
-    message: str
-    status_code: int = -1
-    body: str = ""
-    raw_response: Optional[httpx.Response] = None
+        if len(body_display) > MAX_MESSAGE_LEN:
+            truncated = body_display[:MAX_MESSAGE_LEN]
+            remaining = len(body_display) - MAX_MESSAGE_LEN
+            body_display = f"{truncated}...and {remaining} more chars"
 
-    def __str__(self):
-        body = ""
-        if len(self.body) > 0:
-            body = f"\n{self.body}"
+        message += f". Body: {body_display}"
+        message = message.strip()
 
-        return f"{self.message}: Status {self.status_code}{body}"
+        super().__init__(message, raw_response, body)

mistralai/models/sharingdelete.py

@@ -2,25 +2,56 @@
 
 from __future__ import annotations
 from .entitytype import EntityType
-from mistralai.types import BaseModel
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import validate_open_enum
+from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class SharingDeleteTypedDict(TypedDict):
-    org_id: str
     share_with_uuid: str
     r"""The id of the entity (user, workspace or organization) to share with"""
     share_with_type: EntityType
     r"""The type of entity, used to share a library."""
+    org_id: NotRequired[Nullable[str]]
 
 
 class SharingDelete(BaseModel):
-    org_id: str
-
     share_with_uuid: str
     r"""The id of the entity (user, workspace or organization) to share with"""
 
     share_with_type: Annotated[EntityType, PlainValidator(validate_open_enum(False))]
     r"""The type of entity, used to share a library."""
+
+    org_id: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["org_id"]
+        nullable_fields = ["org_id"]
+        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

mistralai/models/sharingin.py

@@ -3,24 +3,23 @@
 from __future__ import annotations
 from .entitytype import EntityType
 from .shareenum import ShareEnum
-from mistralai.types import BaseModel
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import validate_open_enum
+from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class SharingInTypedDict(TypedDict):
-    org_id: str
     level: ShareEnum
     share_with_uuid: str
     r"""The id of the entity (user, workspace or organization) to share with"""
     share_with_type: EntityType
     r"""The type of entity, used to share a library."""
+    org_id: NotRequired[Nullable[str]]
 
 
 class SharingIn(BaseModel):
-    org_id: str
-
     level: Annotated[ShareEnum, PlainValidator(validate_open_enum(False))]
 
     share_with_uuid: str
@@ -28,3 +27,35 @@ class SharingIn(BaseModel):
 
     share_with_type: Annotated[EntityType, PlainValidator(validate_open_enum(False))]
     r"""The type of entity, used to share a library."""
+
+    org_id: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["org_id"]
+        nullable_fields = ["org_id"]
+        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

mistralai/models/sharingout.py

@@ -11,7 +11,7 @@ class SharingOutTypedDict(TypedDict):
     org_id: str
     role: str
     share_with_type: str
-    share_with_uuid: str
+    share_with_uuid: Nullable[str]
     user_id: NotRequired[Nullable[str]]
 
 
@@ -24,14 +24,14 @@ class SharingOut(BaseModel):
 
     share_with_type: str
 
-    share_with_uuid: str
+    share_with_uuid: Nullable[str]
 
     user_id: OptionalNullable[str] = UNSET
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["user_id"]
-        nullable_fields = ["user_id"]
+        nullable_fields = ["user_id", "share_with_uuid"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/toolexecutiondeltaevent.py

@@ -4,16 +4,25 @@ from __future__ import annotations
 from .builtinconnectors import BuiltInConnectors
 from datetime import datetime
 from mistralai.types import BaseModel
-from typing import Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 ToolExecutionDeltaEventType = Literal["tool.execution.delta"]
 
+ToolExecutionDeltaEventNameTypedDict = TypeAliasType(
+    "ToolExecutionDeltaEventNameTypedDict", Union[BuiltInConnectors, str]
+)
+
+
+ToolExecutionDeltaEventName = TypeAliasType(
+    "ToolExecutionDeltaEventName", Union[BuiltInConnectors, str]
+)
+
 
 class ToolExecutionDeltaEventTypedDict(TypedDict):
     id: str
-    name: BuiltInConnectors
+    name: ToolExecutionDeltaEventNameTypedDict
     arguments: str
     type: NotRequired[ToolExecutionDeltaEventType]
     created_at: NotRequired[datetime]
@@ -23,7 +32,7 @@ class ToolExecutionDeltaEventTypedDict(TypedDict):
 class ToolExecutionDeltaEvent(BaseModel):
     id: str
 
-    name: BuiltInConnectors
+    name: ToolExecutionDeltaEventName
 
     arguments: str
 

mistralai/models/toolexecutiondoneevent.py

@@ -4,16 +4,25 @@ from __future__ import annotations
 from .builtinconnectors import BuiltInConnectors
 from datetime import datetime
 from mistralai.types import BaseModel
-from typing import Any, Dict, Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Any, Dict, Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 ToolExecutionDoneEventType = Literal["tool.execution.done"]
 
+ToolExecutionDoneEventNameTypedDict = TypeAliasType(
+    "ToolExecutionDoneEventNameTypedDict", Union[BuiltInConnectors, str]
+)
+
+
+ToolExecutionDoneEventName = TypeAliasType(
+    "ToolExecutionDoneEventName", Union[BuiltInConnectors, str]
+)
+
 
 class ToolExecutionDoneEventTypedDict(TypedDict):
     id: str
-    name: BuiltInConnectors
+    name: ToolExecutionDoneEventNameTypedDict
     type: NotRequired[ToolExecutionDoneEventType]
     created_at: NotRequired[datetime]
     output_index: NotRequired[int]
@@ -23,7 +32,7 @@ class ToolExecutionDoneEventTypedDict(TypedDict):
 class ToolExecutionDoneEvent(BaseModel):
     id: str
 
-    name: BuiltInConnectors
+    name: ToolExecutionDoneEventName
 
     type: Optional[ToolExecutionDoneEventType] = "tool.execution.done"
 

mistralai/models/toolexecutionentry.py

@@ -5,17 +5,22 @@ from .builtinconnectors import BuiltInConnectors
 from datetime import datetime
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Any, Dict, Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Any, Dict, Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 ToolExecutionEntryObject = Literal["entry"]
 
 ToolExecutionEntryType = Literal["tool.execution"]
 
+NameTypedDict = TypeAliasType("NameTypedDict", Union[BuiltInConnectors, str])
+
+
+Name = TypeAliasType("Name", Union[BuiltInConnectors, str])
+
 
 class ToolExecutionEntryTypedDict(TypedDict):
-    name: BuiltInConnectors
+    name: NameTypedDict
     arguments: str
     object: NotRequired[ToolExecutionEntryObject]
     type: NotRequired[ToolExecutionEntryType]
@@ -26,7 +31,7 @@ class ToolExecutionEntryTypedDict(TypedDict):
 
 
 class ToolExecutionEntry(BaseModel):
-    name: BuiltInConnectors
+    name: Name
 
     arguments: str
 

mistralai/models/toolexecutionstartedevent.py

@@ -4,16 +4,25 @@ from __future__ import annotations
 from .builtinconnectors import BuiltInConnectors
 from datetime import datetime
 from mistralai.types import BaseModel
-from typing import Literal, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 ToolExecutionStartedEventType = Literal["tool.execution.started"]
 
+ToolExecutionStartedEventNameTypedDict = TypeAliasType(
+    "ToolExecutionStartedEventNameTypedDict", Union[BuiltInConnectors, str]
+)
+
+
+ToolExecutionStartedEventName = TypeAliasType(
+    "ToolExecutionStartedEventName", Union[BuiltInConnectors, str]
+)
+
 
 class ToolExecutionStartedEventTypedDict(TypedDict):
     id: str
-    name: BuiltInConnectors
+    name: ToolExecutionStartedEventNameTypedDict
     arguments: str
     type: NotRequired[ToolExecutionStartedEventType]
     created_at: NotRequired[datetime]
@@ -23,7 +32,7 @@ class ToolExecutionStartedEventTypedDict(TypedDict):
 class ToolExecutionStartedEvent(BaseModel):
     id: str
 
-    name: BuiltInConnectors
+    name: ToolExecutionStartedEventName
 
     arguments: str