@@ -64,7 +64,9 @@ def setup_once() -> None:
6464
6565def _wrap_generate_content_stream (f : "Callable[..., Any]" ) -> "Callable[..., Any]" :
6666 @wraps (f )
67- def new_generate_content_stream (self : "Any" , * args : "Any" , ** kwargs : "Any" ) -> "Any" :
67+ def new_generate_content_stream (
68+ self : "Any" , * args : "Any" , ** kwargs : "Any"
69+ ) -> "Any" :
6870 integration = sentry_sdk .get_client ().get_integration (GoogleGenAIIntegration )
6971 if integration is None :
7072 return f (self , * args , ** kwargs )
@@ -256,7 +258,9 @@ def new_generate_content(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
256258
257259def _wrap_async_generate_content (f : "Callable[..., Any]" ) -> "Callable[..., Any]" :
258260 @wraps (f )
259- async def new_async_generate_content (self : "Any" , * args : "Any" , ** kwargs : "Any" ) -> "Any" :
261+ async def new_async_generate_content (
262+ self : "Any" , * args : "Any" , ** kwargs : "Any"
263+ ) -> "Any" :
260264 integration = sentry_sdk .get_client ().get_integration (GoogleGenAIIntegration )
261265 if integration is None :
262266 return await f (self , * args , ** kwargs )
@@ -333,7 +337,9 @@ def new_embed_content(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
333337
334338def _wrap_async_embed_content (f : "Callable[..., Any]" ) -> "Callable[..., Any]" :
335339 @wraps (f )
336- async def new_async_embed_content (self : "Any" , * args : "Any" , ** kwargs : "Any" ) -> "Any" :
340+ async def new_async_embed_content (
341+ self : "Any" , * args : "Any" , ** kwargs : "Any"
342+ ) -> "Any" :
337343 integration = sentry_sdk .get_client ().get_integration (GoogleGenAIIntegration )
338344 if integration is None :
339345 return await f (self , * args , ** kwargs )
0 commit comments