o i @s|dZddlZddlmZmZerddlmZddlmZddl m Z e r5ddlmZddlm Z mZm Z Gdd d Z dS) 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_contextc@seZdZdZddZ ddeddeefdd Zd ed fd d ZdedfddZ d ed fddZ e d eefddZ e d efddZdS)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. cCsd|_d|_d|_dS)z0Initialize the tracing context with empty state.N)_conversation_context _turn_context_conversation_idselfrY/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/utils/tracing/tracing_context.py__init__$s zTracingContext.__init__N span_contextrconversation_idcCs4tsdS||_|rt|}t||_dSd|_dS)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)rr rrr )rrrnon_recording_spanrrrset_conversation_context*s  z'TracingContext.set_conversation_contextreturnrcC|jS)zGet the OpenTelemetry context for the current conversation. Returns: The current conversation context or None if not available. )r r rrrget_conversation_context>z'TracingContext.get_conversation_contextcCs.tsdS|rt|}t||_dSd|_dS)zSet the current turn context. Args: span_context: The span context for the current turn or None to clear it. N)rrrr )rrrrrrset_turn_contextFs  zTracingContext.set_turn_contextcCr)zGet the OpenTelemetry context for the current turn. Returns: The current turn context or None if not available. )r r rrrget_turn_contextUrzTracingContext.get_turn_contextcCr)zGet the ID for the current conversation. Returns: The current conversation ID or None if not available. )r r rrrr]szTracingContext.conversation_idcCs ttS)zlGenerate a new conversation ID. Returns: A new randomly generated UUID string. )struuiduuid4rrrrgenerate_conversation_idfs z'TracingContext.generate_conversation_id)N)__name__ __module__ __qualname____doc__rrrrrrrpropertyr staticmethodrrrrrr s  r )r#rtypingrropentelemetry.contextropentelemetry.tracerpipecat.utils.tracing.setuprrrr rrrrs