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

mistralai/models/files_api_routes_upload_fileop.py

@@ -2,13 +2,16 @@
 
 from __future__ import annotations
 import io
-from mistralai.types import BaseModel
-from mistralai.utils import FieldMetadata, MultipartFormMetadata
+from mistralai.types import BaseModel, UnrecognizedStr
+from mistralai.utils import FieldMetadata, MultipartFormMetadata, validate_open_enum
 import pydantic
-from typing import Final, IO, Optional, TypedDict, Union
+from pydantic.functional_validators import PlainValidator
+from typing import Final, IO, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+FilesAPIRoutesUploadFilePurpose = Union[Literal["fine-tune"], UnrecognizedStr]
+
 class FileTypedDict(TypedDict):
     file_name: str
     content: Union[bytes, IO[bytes], io.BufferedReader]
@@ -47,5 +50,5 @@ class FilesAPIRoutesUploadFileMultiPartBodyParams(BaseModel):
     file=@path/to/your/file.jsonl
     ```
     """
-    PURPOSE: Annotated[Final[Optional[str]], pydantic.Field(alias="purpose"), FieldMetadata(multipart=True)] = "fine-tune" # type: ignore
+    PURPOSE: Annotated[Final[Annotated[Optional[FilesAPIRoutesUploadFilePurpose], PlainValidator(validate_open_enum(False))]], pydantic.Field(alias="purpose"), FieldMetadata(multipart=True)] = "fine-tune" # type: ignore
     

mistralai/models/fileschema.py

@@ -3,13 +3,18 @@
 from __future__ import annotations
 from .sampletype import SampleType
 from .source import Source
-from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+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
-from typing import Final, TypedDict
+from pydantic.functional_validators import PlainValidator
+from typing import Final, Literal, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+FileSchemaPurpose = Union[Literal["fine-tune"], UnrecognizedStr]
+r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
+
 class FileSchemaTypedDict(TypedDict):
     id: str
     r"""The unique identifier of the file."""
@@ -39,7 +44,7 @@ class FileSchema(BaseModel):
     r"""The name of the uploaded file."""
     sample_type: SampleType
     source: Source
-    PURPOSE: Annotated[Final[str], pydantic.Field(alias="purpose")] = "fine-tune" # type: ignore
+    PURPOSE: Annotated[Final[Annotated[FileSchemaPurpose, 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."""
     num_lines: OptionalNullable[int] = UNSET
     

mistralai/models/ftmodelout.py

@@ -5,10 +5,12 @@ from .ftmodelcapabilitiesout import FTModelCapabilitiesOut, FTModelCapabilitiesO
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
 from pydantic import model_serializer
-from typing import Final, List, Optional, TypedDict
+from typing import Final, List, Literal, Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
+FTModelOutObject = Literal["model"]
+
 class FTModelOutTypedDict(TypedDict):
     id: str
     created: int
@@ -31,7 +33,7 @@ class FTModelOut(BaseModel):
     archived: bool
     capabilities: FTModelCapabilitiesOut
     job: str
-    OBJECT: Annotated[Final[Optional[str]], pydantic.Field(alias="object")] = "model" # type: ignore
+    OBJECT: Annotated[Final[Optional[FTModelOutObject]], pydantic.Field(alias="object")] = "model" # type: ignore
     name: OptionalNullable[str] = UNSET
     description: OptionalNullable[str] = UNSET
     max_context_length: Optional[int] = 32768

mistralai/models/githubrepositoryin.py

@@ -4,10 +4,12 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Optional, TypedDict
+from typing import Final, Literal, Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
+GithubRepositoryInType = Literal["github"]
+
 class GithubRepositoryInTypedDict(TypedDict):
     name: str
     owner: str
@@ -20,7 +22,7 @@ class GithubRepositoryIn(BaseModel):
     name: str
     owner: str
     token: str
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "github" # type: ignore
+    TYPE: Annotated[Final[Optional[GithubRepositoryInType]], pydantic.Field(alias="type")] = "github" # type: ignore
     ref: OptionalNullable[str] = UNSET
     weight: Optional[float] = 1
     

mistralai/models/githubrepositoryout.py

@@ -4,10 +4,12 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Optional, TypedDict
+from typing import Final, Literal, Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
+GithubRepositoryOutType = Literal["github"]
+
 class GithubRepositoryOutTypedDict(TypedDict):
     name: str
     owner: str
@@ -20,7 +22,7 @@ class GithubRepositoryOut(BaseModel):
     name: str
     owner: str
     commit_id: str
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "github" # type: ignore
+    TYPE: Annotated[Final[Optional[GithubRepositoryOutType]], pydantic.Field(alias="type")] = "github" # type: ignore
     ref: OptionalNullable[str] = UNSET
     weight: Optional[float] = 1
     

mistralai/models/jobin.py

@@ -12,6 +12,18 @@ from typing import List, Optional, TypedDict
 from typing_extensions import NotRequired
 
 
+JobInIntegrationsTypedDict = WandbIntegrationTypedDict
+
+
+JobInIntegrations = WandbIntegration
+
+
+JobInRepositoriesTypedDict = GithubRepositoryInTypedDict
+
+
+JobInRepositories = GithubRepositoryIn
+
+
 class JobInTypedDict(TypedDict):
     model: FineTuneableModel
     r"""The name of the model to fine-tune."""
@@ -22,9 +34,9 @@ class JobInTypedDict(TypedDict):
     r"""A list containing the IDs of uploaded files that contain validation data. If you provide these files, the data is used to generate validation metrics periodically during fine-tuning. These metrics can be viewed in `checkpoints` when getting the status of a running fine-tuning job. The same data should not be present in both train and validation files."""
     suffix: NotRequired[Nullable[str]]
     r"""A string that will be added to your fine-tuning model name. For example, a suffix of \"my-great-model\" would produce a model name like `ft:open-mistral-7b:my-great-model:xxx...`"""
-    integrations: NotRequired[Nullable[List[WandbIntegrationTypedDict]]]
+    integrations: NotRequired[Nullable[List[JobInIntegrationsTypedDict]]]
     r"""A list of integrations to enable for your fine-tuning job."""
-    repositories: NotRequired[List[GithubRepositoryInTypedDict]]
+    repositories: NotRequired[List[JobInRepositoriesTypedDict]]
     auto_start: NotRequired[bool]
     r"""This field will be required in a future release."""
     
@@ -39,9 +51,9 @@ class JobIn(BaseModel):
     r"""A list containing the IDs of uploaded files that contain validation data. If you provide these files, the data is used to generate validation metrics periodically during fine-tuning. These metrics can be viewed in `checkpoints` when getting the status of a running fine-tuning job. The same data should not be present in both train and validation files."""
     suffix: OptionalNullable[str] = UNSET
     r"""A string that will be added to your fine-tuning model name. For example, a suffix of \"my-great-model\" would produce a model name like `ft:open-mistral-7b:my-great-model:xxx...`"""
-    integrations: OptionalNullable[List[WandbIntegration]] = UNSET
+    integrations: OptionalNullable[List[JobInIntegrations]] = UNSET
     r"""A list of integrations to enable for your fine-tuning job."""
-    repositories: Optional[List[GithubRepositoryIn]] = None
+    repositories: Optional[List[JobInRepositories]] = None
     auto_start: Optional[bool] = None
     r"""This field will be required in a future release."""
     

mistralai/models/jobout.py

@@ -16,6 +16,21 @@ from typing_extensions import Annotated, NotRequired
 Status = Literal["QUEUED", "STARTED", "VALIDATING", "VALIDATED", "RUNNING", "FAILED_VALIDATION", "FAILED", "SUCCESS", "CANCELLED", "CANCELLATION_REQUESTED"]
 r"""The current status of the fine-tuning job."""
 
+Object = Literal["job"]
+r"""The object type of the fine-tuning job."""
+
+IntegrationsTypedDict = WandbIntegrationOutTypedDict
+
+
+Integrations = WandbIntegrationOut
+
+
+RepositoriesTypedDict = GithubRepositoryOutTypedDict
+
+
+Repositories = GithubRepositoryOut
+
+
 class JobOutTypedDict(TypedDict):
     id: str
     r"""The ID of the job."""
@@ -39,11 +54,11 @@ class JobOutTypedDict(TypedDict):
     r"""The name of the fine-tuned model that is being created. The value will be `null` if the fine-tuning job is still running."""
     suffix: NotRequired[Nullable[str]]
     r"""Optional text/code that adds more context for the model. When given a `prompt` and a `suffix` the model will fill what is between them. When `suffix` is not provided, the model will simply execute completion starting with `prompt`."""
-    integrations: NotRequired[Nullable[List[WandbIntegrationOutTypedDict]]]
+    integrations: NotRequired[Nullable[List[IntegrationsTypedDict]]]
     r"""A list of integrations enabled for your fine-tuning job."""
     trained_tokens: NotRequired[Nullable[int]]
     r"""Total number of tokens trained."""
-    repositories: NotRequired[List[GithubRepositoryOutTypedDict]]
+    repositories: NotRequired[List[RepositoriesTypedDict]]
     metadata: NotRequired[Nullable[JobMetadataOutTypedDict]]
     
 
@@ -66,17 +81,17 @@ class JobOut(BaseModel):
     r"""A list containing the IDs of uploaded files that contain training data."""
     validation_files: OptionalNullable[List[str]] = UNSET
     r"""A list containing the IDs of uploaded files that contain validation data."""
-    OBJECT: Annotated[Final[Optional[str]], pydantic.Field(alias="object")] = "job" # type: ignore
+    OBJECT: Annotated[Final[Optional[Object]], pydantic.Field(alias="object")] = "job" # type: ignore
     r"""The object type of the fine-tuning job."""
     fine_tuned_model: OptionalNullable[str] = UNSET
     r"""The name of the fine-tuned model that is being created. The value will be `null` if the fine-tuning job is still running."""
     suffix: OptionalNullable[str] = UNSET
     r"""Optional text/code that adds more context for the model. When given a `prompt` and a `suffix` the model will fill what is between them. When `suffix` is not provided, the model will simply execute completion starting with `prompt`."""
-    integrations: OptionalNullable[List[WandbIntegrationOut]] = UNSET
+    integrations: OptionalNullable[List[Integrations]] = UNSET
     r"""A list of integrations enabled for your fine-tuning job."""
     trained_tokens: OptionalNullable[int] = UNSET
     r"""Total number of tokens trained."""
-    repositories: Optional[List[GithubRepositoryOut]] = None
+    repositories: Optional[List[Repositories]] = None
     metadata: OptionalNullable[JobMetadataOut] = UNSET
     
     @model_serializer(mode="wrap")

mistralai/models/jobsout.py

@@ -4,10 +4,12 @@ from __future__ import annotations
 from .jobout import JobOut, JobOutTypedDict
 from mistralai.types import BaseModel
 import pydantic
-from typing import Final, List, Optional, TypedDict
+from typing import Final, List, Literal, Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
+JobsOutObject = Literal["list"]
+
 class JobsOutTypedDict(TypedDict):
     total: int
     data: NotRequired[List[JobOutTypedDict]]
@@ -16,5 +18,5 @@ class JobsOutTypedDict(TypedDict):
 class JobsOut(BaseModel):
     total: int
     data: Optional[List[JobOut]] = None
-    OBJECT: Annotated[Final[Optional[str]], pydantic.Field(alias="object")] = "list" # type: ignore
+    OBJECT: Annotated[Final[Optional[JobsOutObject]], pydantic.Field(alias="object")] = "list" # type: ignore
     

mistralai/models/legacyjobmetadataout.py

@@ -4,10 +4,12 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Optional, TypedDict
+from typing import Final, Literal, Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
+LegacyJobMetadataOutObject = Literal["job.metadata"]
+
 class LegacyJobMetadataOutTypedDict(TypedDict):
     details: str
     expected_duration_seconds: NotRequired[Nullable[int]]
@@ -50,7 +52,7 @@ class LegacyJobMetadataOut(BaseModel):
     r"""The number of complete passes through the entire training dataset."""
     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."""
-    OBJECT: Annotated[Final[Optional[str]], pydantic.Field(alias="object")] = "job.metadata" # type: ignore
+    OBJECT: Annotated[Final[Optional[LegacyJobMetadataOutObject]], pydantic.Field(alias="object")] = "job.metadata" # type: ignore
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):

mistralai/models/retrievefileout.py

@@ -3,13 +3,18 @@
 from __future__ import annotations
 from .sampletype import SampleType
 from .source import Source
-from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+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
-from typing import Final, TypedDict
+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."""
+
 class RetrieveFileOutTypedDict(TypedDict):
     id: str
     r"""The unique identifier of the file."""
@@ -39,7 +44,7 @@ class RetrieveFileOut(BaseModel):
     r"""The name of the uploaded file."""
     sample_type: SampleType
     source: Source
-    PURPOSE: Annotated[Final[str], pydantic.Field(alias="purpose")] = "fine-tune" # type: ignore
+    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."""
     num_lines: OptionalNullable[int] = UNSET
     

mistralai/models/tool.py

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

mistralai/models/toolcall.py

@@ -2,19 +2,23 @@
 
 from __future__ import annotations
 from .functioncall import FunctionCall, FunctionCallTypedDict
-from mistralai.types import BaseModel
-import pydantic
-from typing import Final, Optional, TypedDict
+from mistralai.types import BaseModel, UnrecognizedStr
+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
 
 
+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[Final[Optional[str]], pydantic.Field(alias="type")] = "function" # type: ignore
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = "function"
     

mistralai/models/trainingparameters.py

@@ -10,6 +10,8 @@ from typing_extensions import NotRequired
 class TrainingParametersTypedDict(TypedDict):
     training_steps: NotRequired[Nullable[int]]
     learning_rate: NotRequired[float]
+    weight_decay: NotRequired[Nullable[float]]
+    warmup_fraction: NotRequired[Nullable[float]]
     epochs: NotRequired[Nullable[float]]
     fim_ratio: NotRequired[Nullable[float]]
     
@@ -17,13 +19,15 @@ class TrainingParametersTypedDict(TypedDict):
 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", "epochs", "fim_ratio"]
-        nullable_fields = ["training_steps", "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)

mistralai/models/trainingparametersin.py

@@ -14,6 +14,10 @@ class TrainingParametersInTypedDict(TypedDict):
     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]
     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: NotRequired[Nullable[float]]
+    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: NotRequired[Nullable[float]]
+    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]]
     
@@ -25,13 +29,17 @@ class TrainingParametersIn(BaseModel):
     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", "epochs", "fim_ratio"]
-        nullable_fields = ["training_steps", "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)

mistralai/models/unarchiveftmodelout.py

@@ -3,10 +3,12 @@
 from __future__ import annotations
 from mistralai.types import BaseModel
 import pydantic
-from typing import Final, Optional, TypedDict
+from typing import Final, Literal, Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
+UnarchiveFTModelOutObject = Literal["model"]
+
 class UnarchiveFTModelOutTypedDict(TypedDict):
     id: str
     archived: NotRequired[bool]
@@ -14,6 +16,6 @@ class UnarchiveFTModelOutTypedDict(TypedDict):
 
 class UnarchiveFTModelOut(BaseModel):
     id: str
-    OBJECT: Annotated[Final[Optional[str]], pydantic.Field(alias="object")] = "model" # type: ignore
+    OBJECT: Annotated[Final[Optional[UnarchiveFTModelOutObject]], pydantic.Field(alias="object")] = "model" # type: ignore
     archived: Optional[bool] = False
     

mistralai/models/uploadfileout.py

@@ -3,13 +3,18 @@
 from __future__ import annotations
 from .sampletype import SampleType
 from .source import Source
-from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+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
-from typing import Final, TypedDict
+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."""
+
 class UploadFileOutTypedDict(TypedDict):
     id: str
     r"""The unique identifier of the file."""
@@ -39,7 +44,7 @@ class UploadFileOut(BaseModel):
     r"""The name of the uploaded file."""
     sample_type: SampleType
     source: Source
-    PURPOSE: Annotated[Final[str], pydantic.Field(alias="purpose")] = "fine-tune" # type: ignore
+    PURPOSE: Annotated[Final[Annotated[Purpose, 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."""
     num_lines: OptionalNullable[int] = UNSET
     

mistralai/models/wandbintegration.py

@@ -4,10 +4,12 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Optional, TypedDict
+from typing import Final, Literal, Optional, TypedDict
 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."""
@@ -23,7 +25,7 @@ class WandbIntegration(BaseModel):
     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."""
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "wandb" # type: ignore
+    TYPE: Annotated[Final[Optional[WandbIntegrationType]], pydantic.Field(alias="type")] = "wandb" # type: ignore
     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

mistralai/models/wandbintegrationout.py

@@ -4,10 +4,12 @@ from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Optional, TypedDict
+from typing import Final, Literal, Optional, TypedDict
 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."""
@@ -19,7 +21,7 @@ class WandbIntegrationOutTypedDict(TypedDict):
 class WandbIntegrationOut(BaseModel):
     project: str
     r"""The name of the project that the new run will be created under."""
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "wandb" # type: ignore
+    TYPE: Annotated[Final[Optional[Type]], pydantic.Field(alias="type")] = "wandb" # type: ignore
     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

mistralai/sdk.py

@@ -3,7 +3,7 @@
 from .basesdk import BaseSDK
 from .httpclient import AsyncHttpClient, HttpClient
 from .sdkconfiguration import SDKConfiguration
-from .utils.logger import Logger, NoOpLogger
+from .utils.logger import Logger, get_default_logger
 from .utils.retries import RetryConfig
 import httpx
 from mistralai import models, utils
@@ -67,7 +67,7 @@ class Mistral(BaseSDK):
             async_client = httpx.AsyncClient()
 
         if debug_logger is None:
-            debug_logger = NoOpLogger()
+            debug_logger = get_default_logger()
 
         assert issubclass(
             type(async_client), AsyncHttpClient

mistralai/sdkconfiguration.py

@@ -29,9 +29,9 @@ class SDKConfiguration:
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.0.1"
-    gen_version: str = "2.390.6"
-    user_agent: str = "speakeasy-sdk/python 1.0.1 2.390.6 0.0.2 mistralai"
+    sdk_version: str = "1.0.3"
+    gen_version: str = "2.404.11"
+    user_agent: str = "speakeasy-sdk/python 1.0.3 2.404.11 0.0.2 mistralai"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 

mistralai/utils/__init__.py

@@ -35,7 +35,7 @@ from .serializers import (
 )
 from .url import generate_url, template_url, remove_suffix
 from .values import get_global_from_env, match_content_type, match_status_codes, match_response
-from .logger import Logger, get_body_content, NoOpLogger
+from .logger import Logger, get_body_content, get_default_logger
 
 __all__ = [
     "BackoffStrategy",
@@ -44,6 +44,7 @@ __all__ = [
     "FormMetadata",
     "generate_url",
     "get_body_content",
+    "get_default_logger",
     "get_discriminator",
     "get_global_from_env",
     "get_headers",
@@ -59,7 +60,6 @@ __all__ = [
     "match_status_codes",
     "match_response",
     "MultipartFormMetadata",
-    "NoOpLogger",
     "OpenEnumMeta",
     "PathParamMetadata",
     "QueryParamMetadata",