mistralai 1.0.2__py3-none-any.whl → 1.1.0__py3-none-any.whl

mistralai_gcp/basesdk.py

@@ -3,10 +3,15 @@
 from .sdkconfiguration import SDKConfiguration
 import httpx
 from mistralai_gcp import models, utils
-from mistralai_gcp._hooks import AfterErrorContext, AfterSuccessContext, BeforeRequestContext
+from mistralai_gcp._hooks import (
+    AfterErrorContext,
+    AfterSuccessContext,
+    BeforeRequestContext,
+)
 from mistralai_gcp.utils import RetryConfig, SerializedRequestBody, get_body_content
 from typing import Callable, List, Optional, Tuple
 
+
 class BaseSDK:
     sdk_configuration: SDKConfiguration
 
@@ -24,6 +29,46 @@ class BaseSDK:
 
         return utils.template_url(base_url, url_variables)
 
+    def build_request_async(
+        self,
+        method,
+        path,
+        base_url,
+        url_variables,
+        request,
+        request_body_required,
+        request_has_path_params,
+        request_has_query_params,
+        user_agent_header,
+        accept_header_value,
+        _globals=None,
+        security=None,
+        timeout_ms: Optional[int] = None,
+        get_serialized_body: Optional[
+            Callable[[], Optional[SerializedRequestBody]]
+        ] = None,
+        url_override: Optional[str] = None,
+    ) -> httpx.Request:
+        client = self.sdk_configuration.async_client
+        return self.build_request_with_client(
+            client,
+            method,
+            path,
+            base_url,
+            url_variables,
+            request,
+            request_body_required,
+            request_has_path_params,
+            request_has_query_params,
+            user_agent_header,
+            accept_header_value,
+            _globals,
+            security,
+            timeout_ms,
+            get_serialized_body,
+            url_override,
+        )
+
     def build_request(
         self,
         method,
@@ -45,7 +90,46 @@ class BaseSDK:
         url_override: Optional[str] = None,
     ) -> httpx.Request:
         client = self.sdk_configuration.client
+        return self.build_request_with_client(
+            client,
+            method,
+            path,
+            base_url,
+            url_variables,
+            request,
+            request_body_required,
+            request_has_path_params,
+            request_has_query_params,
+            user_agent_header,
+            accept_header_value,
+            _globals,
+            security,
+            timeout_ms,
+            get_serialized_body,
+            url_override,
+        )
 
+    def build_request_with_client(
+        self,
+        client,
+        method,
+        path,
+        base_url,
+        url_variables,
+        request,
+        request_body_required,
+        request_has_path_params,
+        request_has_query_params,
+        user_agent_header,
+        accept_header_value,
+        _globals=None,
+        security=None,
+        timeout_ms: Optional[int] = None,
+        get_serialized_body: Optional[
+            Callable[[], Optional[SerializedRequestBody]]
+        ] = None,
+        url_override: Optional[str] = None,
+    ) -> httpx.Request:
         query_params = {}
 
         url = url_override
@@ -69,7 +153,7 @@ class BaseSDK:
         if security is not None:
             if callable(security):
                 security = security()
-        
+
         if security is not None:
             security_headers, security_query_params = utils.get_security(security)
             headers = {**headers, **security_headers}
@@ -129,7 +213,7 @@ class BaseSDK:
                     req.method,
                     req.url,
                     req.headers,
-                    get_body_content(req)
+                    get_body_content(req),
                 )
                 http_res = client.send(req, stream=stream)
             except Exception as e:
@@ -149,7 +233,7 @@ class BaseSDK:
                 http_res.status_code,
                 http_res.url,
                 http_res.headers,
-                "<streaming response>" if stream else http_res.text
+                "<streaming response>" if stream else http_res.text,
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
@@ -189,6 +273,7 @@ class BaseSDK:
     ) -> httpx.Response:
         client = self.sdk_configuration.async_client
         logger = self.sdk_configuration.debug_logger
+
         async def do():
             http_res = None
             try:
@@ -200,7 +285,7 @@ class BaseSDK:
                     req.method,
                     req.url,
                     req.headers,
-                    get_body_content(req)
+                    get_body_content(req),
                 )
                 http_res = await client.send(req, stream=stream)
             except Exception as e:
@@ -220,7 +305,7 @@ class BaseSDK:
                 http_res.status_code,
                 http_res.url,
                 http_res.headers,
-                "<streaming response>" if stream else http_res.text
+                "<streaming response>" if stream else http_res.text,
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):