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

mistralai/agents.py

@@ -27,7 +27,7 @@ class Agents(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
     ) -> Optional[models.ChatCompletionResponse]:
-        r"""Chat Completion
+        r"""Agents Completion
 
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param agent_id: The ID of the agent to use for this completion.
@@ -131,7 +131,7 @@ class Agents(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
     ) -> Optional[models.ChatCompletionResponse]:
-        r"""Chat Completion
+        r"""Agents Completion
 
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param agent_id: The ID of the agent to use for this completion.

mistralai/client.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-MIGRATION_MESSAGE = "This client is deprecated starting v1.0.0, pin your version to 0.4.2. Or migrate by following this documentation: TODO add link"
+MIGRATION_MESSAGE = "This client is deprecated. To migrate to the new client, please refer to this guide: https://github.com/mistralai/client-python/blob/main/MIGRATION.md. If you need to use this client anyway, pin your version to 0.4.2."
 
 
 class MistralClient:

mistralai/jobs.py

@@ -211,7 +211,6 @@ class Jobs(BaseSDK):
         self, *,
         model: models.FineTuneableModel,
         hyperparameters: Union[models.TrainingParametersIn, models.TrainingParametersInTypedDict],
-        dry_run: OptionalNullable[bool] = UNSET,
         training_files: Optional[Union[List[models.TrainingFile], List[models.TrainingFileTypedDict]]] = None,
         validation_files: OptionalNullable[List[str]] = UNSET,
         suffix: OptionalNullable[str] = UNSET,
@@ -228,7 +227,6 @@ class Jobs(BaseSDK):
 
         :param model: The name of the model to fine-tune.
         :param hyperparameters: The fine-tuning hyperparameter settings used in a fine-tune job.
-        :param dry_run: * If `true` the job is not spawned, instead the query returns a handful of useful metadata   for the user to perform sanity checks (see `LegacyJobMetadataOut` response). * Otherwise, the job is started and the query returns the job ID along with some of the   input parameters (see `JobOut` response).
         :param training_files: 
         :param validation_files: 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.
         :param suffix: 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...`
@@ -247,18 +245,15 @@ class Jobs(BaseSDK):
         if server_url is not None:
             base_url = server_url
         
-        request = models.JobsAPIRoutesFineTuningCreateFineTuningJobRequest(
-            dry_run=dry_run,
-            job_in=models.JobIn(
-                model=model,
-                training_files=utils.get_pydantic_model(training_files, Optional[List[models.TrainingFile]]),
-                validation_files=validation_files,
-                hyperparameters=utils.get_pydantic_model(hyperparameters, models.TrainingParametersIn),
-                suffix=suffix,
-                integrations=utils.get_pydantic_model(integrations, OptionalNullable[List[models.WandbIntegration]]),
-                repositories=utils.get_pydantic_model(repositories, Optional[List[models.GithubRepositoryIn]]),
-                auto_start=auto_start,
-            ),
+        request = models.JobIn(
+            model=model,
+            training_files=utils.get_pydantic_model(training_files, Optional[List[models.TrainingFile]]),
+            validation_files=validation_files,
+            hyperparameters=utils.get_pydantic_model(hyperparameters, models.TrainingParametersIn),
+            suffix=suffix,
+            integrations=utils.get_pydantic_model(integrations, OptionalNullable[List[models.WandbIntegration]]),
+            repositories=utils.get_pydantic_model(repositories, Optional[List[models.GithubRepositoryIn]]),
+            auto_start=auto_start,
         )
         
         req = self.build_request(
@@ -273,7 +268,7 @@ class Jobs(BaseSDK):
             user_agent_header="user-agent",
             accept_header_value="application/json",
             security=self.sdk_configuration.security,
-            get_serialized_body=lambda: utils.serialize_request_body(request.job_in, False, False, "json", models.JobIn),
+            get_serialized_body=lambda: utils.serialize_request_body(request, False, False, "json", models.JobIn),
             timeout_ms=timeout_ms,
         )
         
@@ -312,7 +307,6 @@ class Jobs(BaseSDK):
         self, *,
         model: models.FineTuneableModel,
         hyperparameters: Union[models.TrainingParametersIn, models.TrainingParametersInTypedDict],
-        dry_run: OptionalNullable[bool] = UNSET,
         training_files: Optional[Union[List[models.TrainingFile], List[models.TrainingFileTypedDict]]] = None,
         validation_files: OptionalNullable[List[str]] = UNSET,
         suffix: OptionalNullable[str] = UNSET,
@@ -329,7 +323,6 @@ class Jobs(BaseSDK):
 
         :param model: The name of the model to fine-tune.
         :param hyperparameters: The fine-tuning hyperparameter settings used in a fine-tune job.
-        :param dry_run: * If `true` the job is not spawned, instead the query returns a handful of useful metadata   for the user to perform sanity checks (see `LegacyJobMetadataOut` response). * Otherwise, the job is started and the query returns the job ID along with some of the   input parameters (see `JobOut` response).
         :param training_files: 
         :param validation_files: 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.
         :param suffix: 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...`
