mistralai 1.2.5__py3-none-any.whl → 1.2.6__py3-none-any.whl

mistralai/jobs.py

@@ -6,7 +6,7 @@ from mistralai import models, utils
 from mistralai._hooks import HookContext
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
-from typing import List, Optional, Union
+from typing import List, Mapping, Optional, Union
 
 
 class Jobs(BaseSDK):
@@ -25,6 +25,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.JobsOut]:
         r"""Get Fine Tuning Jobs
 
@@ -42,6 +43,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -63,7 +65,7 @@ class Jobs(BaseSDK):
             suffix=suffix,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="GET",
             path="/v1/fine_tuning/jobs",
             base_url=base_url,
@@ -74,6 +76,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -131,6 +134,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.JobsOut]:
         r"""Get Fine Tuning Jobs
 
@@ -148,6 +152,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -169,7 +174,7 @@ class Jobs(BaseSDK):
             suffix=suffix,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="GET",
             path="/v1/fine_tuning/jobs",
             base_url=base_url,
@@ -180,6 +185,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -248,6 +254,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.JobsAPIRoutesFineTuningCreateFineTuningJobResponse]:
         r"""Create Fine Tuning Job
 
@@ -264,6 +271,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -292,7 +300,7 @@ class Jobs(BaseSDK):
             auto_start=auto_start,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/v1/fine_tuning/jobs",
             base_url=base_url,
@@ -303,6 +311,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.JobIn
@@ -377,6 +386,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.JobsAPIRoutesFineTuningCreateFineTuningJobResponse]:
         r"""Create Fine Tuning Job
 
@@ -393,6 +403,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -421,7 +432,7 @@ class Jobs(BaseSDK):
             auto_start=auto_start,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/v1/fine_tuning/jobs",
             base_url=base_url,
@@ -432,6 +443,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.JobIn
@@ -487,6 +499,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.DetailedJobOut]:
         r"""Get Fine Tuning Job
 
@@ -496,6 +509,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -509,7 +523,7 @@ class Jobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="GET",
             path="/v1/fine_tuning/jobs/{job_id}",
             base_url=base_url,
@@ -520,6 +534,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -569,6 +584,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.DetailedJobOut]:
         r"""Get Fine Tuning Job
 
@@ -578,6 +594,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -591,7 +608,7 @@ class Jobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="GET",
             path="/v1/fine_tuning/jobs/{job_id}",
             base_url=base_url,
@@ -602,6 +619,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -651,6 +669,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.DetailedJobOut]:
         r"""Cancel Fine Tuning Job
 
@@ -660,6 +679,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -673,7 +693,7 @@ class Jobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/v1/fine_tuning/jobs/{job_id}/cancel",
             base_url=base_url,
@@ -684,6 +704,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -733,6 +754,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.DetailedJobOut]:
         r"""Cancel Fine Tuning Job
 
@@ -742,6 +764,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -755,7 +778,7 @@ class Jobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/v1/fine_tuning/jobs/{job_id}/cancel",
             base_url=base_url,
@@ -766,6 +789,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -815,6 +839,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.DetailedJobOut]:
         r"""Start Fine Tuning Job
 
@@ -824,6 +849,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -837,7 +863,7 @@ class Jobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/v1/fine_tuning/jobs/{job_id}/start",
             base_url=base_url,
@@ -848,6 +874,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -897,6 +924,7 @@ class Jobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.DetailedJobOut]:
         r"""Start Fine Tuning Job
 
@@ -906,6 +934,7 @@ class Jobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -919,7 +948,7 @@ class Jobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/v1/fine_tuning/jobs/{job_id}/start",
             base_url=base_url,
@@ -930,6 +959,7 @@ class Jobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )

mistralai/mistral_jobs.py

@@ -6,7 +6,7 @@ from mistralai import models, utils
 from mistralai._hooks import HookContext
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Mapping, Optional
 
 
 class MistralJobs(BaseSDK):
@@ -23,6 +23,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobsOut]:
         r"""Get Batch Jobs
 
@@ -38,6 +39,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -57,7 +59,7 @@ class MistralJobs(BaseSDK):
             status=status,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="GET",
             path="/v1/batch/jobs",
             base_url=base_url,
@@ -68,6 +70,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -123,6 +126,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobsOut]:
         r"""Get Batch Jobs
 
@@ -138,6 +142,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -157,7 +162,7 @@ class MistralJobs(BaseSDK):
             status=status,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="GET",
             path="/v1/batch/jobs",
             base_url=base_url,
