qijdZddlZddlmZddlmZdedeefdZdedeefdZdedeefd Z y) zGUtility functions for extracting probability metrics from STT services.N)Optional)TranscriptionFrameframereturnc|jsyt|jdrS|jjr=|jjd}t|dd}|t j |Sy)aExtract probability from Whisper-based TranscriptionFrame result. Works with Groq, OpenAI Whisper, or other Whisper-based services that use verbose_json format with segments containing avg_logprob. Converts avg_logprob to probability. Args: frame: TranscriptionFrame with result from GroqSTTService or OpenAISTTService (when include_prob_metrics=True and using Whisper models). Returns: Probability (0-1) if available, None otherwise. Example:: from pipecat.services.groq.stt import GroqSTTService from pipecat.services.whisper.utils import extract_whisper_probability stt = GroqSTTService(include_prob_metrics=True) # ... use stt in pipeline ... # In your frame processor: if isinstance(frame, TranscriptionFrame): prob = extract_whisper_probability(frame) if prob: print(f"Transcription confidence: {prob:.2%}") Nsegmentsr avg_logprob)resulthasattrrgetattrmathexp)rsegmentr s P/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/whisper/utils.pyextract_whisper_probabilityrsd8 <<u||Z(U\\-B-B,,''*g}d;  "88K( ( c|jsyt|jdrD|jj}|r,t|t |z }t j |Sy)a7Extract probability from OpenAI GPT-4o-transcribe TranscriptionFrame result. Args: frame: TranscriptionFrame with result from OpenAISTTService using GPT-4o-transcribe model (when include_prob_metrics=True). Returns: Probability (0-1) if available, None otherwise. Example:: from pipecat.services.openai.stt import OpenAISTTService from pipecat.services.whisper.utils import extract_openai_gpt4o_probability stt = OpenAISTTService(model="gpt-4o-transcribe", include_prob_metrics=True) # ... use stt in pipeline ... # In your frame processor: if isinstance(frame, TranscriptionFrame): prob = extract_openai_gpt4o_probability(frame) if prob: print(f"Transcription confidence: {prob:.2%}") Nlogprobs)r r rsumlenr r)rrr s r extract_openai_gpt4o_probabilityr8sT. <<u||Z(<<(( h-#h-7K88K( ( rc|jsy|j}t|dr|jrt|jdr|jjr|jjd}t |dd}| t |St |dd}|rN|Dcgc]}t |dd}}|Dcgc]}|| }}|r t t |t|z Sycc}wcc}w)a0Extract probability from Deepgram TranscriptionFrame result. Args: frame: TranscriptionFrame with result from DeepgramSTTService. Returns: Probability (0-1) if available, None otherwise. Returns alternative-level confidence if available, otherwise calculates average confidence from word-level confidences. Example:: from pipecat.services.deepgram.stt import DeepgramSTTService from pipecat.services.whisper.utils import extract_deepgram_probability stt = DeepgramSTTService() # ... use stt in pipeline ... # In your frame processor: if isinstance(frame, TranscriptionFrame): prob = extract_deepgram_probability(frame) if prob: print(f"Transcription confidence: {prob:.2%}") Nchannel alternativesr confidencewords)r r rrr floatrr)rr altconfrw word_confscs rextract_deepgram_probabilityr#]s0 << \\Fvy!fnn 6>>> 2v~~7R7R..--a0C3 d3DT{"C$/EFKLgat<L L)3EAq}aE E Z3z?!BCC  MEs$C/>C4C4) __doc__r typingrpipecat.frames.framesrrrrr#rrr(s^N 4&'9&huo&R",>"8E?"J*(:*x*r