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

mistralai/models/libraries_documents_reprocess_v1op.py

@@ -0,0 +1,21 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesDocumentsReprocessV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+
+
+class LibrariesDocumentsReprocessV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    document_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/libraries_documents_update_v1op.py

@@ -0,0 +1,28 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .documentupdatein import DocumentUpdateIn, DocumentUpdateInTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesDocumentsUpdateV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+    document_update_in: DocumentUpdateInTypedDict
+
+
+class LibrariesDocumentsUpdateV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    document_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    document_update_in: Annotated[
+        DocumentUpdateIn,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

mistralai/models/libraries_documents_upload_v1op.py

@@ -0,0 +1,56 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .file import File, FileTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import (
+    FieldMetadata,
+    MultipartFormMetadata,
+    PathParamMetadata,
+    RequestMetadata,
+)
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesDocumentsUploadV1DocumentUploadTypedDict(TypedDict):
+    file: FileTypedDict
+    r"""The File object (not file name) to be uploaded.
+    To upload a file and specify a custom file name you should format your request as such:
+    ```bash
+    file=@path/to/your/file.jsonl;filename=custom_name.jsonl
+    ```
+    Otherwise, you can just keep the original file name:
+    ```bash
+    file=@path/to/your/file.jsonl
+    ```
+    """
+
+
+class LibrariesDocumentsUploadV1DocumentUpload(BaseModel):
+    file: Annotated[File, FieldMetadata(multipart=MultipartFormMetadata(file=True))]
+    r"""The File object (not file name) to be uploaded.
+    To upload a file and specify a custom file name you should format your request as such:
+    ```bash
+    file=@path/to/your/file.jsonl;filename=custom_name.jsonl
+    ```
+    Otherwise, you can just keep the original file name:
+    ```bash
+    file=@path/to/your/file.jsonl
+    ```
+    """
+
+
+class LibrariesDocumentsUploadV1RequestTypedDict(TypedDict):
+    library_id: str
+    request_body: LibrariesDocumentsUploadV1DocumentUploadTypedDict
+
+
+class LibrariesDocumentsUploadV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    request_body: Annotated[
+        LibrariesDocumentsUploadV1DocumentUpload,
+        FieldMetadata(request=RequestMetadata(media_type="multipart/form-data")),
+    ]

mistralai/models/libraries_get_v1op.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 mistralai.utils import FieldMetadata, PathParamMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesGetV1RequestTypedDict(TypedDict):
+    library_id: str
+
+
+class LibrariesGetV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/libraries_share_create_v1op.py

@@ -0,0 +1,22 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .sharingin import SharingIn, SharingInTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesShareCreateV1RequestTypedDict(TypedDict):
+    library_id: str
+    sharing_in: SharingInTypedDict
+
+
+class LibrariesShareCreateV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    sharing_in: Annotated[
+        SharingIn, FieldMetadata(request=RequestMetadata(media_type="application/json"))
+    ]

mistralai/models/libraries_share_delete_v1op.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .sharingdelete import SharingDelete, SharingDeleteTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesShareDeleteV1RequestTypedDict(TypedDict):
+    library_id: str
+    sharing_delete: SharingDeleteTypedDict
+
+
+class LibrariesShareDeleteV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    sharing_delete: Annotated[
+        SharingDelete,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

mistralai/models/libraries_share_list_v1op.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 mistralai.utils import FieldMetadata, PathParamMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesShareListV1RequestTypedDict(TypedDict):
+    library_id: str
+
+
+class LibrariesShareListV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/libraries_update_v1op.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .libraryinupdate import LibraryInUpdate, LibraryInUpdateTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class LibrariesUpdateV1RequestTypedDict(TypedDict):
+    library_id: str
+    library_in_update: LibraryInUpdateTypedDict
+
+
+class LibrariesUpdateV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    library_in_update: Annotated[
+        LibraryInUpdate,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

mistralai/models/libraryin.py

@@ -0,0 +1,50 @@
+"""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 LibraryInTypedDict(TypedDict):
+    name: str
+    description: NotRequired[Nullable[str]]
+    chunk_size: NotRequired[Nullable[int]]
+
+
+class LibraryIn(BaseModel):
+    name: str
+
+    description: OptionalNullable[str] = UNSET
+
+    chunk_size: OptionalNullable[int] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["description", "chunk_size"]
+        nullable_fields = ["description", "chunk_size"]
+        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/libraryinupdate.py

@@ -0,0 +1,47 @@
+"""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 LibraryInUpdateTypedDict(TypedDict):
+    name: NotRequired[Nullable[str]]
+    description: NotRequired[Nullable[str]]
+
+
+class LibraryInUpdate(BaseModel):
+    name: OptionalNullable[str] = UNSET
+
+    description: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["name", "description"]
+        nullable_fields = ["name", "description"]
+        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/libraryout.py

@@ -0,0 +1,107 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from datetime import datetime
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class LibraryOutTypedDict(TypedDict):
+    id: str
+    name: str
+    created_at: datetime
+    updated_at: datetime
+    owner_id: str
+    owner_type: str
+    total_size: int
+    nb_documents: int
+    chunk_size: Nullable[int]
+    emoji: NotRequired[Nullable[str]]
+    description: NotRequired[Nullable[str]]
+    generated_name: NotRequired[Nullable[str]]
+    generated_description: NotRequired[Nullable[str]]
+    explicit_user_members_count: NotRequired[Nullable[int]]
+    explicit_workspace_members_count: NotRequired[Nullable[int]]
+    org_sharing_role: NotRequired[Nullable[str]]
+
+
+class LibraryOut(BaseModel):
+    id: str
+
+    name: str
+
+    created_at: datetime
+
+    updated_at: datetime
+
+    owner_id: str
+
+    owner_type: str
+
+    total_size: int
+
+    nb_documents: int
+
+    chunk_size: Nullable[int]
+
+    emoji: OptionalNullable[str] = UNSET
+
+    description: OptionalNullable[str] = UNSET
+
+    generated_name: OptionalNullable[str] = UNSET
+
+    generated_description: OptionalNullable[str] = UNSET
+
+    explicit_user_members_count: OptionalNullable[int] = UNSET
+
+    explicit_workspace_members_count: OptionalNullable[int] = UNSET
+
+    org_sharing_role: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "emoji",
+            "description",
+            "generated_name",
+            "generated_description",
+            "explicit_user_members_count",
+            "explicit_workspace_members_count",
+            "org_sharing_role",
+        ]
+        nullable_fields = [
+            "chunk_size",
+            "emoji",
+            "description",
+            "generated_name",
+            "generated_description",
+            "explicit_user_members_count",
+            "explicit_workspace_members_count",
+            "org_sharing_role",
+        ]
+        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/listdocumentout.py

@@ -0,0 +1,19 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .documentout import DocumentOut, DocumentOutTypedDict
+from .paginationinfo import PaginationInfo, PaginationInfoTypedDict
+from mistralai.types import BaseModel
+from typing import List
+from typing_extensions import TypedDict
+
+
+class ListDocumentOutTypedDict(TypedDict):
+    pagination: PaginationInfoTypedDict
+    data: List[DocumentOutTypedDict]
+
+
+class ListDocumentOut(BaseModel):
+    pagination: PaginationInfo
+
+    data: List[DocumentOut]

mistralai/models/listlibraryout.py

@@ -0,0 +1,15 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .libraryout import LibraryOut, LibraryOutTypedDict
+from mistralai.types import BaseModel
+from typing import List
+from typing_extensions import TypedDict
+
+
+class ListLibraryOutTypedDict(TypedDict):
+    data: List[LibraryOutTypedDict]
+
+
+class ListLibraryOut(BaseModel):
+    data: List[LibraryOut]

mistralai/models/listsharingout.py

@@ -0,0 +1,15 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .sharingout import SharingOut, SharingOutTypedDict
+from mistralai.types import BaseModel
+from typing import List
+from typing_extensions import TypedDict
+
+
+class ListSharingOutTypedDict(TypedDict):
+    data: List[SharingOutTypedDict]
+
+
+class ListSharingOut(BaseModel):
+    data: List[SharingOut]

mistralai/models/messageinputentry.py

@@ -39,6 +39,7 @@ class MessageInputEntryTypedDict(TypedDict):
     created_at: NotRequired[datetime]
     completed_at: NotRequired[Nullable[datetime]]
     id: NotRequired[str]
+    prefix: NotRequired[bool]
 
 
 class MessageInputEntry(BaseModel):
@@ -58,9 +59,18 @@ class MessageInputEntry(BaseModel):
 
     id: Optional[str] = None
 
+    prefix: Optional[bool] = False
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["object", "type", "created_at", "completed_at", "id"]
+        optional_fields = [
+            "object",
+            "type",
+            "created_at",
+            "completed_at",
+            "id",
+            "prefix",
+        ]
         nullable_fields = ["completed_at"]
         null_default_fields = []
 
@@ -68,7 +78,7 @@ class MessageInputEntry(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/messageoutputentry.py

@@ -79,7 +79,7 @@ class MessageOutputEntry(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/messageoutputevent.py

@@ -72,7 +72,7 @@ class MessageOutputEvent(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/metricout.py

@@ -33,7 +33,7 @@ class MetricOut(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/modelcapabilities.py

@@ -12,6 +12,7 @@ class ModelCapabilitiesTypedDict(TypedDict):
     function_calling: NotRequired[bool]
     fine_tuning: NotRequired[bool]
     vision: NotRequired[bool]
+    classification: NotRequired[bool]
 
 
 class ModelCapabilities(BaseModel):
@@ -24,3 +25,5 @@ class ModelCapabilities(BaseModel):
     fine_tuning: Optional[bool] = False
 
     vision: Optional[bool] = False
+
+    classification: Optional[bool] = False

mistralai/models/modelconversation.py

@@ -106,7 +106,7 @@ class ModelConversation(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/ocrimageobject.py

@@ -62,7 +62,7 @@ class OCRImageObject(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/ocrpageobject.py

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

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
+from .filechunk import FileChunk, FileChunkTypedDict
 from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
@@ -11,12 +12,13 @@ from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
 DocumentTypedDict = TypeAliasType(
-    "DocumentTypedDict", Union[ImageURLChunkTypedDict, DocumentURLChunkTypedDict]
+    "DocumentTypedDict",
+    Union[FileChunkTypedDict, ImageURLChunkTypedDict, DocumentURLChunkTypedDict],
 )
 r"""Document to run OCR on"""
 
 
-Document = TypeAliasType("Document", Union[ImageURLChunk, DocumentURLChunk])
+Document = TypeAliasType("Document", Union[FileChunk, ImageURLChunk, DocumentURLChunk])
 r"""Document to run OCR on"""
 
 
@@ -91,7 +93,7 @@ class OCRRequest(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/ocrresponse.py

@@ -41,7 +41,7 @@ class OCRResponse(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/ocrusageinfo.py

@@ -30,7 +30,7 @@ class OCRUsageInfo(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/paginationinfo.py

@@ -0,0 +1,25 @@
+"""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 PaginationInfoTypedDict(TypedDict):
+    total_items: int
+    total_pages: int
+    current_page: int
+    page_size: int
+    has_more: bool
+
+
+class PaginationInfo(BaseModel):
+    total_items: int
+
+    total_pages: int
+
+    current_page: int
+
+    page_size: int
+
+    has_more: bool