jyapayne/vocode-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add assembly ai integration

Browse source
This commit is contained in:
Kian 2023-02-27 10:19:44 -08:00
commit f39af91505
4 changed files with 14 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
vocode/models/transcriber.py
View file

@ -8,6 +8,7 @@ class TranscriberType(str, Enum):
BASE = "base"
DEEPGRAM = "deepgram"
GOOGLE = "google"
ASSEMBLY_AI = "assembly_ai"
class TranscriberConfig(TypedModel, type=TranscriberType.BASE):
sampling_rate: int
@ -28,4 +29,9 @@ class DeepgramTranscriberConfig(TranscriberConfig, type=TranscriberType.DEEPGRAM
class GoogleTranscriberConfig(TranscriberConfig, type=TranscriberType.GOOGLE):
model: Optional[str] = None
should_warmup_model: bool = False
should_warmup_model: bool = False
class AssemblyAITranscriberConfig(TranscriberConfig, type=TranscriberType.ASSEMBLY_AI):
model: Optional[str] = None
should_warmup_model: bool = False
version: Optional[str] = None