@@ -348,18 +341,15 @@ class Jobs(BaseSDK):
         if server_url is not None:
             base_url = server_url
         
-        request = models.JobsAPIRoutesFineTuningCreateFineTuningJobRequest(
-            dry_run=dry_run,
-            job_in=models.JobIn(
-                model=model,
-                training_files=utils.get_pydantic_model(training_files, Optional[List[models.TrainingFile]]),
-                validation_files=validation_files,
-                hyperparameters=utils.get_pydantic_model(hyperparameters, models.TrainingParametersIn),
-                suffix=suffix,
-                integrations=utils.get_pydantic_model(integrations, OptionalNullable[List[models.WandbIntegration]]),
-                repositories=utils.get_pydantic_model(repositories, Optional[List[models.GithubRepositoryIn]]),
-                auto_start=auto_start,
-            ),
+        request = models.JobIn(
+            model=model,
+            training_files=utils.get_pydantic_model(training_files, Optional[List[models.TrainingFile]]),
+            validation_files=validation_files,
+            hyperparameters=utils.get_pydantic_model(hyperparameters, models.TrainingParametersIn),
+            suffix=suffix,
+            integrations=utils.get_pydantic_model(integrations, OptionalNullable[List[models.WandbIntegration]]),
+            repositories=utils.get_pydantic_model(repositories, Optional[List[models.GithubRepositoryIn]]),
+            auto_start=auto_start,
         )
         
         req = self.build_request(
@@ -374,7 +364,7 @@ class Jobs(BaseSDK):
             user_agent_header="user-agent",
             accept_header_value="application/json",
             security=self.sdk_configuration.security,
-            get_serialized_body=lambda: utils.serialize_request_body(request.job_in, False, False, "json", models.JobIn),
+            get_serialized_body=lambda: utils.serialize_request_body(request, False, False, "json", models.JobIn),
             timeout_ms=timeout_ms,
         )
         

mistralai/models/__init__.py

@@ -33,7 +33,7 @@ from .finetuneablemodel import FineTuneableModel
 from .ftmodelcapabilitiesout import FTModelCapabilitiesOut, FTModelCapabilitiesOutTypedDict
 from .ftmodelout import FTModelOut, FTModelOutTypedDict
 from .function import Function, FunctionTypedDict
-from .functioncall import FunctionCall, FunctionCallTypedDict
+from .functioncall import Arguments, ArgumentsTypedDict, FunctionCall, FunctionCallTypedDict
 from .githubrepositoryin import GithubRepositoryIn, GithubRepositoryInTypedDict
 from .githubrepositoryout import GithubRepositoryOut, GithubRepositoryOutTypedDict
 from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
@@ -42,7 +42,7 @@ from .jobmetadataout import JobMetadataOut, JobMetadataOutTypedDict
 from .jobout import JobOut, JobOutTypedDict, Status
 from .jobs_api_routes_fine_tuning_archive_fine_tuned_modelop import JobsAPIRoutesFineTuningArchiveFineTunedModelRequest, JobsAPIRoutesFineTuningArchiveFineTunedModelRequestTypedDict
 from .jobs_api_routes_fine_tuning_cancel_fine_tuning_jobop import JobsAPIRoutesFineTuningCancelFineTuningJobRequest, JobsAPIRoutesFineTuningCancelFineTuningJobRequestTypedDict
-from .jobs_api_routes_fine_tuning_create_fine_tuning_jobop import JobsAPIRoutesFineTuningCreateFineTuningJobRequest, JobsAPIRoutesFineTuningCreateFineTuningJobRequestTypedDict, JobsAPIRoutesFineTuningCreateFineTuningJobResponse, JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict
+from .jobs_api_routes_fine_tuning_create_fine_tuning_jobop import JobsAPIRoutesFineTuningCreateFineTuningJobResponse, JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict
 from .jobs_api_routes_fine_tuning_get_fine_tuning_jobop import JobsAPIRoutesFineTuningGetFineTuningJobRequest, JobsAPIRoutesFineTuningGetFineTuningJobRequestTypedDict
 from .jobs_api_routes_fine_tuning_get_fine_tuning_jobsop import JobsAPIRoutesFineTuningGetFineTuningJobsRequest, JobsAPIRoutesFineTuningGetFineTuningJobsRequestTypedDict, QueryParamStatus
 from .jobs_api_routes_fine_tuning_start_fine_tuning_jobop import JobsAPIRoutesFineTuningStartFineTuningJobRequest, JobsAPIRoutesFineTuningStartFineTuningJobRequestTypedDict
@@ -79,4 +79,4 @@ from .validationerror import Loc, LocTypedDict, ValidationError, ValidationError
 from .wandbintegration import WandbIntegration, WandbIntegrationTypedDict
 from .wandbintegrationout import WandbIntegrationOut, WandbIntegrationOutTypedDict
 
-__all__ = ["AgentsCompletionRequest", "AgentsCompletionRequestMessages", "AgentsCompletionRequestMessagesTypedDict", "AgentsCompletionRequestStop", "AgentsCompletionRequestStopTypedDict", "AgentsCompletionRequestToolChoice", "AgentsCompletionRequestTypedDict", "AgentsCompletionStreamRequest", "AgentsCompletionStreamRequestStop", "AgentsCompletionStreamRequestStopTypedDict", "AgentsCompletionStreamRequestTypedDict", "ArchiveFTModelOut", "ArchiveFTModelOutTypedDict", "AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestMessages", "ChatCompletionStreamRequestMessagesTypedDict", "ChatCompletionStreamRequestStop", "ChatCompletionStreamRequestStopTypedDict", "ChatCompletionStreamRequestToolChoice", "ChatCompletionStreamRequestTypedDict", "CheckpointOut", "CheckpointOutTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceFinishReason", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeleteFileOut", "DeleteFileOutTypedDict", "DeleteModelOut", "DeleteModelOutTypedDict", "DeleteModelV1ModelsModelIDDeleteRequest", "DeleteModelV1ModelsModelIDDeleteRequestTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "DetailedJobOut", "DetailedJobOutStatus", "DetailedJobOutTypedDict", "EmbeddingRequest", "EmbeddingRequestTypedDict", "EmbeddingResponse", "EmbeddingResponseData", "EmbeddingResponseDataTypedDict", "EmbeddingResponseTypedDict", "EventOut", "EventOutTypedDict", "FIMCompletionRequest", "FIMCompletionRequestStop", "FIMCompletionRequestStopTypedDict", "FIMCompletionRequestTypedDict", "FIMCompletionResponse", "FIMCompletionResponseTypedDict", "FIMCompletionStreamRequest", "FIMCompletionStreamRequestStop", "FIMCompletionStreamRequestStopTypedDict", "FIMCompletionStreamRequestTypedDict", "FTModelCapabilitiesOut", "FTModelCapabilitiesOutTypedDict", "FTModelOut", "FTModelOutTypedDict", "File", "FileSchema", "FileSchemaTypedDict", "FileTypedDict", "FilesAPIRoutesDeleteFileRequest", "FilesAPIRoutesDeleteFileRequestTypedDict", "FilesAPIRoutesRetrieveFileRequest", "FilesAPIRoutesRetrieveFileRequestTypedDict", "FilesAPIRoutesUploadFileMultiPartBodyParams", "FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict", "FineTuneableModel", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "GithubRepositoryIn", "GithubRepositoryInTypedDict", "GithubRepositoryOut", "GithubRepositoryOutTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Inputs", "InputsTypedDict", "JobIn", "JobInTypedDict", "JobMetadataOut", "JobMetadataOutTypedDict", "JobOut", "JobOutTypedDict", "JobsAPIRoutesFineTuningArchiveFineTunedModelRequest", "JobsAPIRoutesFineTuningArchiveFineTunedModelRequestTypedDict", "JobsAPIRoutesFineTuningCancelFineTuningJobRequest", "JobsAPIRoutesFineTuningCancelFineTuningJobRequestTypedDict", "JobsAPIRoutesFineTuningCreateFineTuningJobRequest", "JobsAPIRoutesFineTuningCreateFineTuningJobRequestTypedDict", "JobsAPIRoutesFineTuningCreateFineTuningJobResponse", "JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict", "JobsAPIRoutesFineTuningGetFineTuningJobRequest", "JobsAPIRoutesFineTuningGetFineTuningJobRequestTypedDict", "JobsAPIRoutesFineTuningGetFineTuningJobsRequest", "JobsAPIRoutesFineTuningGetFineTuningJobsRequestTypedDict", "JobsAPIRoutesFineTuningStartFineTuningJobRequest", "JobsAPIRoutesFineTuningStartFineTuningJobRequestTypedDict", "JobsAPIRoutesFineTuningUnarchiveFineTunedModelRequest", "JobsAPIRoutesFineTuningUnarchiveFineTunedModelRequestTypedDict", "JobsAPIRoutesFineTuningUpdateFineTunedModelRequest", "JobsAPIRoutesFineTuningUpdateFineTunedModelRequestTypedDict", "JobsOut", "JobsOutTypedDict", "LegacyJobMetadataOut", "LegacyJobMetadataOutTypedDict", "ListFilesOut", "ListFilesOutTypedDict", "Loc", "LocTypedDict", "Messages", "MessagesTypedDict", "MetricOut", "MetricOutTypedDict", "ModelCapabilities", "ModelCapabilitiesTypedDict", "ModelCard", "ModelCardTypedDict", "ModelList", "ModelListTypedDict", "QueryParamStatus", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "RetrieveFileOut", "RetrieveFileOutTypedDict", "RetrieveModelV1ModelsModelIDGetRequest", "RetrieveModelV1ModelsModelIDGetRequestTypedDict", "Role", "SDKError", "SampleType", "Security", "SecurityTypedDict", "Source", "Status", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolTypedDict", "TrainingFile", "TrainingFileTypedDict", "TrainingParameters", "TrainingParametersIn", "TrainingParametersInTypedDict", "TrainingParametersTypedDict", "UnarchiveFTModelOut", "UnarchiveFTModelOutTypedDict", "UpdateFTModelIn", "UpdateFTModelInTypedDict", "UploadFileOut", "UploadFileOutTypedDict", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict", "WandbIntegration", "WandbIntegrationOut", "WandbIntegrationOutTypedDict", "WandbIntegrationTypedDict"]
+__all__ = ["AgentsCompletionRequest", "AgentsCompletionRequestMessages", "AgentsCompletionRequestMessagesTypedDict", "AgentsCompletionRequestStop", "AgentsCompletionRequestStopTypedDict", "AgentsCompletionRequestToolChoice", "AgentsCompletionRequestTypedDict", "AgentsCompletionStreamRequest", "AgentsCompletionStreamRequestStop", "AgentsCompletionStreamRequestStopTypedDict", "AgentsCompletionStreamRequestTypedDict", "ArchiveFTModelOut", "ArchiveFTModelOutTypedDict", "Arguments", "ArgumentsTypedDict", "AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestMessages", "ChatCompletionStreamRequestMessagesTypedDict", "ChatCompletionStreamRequestStop", "ChatCompletionStreamRequestStopTypedDict", "ChatCompletionStreamRequestToolChoice", "ChatCompletionStreamRequestTypedDict", "CheckpointOut", "CheckpointOutTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceFinishReason", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeleteFileOut", "DeleteFileOutTypedDict", "DeleteModelOut", "DeleteModelOutTypedDict", "DeleteModelV1ModelsModelIDDeleteRequest", "DeleteModelV1ModelsModelIDDeleteRequestTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "DetailedJobOut", "DetailedJobOutStatus", "DetailedJobOutTypedDict", "EmbeddingRequest", "EmbeddingRequestTypedDict", "EmbeddingResponse", "EmbeddingResponseData", "EmbeddingResponseDataTypedDict", "EmbeddingResponseTypedDict", "EventOut", "EventOutTypedDict", "FIMCompletionRequest", "FIMCompletionRequestStop", "FIMCompletionRequestStopTypedDict", "FIMCompletionRequestTypedDict", "FIMCompletionResponse", "FIMCompletionResponseTypedDict", "FIMCompletionStreamRequest", "FIMCompletionStreamRequestStop", "FIMCompletionStreamRequestStopTypedDict", "FIMCompletionStreamRequestTypedDict", "FTModelCapabilitiesOut", "FTModelCapabilitiesOutTypedDict", "FTModelOut", "FTModelOutTypedDict", "File", "FileSchema", "FileSchemaTypedDict", "FileTypedDict", "FilesAPIRoutesDeleteFileRequest", "FilesAPIRoutesDeleteFileRequestTypedDict", "FilesAPIRoutesRetrieveFileRequest", "FilesAPIRoutesRetrieveFileRequestTypedDict", "FilesAPIRoutesUploadFileMultiPartBodyParams", "FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict", "FineTuneableModel", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "GithubRepositoryIn", "GithubRepositoryInTypedDict", "GithubRepositoryOut", "GithubRepositoryOutTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Inputs", "InputsTypedDict", "JobIn", "JobInTypedDict", "JobMetadataOut", "JobMetadataOutTypedDict", "JobOut", "JobOutTypedDict", "JobsAPIRoutesFineTuningArchiveFineTunedModelRequest", "JobsAPIRoutesFineTuningArchiveFineTunedModelRequestTypedDict", "JobsAPIRoutesFineTuningCancelFineTuningJobRequest", "JobsAPIRoutesFineTuningCancelFineTuningJobRequestTypedDict", "JobsAPIRoutesFineTuningCreateFineTuningJobResponse", "JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict", "JobsAPIRoutesFineTuningGetFineTuningJobRequest", "JobsAPIRoutesFineTuningGetFineTuningJobRequestTypedDict", "JobsAPIRoutesFineTuningGetFineTuningJobsRequest", "JobsAPIRoutesFineTuningGetFineTuningJobsRequestTypedDict", "JobsAPIRoutesFineTuningStartFineTuningJobRequest", "JobsAPIRoutesFineTuningStartFineTuningJobRequestTypedDict", "JobsAPIRoutesFineTuningUnarchiveFineTunedModelRequest", "JobsAPIRoutesFineTuningUnarchiveFineTunedModelRequestTypedDict", "JobsAPIRoutesFineTuningUpdateFineTunedModelRequest", "JobsAPIRoutesFineTuningUpdateFineTunedModelRequestTypedDict", "JobsOut", "JobsOutTypedDict", "LegacyJobMetadataOut", "LegacyJobMetadataOutTypedDict", "ListFilesOut", "ListFilesOutTypedDict", "Loc", "LocTypedDict", "Messages", "MessagesTypedDict", "MetricOut", "MetricOutTypedDict", "ModelCapabilities", "ModelCapabilitiesTypedDict", "ModelCard", "ModelCardTypedDict", "ModelList", "ModelListTypedDict", "QueryParamStatus", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "RetrieveFileOut", "RetrieveFileOutTypedDict", "RetrieveModelV1ModelsModelIDGetRequest", "RetrieveModelV1ModelsModelIDGetRequestTypedDict", "Role", "SDKError", "SampleType", "Security", "SecurityTypedDict", "Source", "Status", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolTypedDict", "TrainingFile", "TrainingFileTypedDict", "TrainingParameters", "TrainingParametersIn", "TrainingParametersInTypedDict", "TrainingParametersTypedDict", "UnarchiveFTModelOut", "UnarchiveFTModelOutTypedDict", "UpdateFTModelIn", "UpdateFTModelInTypedDict", "UploadFileOut", "UploadFileOutTypedDict", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict", "WandbIntegration", "WandbIntegrationOut", "WandbIntegrationOutTypedDict", "WandbIntegrationTypedDict"]

mistralai/models/agentscompletionrequest.py

@@ -13,6 +13,20 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+AgentsCompletionRequestStopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+AgentsCompletionRequestStop = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+AgentsCompletionRequestMessagesTypedDict = Union[UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
+
+
+AgentsCompletionRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
+
+
 AgentsCompletionRequestToolChoice = Literal["auto", "none", "any"]
 
 class AgentsCompletionRequestTypedDict(TypedDict):
@@ -80,17 +94,3 @@ class AgentsCompletionRequest(BaseModel):
 
         return m
         
-
-AgentsCompletionRequestStopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-AgentsCompletionRequestStop = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-AgentsCompletionRequestMessagesTypedDict = Union[UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
-
-
-AgentsCompletionRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
-

mistralai/models/agentscompletionstreamrequest.py

@@ -7,6 +7,14 @@ from typing import List, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+AgentsCompletionStreamRequestStopTypedDict = Union[str, List[str]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
+AgentsCompletionStreamRequestStop = 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 AgentsCompletionStreamRequestTypedDict(TypedDict):
     model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
@@ -82,11 +90,3 @@ class AgentsCompletionStreamRequest(BaseModel):
 
         return m
         
-
-AgentsCompletionStreamRequestStopTypedDict = Union[str, List[str]]
-r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
-
-
-AgentsCompletionStreamRequestStop = 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/chatcompletionrequest.py

@@ -14,6 +14,20 @@ 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):
@@ -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,6 +14,20 @@ 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):
@@ -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,20 +4,20 @@ 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]]
+    tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
     
 
 class DeltaMessage(BaseModel):
     role: Optional[str] = None
     content: Optional[str] = None
