@@ -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