qiDdZddlmZmZmZmZmZer ddlmZddl m Z ddl m Z e rddlmZde de fdZ d/d d de d e d e dee deddeede deedd fdZ d0d d de d e de dee deedee dee deddedeedd fdZ d1d d de d e dedee dee dee deed ee d!ee d"eee efd#eee efdeedd fd$Z d2d d de d e de d ee dee d%ee deddeeed&ee dee d'eed(ee d)eedd fd*Z d3d d de d e de d+eee efdee d'eed,ee d-eeedeeed&ee d)eedd fd.Zy )4aFunctions for adding attributes to OpenTelemetry spans. This module provides specialized functions for adding service-specific attributes to OpenTelemetry spans, following standard semantic conventions where applicable and Pipecat-specific conventions for additional context. ) TYPE_CHECKINGAnyDictListOptional)Span)ServiceSettings)is_tracing_available service_namereturncddddddd}||vr||S|jdr|dd j}|S|j}|S) aExtract the standardized gen_ai.system value from a service class name. Source: https://opentelemetry.io/docs/specs/semconv/attributes-registry/gen-ai/#gen-ai-system Uses standard OTel names where possible, with special case mappings for service names that don't follow the pattern. Args: service_name: The service class name to extract system name from. Returns: The standardized gen_ai.system value. z aws.bedrockz az.ai.openai gcp.geminiz gcp.vertex_aixai)AWSBedrockLLMServiceAzureLLMServiceGoogleLLMServiceGoogleLLMOpenAIBetaServiceGoogleVertexLLMServiceGrokLLMService LLMServiceNi)endswithlower)r SPECIAL_CASE_MAPPINGSproviders Z/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/utils/tracing/service_attributes.py$_get_gen_ai_system_from_service_namerss"!.)(&2"1 ,,$\22\*%++- O %%' ONspanrmodelvoice_idtextsettingsr character_countoperation_namettfbc  |jd|jddj|jd||jd||jdd|jd||r|jd |||jd |||jd ||r\|jj D];\} } t | t tttfs'|jd | | =| j D]8\} } t | t tttfs'|j| | :y )a#Add TTS-specific attributes to a span. Args: span: The span to add attributes to. service_name: Name of the TTS service (e.g., "cartesia"). model: Model name/identifier. voice_id: Voice identifier. text: The text being synthesized. settings: Service configuration settings. character_count: Number of characters in the text. operation_name: Name of the operation (default: "tts"). ttfb: Time to first byte in seconds. **kwargs: Additional attributes to add. gen_ai.system TTSServicegen_ai.request.modelgen_ai.operation.namegen_ai.output.typespeechr r!Nzmetrics.character_count metrics.ttfb settings. set_attributereplacer given_fieldsitems isinstancestrintfloatbool) rr rr r!r"r#r$r%kwargskeyvalues radd_tts_span_attributesr=CsA6  (<(<\2(N(T(T(VW-u5.?+X6z8,  64(" 4oF  >40"//1779 =JC%#sE4!89""Yse#4e< = lln+ U ec3t4 5   sE *+r transcriptis_finallanguageuser_id vad_enabledc  |jd|jddj|jd||jd||jd| |r|jd|||jd ||r|jd ||r|jd || |jd | |r\|jj D];\} } t | t tttfs'|jd | | =| j D]8\} } t | t tttfs'|j| | :y)aAdd STT-specific attributes to a span. Args: span: The span to add attributes to. service_name: Name of the STT service (e.g., "deepgram"). model: Model name/identifier. operation_name: Name of the operation (default: "stt"). transcript: The transcribed text. is_final: Whether this is a final transcript. language: Detected or configured language. user_id: User ID associated with the audio being transcribed. settings: Service configuration settings. vad_enabled: Whether voice activity detection is enabled. ttfb: Time to first byte in seconds. **kwargs: Additional attributes to add. r' STTServicer)r*r+rBr>Nr?r@rAr.r/r0)rr rr$r>r?r@rAr"rBr%r:r;r<s radd_stt_span_attributesrEzsU>  (<(<\2(N(T(T(VW-u5.?}k2 <4 :x0 :x0 9g.  >40"//1779 =JC%#sE4!89""Yse#4e< = lln+ U ec3t4 5   sE *+rstreammessagesoutputtools tool_count tool_choicesystem parametersextra_parametersc  |jdt||jd||jdd|jdd|jd||r|jd||r|jd ||r|jd |||jd ||r|jd || r|jd| | |jd| | rh| jD]U\}}t|tt t tfs'|dvr|jd||A|jd||W| rN| jD];\}}t|tt t tfs'|jd||=| jD]8\}}t|tt t tfs'|j||:y )aAdd LLM-specific attributes to a span. Args: span: The span to add attributes to. service_name: Name of the LLM service (e.g., "openai"). model: Model name/identifier. stream: Whether streaming is enabled. messages: JSON-serialized messages. output: Aggregated output text from the LLM. tools: JSON-serialized tools configuration. tool_count: Number of tools available. tool_choice: Tool selection configuration. system: System message. parameters: Service parameters. extra_parameters: Additional parameters. ttfb: Time to first byte in seconds. **kwargs: Additional attributes to add. r'r*r+chatr,r!rFinputrHrINrJrKrLr.) temperature max_tokensmax_completion_tokenstop_ptop_kfrequency_penaltypresence_penaltyseedzgen_ai.request.zparam.zextra.)r1rr4r5r6r7r8r9)rr rrFrGrHrIrJrKrLrMrNr%r:r;r<s radd_llm_span_attributesrZsF (L\(Z[-u5.7+V4x( 7H-  8V,  7E* <4 =+6  8V,  >40$**, >JC%#sE4!89  &&'>F&&u~u= >"*002 :JC%#sE4!89""VC5>59 : lln+ U ec3t4 5   sE *+r modalitiestools_serializedis_input text_outputaudio_data_sizec |jdd|jd||jd||jd||r|jd||r|jd||r|jd|| r&|jd | | |jd | | r|jd | | |jd | |r|jdt||jddg}|D]Y}t|trd|vr|j |d,t |ds9t |dd }|I|j |[|r!|jddj|| r|jd| |r|jjD]\}}t|ttttfr|jd||<|dk(sB|sEt |dr(|j|jd|jt |dr2|jr&|jd|jj t |ds|j"s|jd|j"j |jD]8\}}t|ttttfs'|j||:y )adAdd Gemini Live specific attributes to a span. Args: span: The span to add attributes to. service_name: Name of the service. model: Model name/identifier. operation_name: Name of the operation (setup, model_turn, tool_call, etc.). voice_id: Voice identifier used for output. language: Language code for the session. modalities: Supported modalities (e.g., "AUDIO", "TEXT"). settings: Service configuration settings. tools: Available tools/functions list. tools_serialized: JSON-serialized tools for detailed inspection. transcript: Transcription text. is_input: Whether transcript is input (True) or output (False). text_output: Text output from model. audio_data_size: Size of audio data in bytes. **kwargs: Additional attributes to add. r'rr*r+service.operationr r@r[r>Ntranscript.is_inputr^audio.data_size_bytes tools.counttools.availableTname tools.names,tools.definitionsr/vaddisabledzsettings.vad.disabledstart_sensitivityzsettings.vad.start_sensitivityend_sensitivityzsettings.vad.end_sensitivity)r1lenr5dictappendhasattrgetattrjoinr3r4r6r7r8r9rkrlr<rm)rr rr$r r@r[r"rIr\r>r]r^r_r: tool_namestool tool_namer;r<s radd_gemini_live_span_attributesrwsJ  5-u5.?*N; :x0 :x0 <4 <4     4h ? =+6" 2OD  =#e*5 ,d3  1D$%&D.!!$v,/v&#D&$7 (%%i0  1    }chhz.B C .0@A"//1779 dJC%#sE4!89""Yse#4e<%5*-%..2L&&'>O5"565;R;R&&8%:Q:Q:W:W5"349N9N&&'EuG\G\GbGbc dlln+ U ec3t4 5   sE *+rsession_propertiescontext_messagesfunction_callsc z|jdd|jd||jd||jd||r&|jd|||jd||r|jd || |jd | | r|jd t| |jd d g} | D]}t|trd|vr| j |d,t |dr| j |j Tt|tsed|vsjd|dvsr| j |dd| r!|jddj| | r|jd| |rt|jdt||rW|d}t |dr|jd|j n)t|trd|vr|jd|d|r|jD]\}}t|ttttfr|jd||<|dk(sB|Et|tr|jd|ht|tsy|jdd |jD];\}}t|ttttfs'|jd||=| jD]8\}}t|ttttfs'|j||:y)aAdd OpenAI Realtime specific attributes to a span. Args: span: The span to add attributes to. service_name: Name of the service. model: Model name/identifier. operation_name: Name of the operation (setup, transcription, response, etc.). session_properties: Session configuration properties. transcript: Transcription text. is_input: Whether transcript is input (True) or output (False). context_messages: JSON-serialized context messages. function_calls: Function calls being made. tools: Available tools/functions list. tools_serialized: JSON-serialized tools for detailed inspection. audio_data_size: Size of audio data in bytes. **kwargs: Additional attributes to add. r'openair*r+rar>NrbrQrcrdreTrffunctionrgrhrizfunction_calls.countrzfunction_calls.first_namezsession.turn_detectionzsession.turn_detection.enabledzsession.turn_detection.) r1rnr5rorprqrfrsr4r6r7r8r9)rr rr$rxr>r]ryrzrIr\r_r:rtrucallr;r<td_keytd_values r#add_openai_realtime_span_attributesrsB 1-u5.?*N; <4     4h ? 7$45" 2OD  =#e*5 ,d3  6TR\M]C]!!$z"26":;  <    }chhz.B C .0@A 13~3FG !!$DtV$""#> JD$'FdN""#>V M,224 ]JC%#sE4!89""XcU#3U;((U->eT*&&'GOt,&&'GN,1KKM](%hc5$0GH ..1H/QS[\] ]lln+ U ec3t4 5   sE *+r)NNNttsN)sttNNNNNFN) TNNNNNNNNN) NNNNNNNNNN)NNNNNNNN)__doc__typingrrrrropentelemetry.tracerpipecat.services.settingsr pipecat.utils.tracing.setupr r6rr7r8r=r9rErZrwrrrrs<<;(9<($s$s$X,0%) 4+ 4+4+ 4+ 4+ 3- 4+ () 4+c]4+4+ 5/4+ 4+v $#"!,0 =+ =+=+ =+ =+  =+ tn =+sm=+c]=+()=+=+ 5/=+ =+H"  $!% +/15 Z+ Z+Z+ Z+  Z+ sm Z+ SM Z+ C=Z+ Z+#Z+ SMZ+c3h(Z+tCH~.Z+ 5/Z+ Z+D#" $,0"&&* $#!%%)f+ f+f+ f+ f+ sm f+ sm f+ f+()f+ DJ f+smf+ f+tnf+#f+c]f+  !f+\48 $#&*+/"&&*%)`+ `+`+ `+ `+ !c3h0 `+  `+tn`+sm`+T$Z(`+ DJ `+sm`+c]`+ `+r