-    tool_calls: OptionalNullable[ToolCall] = UNSET
+    tool_calls: OptionalNullable[List[ToolCall]] = UNSET
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):

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/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/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/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."""
     

mistralai/models/jobs_api_routes_fine_tuning_create_fine_tuning_jobop.py

@@ -1,64 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .jobin import JobIn, JobInTypedDict
 from .jobout import JobOut, JobOutTypedDict
 from .legacyjobmetadataout import LegacyJobMetadataOut, LegacyJobMetadataOutTypedDict
-from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import FieldMetadata, QueryParamMetadata, RequestMetadata
-from pydantic import model_serializer
-from typing import TypedDict, Union
-from typing_extensions import Annotated, NotRequired
+from typing import Union
 
 
-class JobsAPIRoutesFineTuningCreateFineTuningJobRequestTypedDict(TypedDict):
-    job_in: JobInTypedDict
-    dry_run: NotRequired[Nullable[bool]]
-    r"""* If `true` the job is not spawned, instead the query returns a handful of useful metadata
-    for the user to perform sanity checks (see `LegacyJobMetadataOut` response).
-    * Otherwise, the job is started and the query returns the job ID along with some of the
-    input parameters (see `JobOut` response).
-
-    """
-    
-
-class JobsAPIRoutesFineTuningCreateFineTuningJobRequest(BaseModel):
-    job_in: Annotated[JobIn, FieldMetadata(request=RequestMetadata(media_type="application/json"))]
-    dry_run: Annotated[OptionalNullable[bool], FieldMetadata(query=QueryParamMetadata(style="form", explode=True))] = UNSET
-    r"""* If `true` the job is not spawned, instead the query returns a handful of useful metadata
-    for the user to perform sanity checks (see `LegacyJobMetadataOut` response).
-    * Otherwise, the job is started and the query returns the job ID along with some of the
-    input parameters (see `JobOut` response).
-
-    """
-    
-    @model_serializer(mode="wrap")
-    def serialize_model(self, handler):
-        optional_fields = ["dry_run"]
-        nullable_fields = ["dry_run"]
-        null_default_fields = []
-
-        serialized = handler(self)
-
-        m = {}
-
-        for n, f in self.model_fields.items():
-            k = f.alias or n
-            val = serialized.get(k)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (self.__pydantic_fields_set__.intersection({n}) or k in null_default_fields) # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
-
-        return m
-        
-
 JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict = Union[LegacyJobMetadataOutTypedDict, JobOutTypedDict]
 r"""OK"""
 

mistralai/models/jobs_api_routes_fine_tuning_unarchive_fine_tuned_modelop.py

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

mistralai/models/jobs_api_routes_fine_tuning_update_fine_tuned_modelop.py

@@ -10,10 +10,12 @@ from typing_extensions import Annotated
 
 class JobsAPIRoutesFineTuningUpdateFineTunedModelRequestTypedDict(TypedDict):
     model_id: str
+    r"""The ID of the model to update."""
     update_ft_model_in: UpdateFTModelInTypedDict
     
 
 class JobsAPIRoutesFineTuningUpdateFineTunedModelRequest(BaseModel):
     model_id: Annotated[str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))]
+    r"""The ID of the model to update."""
     update_ft_model_in: Annotated[UpdateFTModelIn, FieldMetadata(request=RequestMetadata(media_type="application/json"))]
     

mistralai/models/retrieve_model_v1_models_model_id_getop.py

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

mistralai/models/systemmessage.py

@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+
+
+Content = Union[str, List[ContentChunk]]
+
+
 Role = Literal["system"]
 
 class SystemMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class SystemMessage(BaseModel):
     content: Content
     role: Optional[Role] = "system"
     
-
-ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
-
-
-Content = Union[str, List[ContentChunk]]
-

mistralai/models/usermessage.py

@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+
+
+UserMessageContent = Union[str, List[TextChunk]]
+
+
 UserMessageRole = Literal["user"]
 
 class UserMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class UserMessage(BaseModel):
     content: UserMessageContent
     role: Optional[UserMessageRole] = "user"
     
-
-UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
-
-
-UserMessageContent = Union[str, List[TextChunk]]
-