Update type_system.py

SLR722 · Jan 13, 2025 · 9b4d330 · 9b4d330
1 parent e8f905d
commit 9b4d330
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/llama_stack/apis/common/type_system.py b/llama_stack/apis/common/type_system.py
@@ -53,6 +53,10 @@ class AgentTurnInputType(BaseModel):
     # expects List[Message] for messages (may also include attachments?)
     type: Literal["agent_turn_input"] = "agent_turn_input"
 
+class DialogType(BaseModel):
+    # expects List[Message] for messages
+    type: Literal["dialog"] = "dialog"
+
 
 ParamType = register_schema(
     Annotated[