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

mistralai_gcp/chat.py

@@ -5,7 +5,7 @@ from mistralai_gcp import models, utils
 from mistralai_gcp._hooks import HookContext
 from mistralai_gcp.types import Nullable, OptionalNullable, UNSET
 from mistralai_gcp.utils import eventstreaming
-from typing import Any, List, Optional, Union
+from typing import Any, List, Mapping, Optional, Union
 
 
 class Chat(BaseSDK):
@@ -17,7 +17,7 @@ class Chat(BaseSDK):
         model: Nullable[str],
         messages: Union[List[models.Messages], List[models.MessagesTypedDict]],
         temperature: OptionalNullable[float] = UNSET,
-        top_p: Optional[float] = 1,
+        top_p: Optional[float] = None,
         max_tokens: OptionalNullable[int] = UNSET,
         stream: Optional[bool] = True,
         stop: Optional[Union[models.Stop, models.StopTypedDict]] = None,
@@ -40,6 +40,7 @@ class Chat(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
         r"""Stream chat completion
 
@@ -62,6 +63,7 @@ class Chat(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
@@ -92,7 +94,7 @@ class Chat(BaseSDK):
             n=n,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/streamRawPredict",
             base_url=base_url,
@@ -103,6 +105,7 @@ class Chat(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="text/event-stream",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.ChatCompletionStreamRequest
@@ -162,7 +165,7 @@ class Chat(BaseSDK):
         model: Nullable[str],
         messages: Union[List[models.Messages], List[models.MessagesTypedDict]],
         temperature: OptionalNullable[float] = UNSET,
-        top_p: Optional[float] = 1,
+        top_p: Optional[float] = None,
         max_tokens: OptionalNullable[int] = UNSET,
         stream: Optional[bool] = True,
         stop: Optional[Union[models.Stop, models.StopTypedDict]] = None,
@@ -185,6 +188,7 @@ class Chat(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
         r"""Stream chat completion
 
@@ -207,6 +211,7 @@ class Chat(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
@@ -237,7 +242,7 @@ class Chat(BaseSDK):
             n=n,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/streamRawPredict",
             base_url=base_url,
@@ -248,6 +253,7 @@ class Chat(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="text/event-stream",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.ChatCompletionStreamRequest
@@ -310,7 +316,7 @@ class Chat(BaseSDK):
             List[models.ChatCompletionRequestMessagesTypedDict],
         ],
         temperature: OptionalNullable[float] = UNSET,
-        top_p: Optional[float] = 1,
+        top_p: Optional[float] = None,
         max_tokens: OptionalNullable[int] = UNSET,
         stream: Optional[bool] = False,
         stop: Optional[
@@ -338,6 +344,7 @@ class Chat(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.ChatCompletionResponse]:
         r"""Chat Completion
 
@@ -358,6 +365,7 @@ class Chat(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
@@ -390,7 +398,7 @@ class Chat(BaseSDK):
             n=n,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/rawPredict",
             base_url=base_url,
@@ -401,6 +409,7 @@ class Chat(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.ChatCompletionRequest
@@ -459,7 +468,7 @@ class Chat(BaseSDK):
             List[models.ChatCompletionRequestMessagesTypedDict],
         ],
         temperature: OptionalNullable[float] = UNSET,
-        top_p: Optional[float] = 1,
+        top_p: Optional[float] = None,
         max_tokens: OptionalNullable[int] = UNSET,
         stream: Optional[bool] = False,
         stop: Optional[
@@ -487,6 +496,7 @@ class Chat(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.ChatCompletionResponse]:
         r"""Chat Completion
 
@@ -507,6 +517,7 @@ class Chat(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
@@ -539,7 +550,7 @@ class Chat(BaseSDK):
             n=n,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/rawPredict",
             base_url=base_url,
@@ -550,6 +561,7 @@ class Chat(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.ChatCompletionRequest

mistralai_gcp/fim.py

@@ -5,7 +5,7 @@ from mistralai_gcp import models, utils
 from mistralai_gcp._hooks import HookContext
 from mistralai_gcp.types import Nullable, OptionalNullable, UNSET
 from mistralai_gcp.utils import eventstreaming
-from typing import Any, Optional, Union
+from typing import Any, Mapping, Optional, Union
 
 
 class Fim(BaseSDK):
@@ -32,6 +32,7 @@ class Fim(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
         r"""Stream fim completion
 
@@ -50,6 +51,7 @@ class Fim(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
@@ -72,7 +74,7 @@ class Fim(BaseSDK):
             min_tokens=min_tokens,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/streamRawPredict#fim",
             base_url=base_url,
@@ -83,6 +85,7 @@ class Fim(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="text/event-stream",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.FIMCompletionStreamRequest
@@ -157,6 +160,7 @@ class Fim(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
     ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
         r"""Stream fim completion
 
@@ -175,6 +179,7 @@ class Fim(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
@@ -197,7 +202,7 @@ class Fim(BaseSDK):
             min_tokens=min_tokens,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/streamRawPredict#fim",
             base_url=base_url,
@@ -208,6 +213,7 @@ class Fim(BaseSDK):
             request_has_query_params=True,
             user_agent_header="user-agent",
             accept_header_value="text/event-stream",
+            http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.FIMCompletionStreamRequest
@@ -282,6 +288,7 @@ class Fim(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.FIMCompletionResponse]:
         r"""Fim Completion
 
@@ -300,6 +307,7 @@ class Fim(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
@@ -322,7 +330,7 @@ class Fim(BaseSDK):
             min_tokens=min_tokens,
         )
 
-        req = self.build_request(
+        req = self._build_request(
             method="POST",
             path="/rawPredict#fim",
             base_url=base_url,
@@ -333,6 +341,7 @@ class Fim(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.FIMCompletionRequest
@@ -403,6 +412,7 @@ class Fim(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.FIMCompletionResponse]:
         r"""Fim Completion
 
@@ -421,6 +431,7 @@ class Fim(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
@@ -443,7 +454,7 @@ class Fim(BaseSDK):
             min_tokens=min_tokens,
         )
 
-        req = self.build_request_async(
+        req = self._build_request_async(
             method="POST",
             path="/rawPredict#fim",
             base_url=base_url,
@@ -454,6 +465,7 @@ class Fim(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.FIMCompletionRequest

mistralai_gcp/models/chatcompletionrequest.py

@@ -105,7 +105,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

mistralai_gcp/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

mistralai_gcp/sdkconfiguration.py

@@ -28,9 +28,9 @@ class SDKConfiguration:
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.2.3"
-    gen_version: str = "2.470.1"
-    user_agent: str = "speakeasy-sdk/python 1.2.3 2.470.1 0.0.2 mistralai-gcp"
+    sdk_version: str = "1.2.6"
+    gen_version: str = "2.486.1"
+    user_agent: str = "speakeasy-sdk/python 1.2.6 2.486.1 0.0.2 mistralai-gcp"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 

mistralai_gcp/utils/forms.py

@@ -109,13 +109,12 @@ def serialize_multipart_form(
         if not field_metadata:
             continue
 
-        f_name = field.alias if field.alias is not None else name
+        f_name = field.alias if field.alias else name
 
         if field_metadata.file:
             file_fields: Dict[str, FieldInfo] = val.__class__.model_fields
 
             file_name = ""
-            field_name = ""
             content = None
             content_type = None
 
@@ -131,20 +130,15 @@ def serialize_multipart_form(
                 elif file_field_name == "content_type":
                     content_type = getattr(val, file_field_name, None)
                 else:
-                    field_name = (
-                        file_field.alias
-                        if file_field.alias is not None
-                        else file_field_name
-                    )
                     file_name = getattr(val, file_field_name)
 
-            if field_name == "" or file_name == "" or content is None:
+            if file_name == "" or content is None:
                 raise ValueError("invalid multipart/form-data file")
 
             if content_type is not None:
-                files[field_name] = (file_name, content, content_type)
+                files[f_name] = (file_name, content, content_type)
             else:
-                files[field_name] = (file_name, content)
+                files[f_name] = (file_name, content)
         elif field_metadata.json:
             files[f_name] = (
                 None,

mistralai_gcp/utils/requestbodies.py

@@ -23,7 +23,7 @@ SERIALIZATION_METHOD_TO_CONTENT_TYPE = {
 
 @dataclass
 class SerializedRequestBody:
-    media_type: str
+    media_type: Optional[str] = None
     content: Optional[Any] = None
     data: Optional[Any] = None
     files: Optional[Any] = None