o ÿiã@sjdZddlZddlmZddlmZdedeefdd„Zdedeefd d „Zdedeefd d „Z dS) zGUtility functions for extracting probability metrics from STT services.éN)ÚOptional)ÚTranscriptionFrameÚframeÚreturncCsL|jsdSt|jdƒr$|jjr$|jjd}t|ddƒ}|dur$t |¡SdS)aÅExtract 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%}") NÚsegmentsrÚ avg_logprob)ÚresultÚhasattrrÚgetattrÚmathÚexp)rÚsegmentr©rúR/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/whisper/utils.pyÚextract_whisper_probabilitys   rcCs@|jsdSt|jdƒr|jj}|rt|ƒt|ƒ}t |¡SdS)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%}") NÚlogprobs)rr rÚsumÚlenr r )rrrrrrÚ extract_openai_gpt4o_probability8s  rcCs¤|jsdS|j}t|dƒrP|jrPt|jdƒrP|jjrP|jjd}t|ddƒ}|dur.t|ƒSt|ddƒ}|rPdd„|Dƒ}d d„|Dƒ}|rPtt|ƒt|ƒƒSdS) 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%}") NÚchannelÚ alternativesrÚ confidenceÚwordscSsg|]}t|ddƒ‘qS)rN)r )Ú.0ÚwrrrÚ ‚óz0extract_deepgram_probability..cSsg|]}|dur|‘qS)Nr)rÚcrrrrƒr)rr rrr Úfloatrr)rrÚaltÚconfrÚ word_confsrrrÚextract_deepgram_probability]s    r") Ú__doc__r ÚtypingrÚpipecat.frames.framesrrrrr"rrrrÚs  )%