REDROOM

Edit File Size: 1.47 KB

//opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/openai_agents/spans/execute_tool.py

Text Base64

from typing import TYPE_CHECKING

import sentry_sdk
from sentry_sdk.consts import OP, SPANDATA, SPANSTATUS
from sentry_sdk.scope import should_send_default_pii

from ..consts import SPAN_ORIGIN
from ..utils import _set_agent_data

if TYPE_CHECKING:
    from typing import Any

import agents

def execute_tool_span(
    tool: "agents.Tool", *args: "Any", **kwargs: "Any"
) -> "sentry_sdk.tracing.Span":
    span = sentry_sdk.start_span(
        op=OP.GEN_AI_EXECUTE_TOOL,
        name=f"execute_tool {tool.name}",
        origin=SPAN_ORIGIN,
    )

span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "execute_tool")

span.set_data(SPANDATA.GEN_AI_TOOL_NAME, tool.name)
    span.set_data(SPANDATA.GEN_AI_TOOL_DESCRIPTION, tool.description)

if should_send_default_pii():
        input = args[1]
        span.set_data(SPANDATA.GEN_AI_TOOL_INPUT, input)

return span

def update_execute_tool_span(
    span: "sentry_sdk.tracing.Span",
    agent: "agents.Agent",
    tool: "agents.Tool",
    result: "Any",
) -> None:
    _set_agent_data(span, agent)

if isinstance(result, str) and result.startswith(
        "An error occurred while running the tool"
    ):
        span.set_status(SPANSTATUS.INTERNAL_ERROR)

if should_send_default_pii():
        span.set_data(SPANDATA.GEN_AI_TOOL_OUTPUT, result)

# Add conversation ID from agent
    conv_id = getattr(agent, "_sentry_conversation_id", None)
    if conv_id:
        span.set_data(SPANDATA.GEN_AI_CONVERSATION_ID, conv_id)

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-06-11 06:30:31	Edit Download recursive recursive
agent_workflow.py	459 B	lrw-r--r--	2026-06-11 06:30:31	Edit Download recursive recursive
ai_client.py	2.12 KB	lrw-r--r--	2026-06-11 06:30:31	Edit Download recursive recursive
execute_tool.py	1.47 KB	lrw-r--r--	2026-06-11 06:30:31	Edit Download recursive recursive
handoff.py	723 B	lrw-r--r--	2026-06-11 06:30:31	Edit Download recursive recursive
invoke_agent.py	3.09 KB	lrw-r--r--	2026-06-11 06:30:31	Edit Download recursive recursive
__init__.py	380 B	lrw-r--r--	2026-06-11 06:30:31	Edit Download recursive recursive

Directory Contents