mistralai 1.8.2__py3-none-any.whl → 1.9.2__py3-none-any.whl

mistralai/models/processingstatusout.py

@@ -0,0 +1,16 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class ProcessingStatusOutTypedDict(TypedDict):
+    document_id: str
+    processing_status: str
+
+
+class ProcessingStatusOut(BaseModel):
+    document_id: str
+
+    processing_status: str

mistralai/models/responseformat.py

@@ -31,7 +31,7 @@ class ResponseFormat(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/models/retrievefileout.py

@@ -66,7 +66,7 @@ class RetrieveFileOut(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/models/shareenum.py

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

mistralai/models/sharingdelete.py

@@ -0,0 +1,26 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .entitytype import EntityType
+from mistralai.types import BaseModel
+from mistralai.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing_extensions import Annotated, 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."""
+
+
+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."""

mistralai/models/sharingin.py

@@ -0,0 +1,30 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .entitytype import EntityType
+from .shareenum import ShareEnum
+from mistralai.types import BaseModel
+from mistralai.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing_extensions import Annotated, 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."""
+
+
+class SharingIn(BaseModel):
+    org_id: str
+
+    level: Annotated[ShareEnum, PlainValidator(validate_open_enum(False))]
+
+    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."""

mistralai/models/sharingout.py

@@ -0,0 +1,59 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class SharingOutTypedDict(TypedDict):
+    library_id: str
+    org_id: str
+    role: str
+    share_with_type: str
+    share_with_uuid: str
+    user_id: NotRequired[Nullable[str]]
+
+
+class SharingOut(BaseModel):
+    library_id: str
+
+    org_id: str
+
+    role: str
+
+    share_with_type: str
+
+    share_with_uuid: str
+
+    user_id: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["user_id"]
+        nullable_fields = ["user_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/ssetypes.py

@@ -10,6 +10,7 @@ SSETypes = Literal[
     "conversation.response.error",
     "message.output.delta",
     "tool.execution.started",
+    "tool.execution.delta",
     "tool.execution.done",
     "agent.handoff.started",
     "agent.handoff.done",

mistralai/models/toolexecutiondeltaevent.py

@@ -0,0 +1,34 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+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
+
+
+ToolExecutionDeltaEventType = Literal["tool.execution.delta"]
+
+
+class ToolExecutionDeltaEventTypedDict(TypedDict):
+    id: str
+    name: BuiltInConnectors
+    arguments: str
+    type: NotRequired[ToolExecutionDeltaEventType]
+    created_at: NotRequired[datetime]
+    output_index: NotRequired[int]
+
+
+class ToolExecutionDeltaEvent(BaseModel):
+    id: str
+
+    name: BuiltInConnectors
+
+    arguments: str
+
+    type: Optional[ToolExecutionDeltaEventType] = "tool.execution.delta"
+
+    created_at: Optional[datetime] = None
+
+    output_index: Optional[int] = 0

mistralai/models/toolexecutionentry.py

@@ -16,6 +16,7 @@ ToolExecutionEntryType = Literal["tool.execution"]
 
 class ToolExecutionEntryTypedDict(TypedDict):
     name: BuiltInConnectors
+    arguments: str
     object: NotRequired[ToolExecutionEntryObject]
     type: NotRequired[ToolExecutionEntryType]
     created_at: NotRequired[datetime]
@@ -27,6 +28,8 @@ class ToolExecutionEntryTypedDict(TypedDict):
 class ToolExecutionEntry(BaseModel):
     name: BuiltInConnectors
 
+    arguments: str
+
     object: Optional[ToolExecutionEntryObject] = "entry"
 
     type: Optional[ToolExecutionEntryType] = "tool.execution"
@@ -49,7 +52,7 @@ class ToolExecutionEntry(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/models/toolexecutionstartedevent.py

@@ -14,6 +14,7 @@ ToolExecutionStartedEventType = Literal["tool.execution.started"]
 class ToolExecutionStartedEventTypedDict(TypedDict):
     id: str
     name: BuiltInConnectors
+    arguments: str
     type: NotRequired[ToolExecutionStartedEventType]
     created_at: NotRequired[datetime]
     output_index: NotRequired[int]
@@ -24,6 +25,8 @@ class ToolExecutionStartedEvent(BaseModel):
 
     name: BuiltInConnectors
 
+    arguments: str
+
     type: Optional[ToolExecutionStartedEventType] = "tool.execution.started"
 
     created_at: Optional[datetime] = None

mistralai/models/toolfilechunk.py

@@ -40,7 +40,7 @@ class ToolFileChunk(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/models/toolmessage.py

@@ -45,7 +45,7 @@ class ToolMessage(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/models/toolreferencechunk.py

@@ -16,7 +16,8 @@ class ToolReferenceChunkTypedDict(TypedDict):
     title: str
     type: NotRequired[ToolReferenceChunkType]
     url: NotRequired[Nullable[str]]
-    source: NotRequired[Nullable[str]]
+    favicon: NotRequired[Nullable[str]]
+    description: NotRequired[Nullable[str]]
 
 
 class ToolReferenceChunk(BaseModel):
@@ -28,19 +29,21 @@ class ToolReferenceChunk(BaseModel):
 
     url: OptionalNullable[str] = UNSET
 
-    source: OptionalNullable[str] = UNSET
+    favicon: OptionalNullable[str] = UNSET
+
+    description: OptionalNullable[str] = UNSET
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["type", "url", "source"]
-        nullable_fields = ["url", "source"]
+        optional_fields = ["type", "url", "favicon", "description"]
+        nullable_fields = ["url", "favicon", "description"]
         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/models/updateftmodelin.py

@@ -26,7 +26,7 @@ class UpdateFTModelIn(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/models/uploadfileout.py

@@ -63,7 +63,7 @@ class UploadFileOut(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/models/usermessage.py

@@ -39,7 +39,7 @@ class UserMessage(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/models/wandbintegration.py

@@ -45,7 +45,7 @@ class WandbIntegration(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/models/wandbintegrationout.py

@@ -43,7 +43,7 @@ class WandbIntegrationOut(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/models_.py

@@ -63,6 +63,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_models_v1_models_get",
                 oauth2_scopes=[],
@@ -155,6 +156,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_models_v1_models_get",
                 oauth2_scopes=[],
@@ -206,7 +208,7 @@ class Models(BaseSDK):
     ) -> models.RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGet:
         r"""Retrieve Model
 
-        Retrieve a model information.
+        Retrieve information about a model.
 
         :param model_id: The ID of the model to retrieve.
         :param retries: Override the default retry configuration for this method
@@ -254,6 +256,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="retrieve_model_v1_models__model_id__get",
                 oauth2_scopes=[],
@@ -308,7 +311,7 @@ class Models(BaseSDK):
     ) -> models.RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGet:
         r"""Retrieve Model
 
-        Retrieve a model information.
+        Retrieve information about a model.
 
         :param model_id: The ID of the model to retrieve.
         :param retries: Override the default retry configuration for this method
@@ -356,6 +359,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="retrieve_model_v1_models__model_id__get",
                 oauth2_scopes=[],
@@ -458,6 +462,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_model_v1_models__model_id__delete",
                 oauth2_scopes=[],
@@ -557,6 +562,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_model_v1_models__model_id__delete",
                 oauth2_scopes=[],
@@ -667,6 +673,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model",
                 oauth2_scopes=[],
@@ -774,6 +781,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model",
                 oauth2_scopes=[],
@@ -870,6 +878,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model",
                 oauth2_scopes=[],
@@ -963,6 +972,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model",
                 oauth2_scopes=[],
@@ -1056,6 +1066,7 @@ class Models(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model",
                 oauth2_scopes=[],
@@ -1149,6 +1160,7 @@ class Models(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model",
                 oauth2_scopes=[],

mistralai/ocr.py

@@ -103,6 +103,7 @@ class Ocr(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="ocr_v1_ocr_post",
                 oauth2_scopes=[],
@@ -235,6 +236,7 @@ class Ocr(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="ocr_v1_ocr_post",
                 oauth2_scopes=[],

mistralai/sdk.py

@@ -6,46 +6,62 @@ from .sdkconfiguration import SDKConfiguration
 from .utils.logger import Logger, get_default_logger
 from .utils.retries import RetryConfig
 import httpx
+import importlib
 from mistralai import models, utils
 from mistralai._hooks import SDKHooks
-from mistralai.agents import Agents
-from mistralai.batch import Batch
-from mistralai.beta import Beta
-from mistralai.chat import Chat
-from mistralai.classifiers import Classifiers
-from mistralai.embeddings import Embeddings
-from mistralai.files import Files
-from mistralai.fim import Fim
-from mistralai.fine_tuning import FineTuning
-from mistralai.models_ import Models
-from mistralai.ocr import Ocr
 from mistralai.types import OptionalNullable, UNSET
-from typing import Any, Callable, Dict, Optional, Union, cast
+from typing import Any, Callable, Dict, Optional, TYPE_CHECKING, Union, cast
 import weakref
 
+if TYPE_CHECKING:
+    from mistralai.agents import Agents
+    from mistralai.batch import Batch
+    from mistralai.beta import Beta
+    from mistralai.chat import Chat
+    from mistralai.classifiers import Classifiers
+    from mistralai.embeddings import Embeddings
+    from mistralai.files import Files
+    from mistralai.fim import Fim
+    from mistralai.fine_tuning import FineTuning
+    from mistralai.models_ import Models
+    from mistralai.ocr import Ocr
+
 
 class Mistral(BaseSDK):
     r"""Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create your account on [La Plateforme](https://console.mistral.ai) to get access and read the [docs](https://docs.mistral.ai) to learn how to use it."""
 
-    models: Models
+    models: "Models"
     r"""Model Management API"""
-    beta: Beta
-    files: Files
+    beta: "Beta"
+    files: "Files"
     r"""Files API"""
-    fine_tuning: FineTuning
-    batch: Batch
-    chat: Chat
+    fine_tuning: "FineTuning"
+    batch: "Batch"
+    chat: "Chat"
     r"""Chat Completion API."""
-    fim: Fim
+    fim: "Fim"
     r"""Fill-in-the-middle API."""
-    agents: Agents
+    agents: "Agents"
     r"""Agents API."""
-    embeddings: Embeddings
+    embeddings: "Embeddings"
     r"""Embeddings API."""
-    classifiers: Classifiers
+    classifiers: "Classifiers"
     r"""Classifiers API."""
-    ocr: Ocr
+    ocr: "Ocr"
     r"""OCR API"""
+    _sub_sdk_map = {
+        "models": ("mistralai.models_", "Models"),
+        "beta": ("mistralai.beta", "Beta"),
+        "files": ("mistralai.files", "Files"),
+        "fine_tuning": ("mistralai.fine_tuning", "FineTuning"),
+        "batch": ("mistralai.batch", "Batch"),
+        "chat": ("mistralai.chat", "Chat"),
+        "fim": ("mistralai.fim", "Fim"),
+        "agents": ("mistralai.agents", "Agents"),
+        "embeddings": ("mistralai.embeddings", "Embeddings"),
+        "classifiers": ("mistralai.classifiers", "Classifiers"),
+        "ocr": ("mistralai.ocr", "Ocr"),
+    }
 
     def __init__(
         self,
@@ -120,6 +136,9 @@ class Mistral(BaseSDK):
 
         hooks = SDKHooks()
 
+        # pylint: disable=protected-access
+        self.sdk_configuration.__dict__["_hooks"] = hooks
+
         current_server_url, *_ = self.sdk_configuration.get_server_details()
         server_url, self.sdk_configuration.client = hooks.sdk_init(
             current_server_url, client
@@ -127,9 +146,6 @@ class Mistral(BaseSDK):
         if current_server_url != server_url:
             self.sdk_configuration.server_url = server_url
 
-        # pylint: disable=protected-access
-        self.sdk_configuration.__dict__["_hooks"] = hooks
-
         weakref.finalize(
             self,
             close_clients,
@@ -140,20 +156,32 @@ class Mistral(BaseSDK):
             self.sdk_configuration.async_client_supplied,
         )
 
-        self._init_sdks()
-
-    def _init_sdks(self):
-        self.models = Models(self.sdk_configuration)
-        self.beta = Beta(self.sdk_configuration)
-        self.files = Files(self.sdk_configuration)
-        self.fine_tuning = FineTuning(self.sdk_configuration)
-        self.batch = Batch(self.sdk_configuration)
-        self.chat = Chat(self.sdk_configuration)
-        self.fim = Fim(self.sdk_configuration)
-        self.agents = Agents(self.sdk_configuration)
-        self.embeddings = Embeddings(self.sdk_configuration)
-        self.classifiers = Classifiers(self.sdk_configuration)
-        self.ocr = Ocr(self.sdk_configuration)
+    def __getattr__(self, name: str):
+        if name in self._sub_sdk_map:
+            module_path, class_name = self._sub_sdk_map[name]
+            try:
+                module = importlib.import_module(module_path)
+                klass = getattr(module, class_name)
+                instance = klass(self.sdk_configuration)
+                setattr(self, name, instance)
+                return instance
+            except ImportError as e:
+                raise AttributeError(
+                    f"Failed to import module {module_path} for attribute {name}: {e}"
+                ) from e
+            except AttributeError as e:
+                raise AttributeError(
+                    f"Failed to find class {class_name} in module {module_path} for attribute {name}: {e}"
+                ) from e
+
+        raise AttributeError(
+            f"'{type(self).__name__}' object has no attribute '{name}'"
+        )
+
+    def __dir__(self):
+        default_attrs = list(super().__dir__())
+        lazy_attrs = list(self._sub_sdk_map.keys())
+        return sorted(list(set(default_attrs + lazy_attrs)))
 
     def __enter__(self):
         return self

mistralai/sdkconfiguration.py

@@ -1,6 +1,5 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-from ._hooks import SDKHooks
 from ._version import (
     __gen_version__,
     __openapi_doc_version__,
@@ -42,9 +41,6 @@ class SDKConfiguration:
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 
-    def __post_init__(self):
-        self._hooks = SDKHooks()
-
     def get_server_details(self) -> Tuple[str, Dict[str, str]]:
         if self.server_url is not None and self.server_url:
             return remove_suffix(self.server_url, "/"), {}
@@ -55,6 +51,3 @@ class SDKConfiguration:
             raise ValueError(f'Invalid server "{self.server}"')
 
         return SERVERS[self.server], {}
-
-    def get_hooks(self) -> SDKHooks:
-        return self._hooks