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

feat(assemblyai): add domain parameter for Medical Mode

Browse Source 
Add `domain` field to AssemblyAISTTSettings to support AssemblyAI's
streaming API `domain` query parameter, enabling specialized recognition
modes like Medical Mode (`medical-v1`).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Martin Schweiger
2026-03-24 13:09:42 +08:00
parent 7414b30308
commit f21b262969
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/pipecat/services/assemblyai/stt.py
View File

@@ -101,6 +101,8 @@ class AssemblyAISTTSettings(STTSettings):
speaker_labels: Enable speaker diarization. speaker_labels: Enable speaker diarization.
vad_threshold: VAD confidence threshold (0.0–1.0) for classifying vad_threshold: VAD confidence threshold (0.0–1.0) for classifying
audio frames as silence. Only applicable to u3-rt-pro. audio frames as silence. Only applicable to u3-rt-pro.
domain: Optional domain for specialized recognition modes. For example,
set to "medical-v1" to enable Medical Mode for healthcare transcription.
""" """
formatted_finals: bool | _NotGiven = field(default_factory=lambda: NOT_GIVEN) formatted_finals: bool | _NotGiven = field(default_factory=lambda: NOT_GIVEN)
@@ -118,6 +120,7 @@ class AssemblyAISTTSettings(STTSettings):
format_turns: bool | _NotGiven = field(default_factory=lambda: NOT_GIVEN) format_turns: bool | _NotGiven = field(default_factory=lambda: NOT_GIVEN)
speaker_labels: bool | None | _NotGiven = field(default_factory=lambda: NOT_GIVEN) speaker_labels: bool | None | _NotGiven = field(default_factory=lambda: NOT_GIVEN)
vad_threshold: float | None | _NotGiven = field(default_factory=lambda: NOT_GIVEN) vad_threshold: float | None | _NotGiven = field(default_factory=lambda: NOT_GIVEN)
domain: str | None | _NotGiven = field(default_factory=lambda: NOT_GIVEN)
class AssemblyAISTTService(WebsocketSTTService): class AssemblyAISTTService(WebsocketSTTService):
@@ -204,6 +207,7 @@ class AssemblyAISTTService(WebsocketSTTService):
format_turns=True, format_turns=True,
speaker_labels=None, speaker_labels=None,
vad_threshold=None, vad_threshold=None,
domain=None,
) )
# 2. Apply direct init arg overrides (deprecated) # 2. Apply direct init arg overrides (deprecated)
@@ -470,6 +474,7 @@ class AssemblyAISTTService(WebsocketSTTService):
"format_turns": s.format_turns, "format_turns": s.format_turns,
"speaker_labels": s.speaker_labels, "speaker_labels": s.speaker_labels,
"vad_threshold": s.vad_threshold, "vad_threshold": s.vad_threshold,
"domain": s.domain,
} }
for k, v in optional_fields.items(): for k, v in optional_fields.items():