/UiSFddlmZddlmZmZddlmZmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZdd lmZdd lmZerdd lmZ d dd ZddZddZ d d dZ d! d"dZ! d# d$dZ" d d%dZ# d! d&dZ$ d d'dZ% d( d)dZ& d* d+dZ' d, d-dZ( d d.dZ) d! d/dZ*y )0) annotations)MappingSequence) TYPE_CHECKINGAny)logger)get_trace_provider) AgentSpanDataCustomSpanDataFunctionSpanDataGenerationSpanDataGuardrailSpanDataHandoffSpanDataMCPListToolsSpanDataResponseSpanDataSpeechGroupSpanDataSpeechSpanDataTranscriptionSpanData)Span)Trace)ResponseNctj}|rtjdtj |||||S)a Create a new trace. The trace will not be started automatically; you should either use it as a context manager (`with trace(...):`) or call `trace.start()` + `trace.finish()` manually. In addition to the workflow name and optional grouping identifier, you can provide an arbitrary metadata dictionary to attach additional user-defined information to the trace. Args: workflow_name: The name of the logical app or workflow. For example, you might provide "code_bot" for a coding agent, or "customer_support_agent" for a customer support agent. trace_id: The ID of the trace. Optional. If not provided, we will generate an ID. We recommend using `util.gen_trace_id()` to generate a trace ID, to guarantee that IDs are correctly formatted. group_id: Optional grouping identifier to link multiple traces from the same conversation or process. For instance, you might use a chat thread ID. metadata: Optional dictionary of additional metadata to attach to the trace. disabled: If True, we will return a Trace but the Trace will not be recorded. Returns: The newly created trace object. zKTrace already exists. Creating a new trace, but this is probably a mistake.)nametrace_idgroup_idmetadatadisabled)r get_current_tracer warning create_trace) workflow_namerrrr current_traces q/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/tracing/create.pytracer&sS<'(::sBD   + +$%   ,  r'cPtjt||||S)aCreate a new response span. The span will not be started automatically, you should either do `with response_span() ...` or call `span.start()` + `span.finish()` manually. Args: response: The OpenAI Response object. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. )responser0)r r4r)r@r2r3rs r% response_spanrAs0$   + +"H5 , r'cRtjt|||||S)a1Create a new handoff span. The span will not be started automatically, you should either do `with handoff_span() ...` or call `span.start()` + `span.finish()` manually. Args: from_agent: The name of the agent that is handing off. to_agent: The name of the agent that is receiving the handoff. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. Returns: The newly created handoff span. ) from_agentto_agentr0)r r4r)rCrDr2r3rs r% handoff_spanrEs2.   + +!Z(K , r'cZtjt||xsi|||S)aCCreate a new custom span, to which you can add your own metadata. The span will not be started automatically, you should either do `with custom_span() ...` or call `span.start()` + `span.finish()` manually. Args: name: The name of the custom span. data: Arbitrary structured data to associate with the span. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. Returns: The newly created custom span. )rdatar0)r r4r )rrGr2r3rs r% custom_spanrHs60   + + d< , r'cRtjt|||||S)aCreate a new guardrail span. The span will not be started automatically, you should either do `with guardrail_span() ...` or call `span.start()` + `span.finish()` manually. Args: name: The name of the guardrail. triggered: Whether the guardrail was triggered. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. )r triggeredr0)r r4r)rrJr2r3rs r%guardrail_spanrK s2(   + +#C , r'c Xtjt||||||||S)aJCreate a new transcription span. The span will not be started automatically, you should either do `with transcription_span() ...` or call `span.start()` + `span.finish()` manually. Args: model: The name of the model used for the speech-to-text. input: The audio input of the speech-to-text transcription, as a base64 encoded string of audio bytes. input_format: The format of the audio input (defaults to "pcm"). output: The output of the speech-to-text transcription. model_config: The model configuration (hyperparameters) used. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. Returns: The newly created speech-to-text span. )r7 input_formatr8r;r<r0)r r4r)r;r7rMr8r<r2r3rs r%transcription_spanrN<sA<   + +'%%   ,  r'c Ztjt|||||||||S)a)Create a new speech span. The span will not be started automatically, you should either do `with speech_span() ...` or call `span.start()` + `span.finish()` manually. Args: model: The name of the model used for the text-to-speech. input: The text input of the text-to-speech. output: The audio output of the text-to-speech as base64 encoded string of PCM audio bytes. output_format: The format of the audio output (defaults to "pcm"). model_config: The model configuration (hyperparameters) used. first_content_at: The time of the first byte of the audio output. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. )r;r7r8 output_formatr<first_content_atr0)r r4r) r;r7r8rPr<rQr2r3rs r% speech_spanrRhsD8   + + '%-   ,  r'cPtjt||||S)aCreate a new speech group span. The span will not be started automatically, you should either do `with speech_group_span() ...` or call `span.start()` + `span.finish()` manually. Args: input: The input text used for the speech request. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. )r7r0)r r4r)r7r2r3rs r%speech_group_spanrTs0$   + +%E2 , r'cRtjt|||||S)aCreate a new MCP list tools span. The span will not be started automatically, you should either do `with mcp_tools_span() ...` or call `span.start()` + `span.finish()` manually. Args: server: The name of the MCP server. result: The result of the MCP list tools call. span_id: The ID of the span. Optional. If not provided, we will generate an ID. We recommend using `util.gen_span_id()` to generate a span ID, to guarantee that IDs are correctly formatted. parent: The parent span or trace. If not provided, we will automatically use the current trace/span as the parent. disabled: If True, we will return a Span but the Span will not be recorded. )serverresultr0)r r4r)rVrWr2r3rs r%mcp_tools_spanrXs2(   + +&fVD , r')NNNF) r#strr str | NonerrZrdict[str, Any] | Nonerboolreturnr)r]z Trace | None)r]zSpan[Any] | None)NNNNNF)rrYr-list[str] | Noner.r^r/rZr2rZr3Trace | Span[Any] | Nonerr\r]zSpan[AgentSpanData])NNNNF)rrYr7rZr8rZr2rZr3r_rr\r]zSpan[FunctionSpanData])NNNNNNNF)r7"Sequence[Mapping[str, Any]] | Noner8r`r;rZr<Mapping[str, Any] | Noner=r[r2rZr3r_rr\r]zSpan[GenerationSpanData]) r@zResponse | Noner2rZr3r_rr\r]zSpan[ResponseSpanData]) rCrZrDrZr2rZr3r_rr\r]zSpan[HandoffSpanData]) rrYrGr[r2rZr3r_rr\r]zSpan[CustomSpanData])FNNF) rrYrJr\r2rZr3r_rr\r]zSpan[GuardrailSpanData])NNpcmNNNNF)r;rZr7rZrMrZr8rZr<rar2rZr3r_rr\r]zSpan[TranscriptionSpanData]) NNNrbNNNNF)r;rZr7rZr8rZrPrZr<rarQrZr2rZr3r_rr\r]zSpan[SpeechSpanData]) r7rZr2rZr3r_rr\r]zSpan[SpeechGroupSpanData]) rVrZrWr^r2rZr3r_rr\r]zSpan[MCPListToolsSpanData])+ __future__rcollections.abcrrtypingrrr setupr r1r r rrrrrrrrrspansrtracesropenai.types.responsesrr&r r+r5r9r>rArErHrKrNrRrTrXr)r'r%rjsp"-%%    /  &* ****$ *  *  *Z4 3"&""'+          %   J'+      %   D1515-1#''+- -- .- -+ - ! -  - %---b!%'+   %   6"'+  %     B#''+    %     D'+   %     :$-1'+) ) ))  ) + )  ) %))!)Z %-1#''+( ( ( ( ( + ( ! (( %(((X'+    %   6#'+    %     r'