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

add punctuation based endpointing

Browse source
This commit is contained in:
Kian 2023-03-04 10:05:59 -08:00
commit 5c6f4cb74a
1 changed files with 18 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

20
vocode/models/transcriber.py
View file

@ -12,8 +12,24 @@ class TranscriberType(str, Enum):
ASSEMBLY_AI = "assembly_ai" ASSEMBLY_AI = "assembly_ai"
class EndpointingConfig(BaseModel): class EndpointingType(str, Enum):
time_cutoff_seconds: Optional[float] = None BASE = "base"
TIME_BASED = "time_based"
PUNCTUATION_BASED = "punctuation_based"
class EndpointingConfig(TypedModel, type=EndpointingType.BASE):
pass
class TimeEndpointingConfig(EndpointingConfig, type=EndpointingType.TIME_BASED):
time_cutoff_seconds: float = 0.4
class PunctuationEndpointingConfig(
EndpointingConfig, type=EndpointingType.PUNCTUATION_BASED
):
time_cutoff_seconds: float = 0.4
class TranscriberConfig(TypedModel, type=TranscriberType.BASE): class TranscriberConfig(TypedModel, type=TranscriberType.BASE):