o i @sVddlmZddlmZddlmZerddlmZdZ eddGdd d Z d S) ) annotations) dataclass) TYPE_CHECKING)SelfT)frozenc@sneZdZUdZded<ded<ded<ded<ded<ded<edd dZdddZdddZdddZ dS)InstrumentationNameszMConfiguration for instrumentation span names and attributes based on version.stragent_run_span_nameagent_name_attrtool_span_nametool_arguments_attrtool_result_attroutput_tool_span_nameversionintreturnrcCs0|dkr|dddddddS|d d d d d d dS)zCreate instrumentation configuration for a specific version. Args: version: The instrumentation version (1, 2, or 3+) Returns: InstrumentationConfig instance with version-appropriate settings rz agent run agent_namez running tooltool_arguments tool_responsezrunning output function)r r r r rr invoke_agentzgen_ai.agent.name execute_toolzgen_ai.tool.call.argumentszgen_ai.tool.call.result)clsrrrb/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/_instrumentation.py for_versions"  z InstrumentationNames.for_versionrcC|jdkr d|S|jS)zGet the formatted agent span name. Args: agent_name: Name of the agent being executed Returns: Formatted span name rz invoke_agent )r )selfrrrrget_agent_run_span_name: z,InstrumentationNames.get_agent_run_span_name tool_namecCr)zGet the formatted tool span name. Args: tool_name: Name of the tool being executed Returns: Formatted span name r execute_tool )r rr rrrget_tool_span_nameGrz'InstrumentationNames.get_tool_span_namecCr)zGet the formatted output tool span name. Args: tool_name: Name of the tool being executed Returns: Formatted span name rr!)rr"rrrget_output_tool_span_nameTrz.InstrumentationNames.get_output_tool_span_nameN)rrrr)rr rr )r r rr ) __name__ __module__ __qualname____doc____annotations__ classmethodrrr#r$rrrrr s     rN) __future__r dataclassesrtypingrtyping_extensionsrDEFAULT_INSTRUMENTATION_VERSIONrrrrrs