.

Author: antonpirker

sentry_sdk/integrations/openai_agents/patches/models.py

@@ -22,6 +22,7 @@ def _create_get_model_wrapper(original_get_model):
     Wraps the agents.Runner._get_model method to wrap the get_response method of the model to create a AI client span.
     """
 
+    @classmethod  # type: ignore[misc]
     @wraps(original_get_model)
     def wrapped_get_model(cls, agent, run_config):
         # type: (agents.Runner, agents.Agent, agents.RunConfig) -> agents.Model

sentry_sdk/integrations/openai_agents/patches/runner.py

@@ -25,6 +25,7 @@ def _create_run_wrapper(original_func):
     """
     is_async = asyncio.iscoroutinefunction(original_func)
 
+    @classmethod  # type: ignore[misc]
     @wraps(original_func)
     async def async_wrapper(cls, *args, **kwargs):
         # type: (agents.Runner, *Any, **Any) -> Any
@@ -36,6 +37,7 @@ async def async_wrapper(cls, *args, **kwargs):
 
         return result
 
+    @classmethod  # type: ignore[misc]
     @wraps(original_func)
     def sync_wrapper(cls, *args, **kwargs):
         # type: (agents.Runner, *Any, **Any) -> Any

sentry_sdk/integrations/openai_agents/patches/tools.py

@@ -21,6 +21,7 @@ def _create_get_all_tools_wrapper(original_get_all_tools):
     Wraps the agents.Runner._get_all_tools method of the Runner class to wrap all function tools with Sentry instrumentation.
     """
 
+    @classmethod  # type: ignore[misc]
     @wraps(original_get_all_tools)
     async def wrapped_get_all_tools(cls, agent, context_wrapper):
         # type: (agents.Runner, agents.Agent, agents.RunContextWrapper) -> list[agents.Tool]

sentry_sdk/integrations/openai_agents/spans/agent_workflow.py

@@ -10,9 +10,9 @@
     from typing import Any
 
 
-def agent_workflow_span(*args, **kwargs):
-    # type: (*Any, **Any) -> sentry_sdk.tracing.Span
-    agent = args[0]
+def agent_workflow_span(agent):
+    # type: (Agent) -> sentry_sdk.tracing.Span
+
     # Create a transaction or a span if an transaction is already active
     span = _get_start_span_function()(
         name=f"{agent.name} workflow",