o iD@sdZddlmZmZmZmZmZerddlmZddl m Z ddl m Z e r-ddlmZde de fdd Z d9d d de de de dee deddeede deedd fddZ   d:d d de de de dee deedee dee deddedeedd fdd Z ! d;d d de de d"ed#ee d$ee d%ee d&eed'ee d(ee d)eee efd*eee efdeedd fd+d,Z daFunctions 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_namereturncCsLddddddd}||vr||S|dr |dd }|S|}|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_MAPPINGSproviderr\/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/utils/tracing/service_attributes.py$_get_gen_ai_system_from_service_names  rNttsspanrmodelvoice_idtextsettingsrcharacter_countoperation_namettfbc Ks|d|dd|d||d||dd|d||r,|d ||d ur6|d ||d ur@|d ||r_|D]\} } t| tttt fr^|d | | qH| D]\} } t| tttt frv|| | qcd S)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.typespeechrr Nzmetrics.character_count metrics.ttfb settings. set_attributereplacer given_fieldsitems isinstancestrintfloatbool) rr rrr r!r"r#r$kwargskeyvaluerrradd_tts_span_attributesCs*        r;sttF transcriptis_finallanguageuser_id vad_enabledc Ks|d|dd|d||d||d| |r&|d||dur0|d ||r8|d ||r@|d || durJ|d | |ri|D]\} } t| tttt frh|d | | qR| D]\} } t| tttt fr|| | qmdS)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)rAr=Nr>r?r@r,r-r.)rr rr#r=r>r?r@r!rAr$r8r9r:rrradd_stt_span_attributeszs0         rCTstreammessagesoutputtools tool_count tool_choicesystem parametersextra_parametersc Ks||dt||d||dd|dd|d||r(|d||r0|d ||r8|d ||d urB|d ||rJ|d || rR|d| | d ur\|d| | r| D]$\}}t|ttttfr|dvr}|d||qb|d||qb| r| D]\}}t|ttttfr|d||q| D]\}}t|ttttfr|||qd S)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 rDinputrFrGNrHrIrJr,) temperature max_tokensmax_completion_tokenstop_ptop_kfrequency_penaltypresence_penaltyseedzgen_ai.request.zparam.zextra.)r/rr2r3r4r5r6r7)rr rrDrErFrGrHrIrJrKrLr$r8r9r:rrradd_llm_span_attributessH#             rW modalitiestools_serializedis_input text_outputaudio_data_sizecKs*|dd|d||d||d||r |d||r(|d||r0|d|| rB|d | | d urB|d | | rJ|d | | d urT|d | |r|dt||ddg}|D]'}t|tr{d|vr{||dqht|drt|dd }|d ur||qh|r|dd|| r|d| |r| D]N\}}t|t t t t fr|d||q|dkr|rt|dr|jd ur|d|jt|dr|jr|d|jjt|dr|jr|d|jjq| D]\}}t|t t t t fr|||qd S)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%r r(r)service.operationrr?rXr=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)r/lenr3dictappendhasattrgetattrjoinr1r2r4r5r6r7rgrhr:ri)rr rr#rr?rXr!rGrYr=rZr[r\r8 tool_namestool tool_namer9r:rrradd_gemini_live_span_attributessj %                 rssession_propertiescontext_messagesfunction_callsc Ks|dd|d||d||d||r*|d||dur*|d||r2|d || dur<|d | | r|d t| |d d g} | D]7}t|trcd|vrc| |dqPt|dro| |jqPt|trd|vrd|dvr| |ddqP| r|dd| | r|d| |r|dt||r|d}t|dr|d|jnt|trd|vr|d|d|r'|D]V\}}t|t t t t fr|d||q|dkr&|dur&t|t r|d|qt|tr&|dd |D]\}}t|t t t t fr$|d||q q| D]\}}t|t t t t fr?|||q+dS)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)r]r=Nr^rNr_r`raTrbfunctionrcrdrezfunction_calls.countrzfunction_calls.first_namezsession.turn_detectionzsession.turn_detection.enabledzsession.turn_detection.) r/rjr3rkrlrmrbror2r4r5r6r7)rr rr#rtr=rZrurvrGrYr\r8rprqcallr9r:td_keytd_valuerrr#add_openai_realtime_span_attributessl !               r})NNNrN)r<NNNNNFN) TNNNNNNNNN) NNNNNNNNNN)NNNNNNNN)__doc__typingrrrrropentelemetry.tracerpipecat.services.settingsrpipecat.utils.tracing.setupr r4rr5r6r;r7rCrWrsr}rrrrsn    ,  ;    D      b      n