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

adds ability to pass in api key for every transcriber agent and synthesizer

Browse source
This commit is contained in:
Ajay Raj 2023-03-28 23:33:36 -07:00
commit ecebe4c1a5
11 changed files with 71 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

7
vocode/streaming/transcriber/assembly_ai_transcriber.py
View file

@ -22,9 +22,14 @@ class AssemblyAITranscriber(BaseTranscriber):
self,
transcriber_config: AssemblyAITranscriberConfig,
logger: logging.Logger = None,
api_key: str = None,
):
super().__init__(transcriber_config)
self.api_key = getenv("ASSEMBLY_AI_API_KEY")
self.api_key = api_key or getenv("ASSEMBLY_AI_API_KEY")
if not self.api_key:
raise Exception(
"Please set ASSEMBLY_AI_API_KEY environment variable or pass it as a parameter"
)
self._ended = False
self.is_ready = False
self.logger = logger or logging.getLogger(__name__)

7
vocode/streaming/transcriber/deepgram_transcriber.py
View file

@ -28,9 +28,14 @@ class DeepgramTranscriber(BaseTranscriber):
self,
transcriber_config: DeepgramTranscriberConfig,
logger: logging.Logger = None,
api_key: str = None,
):
super().__init__(transcriber_config)
self.api_key = getenv("DEEPGRAM_API_KEY")
self.api_key = api_key or getenv("DEEPGRAM_API_KEY")
if not self.api_key:
raise Exception(
"Please set DEEPGRAM_API_KEY environment variable or pass it as a parameter"
)
self.transcriber_config = transcriber_config
self._ended = False
self.warmed_up = False

5
vocode/streaming/transcriber/google_transcriber.py
View file

@ -3,6 +3,7 @@ import time
import queue
from google.cloud import speech
import threading
from vocode import getenv
from vocode.streaming.models.audio_encoding import AudioEncoding
from vocode.streaming.transcriber.base_transcriber import (
@ -18,6 +19,10 @@ class GoogleTranscriber(BaseTranscriber):
super().__init__(transcriber_config)
self._queue = queue.Queue()
self._ended = False
if not getenv("GOOGLE_APPLICATION_CREDENTIALS"):
raise Exception(
"Please set GOOGLE_APPLICATION_CREDENTIALS environment variable"
)
self.google_streaming_config = self.create_google_streaming_config()
self.client = speech.SpeechClient()
self.warmed_up = False