Annotate NewMessage.promptState as Optional

NeonGeckoCom · Jan 9, 2025 · 2f03c80 · 2f03c80
1 parent 08269a8
commit 2f03c80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neon_data_models/models/api/klat/socketio.py b/neon_data_models/models/api/klat/socketio.py
@@ -58,7 +58,7 @@ class NewMessage(BaseModel):
     userID: str = Field(description="User ID associated with the message")
     prompt_id: Optional[str] = Field(
         default=None, description="Prompt ID this message is in response to")
-    promptState: CcaiPromptStates = Field(
+    promptState: Optional[CcaiPromptStates] = Field(
         default=None,
         description="Associated CCAI state if `prompt_id` is defined")
     source: str = Field(description="Username associated with the message")