fix: Validates session factory type in repository (#18497)

Signed-off-by: -LAN- <laipz8200@outlook.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

api/repositories/workflow_node_execution/sqlalchemy_repository.py

@@ -37,8 +37,12 @@ class SQLAlchemyWorkflowNodeExecutionRepository:
         # If an engine is provided, create a sessionmaker from it
         if isinstance(session_factory, Engine):
             self._session_factory = sessionmaker(bind=session_factory, expire_on_commit=False)
-        else:
+        elif isinstance(session_factory, sessionmaker):
             self._session_factory = session_factory
+        else:
+            raise ValueError(
+                f"Invalid session_factory type {type(session_factory).__name__}; expected sessionmaker or Engine"
+            )
 
         self._tenant_id = tenant_id
         self._app_id = app_id

api/services/workflow_run_service.py

@@ -133,7 +133,7 @@ class WorkflowRunService:
             params={
                 "tenant_id": app_model.tenant_id,
                 "app_id": app_model.id,
-                "session_factory": db.session.get_bind,
+                "session_factory": db.session.get_bind(),
             }
         )
 

api/tasks/remove_app_and_related_data_task.py

@@ -193,7 +193,7 @@ def _delete_app_workflow_node_executions(tenant_id: str, app_id: str):
         params={
             "tenant_id": tenant_id,
             "app_id": app_id,
-            "session_factory": db.session.get_bind,
+            "session_factory": db.session.get_bind(),
         }
     )