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

mistralai/models/jobsout.py

@@ -3,9 +3,11 @@
 from __future__ import annotations
 from .jobout import JobOut, JobOutTypedDict
 from mistralai.types import BaseModel
+from mistralai.utils import validate_const
 import pydantic
-from typing import Final, List, Literal, Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from pydantic.functional_validators import AfterValidator
+from typing import List, Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 JobsOutObject = Literal["list"]
@@ -14,6 +16,7 @@ JobsOutObject = Literal["list"]
 class JobsOutTypedDict(TypedDict):
     total: int
     data: NotRequired[List[JobOutTypedDict]]
+    object: JobsOutObject
 
 
 class JobsOut(BaseModel):
@@ -21,6 +24,7 @@ class JobsOut(BaseModel):
 
     data: Optional[List[JobOut]] = None
 
-    # fmt: off
-    OBJECT: Annotated[Final[Optional[JobsOutObject]], pydantic.Field(alias="object")] = "list" # type: ignore
-    # fmt: on
+    OBJECT: Annotated[
+        Annotated[Optional[JobsOutObject], AfterValidator(validate_const("list"))],
+        pydantic.Field(alias="object"),
+    ] = "list"

mistralai/models/legacyjobmetadataout.py

@@ -2,10 +2,12 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import validate_const
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Literal, Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 LegacyJobMetadataOutObject = Literal["job.metadata"]
@@ -31,6 +33,7 @@ class LegacyJobMetadataOutTypedDict(TypedDict):
     r"""The number of complete passes through the entire training dataset."""
     training_steps: NotRequired[Nullable[int]]
     r"""The number of training steps to perform. A training step refers to a single update of the model weights during the fine-tuning process. This update is typically calculated using a batch of samples from the training dataset."""
+    object: LegacyJobMetadataOutObject
 
 
 class LegacyJobMetadataOut(BaseModel):
@@ -64,9 +67,13 @@ class LegacyJobMetadataOut(BaseModel):
     training_steps: OptionalNullable[int] = UNSET
     r"""The number of training steps to perform. A training step refers to a single update of the model weights during the fine-tuning process. This update is typically calculated using a batch of samples from the training dataset."""
 
-    # fmt: off
-    OBJECT: Annotated[Final[Optional[LegacyJobMetadataOutObject]], pydantic.Field(alias="object")] = "job.metadata" # type: ignore
-    # fmt: on
+    OBJECT: Annotated[
+        Annotated[
+            Optional[LegacyJobMetadataOutObject],
+            AfterValidator(validate_const("job.metadata")),
+        ],
+        pydantic.Field(alias="object"),
+    ] = "job.metadata"
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):

mistralai/models/listfilesout.py

@@ -3,15 +3,19 @@
 from __future__ import annotations
 from .fileschema import FileSchema, FileSchemaTypedDict
 from mistralai.types import BaseModel
-from typing import List, TypedDict
+from typing import List
+from typing_extensions import TypedDict
 
 
 class ListFilesOutTypedDict(TypedDict):
     data: List[FileSchemaTypedDict]
     object: str
+    total: int
 
 
 class ListFilesOut(BaseModel):
     data: List[FileSchema]
 
     object: str
+
+    total: int

mistralai/models/metricout.py

@@ -3,8 +3,7 @@
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import TypedDict
-from typing_extensions import NotRequired
+from typing_extensions import NotRequired, TypedDict
 
 
 class MetricOutTypedDict(TypedDict):

mistralai/models/modelcapabilities.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from typing import Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 class ModelCapabilitiesTypedDict(TypedDict):

mistralai/models/modellist.py

@@ -6,8 +6,8 @@ from .ftmodelcard import FTModelCard, FTModelCardTypedDict
 from mistralai.types import BaseModel
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag
-from typing import List, Optional, TypedDict, Union
-from typing_extensions import Annotated, NotRequired
+from typing import List, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 DataTypedDict = Union[BaseModelCardTypedDict, FTModelCardTypedDict]

mistralai/models/responseformat.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 from .responseformats import ResponseFormats
 from mistralai.types import BaseModel
-from typing import Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 class ResponseFormatTypedDict(TypedDict):

mistralai/models/retrieve_model_v1_models_model_id_getop.py

