mistralai 1.0.0rc2__py3-none-any.whl → 1.0.2__py3-none-any.whl

mistralai/models/archiveftmodelout.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
 
 
+ArchiveFTModelOutObject = Literal["model"]
+
 class ArchiveFTModelOutTypedDict(TypedDict):
     id: str
     archived: NotRequired[bool]
@@ -14,6 +16,6 @@ class ArchiveFTModelOutTypedDict(TypedDict):
 
 class ArchiveFTModelOut(BaseModel):
     id: str
-    OBJECT: Annotated[Final[Optional[str]], pydantic.Field(alias="object")] = "model" # type: ignore
+    OBJECT: Annotated[Final[Optional[ArchiveFTModelOutObject]], pydantic.Field(alias="object")] = "model" # type: ignore
     archived: Optional[bool] = True
     

mistralai/models/chatcompletionchoice.py

@@ -3,20 +3,19 @@
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
 from mistralai.types import BaseModel
-from typing import Literal, Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Literal, TypedDict
 
 
 FinishReason = Literal["stop", "length", "model_length", "error", "tool_calls"]
 
 class ChatCompletionChoiceTypedDict(TypedDict):
     index: int
+    message: AssistantMessageTypedDict
     finish_reason: FinishReason
-    message: NotRequired[AssistantMessageTypedDict]
     
 
 class ChatCompletionChoice(BaseModel):
     index: int
+    message: AssistantMessage
     finish_reason: FinishReason
-    message: Optional[AssistantMessage] = None
     

mistralai/models/chatcompletionrequest.py

@@ -14,11 +14,25 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+StopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+Stop = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+MessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
+
+
+Messages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
+
+
 ToolChoice = Literal["auto", "none", "any"]
 
 class ChatCompletionRequestTypedDict(TypedDict):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
     messages: List[MessagesTypedDict]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
     temperature: NotRequired[float]
@@ -44,7 +58,7 @@ class ChatCompletionRequestTypedDict(TypedDict):
 
 class ChatCompletionRequest(BaseModel):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
     messages: List[Messages]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
     temperature: Optional[float] = 0.7
@@ -93,17 +107,3 @@ class ChatCompletionRequest(BaseModel):
 
         return m
         
-
-StopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-Stop = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-MessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
-
-
-Messages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
-

mistralai/models/chatcompletionstreamrequest.py

@@ -14,11 +14,25 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+ChatCompletionStreamRequestStopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+ChatCompletionStreamRequestStop = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+ChatCompletionStreamRequestMessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
+
+
+ChatCompletionStreamRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
+
+
 ChatCompletionStreamRequestToolChoice = Literal["auto", "none", "any"]
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
     messages: List[ChatCompletionStreamRequestMessagesTypedDict]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
     temperature: NotRequired[float]
@@ -43,7 +57,7 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
 
 class ChatCompletionStreamRequest(BaseModel):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
     messages: List[ChatCompletionStreamRequestMessages]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
     temperature: Optional[float] = 0.7
@@ -91,17 +105,3 @@ class ChatCompletionStreamRequest(BaseModel):
 
         return m
         
-
-ChatCompletionStreamRequestStopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-ChatCompletionStreamRequestStop = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-ChatCompletionStreamRequestMessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
-
-
-ChatCompletionStreamRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
-

mistralai/models/delete_model_v1_models_model_id_deleteop.py

@@ -9,8 +9,10 @@ from typing_extensions import Annotated
 
 class DeleteModelV1ModelsModelIDDeleteRequestTypedDict(TypedDict):
     model_id: str
+    r"""The ID of the model to delete."""
     
 
 class DeleteModelV1ModelsModelIDDeleteRequest(BaseModel):
     model_id: Annotated[str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))]
+    r"""The ID of the model to delete."""
     

mistralai/models/deltamessage.py

@@ -4,25 +4,25 @@ from __future__ import annotations
 from .toolcall import ToolCall, ToolCallTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional, TypedDict
+from typing import List, Optional, TypedDict
 from typing_extensions import NotRequired
 
 
 class DeltaMessageTypedDict(TypedDict):
     role: NotRequired[str]
-    content: NotRequired[str]
-    tool_calls: NotRequired[Nullable[ToolCallTypedDict]]
+    content: NotRequired[Nullable[str]]
+    tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
     
 
 class DeltaMessage(BaseModel):
     role: Optional[str] = None
