mistralai 0.5.5a50__py3-none-any.whl → 1.0.0__py3-none-any.whl

mistralai/models/fimcompletionrequest.py

@@ -8,13 +8,13 @@ from typing_extensions import NotRequired
 
 
 class FIMCompletionRequestTypedDict(TypedDict):
-    prompt: str
-    r"""The text/code to complete."""
-    model: NotRequired[Nullable[str]]
+    model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
     - `codestral-2405`
     - `codestral-latest`
     """
+    prompt: str
+    r"""The text/code to complete."""
     temperature: NotRequired[float]
     r"""What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both."""
     top_p: NotRequired[float]
@@ -34,13 +34,13 @@ class FIMCompletionRequestTypedDict(TypedDict):
     
 
 class FIMCompletionRequest(BaseModel):
-    prompt: str
-    r"""The text/code to complete."""
-    model: OptionalNullable[str] = UNSET
+    model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
     - `codestral-2405`
     - `codestral-latest`
     """
+    prompt: str
+    r"""The text/code to complete."""
     temperature: Optional[float] = 0.7
     r"""What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both."""
     top_p: Optional[float] = 1
@@ -60,7 +60,7 @@ class FIMCompletionRequest(BaseModel):
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["model", "temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "suffix"]
+        optional_fields = ["temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "suffix"]
         nullable_fields = ["model", "max_tokens", "min_tokens", "random_seed", "suffix"]
         null_default_fields = []
 
@@ -72,18 +72,13 @@ class FIMCompletionRequest(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/fimcompletionstreamrequest.py

@@ -8,13 +8,13 @@ from typing_extensions import NotRequired
 
 
 class FIMCompletionStreamRequestTypedDict(TypedDict):
-    prompt: str
-    r"""The text/code to complete."""
-    model: NotRequired[Nullable[str]]
+    model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
     - `codestral-2405`
     - `codestral-latest`
     """
+    prompt: str
+    r"""The text/code to complete."""
     temperature: NotRequired[float]
     r"""What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both."""
     top_p: NotRequired[float]
@@ -33,13 +33,13 @@ class FIMCompletionStreamRequestTypedDict(TypedDict):
     
 
 class FIMCompletionStreamRequest(BaseModel):
-    prompt: str
-    r"""The text/code to complete."""
-    model: OptionalNullable[str] = UNSET
+    model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
     - `codestral-2405`
     - `codestral-latest`
     """
+    prompt: str
+    r"""The text/code to complete."""
     temperature: Optional[float] = 0.7
     r"""What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both."""
     top_p: Optional[float] = 1
@@ -58,7 +58,7 @@ class FIMCompletionStreamRequest(BaseModel):
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["model", "temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "suffix"]
+        optional_fields = ["temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "suffix"]
         nullable_fields = ["model", "max_tokens", "min_tokens", "random_seed", "suffix"]
         null_default_fields = []
 
@@ -70,18 +70,13 @@ class FIMCompletionStreamRequest(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/ftmodelout.py

@@ -51,18 +51,13 @@ class FTModelOut(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

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
 
 
 class FunctionCallTypedDict(TypedDict):
     name: str
-    arguments: str
+    arguments: ArgumentsTypedDict
     
 
 class FunctionCall(BaseModel):
     name: str
-    arguments: str
+    arguments: Arguments
     
+
+ArgumentsTypedDict = Union[Dict[str, Any], str]
+
+
+Arguments = Union[Dict[str, Any], str]
+

mistralai/models/githubrepositoryin.py

@@ -38,18 +38,13 @@ class GithubRepositoryIn(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/githubrepositoryout.py

@@ -38,18 +38,13 @@ class GithubRepositoryOut(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/httpvalidationerror.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 from .validationerror import ValidationError
+from mistralai import utils
 from mistralai.types import BaseModel
-import mistralai.utils as utils
 from typing import List, Optional
 
 class HTTPValidationErrorData(BaseModel):

mistralai/models/jobin.py

@@ -59,18 +59,13 @@ class JobIn(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/jobmetadataout.py

@@ -40,18 +40,13 @@ class JobMetadataOut(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/jobout.py

@@ -93,18 +93,13 @@ class JobOut(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

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,69 +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)
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields
-                or (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
-            ):
-                m[k] = val
-
-        return m
-        
-
 JobsAPIRoutesFineTuningCreateFineTuningJobResponseTypedDict = Union[LegacyJobMetadataOutTypedDict, JobOutTypedDict]
 r"""OK"""
 

mistralai/models/jobs_api_routes_fine_tuning_get_fine_tuning_jobsop.py

@@ -67,18 +67,13 @@ class JobsAPIRoutesFineTuningGetFineTuningJobsRequest(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

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

@@ -66,18 +66,13 @@ class LegacyJobMetadataOut(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/metricout.py

@@ -36,18 +36,13 @@ class MetricOut(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/modelcard.py

@@ -52,18 +52,13 @@ class ModelCard(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

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

@@ -57,18 +57,13 @@ class RetrieveFileOut(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/security.py

@@ -3,14 +3,14 @@
 from __future__ import annotations
 from mistralai.types import BaseModel
 from mistralai.utils import FieldMetadata, SecurityMetadata
-from typing import TypedDict
-from typing_extensions import Annotated
+from typing import Optional, TypedDict
+from typing_extensions import Annotated, NotRequired
 
 
 class SecurityTypedDict(TypedDict):
-    api_key: str
+    api_key: NotRequired[str]
     
 
 class Security(BaseModel):
-    api_key: Annotated[str, FieldMetadata(security=SecurityMetadata(scheme=True, scheme_type="http", sub_type="bearer", field_name="Authorization"))]
+    api_key: Annotated[Optional[str], FieldMetadata(security=SecurityMetadata(scheme=True, scheme_type="http", sub_type="bearer", field_name="Authorization"))] = None
     

mistralai/models/toolmessage.py

@@ -36,18 +36,13 @@ class ToolMessage(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/trainingparameters.py

@@ -34,18 +34,13 @@ class TrainingParameters(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/trainingparametersin.py

@@ -42,18 +42,13 @@ class TrainingParametersIn(BaseModel):
             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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val