vocode-python/simple_conversation.py
2023-03-14 00:32:20 -07:00

59 lines
1.7 KiB
Python

import asyncio
import logging
import signal
from dotenv import load_dotenv
import os
load_dotenv()
import vocode
vocode.api_key = os.getenv("VOCODE_API_KEY")
from vocode.conversation import Conversation
from vocode.helpers import create_microphone_input_and_speaker_output
from vocode.models.transcriber import (
DeepgramTranscriberConfig,
PunctuationEndpointingConfig,
GoogleTranscriberConfig,
)
from vocode.models.agent import (
ChatGPTAgentConfig,
FillerAudioConfig,
RESTfulUserImplementedAgentConfig,
WebSocketUserImplementedAgentConfig,
EchoAgentConfig,
ChatGPTAlphaAgentConfig,
LLMAgentConfig,
ChatGPTAgentConfig,
)
from vocode.models.message import BaseMessage
from vocode.models.synthesizer import AzureSynthesizerConfig
from vocode.user_implemented_agent.restful_agent import RESTfulAgent
logging.basicConfig()
logging.root.setLevel(logging.INFO)
if __name__ == "__main__":
microphone_input, speaker_output = create_microphone_input_and_speaker_output(
use_default_devices=False
)
conversation = Conversation(
input_device=microphone_input,
output_device=speaker_output,
transcriber_config=DeepgramTranscriberConfig.from_input_device(
microphone_input
),
agent_config=WebSocketUserImplementedAgentConfig(
initial_message=BaseMessage(text="Hello!"),
generate_responses=True,
respond=WebSocketUserImplementedAgentConfig.RouteConfig(
url="wss://9b1ff0eee874.ngrok.app/respond",
),
),
synthesizer_config=AzureSynthesizerConfig.from_output_device(speaker_output),
)
signal.signal(signal.SIGINT, lambda _0, _1: conversation.deactivate())
asyncio.run(conversation.start())