open source

vocode/streaming/transcriber/base_transcriber.py

@@ -0,0 +1,59 @@
+from dotenv import load_dotenv
+from typing import Callable, Optional, Awaitable
+
+from vocode.streaming.utils import convert_wav
+from vocode.streaming.models.transcriber import EndpointingConfig, TranscriberConfig
+
+load_dotenv()
+
+
+class Transcription:
+    def __init__(
+        self,
+        message: str,
+        confidence: float,
+        is_final: bool,
+        is_interrupt: bool = False,
+    ):
+        self.message = message
+        self.confidence = confidence
+        self.is_final = is_final
+        self.is_interrupt = is_interrupt
+
+    def __str__(self):
+        return f"Transcription({self.message}, {self.confidence}, {self.is_final})"
+
+
+class BaseTranscriber:
+    def __init__(
+        self,
+        transcriber_config: TranscriberConfig,
+    ):
+        self.transcriber_config = transcriber_config
+        self.on_response: Optional[Callable[[Transcription], Awaitable]] = None
+
+    def get_transcriber_config(self) -> TranscriberConfig:
+        return self.transcriber_config
+
+    def set_on_response(self, on_response: Callable[[Transcription], Awaitable]):
+        self.on_response = on_response
+
+    def get_warmup_bytes(self):
+        sampling_rate = self.transcriber_config.sampling_rate
+        return convert_wav(
+            "convo/audio/ajay.wav",
+            sampling_rate,
+            self.transcriber_config.audio_encoding,
+        )
+
+    async def ready(self):
+        return True
+
+    async def run(self):
+        pass
+
+    def send_audio(self, chunk):
+        pass
+
+    def terminate(self):
+        pass