refactor(sqlalchemy_workflow_node_execution_repository): Refactors repository initialization and imports

Signed-off-by: -LAN- <laipz8200@outlook.com>

api/core/repositories/sqlalchemy_workflow_node_execution_repository.py

@@ -44,8 +44,8 @@ class SQLAlchemyWorkflowNodeExecutionRepository(WorkflowNodeExecutionRepository)
         self,
         session_factory: sessionmaker | Engine,
         user: Union[Account, EndUser],
-        app_id: Optional[str] = None,
+        app_id: Optional[str],
-        triggered_from: Optional[WorkflowNodeExecutionTriggeredFrom] = None,
+        triggered_from: Optional[WorkflowNodeExecutionTriggeredFrom],
     ):
         """
         Initialize the repository with a SQLAlchemy sessionmaker or engine and context information.
@@ -53,8 +53,8 @@ class SQLAlchemyWorkflowNodeExecutionRepository(WorkflowNodeExecutionRepository)
         Args:
             session_factory: SQLAlchemy sessionmaker or engine for creating sessions
             user: Account or EndUser object containing tenant_id, user ID, and role information
-            app_id: Optional app ID for filtering by application
+            app_id: App ID for filtering by application (can be None)
-            triggered_from: Source of the execution trigger (single-step or workflow-run)
+            triggered_from: Source of the execution trigger (SINGLE_STEP or WORKFLOW_RUN)
         """
         # If an engine is provided, create a sessionmaker from it
         if isinstance(session_factory, Engine):

api/services/workflow_run_service.py

@@ -7,7 +7,14 @@ from core.repositories import SQLAlchemyWorkflowNodeExecutionRepository
 from core.workflow.repository.workflow_node_execution_repository import OrderConfig
 from extensions.ext_database import db
 from libs.infinite_scroll_pagination import InfiniteScrollPagination
-from models import Account, App, EndUser, WorkflowNodeExecution, WorkflowRun, WorkflowRunTriggeredFrom
+from models import (
+    Account,
+    App,
+    EndUser,
+    WorkflowNodeExecution,
+    WorkflowRun,
+    WorkflowRunTriggeredFrom,
+)
 
 
 class WorkflowRunService:
@@ -130,7 +137,10 @@ class WorkflowRunService:
             return []
 
         repository = SQLAlchemyWorkflowNodeExecutionRepository(
-            session_factory=db.engine, user=user, app_id=app_model.id
+            session_factory=db.engine,
+            user=user,
+            app_id=app_model.id,
+            triggered_from=None,
         )
 
         # Use the repository to get the node executions with ordering

api/tasks/remove_app_and_related_data_task.py

@@ -204,7 +204,12 @@ def _delete_app_workflow_node_executions(tenant_id: str, app_id: str):
         raise ValueError(errmsg)
 
     # Create a repository instance for WorkflowNodeExecution
-    repository = SQLAlchemyWorkflowNodeExecutionRepository(session_factory=db.engine, user=user, app_id=app_id)
+    repository = SQLAlchemyWorkflowNodeExecutionRepository(
+        session_factory=db.engine,
+        user=user,
+        app_id=app_id,
+        triggered_from=None,
+    )
 
     # Use the clear method to delete all records for this tenant_id and app_id
     repository.clear()

api/tests/unit_tests/repositories/workflow_node_execution/test_sqlalchemy_repository.py

@@ -58,7 +58,12 @@ def repository(session, mock_user):
     """Create a repository instance with test data."""
     _, session_factory = session
     app_id = "test-app"
-    return SQLAlchemyWorkflowNodeExecutionRepository(session_factory=session_factory, user=mock_user, app_id=app_id)
+    return SQLAlchemyWorkflowNodeExecutionRepository(
+        session_factory=session_factory,
+        user=mock_user,
+        app_id=app_id,
+        triggered_from=None,
+    )
 
 
 def test_save(repository, session):