checkpoint
This commit is contained in:
parent
94f94cbebf
commit
d9f12ec0de
5 changed files with 63 additions and 6 deletions
|
|
@ -5,25 +5,40 @@ import signal
|
|||
from vocode.conversation import Conversation
|
||||
from vocode.helpers import create_microphone_input_and_speaker_output
|
||||
from vocode.models.transcriber import DeepgramTranscriberConfig
|
||||
from vocode.models.agent import ChatGPTAgentConfig
|
||||
from vocode.models.agent import ChatGPTAgentConfig, RESTfulUserImplementedAgentConfig
|
||||
from vocode.models.synthesizer import AzureSynthesizerConfig
|
||||
from vocode.user_implemented_agent.restful_agent import RESTfulAgent
|
||||
|
||||
logging.basicConfig()
|
||||
logging.root.setLevel(logging.INFO)
|
||||
|
||||
class EchoAgent(RESTfulAgent):
|
||||
|
||||
async def respond(self, input: str) -> str:
|
||||
return input
|
||||
|
||||
if __name__ == "__main__":
|
||||
import threading
|
||||
|
||||
microphone_input, speaker_output = create_microphone_input_and_speaker_output(use_first_available_device=False)
|
||||
user_agent_thread = threading.Thread(target=EchoAgent().run)
|
||||
user_agent_thread.start()
|
||||
|
||||
conversation = Conversation(
|
||||
input_device=microphone_input,
|
||||
output_device=speaker_output,
|
||||
transcriber_config=DeepgramTranscriberConfig.from_input_device(microphone_input),
|
||||
agent_config=ChatGPTAgentConfig(
|
||||
agent_config=RESTfulUserImplementedAgentConfig(
|
||||
initial_message="Hello!",
|
||||
prompt_preamble="The AI is having a pleasant conversation about life."
|
||||
generate_responses=False,
|
||||
respond=RESTfulUserImplementedAgentConfig.EndpointConfig(
|
||||
url="http://localhost:3001/respond",
|
||||
method="POST"
|
||||
)
|
||||
),
|
||||
synthesizer_config=AzureSynthesizerConfig.from_output_device(speaker_output)
|
||||
)
|
||||
signal.signal(signal.SIGINT, lambda _0, _1: conversation.deactivate())
|
||||
asyncio.run(conversation.start())
|
||||
user_agent_thread.join()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue