?HAMA

Edit File Size: 1.51 KB

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

Text Base64

import sentry_sdk
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, _should_send_prompts

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from typing import Any, Optional

def execute_tool_span(tool_name, tool_args, agent, tool_type="function"):
    # type: (str, Any, Any, str) -> sentry_sdk.tracing.Span
    """Create a span for tool execution.

Args:
        tool_name: The name of the tool being executed
        tool_args: The arguments passed to the tool
        agent: The agent executing the tool
        tool_type: The type of tool ("function" for regular tools, "mcp" for MCP services)
    """
    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_TYPE, tool_type)
    span.set_data(SPANDATA.GEN_AI_TOOL_NAME, tool_name)

_set_agent_data(span, agent)

if _should_send_prompts() and tool_args is not None:
        span.set_data(SPANDATA.GEN_AI_TOOL_INPUT, safe_serialize(tool_args))

return span

def update_execute_tool_span(span, result):
    # type: (sentry_sdk.tracing.Span, Any) -> None
    """Update the execute tool span with the result."""
    if not span:
        return

if _should_send_prompts() and result is not None:
        span.set_data(SPANDATA.GEN_AI_TOOL_OUTPUT, safe_serialize(result))

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

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2025-12-03 08:02:53	Edit Download recursive recursive
ai_client.py	8.63 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
execute_tool.py	1.51 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
invoke_agent.py	3.64 KB	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive
__init__.py	243 B	lrw-r--r--	2025-12-03 08:02:53	Edit Download recursive recursive

Directory Contents