refactor workflow runner

api/core/workflow/callbacks/base_workflow_callback.py

@@ -1,34 +1,63 @@
 from abc import ABC, abstractmethod
+from typing import Optional
 
-from models.workflow import WorkflowNodeExecution, WorkflowRun
+from core.workflow.entities.base_node_data_entities import BaseNodeData
+from core.workflow.entities.node_entities import NodeType
 
 
 class BaseWorkflowCallback(ABC):
     @abstractmethod
-    def on_workflow_run_started(self, workflow_run: WorkflowRun) -> None:
+    def on_workflow_run_started(self) -> None:
         """
         Workflow run started
         """
         raise NotImplementedError
 
     @abstractmethod
-    def on_workflow_run_finished(self, workflow_run: WorkflowRun) -> None:
+    def on_workflow_run_succeeded(self) -> None:
         """
-        Workflow run finished
+        Workflow run succeeded
         """
         raise NotImplementedError
 
     @abstractmethod
-    def on_workflow_node_execute_started(self, workflow_node_execution: WorkflowNodeExecution) -> None:
+    def on_workflow_run_failed(self, error: str) -> None:
+        """
+        Workflow run failed
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def on_workflow_node_execute_started(self, node_id: str,
+                                         node_type: NodeType,
+                                         node_data: BaseNodeData,
+                                         node_run_index: int = 1,
+                                         predecessor_node_id: Optional[str] = None) -> None:
         """
         Workflow node execute started
         """
         raise NotImplementedError
 
     @abstractmethod
-    def on_workflow_node_execute_finished(self, workflow_node_execution: WorkflowNodeExecution) -> None:
+    def on_workflow_node_execute_succeeded(self, node_id: str,
+                                           node_type: NodeType,
+                                           node_data: BaseNodeData,
+                                           inputs: Optional[dict] = None,
+                                           process_data: Optional[dict] = None,
+                                           outputs: Optional[dict] = None,
+                                           execution_metadata: Optional[dict] = None) -> None:
         """
-        Workflow node execute finished
+        Workflow node execute succeeded
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def on_workflow_node_execute_failed(self, node_id: str,
+                                        node_type: NodeType,
+                                        node_data: BaseNodeData,
+                                        error: str) -> None:
+        """
+        Workflow node execute failed
         """
         raise NotImplementedError
 
@@ -38,4 +67,3 @@ class BaseWorkflowCallback(ABC):
         Publish text chunk
         """
         raise NotImplementedError
-