diff --git a/api/models/enums.py b/api/models/enums.py
index cc9f28a7bb..bc66b1bb56 100644
--- a/api/models/enums.py
+++ b/api/models/enums.py
@@ -21,6 +21,7 @@ class DraftVariableType(StrEnum):
     NODE = "node"
     SYS = "sys"
     CONVERSATION = "conversation"
+    MEMORY_BLOCK = "memory_block"
 
 
 class MessageStatus(StrEnum):