mistralai 1.10.1__py3-none-any.whl → 1.12.0__py3-none-any.whl

mistralai/files.py

@@ -4,6 +4,12 @@ from .basesdk import BaseSDK
 import httpx
 from mistralai import models, utils
 from mistralai._hooks import HookContext
+from mistralai.models import (
+    file as models_file,
+    filepurpose as models_filepurpose,
+    sampletype as models_sampletype,
+    source as models_source,
+)
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
@@ -16,8 +22,8 @@ class Files(BaseSDK):
     def upload(
         self,
         *,
-        file: Union[models.File, models.FileTypedDict],
-        purpose: Optional[models.FilePurpose] = None,
+        file: Union[models_file.File, models_file.FileTypedDict],
+        purpose: Optional[models_filepurpose.FilePurpose] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -31,7 +37,15 @@ class Files(BaseSDK):
 
         Please contact us if you need to increase these storage limits.
 
-        :param file: The File object (not file name) to be uploaded.  To upload a file and specify a custom file name you should format your request as such:  ```bash  file=@path/to/your/file.jsonl;filename=custom_name.jsonl  ```  Otherwise, you can just keep the original file name:  ```bash  file=@path/to/your/file.jsonl  ```
+        :param file: The File object (not file name) to be uploaded.
+            To upload a file and specify a custom file name you should format your request as such:
+            ```bash
+            file=@path/to/your/file.jsonl;filename=custom_name.jsonl
+            ```
+            Otherwise, you can just keep the original file name:
+            ```bash
+            file=@path/to/your/file.jsonl
+            ```
         :param purpose:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -73,6 +87,7 @@ class Files(BaseSDK):
                 "multipart",
                 models.FilesAPIRoutesUploadFileMultiPartBodyParams,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -89,7 +104,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_upload_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -113,8 +128,8 @@ class Files(BaseSDK):
     async def upload_async(
         self,
         *,
-        file: Union[models.File, models.FileTypedDict],
-        purpose: Optional[models.FilePurpose] = None,
+        file: Union[models_file.File, models_file.FileTypedDict],
+        purpose: Optional[models_filepurpose.FilePurpose] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -128,7 +143,15 @@ class Files(BaseSDK):
 
         Please contact us if you need to increase these storage limits.
 
-        :param file: The File object (not file name) to be uploaded.  To upload a file and specify a custom file name you should format your request as such:  ```bash  file=@path/to/your/file.jsonl;filename=custom_name.jsonl  ```  Otherwise, you can just keep the original file name:  ```bash  file=@path/to/your/file.jsonl  ```
+        :param file: The File object (not file name) to be uploaded.
+            To upload a file and specify a custom file name you should format your request as such:
+            ```bash
+            file=@path/to/your/file.jsonl;filename=custom_name.jsonl
+            ```
+            Otherwise, you can just keep the original file name:
+            ```bash
+            file=@path/to/your/file.jsonl
+            ```
         :param purpose:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -170,6 +193,7 @@ class Files(BaseSDK):
                 "multipart",
                 models.FilesAPIRoutesUploadFileMultiPartBodyParams,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -186,7 +210,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_upload_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -213,10 +237,11 @@ class Files(BaseSDK):
         page: Optional[int] = 0,
         page_size: Optional[int] = 100,
         include_total: Optional[bool] = True,
-        sample_type: OptionalNullable[List[models.SampleType]] = UNSET,
-        source: OptionalNullable[List[models.Source]] = UNSET,
+        sample_type: OptionalNullable[List[models_sampletype.SampleType]] = UNSET,
+        source: OptionalNullable[List[models_source.Source]] = UNSET,
         search: OptionalNullable[str] = UNSET,
-        purpose: OptionalNullable[models.FilePurpose] = UNSET,
+        purpose: OptionalNullable[models_filepurpose.FilePurpose] = UNSET,
+        mimetypes: OptionalNullable[List[str]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -233,6 +258,7 @@ class Files(BaseSDK):
         :param source:
         :param search:
         :param purpose:
+        :param mimetypes:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -256,6 +282,7 @@ class Files(BaseSDK):
             source=source,
             search=search,
             purpose=purpose,
+            mimetypes=mimetypes,
         )
 
         req = self._build_request(
@@ -271,6 +298,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -287,7 +315,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_list_files",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -314,10 +342,11 @@ class Files(BaseSDK):
         page: Optional[int] = 0,
         page_size: Optional[int] = 100,
         include_total: Optional[bool] = True,
-        sample_type: OptionalNullable[List[models.SampleType]] = UNSET,
-        source: OptionalNullable[List[models.Source]] = UNSET,
+        sample_type: OptionalNullable[List[models_sampletype.SampleType]] = UNSET,
+        source: OptionalNullable[List[models_source.Source]] = UNSET,
         search: OptionalNullable[str] = UNSET,
-        purpose: OptionalNullable[models.FilePurpose] = UNSET,
+        purpose: OptionalNullable[models_filepurpose.FilePurpose] = UNSET,
+        mimetypes: OptionalNullable[List[str]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -334,6 +363,7 @@ class Files(BaseSDK):
         :param source:
         :param search:
         :param purpose:
+        :param mimetypes:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -357,6 +387,7 @@ class Files(BaseSDK):
             source=source,
             search=search,
             purpose=purpose,
+            mimetypes=mimetypes,
         )
 
         req = self._build_request_async(
@@ -372,6 +403,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -388,7 +420,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_list_files",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -455,6 +487,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -471,7 +504,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_retrieve_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -538,6 +571,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -554,7 +588,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_retrieve_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -621,6 +655,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -637,7 +672,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_delete_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -704,6 +739,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -720,7 +756,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_delete_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -787,6 +823,7 @@ class Files(BaseSDK):
             accept_header_value="application/octet-stream",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -803,7 +840,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_download_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -872,6 +909,7 @@ class Files(BaseSDK):
             accept_header_value="application/octet-stream",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -888,7 +926,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_download_file",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -958,6 +996,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -974,7 +1013,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_get_signed_url",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1042,6 +1081,7 @@ class Files(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1058,7 +1098,7 @@ class Files(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="files_api_routes_get_signed_url",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),

mistralai/fim.py

@@ -3,6 +3,10 @@
 from .basesdk import BaseSDK
 from mistralai import models, utils
 from mistralai._hooks import HookContext
+from mistralai.models import (
+    fimcompletionrequest as models_fimcompletionrequest,
+    fimcompletionstreamrequest as models_fimcompletionstreamrequest,
+)
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import eventstreaming, get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
@@ -23,8 +27,8 @@ class Fim(BaseSDK):
         stream: Optional[bool] = False,
         stop: Optional[
             Union[
-                models.FIMCompletionRequestStop,
-                models.FIMCompletionRequestStopTypedDict,
+                models_fimcompletionrequest.FIMCompletionRequestStop,
+                models_fimcompletionrequest.FIMCompletionRequestStopTypedDict,
             ]
         ] = None,
         random_seed: OptionalNullable[int] = UNSET,
@@ -96,6 +100,7 @@ class Fim(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.FIMCompletionRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -112,7 +117,7 @@ class Fim(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="fim_completion_v1_fim_completions_post",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -150,8 +155,8 @@ class Fim(BaseSDK):
         stream: Optional[bool] = False,
         stop: Optional[
             Union[
-                models.FIMCompletionRequestStop,
-                models.FIMCompletionRequestStopTypedDict,
+                models_fimcompletionrequest.FIMCompletionRequestStop,
+                models_fimcompletionrequest.FIMCompletionRequestStopTypedDict,
             ]
         ] = None,
         random_seed: OptionalNullable[int] = UNSET,
@@ -223,6 +228,7 @@ class Fim(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.FIMCompletionRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -239,7 +245,7 @@ class Fim(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="fim_completion_v1_fim_completions_post",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -277,8 +283,8 @@ class Fim(BaseSDK):
         stream: Optional[bool] = True,
         stop: Optional[
             Union[
-                models.FIMCompletionStreamRequestStop,
-                models.FIMCompletionStreamRequestStopTypedDict,
+                models_fimcompletionstreamrequest.FIMCompletionStreamRequestStop,
+                models_fimcompletionstreamrequest.FIMCompletionStreamRequestStopTypedDict,
             ]
         ] = None,
         random_seed: OptionalNullable[int] = UNSET,
@@ -350,6 +356,7 @@ class Fim(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.FIMCompletionStreamRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -366,7 +373,7 @@ class Fim(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_fim",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -412,8 +419,8 @@ class Fim(BaseSDK):
         stream: Optional[bool] = True,
         stop: Optional[
             Union[
-                models.FIMCompletionStreamRequestStop,
-                models.FIMCompletionStreamRequestStopTypedDict,
+                models_fimcompletionstreamrequest.FIMCompletionStreamRequestStop,
+                models_fimcompletionstreamrequest.FIMCompletionStreamRequestStopTypedDict,
             ]
         ] = None,
         random_seed: OptionalNullable[int] = UNSET,
@@ -485,6 +492,7 @@ class Fim(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.FIMCompletionStreamRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -501,7 +509,7 @@ class Fim(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_fim",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),

mistralai/httpclient.py

@@ -107,7 +107,6 @@ def close_clients(
     # to them from the owning SDK instance and they can be reaped.
     owner.client = None
     owner.async_client = None
-
     if sync_client is not None and not sync_client_supplied:
         try:
             sync_client.close()