handle state

Author: alexander-alderman-webb

sentry_sdk/integrations/pydantic_ai/__init__.py

@@ -159,12 +159,7 @@ def setup_once() -> None:
         _patch_agent_run()
         _patch_tool_execution()
 
-        try:
-            from pydantic_ai.capabilities import Hooks
-        except ImportError:
-            Hooks = None
-            PydanticAIIntegration.are_request_hooks_available = False
-
+        PydanticAIIntegration.using_request_hooks = False
         try:
             PYDANTIC_AI_VERSION = version("pydantic-ai-slim")
         except PackageNotFoundError:
@@ -182,8 +177,12 @@ def setup_once() -> None:
             _patch_graph_nodes()
             return
 
-        if Hooks is None:
+        try:
+            from pydantic_ai.capabilities import Hooks
+        except ImportError:
+            Hooks = None
             return
 
+        PydanticAIIntegration.using_request_hooks = True
         hooks = Hooks()
         register_hooks(hooks)

sentry_sdk/integrations/pydantic_ai/patches/agent_run.py

@@ -109,7 +109,7 @@ async def wrapper(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
             model = kwargs.get("model")
             model_settings = kwargs.get("model_settings")
 
-            if PydanticAIIntegration.are_request_hooks_available:
+            if PydanticAIIntegration.using_request_hooks:
                 metadata = kwargs.get("metadata")
                 if metadata is None:
                     kwargs["metadata"] = {"_sentry_span": None}
@@ -158,7 +158,7 @@ def wrapper(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
         model = kwargs.get("model")
         model_settings = kwargs.get("model_settings")
 
-        if PydanticAIIntegration.are_request_hooks_available:
+        if PydanticAIIntegration.using_request_hooks:
             metadata = kwargs.get("metadata")
             if metadata is None:
                 kwargs["metadata"] = {"_sentry_span": None}