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

update models

Browse source
This commit is contained in:
Kian 2023-03-03 10:49:07 -08:00
commit 274c1e3acd
1 changed files with 15 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
vocode/models/transcriber.py
View file

@ -12,17 +12,31 @@ class TranscriberType(str, Enum):
ASSEMBLY_AI = "assembly_ai" ASSEMBLY_AI = "assembly_ai"
class EndpointingType(str, Enum):
BASE = "base"
class EndpointingConfig(TypedModel, type=EndpointingType.BASE):
time_cutoff: int
class TranscriberConfig(TypedModel, type=TranscriberType.BASE): class TranscriberConfig(TypedModel, type=TranscriberType.BASE):
sampling_rate: int sampling_rate: int
audio_encoding: AudioEncoding audio_encoding: AudioEncoding
chunk_size: int chunk_size: int
endpointing_config: Optional[EndpointingConfig] = None
@classmethod @classmethod
def from_input_device(cls, input_device: BaseInputDevice): def from_input_device(
cls,
input_device: BaseInputDevice,
endpointing_config: Optional[EndpointingConfig] = None,
):
return cls( return cls(
sampling_rate=input_device.sampling_rate, sampling_rate=input_device.sampling_rate,
audio_encoding=input_device.audio_encoding, audio_encoding=input_device.audio_encoding,
chunk_size=input_device.chunk_size, chunk_size=input_device.chunk_size,
endpointing_config=endpointing_config,
) )