From 7c043ec800a923c00e51a642e149ad33220bad50 Mon Sep 17 00:00:00 2001 From: Kian Date: Wed, 8 Mar 2023 22:04:18 -0800 Subject: [PATCH] add message types --- vocode/models/agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vocode/models/agent.py b/vocode/models/agent.py index 4cf1487..b45082d 100644 --- a/vocode/models/agent.py +++ b/vocode/models/agent.py @@ -1,5 +1,7 @@ from typing import Optional from enum import Enum + +from vocode.models.message import BaseMessage from .model import TypedModel, BaseModel @@ -15,7 +17,7 @@ class AgentType(str, Enum): class AgentConfig(TypedModel, type=AgentType.BASE): - initial_message: Optional[str] = None + initial_message: Optional[BaseMessage] = None generate_responses: bool = True allowed_idle_time_seconds: Optional[float] = None end_conversation_on_goodbye: bool = False