better span origin

Author: antonpirker

sentry_sdk/integrations/openai_agents/__init__.py

@@ -34,7 +34,6 @@ def _patch_tools():
 
 class OpenAIAgentsIntegration(Integration):
     identifier = "openai_agents"
-    origin = f"auto.ai.{identifier}"
 
     @staticmethod
     def setup_once():

sentry_sdk/integrations/openai_agents/consts.py

@@ -0,0 +1 @@
+SPAN_ORIGIN = "auto.ai.openai_agents"

sentry_sdk/integrations/openai_agents/spans/agent_workflow.py

@@ -1,4 +1,6 @@
 import sentry_sdk
+
+from ..consts import SPAN_ORIGIN
 from ..utils import _get_start_span_function
 
 from typing import TYPE_CHECKING
@@ -14,7 +16,7 @@ def agent_workflow_span(*args, **kwargs):
     # Create a transaction or a span if an transaction is already active
     span = _get_start_span_function()(
         name=f"{agent.name} workflow",
-        origin=sentry_sdk.integrations.openai_agents.OpenAIAgentsIntegration.origin,
+        origin=SPAN_ORIGIN,
     )
 
     return span

sentry_sdk/integrations/openai_agents/spans/ai_client.py

@@ -1,6 +1,7 @@
 import sentry_sdk
 from sentry_sdk.consts import OP, SPANDATA
 
+from ..consts import SPAN_ORIGIN
 from ..utils import (
     _set_agent_data,
     _set_input_data,
@@ -21,7 +22,7 @@ def ai_client_span(agent, get_response_kwargs):
     span = sentry_sdk.start_span(
         op=OP.GEN_AI_CHAT,
         description=f"chat {agent.model}",
-        origin=sentry_sdk.integrations.openai_agents.OpenAIAgentsIntegration.origin,
+        origin=SPAN_ORIGIN,
     )
     # TODO-anton: remove hardcoded stuff and replace something that also works for embedding and so on
     span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "chat")

sentry_sdk/integrations/openai_agents/spans/execute_tool.py

@@ -2,6 +2,7 @@
 from sentry_sdk.consts import OP, SPANDATA
 from sentry_sdk.scope import should_send_default_pii
 
+from ..consts import SPAN_ORIGIN
 from ..utils import _set_agent_data
 
 from typing import TYPE_CHECKING
@@ -16,7 +17,7 @@ def execute_tool_span(tool, *args, **kwargs):
     span = sentry_sdk.start_span(
         op=OP.GEN_AI_EXECUTE_TOOL,
         name=f"execute_tool {tool.name}",
-        origin=sentry_sdk.integrations.openai_agents.OpenAIAgentsIntegration.origin,
+        origin=SPAN_ORIGIN,
     )
 
     span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "execute_tool")

sentry_sdk/integrations/openai_agents/spans/invoke_agent.py

@@ -1,6 +1,7 @@
 import sentry_sdk
 from sentry_sdk.consts import OP, SPANDATA
 
+from ..consts import SPAN_ORIGIN
 from ..utils import _set_agent_data
 
 from typing import TYPE_CHECKING
@@ -15,7 +16,7 @@ def invoke_agent_span(context, agent):
     span = sentry_sdk.start_span(
         op=OP.GEN_AI_INVOKE_AGENT,
         name=f"invoke_agent {agent.name}",
-        origin=sentry_sdk.integrations.openai_agents.OpenAIAgentsIntegration.origin,
+        origin=SPAN_ORIGIN,
     )
     span.__enter__()