conversation ID tracking

vocode/models/agent.py

@@ -56,6 +56,7 @@ class RESTfulUserImplementedAgentConfig(AgentConfig, type=AgentType.RESTFUL_USER
     # update_last_bot_message_on_cut_off: Optional[EndpointConfig]
 
 class RESTfulAgentInput(BaseModel):
+    conversation_id: str
     human_input: str
 
 class RESTfulAgentOutputType(str, Enum):
@@ -88,7 +89,8 @@ class WebSocketAgentMessageType(str, Enum):
     READY = 'websocket_agent_ready'
     STOP = 'websocket_agent_stop'
 
-class WebSocketAgentMessage(TypedModel, type=WebSocketAgentMessageType.BASE): pass
+class WebSocketAgentMessage(TypedModel, type=WebSocketAgentMessageType.BASE):
+    conversation_id: Optional[str] = None
 
 class WebSocketAgentTextMessage(WebSocketAgentMessage, type=WebSocketAgentMessageType.TEXT):
     class Payload(BaseModel):
@@ -97,8 +99,8 @@ class WebSocketAgentTextMessage(WebSocketAgentMessage, type=WebSocketAgentMessag
     data: Payload
 
     @classmethod
-    def from_text(cls, text: str):
-        return cls(data=cls.Payload(text=text))
+    def from_text(cls, text: str, conversation_id: Optional[str] = None):
+        return cls(data=cls.Payload(text=text), conversation_id=conversation_id)
 
 
 class WebSocketAgentStartMessage(WebSocketAgentMessage, type=WebSocketAgentMessageType.START):

vocode/models/telephony.py

@@ -11,6 +11,7 @@ class CallEntity(BaseModel):
 class CreateInboundCall(BaseModel):
     agent_config: AgentConfig
     twilio_sid: str
+    conversation_id: Optional[str] = None
 
 
 class CreateOutboundCall(BaseModel):
@@ -18,4 +19,5 @@ class CreateOutboundCall(BaseModel):
     caller: CallEntity
     agent_config: AgentConfig
     synthesizer_config: Optional[SynthesizerConfig] = None
+    conversation_id: Optional[str] = None
     # TODO add IVR/etc.

vocode/models/websocket.py

@@ -1,5 +1,6 @@
 import base64
 from enum import Enum
+from typing import Optional
 from .model import TypedModel
 from .transcriber import TranscriberConfig
 from .agent import AgentConfig
@@ -28,6 +29,7 @@ class StartMessage(WebSocketMessage, type=WebSocketMessageType.START):
     transcriber_config: TranscriberConfig
     agent_config: AgentConfig
     synthesizer_config: SynthesizerConfig
+    conversation_id: Optional[str] = None
 
 class ReadyMessage(WebSocketMessage, type=WebSocketMessageType.READY):
     pass