Zy Filemanager

📁 File Browser

Name	Size	Permissions	Actions
📁 __pycache__	DIR	0755	🗑️
📄 __init__.py (353 bytes)	353 B	0644	✍️ ⬇️ 🗑️
📄 agent_workflow.py (469 bytes)	469 B	0644	✍️ ⬇️ 🗑️
📄 ai_client.py (1,292 bytes)	1.26 KB	0644	✍️ ⬇️ 🗑️
📄 execute_tool.py (1,411 bytes)	1.38 KB	0644	✍️ ⬇️ 🗑️
📄 handoff.py (525 bytes)	525 B	0644	✍️ ⬇️ 🗑️
📄 invoke_agent.py (2,493 bytes)	2.43 KB	0644	✍️ ⬇️ 🗑️

📤 Upload Files

Single File

Multiple Files

ZIP Extract

ZIP file will be deleted after extraction

➕ Create New

New Folder

New File

✍️ Editing: ai_client.py

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

            Size: 1.26 KB | 
            Last Modified: 2025-12-03 14:34:44        

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,
    _create_mcp_execute_tool_spans,
)

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from agents import Agent
    from typing import Any

def ai_client_span(agent, get_response_kwargs):
    # type: (Agent, dict[str, Any]) -> sentry_sdk.tracing.Span
    # TODO-anton: implement other types of operations. Now "chat" is hardcoded.
    model_name = agent.model.model if hasattr(agent.model, "model") else agent.model
    span = sentry_sdk.start_span(
        op=OP.GEN_AI_CHAT,
        description=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, agent, get_response_kwargs, result):
    # type: (sentry_sdk.tracing.Span, Agent, dict[str, Any], Any) -> None
    _set_usage_data(span, result.usage)
    _set_output_data(span, result)
    _create_mcp_execute_tool_spans(span, result)

↩️ Back