REDROOM

Edit File Size: 2.12 KB

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

Text Base64

from typing import TYPE_CHECKING

import sentry_sdk
from sentry_sdk.consts import OP, SPANDATA

from ..consts import SPAN_ORIGIN
from ..utils import (
    _set_agent_data,
    _set_input_data,
    _set_output_data,
    _set_usage_data,
)

if TYPE_CHECKING:
    from typing import Any, Optional

from agents import Agent

def ai_client_span(
    agent: "Agent", get_response_kwargs: "dict[str, Any]"
) -> "sentry_sdk.tracing.Span":
    # TODO-anton: implement other types of operations. Now "chat" is hardcoded.
    # Get model name from agent.model or fall back to request model (for when agent.model is None/default)
    model_name = None
    if agent.model:
        model_name = agent.model.model if hasattr(agent.model, "model") else agent.model
    elif hasattr(agent, "_sentry_request_model"):
        model_name = agent._sentry_request_model

span = sentry_sdk.start_span(
        op=OP.GEN_AI_CHAT,
        name=f"chat {model_name}",
        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")

_set_agent_data(span, agent)
    _set_input_data(span, get_response_kwargs)

return span

def update_ai_client_span(
    span: "sentry_sdk.tracing.Span",
    response: "Any",
    response_model: "Optional[str]" = None,
    agent: "Optional[Agent]" = None,
) -> None:
    """Update AI client span with response data (works for streaming and non-streaming)."""
    if hasattr(response, "usage") and response.usage:
        _set_usage_data(span, response.usage)

if hasattr(response, "output") and response.output:
        _set_output_data(span, response)

if response_model is not None:
        span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, response_model)
    elif hasattr(response, "model") and response.model:
        span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, str(response.model))

# Set conversation ID from agent if available
    if 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