From 5c6f4cb74a5f9d549df88c8a195fafac6aaaacd1 Mon Sep 17 00:00:00 2001 From: Kian Date: Sat, 4 Mar 2023 10:05:59 -0800 Subject: [PATCH] add punctuation based endpointing --- vocode/models/transcriber.py | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/vocode/models/transcriber.py b/vocode/models/transcriber.py index 9b684df..afe8d6e 100644 --- a/vocode/models/transcriber.py +++ b/vocode/models/transcriber.py @@ -12,8 +12,24 @@ class TranscriberType(str, Enum): ASSEMBLY_AI = "assembly_ai" -class EndpointingConfig(BaseModel): - time_cutoff_seconds: Optional[float] = None +class EndpointingType(str, Enum): + 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):