wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update base_stt

Browse Source
This commit is contained in:
Thu Nguyen
2025-11-05 13:26:59 +07:00
parent 0b779a880b
commit 842c4a3485
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/pipecat/services/whisper/base_stt.py
View File

@@ -122,6 +122,7 @@ class BaseWhisperSTTService(SegmentedSTTService):
language: Optional[Language] = Language.EN, language: Optional[Language] = Language.EN,
prompt: Optional[str] = None, prompt: Optional[str] = None,
temperature: Optional[float] = None, temperature: Optional[float] = None,
include_prob_metrics: bool = False,
**kwargs, **kwargs,
): ):
"""Initialize the Whisper STT service. """Initialize the Whisper STT service.
@@ -133,6 +134,9 @@ class BaseWhisperSTTService(SegmentedSTTService):
language: Language of the audio input. Defaults to English. language: Language of the audio input. Defaults to English.
prompt: Optional text to guide the model's style or continue a previous segment. prompt: Optional text to guide the model's style or continue a previous segment.
temperature: Sampling temperature between 0 and 1. Defaults to 0.0. temperature: Sampling temperature between 0 and 1. Defaults to 0.0.
include_prob_metrics: If True, enables probability metrics in API response.
Each service implements this differently (see child classes).
Defaults to False.
**kwargs: Additional arguments passed to SegmentedSTTService. **kwargs: Additional arguments passed to SegmentedSTTService.
""" """
super().__init__(**kwargs) super().__init__(**kwargs)
@@ -141,6 +145,7 @@ class BaseWhisperSTTService(SegmentedSTTService):
self._language = self.language_to_service_language(language or Language.EN) self._language = self.language_to_service_language(language or Language.EN)
self._prompt = prompt self._prompt = prompt
self._temperature = temperature self._temperature = temperature
self._include_prob_metrics = include_prob_metrics
self._settings = { self._settings = {
"base_url": base_url, "base_url": base_url,
@@ -223,6 +228,7 @@ class BaseWhisperSTTService(SegmentedSTTService):
text, text,
self._user_id, self._user_id,
time_now_iso8601(), time_now_iso8601(),
result=response,
) )
else: else:
logger.warning("Received empty transcription from API") logger.warning("Received empty transcription from API")