.h9#ddlmZddlZddlmZmZddlmZmZerddl m Z m Z Gddej Z Gdd e ZGd d e ZGd d e ZGdde ZGdde ZGdde ZGdde ZGdde ZGdde ZGdde ZGdde ZdS)) annotationsN)MappingSequence) TYPE_CHECKINGAny)ResponseResponseInputItemParamcfeZdZdZejddZeejd dZdS) SpanDataz, Represents span data in the trace. returndict[str, Any]cdS)z%Export the span data as a dictionary.Nselfs U/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/tracing/span_data.pyexportzSpanData.exports  strcdS)zReturn the type of the span.Nrrs rtypez SpanData.types  rNr r r r) __name__ __module__ __qualname____doc__abcabstractmethodrpropertyrrrrr r sp        X   rr cFeZdZdZdZ ddd Zedd ZddZdS) AgentSpanDataze Represents an Agent Span in the trace. Includes name, handoffs, tools, and output type. namehandoffstools output_typeNr$rr%list[str] | Noner&r' str | Nonec>||_||_||_||_dSNr#)rr$r%r&r's r__init__zAgentSpanData.__init__$s' *2 ', '2rr cdS)Nagentrrs rrzAgentSpanData.type0swrr cD|j|j|j|j|jdS)Nrr$r%r&r'r0rs rrzAgentSpanData.export4s+II Z+    r)NNN)r$rr%r(r&r(r'r)rr rrrr __slots__r,r rrrrrr"r"s} =I &*"&"& 3 3 3 3 3X      rr"cBeZdZdZdZ ddd ZeddZddZdS)FunctionSpanDatazk Represents a Function Span in the trace. Includes input, output and MCP data (if applicable). r$inputoutputmcp_dataNr$rr6r)r7 Any | Noner8dict[str, Any] | Nonec>||_||_||_||_dSr+r5)rr$r6r7r8s rr,zFunctionSpanData.__init__Fs$     rr cdS)Nfunctionrrs rrzFunctionSpanData.typeRzrr cp|j|j|j|jrt |jnd|jdS)N)rr$r6r7r8)rr$r6r7rr8rs rrzFunctionSpanData.exportVs>IIZ*.+?c$+&&&4    rr+)r$rr6r)r7r9r8r:rrr1rrrr4r4>sw 8I+/ ! ! ! ! !X      rr4cJeZdZdZdZ ddd ZeddZddZdS)GenerationSpanDatazw Represents a Generation Span in the trace. Includes input, output, model, model configuration, and usage. r6r7model model_configusageNr6"Sequence[Mapping[str, Any]] | Noner7rCr)rDMapping[str, Any] | NonerEr:cL||_||_||_||_||_dSr+rB)rr6r7rCrDrEs rr,zGenerationSpanData.__init__ns,   ( rr rcdS)N generationrrs rrzGenerationSpanData.type|s|rr cP|j|j|j|j|j|jdS)Nrr6r7rCrDrErLrs rrzGenerationSpanData.exports0IZkZ -Z    r)NNNNN) r6rFr7rFrCr)rDrGrEr:rrr1rrrrArA`s I5959 15'+     X      rrAcDeZdZdZdZ ddd Zedd ZddZdS)ResponseSpanDatazS Represents a Response Span in the trace. Includes response and input. responser6NrPResponse | Noner6)str | list[ResponseInputItemParam] | Noner Nonec"||_||_dSr+rO)rrPr6s rr,zResponseSpanData.__init__s !  rrcdS)NrPrrs rrzResponseSpanData.typer>rr c<|j|jr |jjnddS)N)r response_id)rrPidrs rrzResponseSpanData.exports*I/3}F4=++$   rNN)rPrQr6rRr rSrrr1rrrrNrNsz &I%);?X      rrNc>eZdZdZdZd dZedd Zdd Zd S)HandoffSpanDataz] Represents a Handoff Span in the trace. Includes source and destination agents.  from_agentto_agentr]r)r^c"||_||_dSr+r\)rr]r^s rr,zHandoffSpanData.__init__s$  rr rcdS)Nhandoffrrs rrzHandoffSpanData.typesyrr c,|j|j|jdS)Nrr]r^rcrs rrzHandoffSpanData.exports I/    rN)r]r)r^r)rrr1rrrr[r[sm +I!!!!X      rr[c>eZdZdZdZd dZed d Zdd Zd S)CustomSpanDatazY Represents a Custom Span in the trace. Includes name and data property bag. r$datar$rrgr c"||_||_dSr+rf)rr$rgs rr,zCustomSpanData.__init__s  rr cdS)Ncustomrrs rrzCustomSpanData.typexrc,|j|j|jdS)Nrr$rgrmrs rrzCustomSpanData.exports III   rN)r$rrgr rrr1rrrreresm !IX      rrec@eZdZdZdZdddZedd Zdd Zd S)GuardrailSpanDataz[ Represents a Guardrail Span in the trace. Includes name and triggered status. r$ triggeredFr$rrqboolc"||_||_dSr+rp)rr$rqs rr,zGuardrailSpanData.__init__s "rr cdS)N guardrailrrs rrzGuardrailSpanData.type{rr c,|j|j|jdS)Nrr$rqrxrs rrzGuardrailSpanData.exports II   rN)F)r$rrqrrrrr1rrrrorosr &I#####X      rrocJeZdZdZdZ ddd ZeddZddZdS)TranscriptionSpanDatazs Represents a Transcription Span in the trace. Includes input, output, model, and model configuration. )r6r7rCrDNpcmr6r) input_formatr7rCrDrGcL||_||_||_||_||_dSr+)r6r|r7rCrD)rr6r|r7rCrDs rr,zTranscriptionSpanData.__init__s/ (  (rr rcdS)N transcriptionrrs rrzTranscriptionSpanData.type srr cX|j|jpd|jd|j|j|jdS)Nrgformat)rr6r7rCrD)rr6r|r7rCrDrs rrzTranscriptionSpanData.exportsBI (b+kZ -   r)Nr{NNN) r6r)r|r)r7r)rCr)rDrGrrr1rrrrzrzs I!#(! 15 ) ) ) ) )X       rrzcLeZdZdZdZ ddd ZeddZddZdS)SpeechSpanDataz Represents a Speech Span in the trace. Includes input, output, model, model configuration, and first content timestamp. )r6r7rCrDfirst_content_atNr{r6r)r7 output_formatrCrDrGrcZ||_||_||_||_||_||_dSr+)r6r7rrCrDr)rr6r7rrCrDrs rr,zSpeechSpanData.__init__$s7  * ( 0rr rcdS)Nspeechrrs rrzSpeechSpanData.type4rkrr cd|j|j|jpd|jd|j|j|jdS)Nrr)rr6r7rCrDr)rr6r7rrCrDrrs rrzSpeechSpanData.export8sHIZ )r,Z - $ 5   r)NNr{NNN) r6r)r7r)rr)rCr)rDrGrr)rrr1rrrrrs QI!!$) 15'+11111 X       rrcBeZdZdZdZ d d dZed dZdd ZdS)SpeechGroupSpanDataz6 Represents a Speech Group Span in the trace. r6Nr)c||_dSr+)r6)rr6s rr,zSpeechGroupSpanData.__init__Ms rr rcdS)N speech_grouprrs rrzSpeechGroupSpanData.typeSs~rr c |j|jdS)Nrr6rrs rrzSpeechGroupSpanData.exportWsIZ   rr+)r6r)rrr1rrrrrFswI! X      rrc@eZdZdZdZdddZedd Zdd ZdS)MCPListToolsSpanDatazY Represents an MCP List Tools Span in the trace. Includes server and result. serverresultNrr)rr(c"||_||_dSr+r)rrrs rr,zMCPListToolsSpanData.__init__is  rr rcdS)N mcp_toolsrrs rrzMCPListToolsSpanData.typemrvrr c,|j|j|jdS)Nrrrrrs rrzMCPListToolsSpanData.exportqs Ikk   rrY)rr)rr(rrr1rrrrr^sr I X      rr) __future__rrcollections.abcrrtypingrropenai.types.responsesrr ABCr r"r4rArNr[rerorzrrrrrrrsz"""""" --------%%%%%%%%HGGGGGGGG     sw   "     H   D     x   D( ( ( ( ( ( ( ( V     x   :     h   0     X   0        0) ) ) ) ) H) ) ) X' ' ' ' ' X' ' ' T     (   0     8     r