removed unused function

Author: antonpirker

sentry_sdk/integrations/openai_agents/patches/runner.py

@@ -4,10 +4,7 @@
 
 import sentry_sdk
 
-from ..spans import (
-    agent_workflow_span,
-    update_agent_workflow_span,
-)
+from ..spans import agent_workflow_span
 from ..utils import _capture_exception
 
 from typing import TYPE_CHECKING
@@ -27,7 +24,7 @@ def _create_run_wrapper(original_func):
     async def async_wrapper(*args, **kwargs):
         # type: (*Any, **Any) -> Any
         agent = args[0]
-        with agent_workflow_span(agent) as span:
+        with agent_workflow_span(agent):
             result = None
             try:
                 result = await original_func(*args, **kwargs)
@@ -41,8 +38,6 @@ async def async_wrapper(*args, **kwargs):
                     current_span.__exit__(None, None, None)
 
                 raise exc from None
-            finally:
-                update_agent_workflow_span(span, agent, result)
 
     @wraps(original_func)
     def sync_wrapper(*args, **kwargs):

sentry_sdk/integrations/openai_agents/spans/__init__.py

@@ -1,7 +1,4 @@
-from .agent_workflow import (  # noqa: F401
-    agent_workflow_span,
-    update_agent_workflow_span,
-)
+from .agent_workflow import agent_workflow_span  # noqa: F401
 from .ai_client import ai_client_span, update_ai_client_span  # noqa: F401
 from .execute_tool import execute_tool_span, update_execute_tool_span  # noqa: F401
 from .handoff import handoff_span  # noqa: F401

sentry_sdk/integrations/openai_agents/spans/agent_workflow.py

@@ -6,12 +6,11 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from agents import Agent
-    from typing import Any
+    import agents
 
 
 def agent_workflow_span(agent):
-    # type: (Agent) -> sentry_sdk.tracing.Span
+    # type: (agents.Agent) -> sentry_sdk.tracing.Span
 
     # Create a transaction or a span if an transaction is already active
     span = _get_start_span_function()(
@@ -20,8 +19,3 @@ def agent_workflow_span(agent):
     )
 
     return span
-
-
-def update_agent_workflow_span(span, agent, result):
-    # type: (sentry_sdk.tracing.Span, Agent, Any) -> None
-    pass