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

mistralai/models/documenttextcontent.py

@@ -0,0 +1,13 @@
+"""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 DocumentTextContentTypedDict(TypedDict):
+    text: str
+
+
+class DocumentTextContent(BaseModel):
+    text: str

mistralai/models/documentupdatein.py

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

@@ -35,7 +35,7 @@ class DocumentURLChunk(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/embeddingrequest.py

@@ -51,7 +51,7 @@ class EmbeddingRequest(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/entitytype.py

@@ -0,0 +1,9 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+EntityType = Union[Literal["User", "Workspace", "Org"], UnrecognizedStr]
+r"""The type of entity, used to share a library."""

mistralai/models/eventout.py

@@ -34,7 +34,7 @@ class EventOut(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/file.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+import io
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, MultipartFormMetadata
+import pydantic
+from typing import IO, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class FileTypedDict(TypedDict):
+    file_name: str
+    content: Union[bytes, IO[bytes], io.BufferedReader]
+    content_type: NotRequired[str]
+
+
+class File(BaseModel):
+    file_name: Annotated[
+        str, pydantic.Field(alias="fileName"), FieldMetadata(multipart=True)
+    ]
+
+    content: Annotated[
+        Union[bytes, IO[bytes], io.BufferedReader],
+        pydantic.Field(alias=""),
+        FieldMetadata(multipart=MultipartFormMetadata(content=True)),
+    ]
+
+    content_type: Annotated[
+        Optional[str],
+        pydantic.Field(alias="Content-Type"),
+        FieldMetadata(multipart=True),
+    ] = None

mistralai/models/filechunk.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import validate_const
+import pydantic
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, TypedDict
+
+
+class FileChunkTypedDict(TypedDict):
+    file_id: str
+    type: Literal["file"]
+
+
+class FileChunk(BaseModel):
+    file_id: str
+
+    TYPE: Annotated[
+        Annotated[Optional[Literal["file"]], AfterValidator(validate_const("file"))],
+        pydantic.Field(alias="type"),
+    ] = "file"

mistralai/models/files_api_routes_list_filesop.py

@@ -75,7 +75,7 @@ class FilesAPIRoutesListFilesRequest(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/files_api_routes_upload_fileop.py

@@ -1,40 +1,15 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .file import File, FileTypedDict
 from .filepurpose import FilePurpose
-import io
 from mistralai.types import BaseModel
 from mistralai.utils import FieldMetadata, MultipartFormMetadata, validate_open_enum
-import pydantic
 from pydantic.functional_validators import PlainValidator
-from typing import IO, Optional, Union
+from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class FileTypedDict(TypedDict):
-    file_name: str
-    content: Union[bytes, IO[bytes], io.BufferedReader]
-    content_type: NotRequired[str]
-
-
-class File(BaseModel):
-    file_name: Annotated[
-        str, pydantic.Field(alias="fileName"), FieldMetadata(multipart=True)
-    ]
-
-    content: Annotated[
-        Union[bytes, IO[bytes], io.BufferedReader],
-        pydantic.Field(alias=""),
-        FieldMetadata(multipart=MultipartFormMetadata(content=True)),
-    ]
-
-    content_type: Annotated[
-        Optional[str],
-        pydantic.Field(alias="Content-Type"),
-        FieldMetadata(multipart=True),
-    ] = None
-
-
 class FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict(TypedDict):
     file: FileTypedDict
     r"""The File object (not file name) to be uploaded.

mistralai/models/fileschema.py

