diff --git a/api/repositories/sqlalchemy_api_workflow_run_repository.py b/api/repositories/sqlalchemy_api_workflow_run_repository.py index 24dd9b5e6b..4b031f1c9b 100644 --- a/api/repositories/sqlalchemy_api_workflow_run_repository.py +++ b/api/repositories/sqlalchemy_api_workflow_run_repository.py @@ -394,9 +394,7 @@ class DifyAPISQLAlchemyWorkflowRunRepository(APIWorkflowRunRepository): ) node_executions_deleted = node_executions_result.rowcount or 0 - app_logs_result = session.execute( - delete(WorkflowAppLog).where(WorkflowAppLog.workflow_run_id.in_(run_ids)) - ) + app_logs_result = session.execute(delete(WorkflowAppLog).where(WorkflowAppLog.workflow_run_id.in_(run_ids))) app_logs_deleted = app_logs_result.rowcount or 0 pause_ids = session.scalars( diff --git a/api/repositories/sqlalchemy_workflow_trigger_log_repository.py b/api/repositories/sqlalchemy_workflow_trigger_log_repository.py index 800f63daa4..6703a83c46 100644 --- a/api/repositories/sqlalchemy_workflow_trigger_log_repository.py +++ b/api/repositories/sqlalchemy_workflow_trigger_log_repository.py @@ -98,7 +98,5 @@ class SQLAlchemyWorkflowTriggerLogRepository(WorkflowTriggerLogRepository): if not run_ids: return 0 - result = self.session.execute( - delete(WorkflowTriggerLog).where(WorkflowTriggerLog.workflow_run_id.in_(run_ids)) - ) + result = self.session.execute(delete(WorkflowTriggerLog).where(WorkflowTriggerLog.workflow_run_id.in_(run_ids))) return result.rowcount or 0 diff --git a/api/tests/unit_tests/repositories/test_sqlalchemy_api_workflow_run_repository.py b/api/tests/unit_tests/repositories/test_sqlalchemy_api_workflow_run_repository.py index 14d197e0ac..90a3ef6985 100644 --- a/api/tests/unit_tests/repositories/test_sqlalchemy_api_workflow_run_repository.py +++ b/api/tests/unit_tests/repositories/test_sqlalchemy_api_workflow_run_repository.py @@ -219,9 +219,7 @@ class TestCreateWorkflowPause(TestDifyAPISQLAlchemyWorkflowRunRepository): class TestDeleteRunsWithRelated(TestDifyAPISQLAlchemyWorkflowRunRepository): - def test_uses_trigger_log_repository( - self, repository: DifyAPISQLAlchemyWorkflowRunRepository, mock_session: Mock - ): + def test_uses_trigger_log_repository(self, repository: DifyAPISQLAlchemyWorkflowRunRepository, mock_session: Mock): node_ids_result = Mock() node_ids_result.all.return_value = [] pause_ids_result = Mock()