From 19f9ae6aebc61618727ef0556a25216b37923508 Mon Sep 17 00:00:00 2001
From: QuantumGhost <obelisk.reg+git@gmail.com>
Date: Mon, 19 May 2025 19:57:44 +0800
Subject: [PATCH] refactor(api): Use UPPER_SNAKE_CASE name for enum members

Change the enum member names for `DraftVariableType`.
---
 api/models/enums.py    |  6 +++---
 api/models/workflow.py | 12 ++++++------
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/api/models/enums.py b/api/models/enums.py
index c055a45bb4..4434c3fec8 100644
--- a/api/models/enums.py
+++ b/api/models/enums.py
@@ -18,6 +18,6 @@ class WorkflowRunTriggeredFrom(StrEnum):
 
 class DraftVariableType(StrEnum):
     # node means that the correspond variable
-    node = "node"
-    sys = "sys"
-    conversation = "conversation"
+    NODE = "node"
+    SYS = "sys"
+    CONVERSATION = "conversation"
diff --git a/api/models/workflow.py b/api/models/workflow.py
index c01bf1ad06..12e0f347f1 100644
--- a/api/models/workflow.py
+++ b/api/models/workflow.py
@@ -918,19 +918,19 @@ class WorkflowDraftVariable(Base):
         self.value_type = value.value_type
 
     def get_node_id(self) -> str | None:
-        if self.get_variable_type() == DraftVariableType.node:
+        if self.get_variable_type() == DraftVariableType.NODE:
             return self.node_id
         else:
             return None
 
     def get_variable_type(self) -> DraftVariableType:
         match self.node_id:
-            case DraftVariableType.conversation:
-                return DraftVariableType.conversation
-            case DraftVariableType.sys:
-                return DraftVariableType.sys
+            case DraftVariableType.CONVERSATION:
+                return DraftVariableType.CONVERSATION
+            case DraftVariableType.SYS:
+                return DraftVariableType.SYS
             case _:
-                return DraftVariableType.node
+                return DraftVariableType.NODE
 
     @classmethod
     def _create(