@@ -6,8 +6,8 @@ from .ftmodelcard import FTModelCard, FTModelCardTypedDict
 from mistralai.types import BaseModel
 from mistralai.utils import FieldMetadata, PathParamMetadata, get_discriminator
 from pydantic import Discriminator, Tag
-from typing import TypedDict, Union
-from typing_extensions import Annotated
+from typing import Union
+from typing_extensions import Annotated, TypedDict
 
 
 class RetrieveModelV1ModelsModelIDGetRequestTypedDict(TypedDict):

mistralai/models/retrievefileout.py

@@ -1,26 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .filepurpose import FilePurpose
 from .sampletype import SampleType
 from .source import Source
-from mistralai.types import (
-    BaseModel,
-    Nullable,
-    OptionalNullable,
-    UNSET,
-    UNSET_SENTINEL,
-    UnrecognizedStr,
-)
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import validate_open_enum
-import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
-from typing import Final, Literal, TypedDict, Union
-from typing_extensions import Annotated, NotRequired
-
-
-RetrieveFileOutPurpose = Union[Literal["fine-tune"], UnrecognizedStr]
-r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class RetrieveFileOutTypedDict(TypedDict):
@@ -34,8 +22,10 @@ class RetrieveFileOutTypedDict(TypedDict):
     r"""The UNIX timestamp (in seconds) of the event."""
     filename: str
     r"""The name of the uploaded file."""
+    purpose: FilePurpose
     sample_type: SampleType
     source: Source
+    deleted: bool
     num_lines: NotRequired[Nullable[int]]
 
 
@@ -55,14 +45,13 @@ class RetrieveFileOut(BaseModel):
     filename: str
     r"""The name of the uploaded file."""
 
-    sample_type: SampleType
+    purpose: Annotated[FilePurpose, PlainValidator(validate_open_enum(False))]
 
-    source: Source
+    sample_type: Annotated[SampleType, PlainValidator(validate_open_enum(False))]
+
+    source: Annotated[Source, PlainValidator(validate_open_enum(False))]
 
-    # fmt: off
-    PURPOSE: Annotated[Final[Annotated[RetrieveFileOutPurpose, PlainValidator(validate_open_enum(False))]], pydantic.Field(alias="purpose")] = "fine-tune" # type: ignore
-    # fmt: on
-    r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
+    deleted: bool
 
     num_lines: OptionalNullable[int] = UNSET
 

mistralai/models/sampletype.py

@@ -1,7 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from typing import Literal
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
 
 
-SampleType = Literal["pretrain", "instruct"]
+SampleType = Union[
+    Literal["pretrain", "instruct", "batch_request", "batch_result", "batch_error"],
+    UnrecognizedStr,
+]

mistralai/models/security.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 from mistralai.types import BaseModel
 from mistralai.utils import FieldMetadata, SecurityMetadata
-from typing import Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class SecurityTypedDict(TypedDict):

mistralai/models/source.py

@@ -1,7 +1,8 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from typing import Literal
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
 
 
-Source = Literal["upload", "repository"]
+Source = Union[Literal["upload", "repository", "mistral"], UnrecognizedStr]

mistralai/models/systemmessage.py

@@ -3,25 +3,25 @@
 from __future__ import annotations
 from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai.types import BaseModel
-from typing import List, Literal, Optional, TypedDict, Union
-from typing_extensions import NotRequired
+from typing import List, Literal, Optional, Union
+from typing_extensions import NotRequired, TypedDict
 
 
-ContentTypedDict = Union[str, List[TextChunkTypedDict]]
+SystemMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
 
 
-Content = Union[str, List[TextChunk]]
+SystemMessageContent = Union[str, List[TextChunk]]
 
 
 Role = Literal["system"]
 
 
 class SystemMessageTypedDict(TypedDict):
-    content: ContentTypedDict
+    content: SystemMessageContentTypedDict
     role: NotRequired[Role]
 
 
 class SystemMessage(BaseModel):
-    content: Content
+    content: SystemMessageContent
 
     role: Optional[Role] = "system"

mistralai/models/textchunk.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
+from mistralai.utils import validate_const
 import pydantic
-from typing import Final, Literal, Optional, TypedDict
-from typing_extensions import Annotated
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, TypedDict
 
 
 TextChunkType = Literal["text"]