-    content: Optional[str] = None
-    tool_calls: OptionalNullable[ToolCall] = UNSET
+    content: OptionalNullable[str] = UNSET
+    tool_calls: OptionalNullable[List[ToolCall]] = UNSET
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["role", "content", "tool_calls"]
-        nullable_fields = ["tool_calls"]
+        nullable_fields = ["content", "tool_calls"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/detailedjobout.py

@@ -17,6 +17,20 @@ from typing_extensions import Annotated, NotRequired
 
 DetailedJobOutStatus = Literal["QUEUED", "STARTED", "VALIDATING", "VALIDATED", "RUNNING", "FAILED_VALIDATION", "FAILED", "SUCCESS", "CANCELLED", "CANCELLATION_REQUESTED"]
 
+DetailedJobOutObject = Literal["job"]
+
+DetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
+
+
+DetailedJobOutIntegrations = WandbIntegrationOut
+
+
+DetailedJobOutRepositoriesTypedDict = GithubRepositoryOutTypedDict
+
+
+DetailedJobOutRepositories = GithubRepositoryOut
+
+
 class DetailedJobOutTypedDict(TypedDict):
     id: str
     auto_start: bool
@@ -31,9 +45,9 @@ class DetailedJobOutTypedDict(TypedDict):
     validation_files: NotRequired[Nullable[List[str]]]
     fine_tuned_model: NotRequired[Nullable[str]]
     suffix: NotRequired[Nullable[str]]
-    integrations: NotRequired[Nullable[List[WandbIntegrationOutTypedDict]]]
+    integrations: NotRequired[Nullable[List[DetailedJobOutIntegrationsTypedDict]]]
     trained_tokens: NotRequired[Nullable[int]]
-    repositories: NotRequired[List[GithubRepositoryOutTypedDict]]
+    repositories: NotRequired[List[DetailedJobOutRepositoriesTypedDict]]
     metadata: NotRequired[Nullable[JobMetadataOutTypedDict]]
     events: NotRequired[List[EventOutTypedDict]]
     r"""Event items are created every time the status of a fine-tuning job changes. The timestamped list of all events is accessible here."""
@@ -52,12 +66,12 @@ class DetailedJobOut(BaseModel):
     modified_at: int
     training_files: List[str]
     validation_files: OptionalNullable[List[str]] = UNSET
-    OBJECT: Annotated[Final[Optional[str]], pydantic.Field(alias="object")] = "job" # type: ignore
+    OBJECT: Annotated[Final[Optional[DetailedJobOutObject]], pydantic.Field(alias="object")] = "job" # type: ignore
     fine_tuned_model: OptionalNullable[str] = UNSET
     suffix: OptionalNullable[str] = UNSET
-    integrations: OptionalNullable[List[WandbIntegrationOut]] = UNSET
+    integrations: OptionalNullable[List[DetailedJobOutIntegrations]] = UNSET
     trained_tokens: OptionalNullable[int] = UNSET
-    repositories: Optional[List[GithubRepositoryOut]] = None
+    repositories: Optional[List[DetailedJobOutRepositories]] = None
     metadata: OptionalNullable[JobMetadataOut] = UNSET
     events: Optional[List[EventOut]] = None
     r"""Event items are created every time the status of a fine-tuning job changes. The timestamped list of all events is accessible here."""

mistralai/models/embeddingrequest.py

@@ -8,6 +8,14 @@ from typing import List, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+InputsTypedDict = Union[str, List[str]]
+r"""Text to embed."""
+
+
+Inputs = Union[str, List[str]]
+r"""Text to embed."""
+
+
 class EmbeddingRequestTypedDict(TypedDict):
     inputs: InputsTypedDict
     r"""Text to embed."""
@@ -51,11 +59,3 @@ class EmbeddingRequest(BaseModel):
 
         return m
         
-
-InputsTypedDict = Union[str, List[str]]
-r"""Text to embed."""
-
-
-Inputs = Union[str, List[str]]
-r"""Text to embed."""
-

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/fimcompletionrequest.py

@@ -7,6 +7,14 @@ from typing import List, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+FIMCompletionRequestStopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+FIMCompletionRequestStop = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
 class FIMCompletionRequestTypedDict(TypedDict):
     model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
@@ -84,11 +92,3 @@ class FIMCompletionRequest(BaseModel):
 
         return m
         
-
-FIMCompletionRequestStopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-FIMCompletionRequestStop = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-

mistralai/models/fimcompletionstreamrequest.py

@@ -7,6 +7,14 @@ from typing import List, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+FIMCompletionStreamRequestStopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+FIMCompletionStreamRequestStop = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
 class FIMCompletionStreamRequestTypedDict(TypedDict):
     model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
@@ -82,11 +90,3 @@ class FIMCompletionStreamRequest(BaseModel):
 
         return m
         
-
-FIMCompletionStreamRequestStopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-FIMCompletionStreamRequestStop = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-

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/functioncall.py

@@ -2,15 +2,21 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from typing import TypedDict
+from typing import Any, Dict, TypedDict, Union
+
+
+ArgumentsTypedDict = Union[Dict[str, Any], str]
+
+
+Arguments = Union[Dict[str, Any], str]
 
 
 class FunctionCallTypedDict(TypedDict):
     name: str
-    arguments: str
+    arguments: ArgumentsTypedDict
     
 
 class FunctionCall(BaseModel):
     name: str
-    arguments: str
+    arguments: Arguments
     

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/jobs_api_routes_fine_tuning_archive_fine_tuned_modelop.py

@@ -9,8 +9,10 @@ from typing_extensions import Annotated
 
 class JobsAPIRoutesFineTuningArchiveFineTunedModelRequestTypedDict(TypedDict):
     model_id: str
+    r"""The ID of the model to archive."""
     
 
 class JobsAPIRoutesFineTuningArchiveFineTunedModelRequest(BaseModel):
     model_id: Annotated[str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))]
+    r"""The ID of the model to archive."""