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

address comments

Browse source
This commit is contained in:
Kian 2023-03-03 12:11:53 -08:00
commit 69caa03c8d
2 changed files with 15 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

26
vocode/conversation.py
View file

@ -16,17 +16,17 @@ from .models.synthesizer import SynthesizerConfig
from .models.websocket import ReadyMessage, AudioMessage, StartMessage, StopMessage from .models.websocket import ReadyMessage, AudioMessage, StartMessage, StopMessage
from . import api_key from . import api_key
VOCODE_WEBSOCKET_URL = f'wss://api.vocode.dev/conversation' VOCODE_WEBSOCKET_URL = f"wss://3fcd-136-24-82-111.ngrok.io/conversation"
class Conversation: class Conversation:
def __init__( def __init__(
self, self,
input_device: BaseInputDevice, input_device: BaseInputDevice,
output_device: BaseOutputDevice, output_device: BaseOutputDevice,
transcriber_config: TranscriberConfig, transcriber_config: TranscriberConfig,
agent_config: AgentConfig, agent_config: AgentConfig,
synthesizer_config: SynthesizerConfig synthesizer_config: SynthesizerConfig,
): ):
self.input_device = input_device self.input_device = input_device
self.output_device = output_device self.output_device = output_device
@ -43,7 +43,7 @@ class Conversation:
while not self.receiver_ready: while not self.receiver_ready:
await asyncio.sleep(0.1) await asyncio.sleep(0.1)
return True return True
def deactivate(self): def deactivate(self):
self.active = False self.active = False
@ -55,16 +55,18 @@ class Conversation:
await self.output_device.send_async(audio) await self.output_device.send_async(audio)
except queue.Empty: except queue.Empty:
continue continue
loop = asyncio.new_event_loop() loop = asyncio.new_event_loop()
loop.run_until_complete(run()) loop.run_until_complete(run())
async def start(self): async def start(self):
async with websockets.connect(f"{VOCODE_WEBSOCKET_URL}?key={api_key}") as ws: async with websockets.connect(f"{VOCODE_WEBSOCKET_URL}?key={api_key}") as ws:
async def sender(ws): async def sender(ws):
start_message = StartMessage( start_message = StartMessage(
transcriber_config=self.transcriber_config, transcriber_config=self.transcriber_config,
agent_config=self.agent_config, agent_config=self.agent_config,
synthesizer_config=self.synthesizer_config synthesizer_config=self.synthesizer_config,
) )
await ws.send(start_message.json()) await ws.send(start_message.json())
await self.wait_for_ready() await self.wait_for_ready()
@ -83,8 +85,6 @@ class Conversation:
audio_message = AudioMessage.parse_raw(msg) audio_message = AudioMessage.parse_raw(msg)
self.output_audio_queue.put_nowait(audio_message.get_bytes()) self.output_audio_queue.put_nowait(audio_message.get_bytes())
output_thread = threading.Thread(target=self.play_audio) output_thread = threading.Thread(target=self.play_audio)
output_thread.start() output_thread.start()
return await asyncio.gather(sender(ws), receiver(ws)) return await asyncio.gather(sender(ws), receiver(ws))

8
vocode/models/transcriber.py
View file

@ -12,12 +12,8 @@ class TranscriberType(str, Enum):
ASSEMBLY_AI = "assembly_ai" ASSEMBLY_AI = "assembly_ai"
class EndpointingType(str, Enum): class EndpointingConfig(TypedModel):
BASE = "base" time_cutoff_seconds: float
class EndpointingConfig(TypedModel, type=EndpointingType.BASE):
time_cutoff: int
class TranscriberConfig(TypedModel, type=TranscriberType.BASE): class TranscriberConfig(TypedModel, type=TranscriberType.BASE):