checkpoint

vocode/conversation.py

@@ -16,7 +16,7 @@ from .models.synthesizer import SynthesizerConfig
 from .models.websocket import ReadyMessage, AudioMessage, StartMessage, StopMessage
 from . import api_key
 
-VOCODE_WEBSOCKET_URL = f'wss://api.vocode.dev/conversation'
+VOCODE_WEBSOCKET_URL = f'wss://a6eb64f4a9b7.ngrok.io/conversation'
 
 class Conversation:
 

vocode/models/agent.py

@@ -1,6 +1,6 @@
 from typing import Optional
 from enum import Enum
-from .model import TypedModel
+from .model import TypedModel, BaseModel
 
 
 class AgentType(str, Enum):
@@ -9,11 +9,12 @@ class AgentType(str, Enum):
     CHAT_GPT = "chat_gpt"
     ECHO = "echo"
     INFORMATION_RETRIEVAL = "information_retrieval"
+    RESTFUL_USER_IMPLEMENTED = "restful_user_implemented"
 
 
 class AgentConfig(TypedModel, type=AgentType.BASE):
     initial_message: Optional[str] = None
-
+    generate_responses: bool = True
 
 class LLMAgentConfig(AgentConfig, type=AgentType.LLM):
     prompt_preamble: str
@@ -35,3 +36,14 @@ class InformationRetrievalAgentConfig(
 
 class EchoAgentConfig(AgentConfig, type=AgentType.ECHO):
     pass
+
+class RESTfulUserImplementedAgentConfig(AgentConfig, type=AgentType.RESTFUL_USER_IMPLEMENTED):
+    class EndpointConfig(BaseModel):
+        url: str
+        method: str = "POST"
+        input_param_name: str = "human_input"
+        output_jsonpath: str = "response"
+
+    respond: EndpointConfig
+    generate_response: Optional[EndpointConfig]
+    update_last_bot_message_on_cut_off: Optional[EndpointConfig]

vocode/user_implemented_agent/base_agent.py

@@ -0,0 +1,13 @@
+from fastapi import FastAPI, APIRouter
+import uvicorn
+
+class BaseAgent():
+
+    def __init__(self):
+        self.app = FastAPI()
+
+    async def respond(self, human_input) -> str:
+        raise NotImplementedError
+    
+    def run(self, host="localhost", port=3001):
+        uvicorn.run(self.app, host=host, port=port)

vocode/user_implemented_agent/restful_agent.py

@@ -0,0 +1,17 @@
+from .base_agent import BaseAgent
+from pydantic import BaseModel
+from fastapi import APIRouter
+
+class RESTfulAgent(BaseAgent):
+
+    class HumanInput(BaseModel):
+        human_input: str
+        
+    def __init__(self):
+        super().__init__()
+        self.app.post("/respond")(self.respond_rest)
+
+    async def respond_rest(self, request: HumanInput):
+        response = await self.respond(request.human_input)
+        return {"response": response}
+