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

mistralai/models/trainingparameters.py

@@ -14,20 +14,38 @@ class TrainingParametersTypedDict(TypedDict):
     warmup_fraction: NotRequired[Nullable[float]]
     epochs: NotRequired[Nullable[float]]
     fim_ratio: NotRequired[Nullable[float]]
-    
+
 
 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
-    
+
     @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",
+        ]
+        nullable_fields = [
+            "training_steps",
+            "weight_decay",
+            "warmup_fraction",
+            "epochs",
+            "fim_ratio",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
@@ -37,9 +55,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 +71,3 @@ class TrainingParameters(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/trainingparametersin.py

@@ -9,7 +9,7 @@ from typing_extensions import NotRequired
 
 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,44 @@ 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]]
-    
+
 
 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
-    
+
     @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",
+        ]
+        nullable_fields = [
+            "training_steps",
+            "weight_decay",
+            "warmup_fraction",
+            "epochs",
+            "fim_ratio",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
@@ -49,9 +67,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 +83,3 @@ class TrainingParametersIn(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/unarchiveftmodelout.py

@@ -9,13 +9,17 @@ from typing_extensions import Annotated, NotRequired
 
 UnarchiveFTModelOutObject = Literal["model"]
 
+
 class UnarchiveFTModelOutTypedDict(TypedDict):
     id: str
     archived: NotRequired[bool]
-    
+
 
 class UnarchiveFTModelOut(BaseModel):
     id: str
+
+    # fmt: off
     OBJECT: Annotated[Final[Optional[UnarchiveFTModelOutObject]], pydantic.Field(alias="object")] = "model" # type: ignore
+    # fmt: on
+
     archived: Optional[bool] = False
-    

mistralai/models/updateftmodelin.py

@@ -10,12 +10,13 @@ from typing_extensions import NotRequired
 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 +30,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 +46,3 @@ class UpdateFTModelIn(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/uploadfileout.py

@@ -3,7 +3,14 @@
 from __future__ import annotations
 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,
+    UnrecognizedStr,
+)
 from mistralai.utils import validate_open_enum
 import pydantic
 from pydantic import model_serializer
@@ -15,6 +22,7 @@ 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."""
 
+
 class UploadFileOutTypedDict(TypedDict):
     id: str
     r"""The unique identifier of the file."""
@@ -29,25 +37,35 @@ class UploadFileOutTypedDict(TypedDict):
     sample_type: SampleType
     source: Source
     num_lines: NotRequired[Nullable[int]]
-    
+
 
 class UploadFileOut(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
+
+    # 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."""
+
     num_lines: OptionalNullable[int] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["num_lines"]
@@ -61,9 +79,13 @@ class UploadFileOut(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 +95,3 @@ class UploadFileOut(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/usageinfo.py

@@ -9,10 +9,11 @@ class UsageInfoTypedDict(TypedDict):
     prompt_tokens: int
     completion_tokens: int
     total_tokens: int
-    
+
 
 class UsageInfo(BaseModel):
     prompt_tokens: int
+
     completion_tokens: int
+
     total_tokens: int
-    

mistralai/models/usermessage.py

@@ -1,26 +1,27 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .textchunk import TextChunk, TextChunkTypedDict
+from .contentchunk import ContentChunk, ContentChunkTypedDict
 from mistralai.types import BaseModel
 from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
-UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+UserMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
 
 
-UserMessageContent = Union[str, List[TextChunk]]
+UserMessageContent = Union[str, List[ContentChunk]]
 
 
 UserMessageRole = Literal["user"]
 
+
 class UserMessageTypedDict(TypedDict):
     content: UserMessageContentTypedDict
     role: NotRequired[UserMessageRole]
-    
+
 
 class UserMessage(BaseModel):
     content: UserMessageContent
+
     role: Optional[UserMessageRole] = "user"
-    

mistralai/models/validationerror.py

@@ -15,10 +15,11 @@ class ValidationErrorTypedDict(TypedDict):
     loc: List[LocTypedDict]
     msg: str
     type: str
-    
+
 
 class ValidationError(BaseModel):
     loc: List[Loc]
+
     msg: str
+
     type: str
-    

mistralai/models/wandbintegration.py

@@ -10,6 +10,7 @@ from typing_extensions import Annotated, NotRequired
 
 WandbIntegrationType = Literal["wandb"]
 
+
 class WandbIntegrationTypedDict(TypedDict):
     project: str
     r"""The name of the project that the new run will be created under."""
@@ -18,18 +19,24 @@ class WandbIntegrationTypedDict(TypedDict):
     name: NotRequired[Nullable[str]]
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
     run_name: NotRequired[Nullable[str]]
-    
+
 
 class WandbIntegration(BaseModel):
     project: str
     r"""The name of the project that the new run will be created under."""
+
     api_key: str
     r"""The WandB API key to use for authentication."""
+
+    # fmt: off
     TYPE: Annotated[Final[Optional[WandbIntegrationType]], pydantic.Field(alias="type")] = "wandb" # type: ignore
+    # fmt: on
+
     name: OptionalNullable[str] = UNSET
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
+
     run_name: OptionalNullable[str] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["type", "name", "run_name"]
@@ -43,9 +50,13 @@ class WandbIntegration(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
@@ -55,4 +66,3 @@ class WandbIntegration(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/wandbintegrationout.py

@@ -10,22 +10,28 @@ from typing_extensions import Annotated, NotRequired
 
 Type = Literal["wandb"]
 
+
 class WandbIntegrationOutTypedDict(TypedDict):
     project: str
     r"""The name of the project that the new run will be created under."""
     name: NotRequired[Nullable[str]]
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
     run_name: NotRequired[Nullable[str]]
-    
+
 
 class WandbIntegrationOut(BaseModel):
     project: str
     r"""The name of the project that the new run will be created under."""
+
+    # fmt: off
     TYPE: Annotated[Final[Optional[Type]], pydantic.Field(alias="type")] = "wandb" # type: ignore
+    # fmt: on
+
     name: OptionalNullable[str] = UNSET
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
+
     run_name: OptionalNullable[str] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["type", "name", "run_name"]
@@ -39,9 +45,13 @@ class WandbIntegrationOut(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
@@ -51,4 +61,3 @@ class WandbIntegrationOut(BaseModel):
                 m[k] = val
 
         return m
-