From 5c6f4cb74a5f9d549df88c8a195fafac6aaaacd1 Mon Sep 17 00:00:00 2001 From: Kian Date: Sat, 4 Mar 2023 10:05:59 -0800 Subject: [PATCH 1/2] 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): From f1d102c66bc5fd4a714d3176cc836c4076b4fec8 Mon Sep 17 00:00:00 2001 From: Kian Date: Sat, 4 Mar 2023 18:14:17 -0800 Subject: [PATCH 2/2] update model names --- vocode/models/transcriber.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vocode/models/transcriber.py b/vocode/models/transcriber.py index afe8d6e..8f45793 100644 --- a/vocode/models/transcriber.py +++ b/vocode/models/transcriber.py @@ -13,9 +13,9 @@ class TranscriberType(str, Enum): class EndpointingType(str, Enum): - BASE = "base" - TIME_BASED = "time_based" - PUNCTUATION_BASED = "punctuation_based" + BASE = "endpointing_base" + TIME_BASED = "endpointing_time_based" + PUNCTUATION_BASED = "endpointing_punctuation_based" class EndpointingConfig(TypedModel, type=EndpointingType.BASE):