@@ -12,11 +14,13 @@ TextChunkType = Literal["text"]
 
 class TextChunkTypedDict(TypedDict):
     text: str
+    type: TextChunkType
 
 
 class TextChunk(BaseModel):
     text: str
 
-    # fmt: off
-    TYPE: Annotated[Final[Optional[TextChunkType]], pydantic.Field(alias="type")] = "text" # type: ignore
-    # fmt: on
+    TYPE: Annotated[
+        Annotated[Optional[TextChunkType], AfterValidator(validate_const("text"))],
+        pydantic.Field(alias="type"),
+    ] = "text"

mistralai/models/tool.py

@@ -6,8 +6,8 @@ from .tooltypes import ToolTypes
 from mistralai.types import BaseModel
 from mistralai.utils import validate_open_enum
 from pydantic.functional_validators import PlainValidator
-from typing import Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class ToolTypedDict(TypedDict):

mistralai/models/toolcall.py

@@ -6,8 +6,8 @@ from .tooltypes import ToolTypes
 from mistralai.types import BaseModel
 from mistralai.utils import validate_open_enum
 from pydantic.functional_validators import PlainValidator
-from typing import Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class ToolCallTypedDict(TypedDict):

mistralai/models/toolchoice.py

@@ -6,8 +6,8 @@ from .tooltypes import ToolTypes
 from mistralai.types import BaseModel
 from mistralai.utils import validate_open_enum
 from pydantic.functional_validators import PlainValidator
-from typing import Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class ToolChoiceTypedDict(TypedDict):

mistralai/models/toolmessage.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Literal, Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 ToolMessageRole = Literal["tool"]

mistralai/models/trainingfile.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from typing import Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 class TrainingFileTypedDict(TypedDict):

mistralai/models/trainingparameters.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 class TrainingParametersTypedDict(TypedDict):
@@ -14,6 +14,7 @@ class TrainingParametersTypedDict(TypedDict):
     warmup_fraction: NotRequired[Nullable[float]]
     epochs: NotRequired[Nullable[float]]
     fim_ratio: NotRequired[Nullable[float]]
+    seq_len: NotRequired[Nullable[int]]
 
 
 class TrainingParameters(BaseModel):
@@ -29,6 +30,8 @@ class TrainingParameters(BaseModel):
 
     fim_ratio: OptionalNullable[float] = UNSET
 
+    seq_len: OptionalNullable[int] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -38,6 +41,7 @@ class TrainingParameters(BaseModel):
             "warmup_fraction",
             "epochs",
             "fim_ratio",
+            "seq_len",
         ]
         nullable_fields = [
             "training_steps",
@@ -45,6 +49,7 @@ class TrainingParameters(BaseModel):
             "warmup_fraction",
             "epochs",
             "fim_ratio",
+            "seq_len",
         ]
         null_default_fields = []
 

mistralai/models/trainingparametersin.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 class TrainingParametersInTypedDict(TypedDict):
@@ -20,6 +20,7 @@ class TrainingParametersInTypedDict(TypedDict):
     r"""(Advanced Usage) A parameter that specifies the percentage of the total training steps at which the learning rate warm-up phase ends. During this phase, the learning rate gradually increases from a small value to the initial learning rate, helping to stabilize the training process and improve convergence. Similar to `pct_start` in [mistral-finetune](https://github.com/mistralai/mistral-finetune)"""
     epochs: NotRequired[Nullable[float]]
     fim_ratio: NotRequired[Nullable[float]]
+    seq_len: NotRequired[Nullable[int]]
 
 
 class TrainingParametersIn(BaseModel):
@@ -41,6 +42,8 @@ class TrainingParametersIn(BaseModel):
 
     fim_ratio: OptionalNullable[float] = UNSET
 
+    seq_len: OptionalNullable[int] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = [
@@ -50,6 +53,7 @@ class TrainingParametersIn(BaseModel):
             "warmup_fraction",
             "epochs",
             "fim_ratio",
+            "seq_len",
         ]
         nullable_fields = [
             "training_steps",
@@ -57,6 +61,7 @@ class TrainingParametersIn(BaseModel):
             "warmup_fraction",
             "epochs",
             "fim_ratio",
+            "seq_len",
         ]
         null_default_fields = []
 

mistralai/models/unarchiveftmodelout.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
+from mistralai.utils import validate_const
 import pydantic
-from typing import Final, Literal, Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 UnarchiveFTModelOutObject = Literal["model"]
@@ -12,14 +14,18 @@ UnarchiveFTModelOutObject = Literal["model"]
 
 class UnarchiveFTModelOutTypedDict(TypedDict):
     id: str
+    object: UnarchiveFTModelOutObject
     archived: NotRequired[bool]
 
 
 class UnarchiveFTModelOut(BaseModel):
     id: str
 
-    # fmt: off
-    OBJECT: Annotated[Final[Optional[UnarchiveFTModelOutObject]], pydantic.Field(alias="object")] = "model" # type: ignore
-    # fmt: on
+    OBJECT: Annotated[
+        Annotated[
+            Optional[UnarchiveFTModelOutObject], AfterValidator(validate_const("model"))
+        ],
+        pydantic.Field(alias="object"),
+    ] = "model"
 
     archived: Optional[bool] = False

mistralai/models/updateftmodelin.py

@@ -3,8 +3,7 @@
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import TypedDict
-from typing_extensions import NotRequired
+from typing_extensions import NotRequired, TypedDict
 
 
 class UpdateFTModelInTypedDict(TypedDict):

mistralai/models/uploadfileout.py

@@ -1,26 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .filepurpose import FilePurpose
 from .sampletype import SampleType
 from .source import Source
-from mistralai.types import (
-    BaseModel,
-    Nullable,
-    OptionalNullable,
-    UNSET,
-    UNSET_SENTINEL,
-    UnrecognizedStr,
-)
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import validate_open_enum
-import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
-from typing import Final, Literal, TypedDict, Union
-from typing_extensions import Annotated, NotRequired
-
-
-Purpose = Union[Literal["fine-tune"], UnrecognizedStr]
-r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class UploadFileOutTypedDict(TypedDict):
@@ -34,6 +22,7 @@ class UploadFileOutTypedDict(TypedDict):
     r"""The UNIX timestamp (in seconds) of the event."""
     filename: str
     r"""The name of the uploaded file."""
+    purpose: FilePurpose
     sample_type: SampleType
     source: Source
     num_lines: NotRequired[Nullable[int]]
@@ -55,14 +44,11 @@ class UploadFileOut(BaseModel):
     filename: str
     r"""The name of the uploaded file."""
 
-    sample_type: SampleType
+    purpose: Annotated[FilePurpose, PlainValidator(validate_open_enum(False))]
 
-    source: Source
+    sample_type: Annotated[SampleType, PlainValidator(validate_open_enum(False))]
 
-    # fmt: off
-    PURPOSE: Annotated[Final[Annotated[Purpose, PlainValidator(validate_open_enum(False))]], pydantic.Field(alias="purpose")] = "fine-tune" # type: ignore
-    # fmt: on
-    r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
+    source: Annotated[Source, PlainValidator(validate_open_enum(False))]
 
     num_lines: OptionalNullable[int] = UNSET
 

mistralai/models/usageinfo.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from typing import TypedDict
+from typing_extensions import TypedDict
 
 
 class UsageInfoTypedDict(TypedDict):

mistralai/models/usermessage.py

@@ -2,9 +2,10 @@
 
 from __future__ import annotations
 from .contentchunk import ContentChunk, ContentChunkTypedDict
-from mistralai.types import BaseModel
-from typing import List, Literal, Optional, TypedDict, Union
-from typing_extensions import NotRequired
+from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import List, Literal, Optional, Union
+from typing_extensions import NotRequired, TypedDict
 
 
 UserMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
@@ -17,11 +18,41 @@ UserMessageRole = Literal["user"]
 
 
 class UserMessageTypedDict(TypedDict):
-    content: UserMessageContentTypedDict
+    content: Nullable[UserMessageContentTypedDict]
     role: NotRequired[UserMessageRole]
 
 
 class UserMessage(BaseModel):
-    content: UserMessageContent
+    content: Nullable[UserMessageContent]
 
     role: Optional[UserMessageRole] = "user"
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["role"]
+        nullable_fields = ["content"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        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
+
+            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/validationerror.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from typing import List, TypedDict, Union
+from typing import List, Union
+from typing_extensions import TypedDict
 
 
 LocTypedDict = Union[str, int]