mistralai 1.0.3__py3-none-any.whl → 1.2.0__py3-none-any.whl

mistralai/models/retrievefileout.py

@@ -1,20 +1,16 @@
 """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
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-RetrieveFileOutPurpose = Union[Literal["fine-tune"], UnrecognizedStr]
-r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
-
 class RetrieveFileOutTypedDict(TypedDict):
     id: str
     r"""The unique identifier of the file."""
@@ -26,28 +22,39 @@ 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]]
-    
+
 
 class RetrieveFileOut(BaseModel):
     id: str
     r"""The unique identifier of the file."""
+
     object: str
     r"""The object type, which is always \"file\"."""
+
     bytes: int
     r"""The size of the file, in bytes."""
+
     created_at: int
     r"""The UNIX timestamp (in seconds) of the event."""
+
     filename: str
     r"""The name of the uploaded file."""
-    sample_type: SampleType
-    source: Source
-    PURPOSE: Annotated[Final[Annotated[RetrieveFileOutPurpose, PlainValidator(validate_open_enum(False))]], pydantic.Field(alias="purpose")] = "fine-tune" # type: ignore
-    r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
+
+    purpose: Annotated[FilePurpose, PlainValidator(validate_open_enum(False))]
+
+    sample_type: Annotated[SampleType, PlainValidator(validate_open_enum(False))]
+
+    source: Annotated[Source, PlainValidator(validate_open_enum(False))]
+
+    deleted: bool
+
     num_lines: OptionalNullable[int] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["num_lines"]
@@ -61,9 +68,13 @@ class RetrieveFileOut(BaseModel):
         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
+            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
@@ -73,4 +84,3 @@ class RetrieveFileOut(BaseModel):
                 m[k] = val
 
         return m
-        

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,14 +3,23 @@
 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):
     api_key: NotRequired[str]
-    
+
 
 class Security(BaseModel):
-    api_key: Annotated[Optional[str], FieldMetadata(security=SecurityMetadata(scheme=True, scheme_type="http", sub_type="bearer", field_name="Authorization"))] = None
-    
+    api_key: Annotated[
+        Optional[str],
+        FieldMetadata(
+            security=SecurityMetadata(
+                scheme=True,
+                scheme_type="http",
+                sub_type="bearer",
+                field_name="Authorization",
+            )
+        ),
+    ] = None

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

@@ -1,26 +1,27 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .contentchunk import ContentChunk, ContentChunkTypedDict
+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[ContentChunkTypedDict]]
+SystemMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
 
 
-Content = Union[str, List[ContentChunk]]
+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,16 +2,25 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
+from mistralai.utils import validate_const
 import pydantic
-from typing import Final, 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"]
 
 
 class TextChunkTypedDict(TypedDict):
     text: str
-    
+    type: TextChunkType
+
 
 class TextChunk(BaseModel):
     text: str
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "text" # type: ignore
-    
+
+    TYPE: Annotated[
+        Annotated[Optional[TextChunkType], AfterValidator(validate_const("text"))],
+        pydantic.Field(alias="type"),
+    ] = "text"

mistralai/models/tool.py

@@ -2,21 +2,22 @@
 
 from __future__ import annotations
 from .function import Function, FunctionTypedDict
-from mistralai.types import BaseModel, UnrecognizedStr
+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 Literal, Optional, TypedDict, Union
-from typing_extensions import Annotated, NotRequired
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-ToolToolTypes = Union[Literal["function"], UnrecognizedStr]
-
 class ToolTypedDict(TypedDict):
     function: FunctionTypedDict
-    type: NotRequired[ToolToolTypes]
-    
+    type: NotRequired[ToolTypes]
+
 
 class Tool(BaseModel):
     function: Function
-    type: Annotated[Optional[ToolToolTypes], PlainValidator(validate_open_enum(False))] = "function"
-    
+
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
+        None
+    )

mistralai/models/toolcall.py

@@ -2,23 +2,25 @@
 
 from __future__ import annotations
 from .functioncall import FunctionCall, FunctionCallTypedDict
-from mistralai.types import BaseModel, UnrecognizedStr
+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 Literal, Optional, TypedDict, Union
-from typing_extensions import Annotated, NotRequired
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-ToolTypes = Union[Literal["function"], UnrecognizedStr]
-
 class ToolCallTypedDict(TypedDict):
     function: FunctionCallTypedDict
     id: NotRequired[str]
     type: NotRequired[ToolTypes]
-    
+
 
 class ToolCall(BaseModel):
     function: FunctionCall
+
     id: Optional[str] = "null"
-    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = "function"
-    
+
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
+        None
+    )

mistralai/models/toolchoice.py

@@ -0,0 +1,29 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .functionname import FunctionName, FunctionNameTypedDict
+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
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class ToolChoiceTypedDict(TypedDict):
+    r"""ToolChoice is either a ToolChoiceEnum or a ToolChoice"""
+
+    function: FunctionNameTypedDict
+    r"""this restriction of `Function` is used to select a specific function to call"""
+    type: NotRequired[ToolTypes]
+
+
+class ToolChoice(BaseModel):
+    r"""ToolChoice is either a ToolChoiceEnum or a ToolChoice"""
+
+    function: FunctionName
+    r"""this restriction of `Function` is used to select a specific function to call"""
+
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = (
+        None
+    )

mistralai/models/toolchoiceenum.py

@@ -0,0 +1,7 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from typing import Literal
+
+
+ToolChoiceEnum = Literal["auto", "none", "any", "required"]

mistralai/models/toolmessage.py

