From 40a94f94be091f36de58fe0ffb51a84eca27183e Mon Sep 17 00:00:00 2001 From: Ajay Raj Date: Thu, 2 Mar 2023 16:15:22 -0800 Subject: [PATCH] fix user implemented agent types --- user_implemented_agent.py | 4 ++-- vocode/models/agent.py | 10 ++++++---- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/user_implemented_agent.py b/user_implemented_agent.py index 65a35c6..9db9ac1 100644 --- a/user_implemented_agent.py +++ b/user_implemented_agent.py @@ -1,11 +1,11 @@ from vocode.user_implemented_agent.restful_agent import RESTfulAgent from vocode.user_implemented_agent.websocket_agent import WebSocketAgent -class EchoAgent(RESTfulAgent): +class EchoAgent(WebSocketAgent): async def respond(self, input: str) -> str: return input if __name__ == "__main__": agent = EchoAgent() - agent.run() \ No newline at end of file + agent.run() diff --git a/vocode/models/agent.py b/vocode/models/agent.py index 0b9267a..2f573f5 100644 --- a/vocode/models/agent.py +++ b/vocode/models/agent.py @@ -44,8 +44,9 @@ class RESTfulUserImplementedAgentConfig(AgentConfig, type=AgentType.RESTFUL_USER method: str = "POST" respond: EndpointConfig - generate_response: Optional[EndpointConfig] - update_last_bot_message_on_cut_off: Optional[EndpointConfig] + generate_responses: bool = False + # generate_response: Optional[EndpointConfig] + # update_last_bot_message_on_cut_off: Optional[EndpointConfig] class RESTfulAgentInput(BaseModel): human_input: str @@ -58,8 +59,9 @@ class WebSocketUserImplementedAgentConfig(AgentConfig, type=AgentType.WEBSOCKET_ url: str respond: RouteConfig - generate_response: Optional[RouteConfig] - send_message_on_cut_off: bool = False + generate_responses: bool = False + # generate_response: Optional[RouteConfig] + # send_message_on_cut_off: bool = False class WebSocketAgentMessageType(str, Enum): AGENT_BASE = 'agent_base'