@@ -168,6 +173,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -221,6 +227,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobOut]:
         r"""Create Batch Job
 
@@ -234,6 +241,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -251,7 +259,7 @@ class MistralJobs(BaseSDK):
             timeout_hours=timeout_hours,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/v1/batch/jobs",
             base_url=base_url,
@@ -262,6 +270,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.BatchJobIn
@@ -318,6 +327,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobOut]:
         r"""Create Batch Job
 
@@ -331,6 +341,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -348,7 +359,7 @@ class MistralJobs(BaseSDK):
             timeout_hours=timeout_hours,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/v1/batch/jobs",
             base_url=base_url,
@@ -359,6 +370,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.BatchJobIn
@@ -411,6 +423,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobOut]:
         r"""Get Batch Job
 
@@ -420,6 +433,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -433,7 +447,7 @@ class MistralJobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="GET",
             path="/v1/batch/jobs/{job_id}",
             base_url=base_url,
@@ -444,6 +458,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -493,6 +508,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobOut]:
         r"""Get Batch Job
 
@@ -502,6 +518,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -515,7 +532,7 @@ class MistralJobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="GET",
             path="/v1/batch/jobs/{job_id}",
             base_url=base_url,
@@ -526,6 +543,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -575,6 +593,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobOut]:
         r"""Cancel Batch Job
 
@@ -584,6 +603,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -597,7 +617,7 @@ class MistralJobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/v1/batch/jobs/{job_id}/cancel",
             base_url=base_url,
@@ -608,6 +628,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )
@@ -657,6 +678,7 @@ class MistralJobs(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[models.BatchJobOut]:
         r"""Cancel Batch Job
 
@@ -666,6 +688,7 @@ class MistralJobs(BaseSDK):
         :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
+        :param http_headers: Additional headers to set or replace on requests.
         """
         base_url = None
         url_variables = None
@@ -679,7 +702,7 @@ class MistralJobs(BaseSDK):
             job_id=job_id,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/v1/batch/jobs/{job_id}/cancel",
             base_url=base_url,
@@ -690,6 +713,7 @@ class MistralJobs(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="application/json",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             timeout_ms=timeout_ms,
         )

mistralai/models/chatcompletionrequest.py

@@ -97,7 +97,7 @@ class ChatCompletionRequest(BaseModel):
     temperature: OptionalNullable[float] = UNSET
     r"""What sampling temperature to use, we recommend between 0.0 and 0.7. Higher values like 0.7 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. The default value varies depending on the model you are targeting. Call the `/models` endpoint to retrieve the appropriate value."""
 
-    top_p: Optional[float] = 1
+    top_p: Optional[float] = None
     r"""Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both."""
 
     max_tokens: OptionalNullable[int] = UNSET
@@ -127,7 +127,7 @@ class ChatCompletionRequest(BaseModel):
     n: OptionalNullable[int] = UNSET
     r"""Number of completions to return for each request, input tokens are only billed once."""
 
-    safe_prompt: Optional[bool] = False
+    safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
     @model_serializer(mode="wrap")

mistralai/models/chatcompletionstreamrequest.py

@@ -100,7 +100,7 @@ class ChatCompletionStreamRequest(BaseModel):
     temperature: OptionalNullable[float] = UNSET
     r"""What sampling temperature to use, we recommend between 0.0 and 0.7. Higher values like 0.7 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. The default value varies depending on the model you are targeting. Call the `/models` endpoint to retrieve the appropriate value."""
 
-    top_p: Optional[float] = 1
+    top_p: Optional[float] = None
     r"""Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both."""
 
     max_tokens: OptionalNullable[int] = UNSET
@@ -129,7 +129,7 @@ class ChatCompletionStreamRequest(BaseModel):
     n: OptionalNullable[int] = UNSET
     r"""Number of completions to return for each request, input tokens are only billed once."""
 
-    safe_prompt: Optional[bool] = False
+    safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
     @model_serializer(mode="wrap")

mistralai/models/files_api_routes_upload_fileop.py

@@ -19,7 +19,7 @@ class FileTypedDict(TypedDict):
 
 class File(BaseModel):
     file_name: Annotated[
-        str, pydantic.Field(alias="file"), FieldMetadata(multipart=True)
+        str, pydantic.Field(alias="fileName"), FieldMetadata(multipart=True)
     ]
 
     content: Annotated[
@@ -51,11 +51,7 @@ class FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict(TypedDict):
 
 
 class FilesAPIRoutesUploadFileMultiPartBodyParams(BaseModel):
-    file: Annotated[
-        File,
-        pydantic.Field(alias=""),
-        FieldMetadata(multipart=MultipartFormMetadata(file=True)),
-    ]
+    file: Annotated[File, FieldMetadata(multipart=MultipartFormMetadata(file=True))]
     r"""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