@@ -3,25 +3,29 @@
 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"]
 
+
 class ToolMessageTypedDict(TypedDict):
     content: str
     tool_call_id: NotRequired[Nullable[str]]
     name: NotRequired[Nullable[str]]
     role: NotRequired[ToolMessageRole]
-    
+
 
 class ToolMessage(BaseModel):
     content: str
+
     tool_call_id: OptionalNullable[str] = UNSET
+
     name: OptionalNullable[str] = UNSET
+
     role: Optional[ToolMessageRole] = "tool"
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["tool_call_id", "name", "role"]
@@ -35,9 +39,13 @@ class ToolMessage(BaseModel):
         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
+            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
@@ -47,4 +55,3 @@ class ToolMessage(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/tooltypes.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
+
+
+ToolTypes = Union[Literal["function"], UnrecognizedStr]

mistralai/models/trainingfile.py

@@ -2,16 +2,16 @@
 
 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):
     file_id: str
     weight: NotRequired[float]
-    
+
 
 class TrainingFile(BaseModel):
     file_id: str
+
     weight: Optional[float] = 1
-    

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,20 +14,43 @@ 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):
     training_steps: OptionalNullable[int] = UNSET
+
     learning_rate: Optional[float] = 0.0001
+
     weight_decay: OptionalNullable[float] = UNSET
+
     warmup_fraction: OptionalNullable[float] = UNSET
+
     epochs: OptionalNullable[float] = UNSET
+
     fim_ratio: OptionalNullable[float] = UNSET
-    
+
+    seq_len: OptionalNullable[int] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["training_steps", "learning_rate", "weight_decay", "warmup_fraction", "epochs", "fim_ratio"]
-        nullable_fields = ["training_steps", "weight_decay", "warmup_fraction", "epochs", "fim_ratio"]
+        optional_fields = [
+            "training_steps",
+            "learning_rate",
+            "weight_decay",
+            "warmup_fraction",
+            "epochs",
+            "fim_ratio",
+            "seq_len",
+        ]
+        nullable_fields = [
+            "training_steps",
+            "weight_decay",
+            "warmup_fraction",
+            "epochs",
+            "fim_ratio",
+            "seq_len",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
@@ -37,9 +60,13 @@ class TrainingParameters(BaseModel):
         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
+            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
@@ -49,4 +76,3 @@ class TrainingParameters(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/trainingparametersin.py

@@ -3,13 +3,13 @@
 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):
     r"""The fine-tuning hyperparameter settings used in a fine-tune job."""
-    
+
     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."""
     learning_rate: NotRequired[float]
@@ -20,26 +20,49 @@ 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):
     r"""The fine-tuning hyperparameter settings used in a fine-tune job."""
-    
+
     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."""
+
     learning_rate: Optional[float] = 0.0001
     r"""A parameter describing how much to adjust the pre-trained model's weights in response to the estimated error each time the weights are updated during the fine-tuning process."""
+
     weight_decay: OptionalNullable[float] = UNSET
     r"""(Advanced Usage) Weight decay adds a term to the loss function that is proportional to the sum of the squared weights. This term reduces the magnitude of the weights and prevents them from growing too large."""
+
     warmup_fraction: OptionalNullable[float] = UNSET
     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: OptionalNullable[float] = UNSET
+
     fim_ratio: OptionalNullable[float] = UNSET
-    
+
+    seq_len: OptionalNullable[int] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["training_steps", "learning_rate", "weight_decay", "warmup_fraction", "epochs", "fim_ratio"]
-        nullable_fields = ["training_steps", "weight_decay", "warmup_fraction", "epochs", "fim_ratio"]
+        optional_fields = [
+            "training_steps",
+            "learning_rate",
+            "weight_decay",
+            "warmup_fraction",
+            "epochs",
+            "fim_ratio",
+            "seq_len",
+        ]
+        nullable_fields = [
+            "training_steps",
+            "weight_decay",
+            "warmup_fraction",
+            "epochs",
+            "fim_ratio",
+            "seq_len",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
@@ -49,9 +72,13 @@ class TrainingParametersIn(BaseModel):
         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
+            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
@@ -61,4 +88,3 @@ class TrainingParametersIn(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/unarchiveftmodelout.py

@@ -2,20 +2,30 @@
 
 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"]
 
+
 class UnarchiveFTModelOutTypedDict(TypedDict):
     id: str
+    object: UnarchiveFTModelOutObject
     archived: NotRequired[bool]
-    
+
 
 class UnarchiveFTModelOut(BaseModel):
     id: str
-    OBJECT: Annotated[Final[Optional[UnarchiveFTModelOutObject]], pydantic.Field(alias="object")] = "model" # type: ignore
+
+    OBJECT: Annotated[
+        Annotated[
+            Optional[UnarchiveFTModelOutObject], AfterValidator(validate_const("model"))
+        ],
+        pydantic.Field(alias="object"),
+    ] = "model"
+
     archived: Optional[bool] = False
-    

mistralai/models/updateftmodelin.py

@@ -3,19 +3,19 @@
 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):
     name: NotRequired[Nullable[str]]
     description: NotRequired[Nullable[str]]
-    
+
 
 class UpdateFTModelIn(BaseModel):
     name: OptionalNullable[str] = UNSET
+
     description: OptionalNullable[str] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["name", "description"]
@@ -29,9 +29,13 @@ class UpdateFTModelIn(BaseModel):
         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
+            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
@@ -41,4 +45,3 @@ class UpdateFTModelIn(BaseModel):
                 m[k] = val
 
         return m
-