o õæh9#ã@sddlmZddlZddlmZmZddlmZmZer$ddl m Z m Z Gdd„dej ƒZ Gdd „d e ƒZGd d „d e ƒZGd d „d e ƒZGdd„de ƒZGdd„de ƒZGdd„de ƒZGdd„de ƒZGdd„de ƒZGdd„de ƒZGdd„de ƒZGdd„de ƒZdS)é)Ú annotationsN)ÚMappingÚSequence)Ú TYPE_CHECKINGÚAny)ÚResponseÚResponseInputItemParamc@s4eZdZdZejd dd„ƒZeejd dd„ƒƒZd S) ÚSpanDataz, Represents span data in the trace. Úreturnúdict[str, Any]cCódS)z%Export the span data as a dictionary.N©©Úselfr r úY/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/tracing/span_data.pyÚexportszSpanData.exportÚstrcCr )zReturn the type of the span.Nr rr r rÚtypesz SpanData.typeN©r r ©r r) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚabcÚabstractmethodrÚpropertyrr r r rr s r c@s>eZdZdZdZ   ddd d „Zeddd„ƒZddd„ZdS)Ú AgentSpanDataze Represents an Agent Span in the trace. Includes name, handoffs, tools, and output type. ©ÚnameÚhandoffsÚtoolsÚ output_typeNrrr úlist[str] | Noner!r"ú str | NonecCó||_||_||_||_dS©Nr)rrr r!r"r r rÚ__init__$ó zAgentSpanData.__init__r cCr )NÚagentr rr r rr0ózAgentSpanData.typer cCs|j|j|j|j|jdœS)N©rrr r!r"r+rr r rr4s ûzAgentSpanData.export)NNN)rrr r#r!r#r"r$rr© rrrrÚ __slots__r'rrrr r r rrs û rc@s:eZdZdZdZ ddd d „Zeddd„ƒZddd„ZdS)ÚFunctionSpanDatazk Represents a Function Span in the trace. Includes input, output and MCP data (if applicable). ©rÚinputÚoutputÚmcp_dataNrrr0r$r1ú Any | Noner2údict[str, Any] | NonecCr%r&r/)rrr0r1r2r r rr'Fr(zFunctionSpanData.__init__r cCr )NÚfunctionr rr r rrRr*zFunctionSpanData.typer cCs(|j|j|j|jrt|jƒnd|jdœS)N)rrr0r1r2)rrr0r1rr2rr r rrVs ûzFunctionSpanData.exportr&)rrr0r$r1r3r2r4rrr,r r r rr.>s û r.c@sBeZdZdZdZ     ddd d„Zeddd„ƒZddd„ZdS)ÚGenerationSpanDatazw Represents a Generation Span in the trace. Includes input, output, model, model configuration, and usage. ©r0r1ÚmodelÚ model_configÚusageNr0ú"Sequence[Mapping[str, Any]] | Noner1r8r$r9úMapping[str, Any] | Noner:r4cCó"||_||_||_||_||_dSr&r7)rr0r1r8r9r:r r rr'nó  zGenerationSpanData.__init__r rcCr )NÚ generationr rr r rr|r*zGenerationSpanData.typer cCs|j|j|j|j|j|jdœS)N©rr0r1r8r9r:r@rr r rr€súzGenerationSpanData.export)NNNNN) r0r;r1r;r8r$r9r<r:r4rrr,r r r rr6`s  ú r6c@s<eZdZdZdZ  ddd d „Zedd d„ƒZddd„ZdS)ÚResponseSpanDatazS Represents a Response Span in the trace. Includes response and input. ©Úresponser0NrCúResponse | Noner0ú)str | list[ResponseInputItemParam] | Noner ÚNonecCó||_||_dSr&rB)rrCr0r r rr'“s zResponseSpanData.__init__rcCr )NrCr rr r rrr*zResponseSpanData.typer cCs|j|jr |jjdœSddœS)N)rÚ response_id)rrCÚidrr r rr¡s  þþzResponseSpanData.export©NN)rCrDr0rEr rFrrr,r r r rrA‹s ý rAc@s6eZdZdZdZddd„Zedd d „ƒZdd d„ZdS)ÚHandoffSpanDataz] Represents a Handoff Span in the trace. Includes source and destination agents. ©Ú from_agentÚto_agentrMr$rNcCrGr&rL)rrMrNr r rr'°ó zHandoffSpanData.__init__r rcCr )NÚhandoffr rr r rr´r*zHandoffSpanData.typer cCó|j|j|jdœS)N©rrMrNrRrr r rr¸óýzHandoffSpanData.exportN)rMr$rNr$rrr,r r r rrK¨ó  rKc@s6eZdZdZdZddd„Zedd d „ƒZdd d „ZdS)ÚCustomSpanDatazY Represents a Custom Span in the trace. Includes name and data property bag. ©rÚdatarrrWr cCrGr&rV)rrrWr r rr'ÈrOzCustomSpanData.__init__r cCr )NÚcustomr rr r rrÌr*zCustomSpanData.typecCrQ)N©rrrWrYrr r rrÐrSzCustomSpanData.exportN)rrrWr rrr,r r r rrUÀrTrUc@s8eZdZdZdZdddd „Zedd d „ƒZddd„ZdS)ÚGuardrailSpanDataz[ Represents a Guardrail Span in the trace. Includes name and triggered status. ©rÚ triggeredFrrr\ÚboolcCrGr&r[)rrr\r r rr'àrOzGuardrailSpanData.__init__r cCr )NÚ guardrailr rr r rrär*zGuardrailSpanData.typer cCrQ)N©rrr\r_rr r rrèrSzGuardrailSpanData.exportN)F)rrr\r]rrr,r r r rrZØs  rZc@sBeZdZdZdZ     ddd d „Zeddd„ƒZddd„ZdS)ÚTranscriptionSpanDatazs Represents a Transcription Span in the trace. Includes input, output, model, and model configuration. )r0r1r8r9NÚpcmr0r$Ú input_formatr1r8r9r<cCr=r&)r0rbr1r8r9)rr0rbr1r8r9r r rr'ýr>zTranscriptionSpanData.__init__r rcCr )NÚ transcriptionr rr r rr r*zTranscriptionSpanData.typer cCs&|j|jpd|jdœ|j|j|jdœS)NÚ©rWÚformat)rr0r1r8r9)rr0rbr1r8r9rr r rrsþøzTranscriptionSpanData.export)NraNNN) r0r$rbr$r1r$r8r$r9r<rrr,r r r rr`ðs  ú r`c@sDeZdZdZdZ      ddd d„Zeddd„ƒZddd„ZdS)ÚSpeechSpanDataz… Represents a Speech Span in the trace. Includes input, output, model, model configuration, and first content timestamp. )r0r1r8r9Úfirst_content_atNrar0r$r1Ú output_formatr8r9r<rhcCs(||_||_||_||_||_||_dSr&)r0r1rir8r9rh)rr0r1rir8r9rhr r rr'$s   zSpeechSpanData.__init__r rcCr )NÚspeechr rr r rr4r*zSpeechSpanData.typer cCs*|j|j|jpd|jdœ|j|j|jdœS)Nrdre)rr0r1r8r9rh)rr0r1rir8r9rhrr r rr8sþ÷zSpeechSpanData.export)NNraNNN) r0r$r1r$rir$r8r$r9r<rhr$rrr,r r r rrgs ù rgc@s:eZdZdZdZ dddd„Zedd d „ƒZdd d „ZdS)ÚSpeechGroupSpanDataz6 Represents a Speech Group Span in the trace. r0Nr$cCs ||_dSr&)r0)rr0r r rr'Ms zSpeechGroupSpanData.__init__r rcCr )NÚ speech_groupr rr r rrSr*zSpeechGroupSpanData.typer cCs|j|jdœS)N©rr0rmrr r rrWsþzSpeechGroupSpanData.exportr&)r0r$rrr,r r r rrkFs þ rkc@s8eZdZdZdZdddd „Zedd d „ƒZddd„ZdS)ÚMCPListToolsSpanDatazY Represents an MCP List Tools Span in the trace. Includes server and result. ©ÚserverÚresultNrpr$rqr#cCrGr&ro)rrprqr r rr'irOzMCPListToolsSpanData.__init__r rcCr )NÚ mcp_toolsr rr r rrmr*zMCPListToolsSpanData.typer cCrQ)N©rrprqrsrr r rrqrSzMCPListToolsSpanData.exportrJ)rpr$rqr#rrr,r r r rrn^s  rn)Ú __future__rrÚcollections.abcrrÚtypingrrÚopenai.types.responsesrrÚABCr rr.r6rArKrUrZr`rgrkrnr r r rÚs$ ""+,*