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

mistralai/_hooks/sdkhooks.py

@@ -1,11 +1,21 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 import httpx
-from .types import SDKInitHook, BeforeRequestContext, BeforeRequestHook, AfterSuccessContext, AfterSuccessHook, AfterErrorContext, AfterErrorHook, Hooks
+from .types import (
+    SDKInitHook,
+    BeforeRequestContext,
+    BeforeRequestHook,
+    AfterSuccessContext,
+    AfterSuccessHook,
+    AfterErrorContext,
+    AfterErrorHook,
+    Hooks,
+)
 from .registration import init_hooks
 from typing import List, Optional, Tuple
 from mistralai.httpclient import HttpClient
 
+
 class SDKHooks(Hooks):
     def __init__(self) -> None:
         self.sdk_init_hooks: List[SDKInitHook] = []
@@ -31,7 +41,9 @@ class SDKHooks(Hooks):
             base_url, client = hook.sdk_init(base_url, client)
         return base_url, client
 
-    def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> httpx.Request:
+    def before_request(
+        self, hook_ctx: BeforeRequestContext, request: httpx.Request
+    ) -> httpx.Request:
         for hook in self.before_request_hooks:
             out = hook.before_request(hook_ctx, request)
             if isinstance(out, Exception):
@@ -40,7 +52,9 @@ class SDKHooks(Hooks):
 
         return request
 
-    def after_success(self, hook_ctx: AfterSuccessContext, response: httpx.Response) -> httpx.Response:
+    def after_success(
+        self, hook_ctx: AfterSuccessContext, response: httpx.Response
+    ) -> httpx.Response:
         for hook in self.after_success_hooks:
             out = hook.after_success(hook_ctx, response)
             if isinstance(out, Exception):
@@ -48,7 +62,12 @@ class SDKHooks(Hooks):
             response = out
         return response
 
-    def after_error(self, hook_ctx: AfterErrorContext, response: Optional[httpx.Response], error: Optional[Exception]) -> Tuple[Optional[httpx.Response], Optional[Exception]]:
+    def after_error(
+        self,
+        hook_ctx: AfterErrorContext,
+        response: Optional[httpx.Response],
+        error: Optional[Exception],
+    ) -> Tuple[Optional[httpx.Response], Optional[Exception]]:
         for hook in self.after_error_hooks:
             result = hook.after_error(hook_ctx, response, error)
             if isinstance(result, Exception):

mistralai/_hooks/types.py

@@ -1,6 +1,5 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-
 from abc import ABC, abstractmethod
 import httpx
 from mistralai.httpclient import HttpClient
@@ -12,7 +11,12 @@ class HookContext:
     oauth2_scopes: Optional[List[str]] = None
     security_source: Optional[Union[Any, Callable[[], Any]]] = None
 
-    def __init__(self, operation_id: str, oauth2_scopes: Optional[List[str]], security_source: Optional[Union[Any, Callable[[], Any]]]):
+    def __init__(
+        self,
+        operation_id: str,
+        oauth2_scopes: Optional[List[str]],
+        security_source: Optional[Union[Any, Callable[[], Any]]],
+    ):
         self.operation_id = operation_id
         self.oauth2_scopes = oauth2_scopes
         self.security_source = security_source
@@ -20,18 +24,23 @@ class HookContext:
 
 class BeforeRequestContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
-        super().__init__(hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source)
+        super().__init__(
+            hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source
+        )
 
 
 class AfterSuccessContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
-        super().__init__(hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source)
-    
+        super().__init__(
+            hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source
+        )
 
 
 class AfterErrorContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
-        super().__init__(hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source)
+        super().__init__(
+            hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source
+        )
 
 
 class SDKInitHook(ABC):
@@ -42,19 +51,28 @@ class SDKInitHook(ABC):
 
 class BeforeRequestHook(ABC):
     @abstractmethod
-    def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> Union[httpx.Request, Exception]:
+    def before_request(
+        self, hook_ctx: BeforeRequestContext, request: httpx.Request
+    ) -> Union[httpx.Request, Exception]:
         pass
 
 
 class AfterSuccessHook(ABC):
     @abstractmethod
-    def after_success(self, hook_ctx: AfterSuccessContext, response: httpx.Response) -> Union[httpx.Response, Exception]:
+    def after_success(
+        self, hook_ctx: AfterSuccessContext, response: httpx.Response
+    ) -> Union[httpx.Response, Exception]:
         pass
 
 
 class AfterErrorHook(ABC):
     @abstractmethod
-    def after_error(self, hook_ctx: AfterErrorContext, response: Optional[httpx.Response], error: Optional[Exception]) -> Union[Tuple[Optional[httpx.Response], Optional[Exception]], Exception]:
+    def after_error(
+        self,
+        hook_ctx: AfterErrorContext,
+        response: Optional[httpx.Response],
+        error: Optional[Exception],
+    ) -> Union[Tuple[Optional[httpx.Response], Optional[Exception]], Exception]:
         pass