revert

letta-ai · Dec 11, 2024 · 277db12 · 277db12
1 parent 2bce03f
commit 277db12
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/letta/schemas/message.py b/letta/schemas/message.py
@@ -58,7 +58,10 @@ class MessageCreate(BaseModel):
     """Request to create a message"""
 
     # In the simplified format, only allow simple roles
-    role: MessageRole = Field(..., description="The role of the participant.")
+    role: Literal[
+        MessageRole.user,
+        MessageRole.system,
+    ] = Field(..., description="The role of the participant.")
     text: str = Field(..., description="The text of the message.")
     name: Optional[str] = Field(None, description="The name of the participant.")