@@ -63,7 +63,7 @@ class FileSchema(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/fimcompletionrequest.py

@@ -104,7 +104,7 @@ class FIMCompletionRequest(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/fimcompletionstreamrequest.py

@@ -102,7 +102,7 @@ class FIMCompletionStreamRequest(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/ftmodelcard.py

@@ -30,6 +30,7 @@ class FTModelCardTypedDict(TypedDict):
     max_context_length: NotRequired[int]
     aliases: NotRequired[List[str]]
     deprecation: NotRequired[Nullable[datetime]]
+    deprecation_replacement_model: NotRequired[Nullable[str]]
     default_model_temperature: NotRequired[Nullable[float]]
     type: FTModelCardType
     archived: NotRequired[bool]
@@ -62,6 +63,8 @@ class FTModelCard(BaseModel):
 
     deprecation: OptionalNullable[datetime] = UNSET
 
+    deprecation_replacement_model: OptionalNullable[str] = UNSET
+
     default_model_temperature: OptionalNullable[float] = UNSET
 
     TYPE: Annotated[
@@ -84,6 +87,7 @@ class FTModelCard(BaseModel):
             "max_context_length",
             "aliases",
             "deprecation",
+            "deprecation_replacement_model",
             "default_model_temperature",
             "type",
             "archived",
@@ -92,6 +96,7 @@ class FTModelCard(BaseModel):
             "name",
             "description",
             "deprecation",
+            "deprecation_replacement_model",
             "default_model_temperature",
         ]
         null_default_fields = []
@@ -100,7 +105,7 @@ class FTModelCard(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/functioncallentry.py

@@ -55,7 +55,7 @@ class FunctionCallEntry(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/functionresultentry.py

@@ -48,7 +48,7 @@ class FunctionResultEntry(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/githubrepositoryin.py

@@ -42,7 +42,7 @@ class GithubRepositoryIn(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/githubrepositoryout.py

@@ -42,7 +42,7 @@ class GithubRepositoryOut(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/imageurl.py

@@ -26,7 +26,7 @@ class ImageURL(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/inputentries.py

@@ -1,18 +1,37 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .agenthandoffentry import AgentHandoffEntry, AgentHandoffEntryTypedDict
+from .functioncallentry import FunctionCallEntry, FunctionCallEntryTypedDict
 from .functionresultentry import FunctionResultEntry, FunctionResultEntryTypedDict
 from .messageinputentry import MessageInputEntry, MessageInputEntryTypedDict
+from .messageoutputentry import MessageOutputEntry, MessageOutputEntryTypedDict
+from .toolexecutionentry import ToolExecutionEntry, ToolExecutionEntryTypedDict
 from typing import Union
 from typing_extensions import TypeAliasType
 
 
 InputEntriesTypedDict = TypeAliasType(
     "InputEntriesTypedDict",
-    Union[MessageInputEntryTypedDict, FunctionResultEntryTypedDict],
+    Union[
+        FunctionResultEntryTypedDict,
+        MessageInputEntryTypedDict,
+        FunctionCallEntryTypedDict,
+        ToolExecutionEntryTypedDict,
+        MessageOutputEntryTypedDict,
+        AgentHandoffEntryTypedDict,
+    ],
 )
 
 
 InputEntries = TypeAliasType(
-    "InputEntries", Union[MessageInputEntry, FunctionResultEntry]
+    "InputEntries",
+    Union[
+        FunctionResultEntry,
+        MessageInputEntry,
+        FunctionCallEntry,
+        ToolExecutionEntry,
+        MessageOutputEntry,
+        AgentHandoffEntry,
+    ],
 )

mistralai/models/jobin.py

@@ -120,7 +120,7 @@ class JobIn(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/jobmetadataout.py

@@ -57,7 +57,7 @@ class JobMetadataOut(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/jobs_api_routes_batch_get_batch_jobsop.py

@@ -74,7 +74,7 @@ class JobsAPIRoutesBatchGetBatchJobsRequest(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/jobs_api_routes_fine_tuning_get_fine_tuning_jobsop.py

@@ -135,7 +135,7 @@ class JobsAPIRoutesFineTuningGetFineTuningJobsRequest(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/jsonschema.py

@@ -34,7 +34,7 @@ class JSONSchema(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/legacyjobmetadataout.py

@@ -98,7 +98,7 @@ class LegacyJobMetadataOut(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/libraries_delete_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 LibrariesDeleteV1RequestTypedDict(TypedDict):
+    library_id: str
+
+
+class LibrariesDeleteV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/libraries_documents_delete_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 LibrariesDocumentsDeleteV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+
+
+class LibrariesDocumentsDeleteV1Request(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_get_extracted_text_signed_url_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 LibrariesDocumentsGetExtractedTextSignedURLV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+
+
+class LibrariesDocumentsGetExtractedTextSignedURLV1Request(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_get_signed_url_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 LibrariesDocumentsGetSignedURLV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+
+
+class LibrariesDocumentsGetSignedURLV1Request(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_get_status_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 LibrariesDocumentsGetStatusV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+
+
+class LibrariesDocumentsGetStatusV1Request(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_get_text_content_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 LibrariesDocumentsGetTextContentV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+
+
+class LibrariesDocumentsGetTextContentV1Request(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_get_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 LibrariesDocumentsGetV1RequestTypedDict(TypedDict):
+    library_id: str
+    document_id: str
+
+
+class LibrariesDocumentsGetV1Request(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_list_v1op.py

@@ -0,0 +1,78 @@
+"""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 mistralai.utils import FieldMetadata, PathParamMetadata, QueryParamMetadata
+from pydantic import model_serializer
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class LibrariesDocumentsListV1RequestTypedDict(TypedDict):
+    library_id: str
+    search: NotRequired[Nullable[str]]
+    page_size: NotRequired[int]
+    page: NotRequired[int]
+    sort_by: NotRequired[str]
+    sort_order: NotRequired[str]
+
+
+class LibrariesDocumentsListV1Request(BaseModel):
+    library_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    search: Annotated[
+        OptionalNullable[str],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = UNSET
+
+    page_size: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 100
+
+    page: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 0
+
+    sort_by: Annotated[
+        Optional[str],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = "created_at"
+
+    sort_order: Annotated[
+        Optional[str],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = "desc"
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["search", "page_size", "page", "sort_by", "sort_order"]
+        nullable_fields = ["search"]
+        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