qi dZddlZddlmZmZer ddlmZddlmZddl m Z e rddlmZddlm Z mZm Z Gdd Z y) a)Pipeline-scoped tracing context for OpenTelemetry tracing in Pipecat. This module provides a per-pipeline tracing context that holds the current conversation and turn span contexts. Each PipelineTask creates its own TracingContext, ensuring concurrent pipelines do not interfere with each other. N) TYPE_CHECKINGOptional)Context) SpanContext)is_tracing_available)NonRecordingSpanrset_span_in_contextceZdZdZdZ ddeddeefdZded fd Zdedfd Z ded fd Z e deefd Z e defdZy)TracingContextzPipeline-scoped tracing context. Holds the current conversation and turn span contexts for a single pipeline. Created by PipelineTask, passed to TurnTraceObserver (writer) and services (readers) via StartFrame. c.d|_d|_d|_y)z0Initialize the tracing context with empty state.N)_conversation_context _turn_context_conversation_idselfs W/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/utils/tracing/tracing_context.py__init__zTracingContext.__init__$s:>"26/3N span_contextrconversation_idcrtsy||_|rt|}t||_yd|_y)zSet the current conversation context. Args: span_context: The span context for the current conversation or None to clear it. conversation_id: Optional ID for the conversation. N)rrrr r )rrrnon_recording_spans rset_conversation_contextz'TracingContext.set_conversation_context*s9$%  / !1,!? )<=O)PD &)-D &rreturnrc|jS)zGet the OpenTelemetry context for the current conversation. Returns: The current conversation context or None if not available. )r rs rget_conversation_contextz'TracingContext.get_conversation_context>s )))rcdtsy|rt|}t||_yd|_y)zSet the current turn context. Args: span_context: The span context for the current turn or None to clear it. N)rrr r)rrrs rset_turn_contextzTracingContext.set_turn_contextFs1 $%  !1,!? !45G!HD !%D rc|jS)zGet the OpenTelemetry context for the current turn. Returns: The current turn context or None if not available. )rrs rget_turn_contextzTracingContext.get_turn_contextUs !!!rc|jS)zGet the ID for the current conversation. Returns: The current conversation ID or None if not available. )rrs rrzTracingContext.conversation_id]s$$$rc<ttjS)zlGenerate a new conversation ID. Returns: A new randomly generated UUID string. )struuiduuid4rrgenerate_conversation_idz'TracingContext.generate_conversation_idfs4::<  r)N)__name__ __module__ __qualname____doc__rrr#rrrr propertyr staticmethodr'r&rrr r s4W[.$]3.FNsm.(*(9*=* &Xm-D &"(9"5"%#%%!c!!rr )r+r$typingrropentelemetry.contextropentelemetry.tracerpipecat.utils.tracing.setuprrr r r&rrr2s9 *-/<-VVQ!Q!r