fix: working in feedback

Author: constantinius

sentry_sdk/integrations/pydantic_ai/patches/agent_run.py

@@ -13,6 +13,7 @@
 
 try:
     import pydantic_ai
+    from pydantic_ai.agent import Agent
 except ImportError:
     raise DidNotEnable("pydantic-ai not installed")
 
@@ -202,8 +203,6 @@ def _patch_agent_run():
     This patches both non-streaming (run, run_sync) and streaming
     (run_stream, run_stream_events) methods.
     """
-    # Import here to avoid circular imports
-    from pydantic_ai.agent import Agent
 
     # Store original methods
     original_run = Agent.run

sentry_sdk/integrations/pydantic_ai/spans/invoke_agent.py

@@ -1,6 +1,7 @@
 import sentry_sdk
 from sentry_sdk.ai.utils import get_start_span_function, set_data_normalized
 from sentry_sdk.consts import OP, SPANDATA
+from sentry_sdk.utils import safe_serialize
 
 from ..consts import SPAN_ORIGIN
 from ..utils import _set_agent_data, _set_model_data, _should_send_prompts
@@ -36,8 +37,6 @@ def invoke_agent_span(user_prompt, agent, model, model_settings):
     # Add available tools if present
     if agent and hasattr(agent, "_function_toolset"):
         try:
-            from sentry_sdk.utils import safe_serialize
-
             tools = []
             # Get tools from the function toolset
             if hasattr(agent._function_toolset, "tools"):