diff --git a/api/core/tools/__base/tool.py b/api/core/tools/__base/tool.py
index 65f5f33ba5..8ca4eabb7a 100644
--- a/api/core/tools/__base/tool.py
+++ b/api/core/tools/__base/tool.py
@@ -231,16 +231,3 @@ class Tool(ABC):
                 variable_name=variable_name, variable_value=variable_value, stream=stream
             ),
         )
-
-    def create_variable_message(
-        self, variable_name: str, variable_value: Any, stream: bool = False
-    ) -> ToolInvokeMessage:
-        """
-        create a variable message
-        """
-        return ToolInvokeMessage(
-            type=ToolInvokeMessage.MessageType.VARIABLE,
-            message=ToolInvokeMessage.VariableMessage(
-                variable_name=variable_name, variable_value=variable_value, stream=stream
-            ),
-        )
diff --git a/api/models/workflow.py b/api/models/workflow.py
index b3ce06544c..1364cbb471 100644
--- a/api/models/workflow.py
+++ b/api/models/workflow.py
@@ -21,7 +21,7 @@ from core.workflow.constants import (
     MEMORY_BLOCK_VARIABLE_NODE_ID,
     SYSTEM_VARIABLE_NODE_ID,
 )
-from core.workflow.enums import NodeType
+from core.workflow.enums import NodeType, WorkflowExecutionStatus
 from extensions.ext_storage import Storage
 from factories.variable_factory import TypeMismatchError, build_segment_with_type
 from libs.datetime_utils import naive_utc_now