diff --git a/api/repositories/sqlalchemy_api_workflow_node_execution_repository.py b/api/repositories/sqlalchemy_api_workflow_node_execution_repository.py index 9d0a42f518..ab116611b8 100644 --- a/api/repositories/sqlalchemy_api_workflow_node_execution_repository.py +++ b/api/repositories/sqlalchemy_api_workflow_node_execution_repository.py @@ -328,20 +328,26 @@ class DifyAPISQLAlchemyWorkflowNodeExecutionRepository(DifyAPIWorkflowNodeExecut if not node_execution_ids: return 0, 0 - offloads_deleted = cast( - CursorResult, - session.execute( - delete(WorkflowNodeExecutionOffload).where( - WorkflowNodeExecutionOffload.node_execution_id.in_(node_execution_ids) - ) - ), - ).rowcount or 0 + offloads_deleted = ( + cast( + CursorResult, + session.execute( + delete(WorkflowNodeExecutionOffload).where( + WorkflowNodeExecutionOffload.node_execution_id.in_(node_execution_ids) + ) + ), + ).rowcount + or 0 + ) - node_executions_deleted = cast( - CursorResult, - session.execute( - delete(WorkflowNodeExecutionModel).where(WorkflowNodeExecutionModel.id.in_(node_execution_ids)) - ), - ).rowcount or 0 + node_executions_deleted = ( + cast( + CursorResult, + session.execute( + delete(WorkflowNodeExecutionModel).where(WorkflowNodeExecutionModel.id.in_(node_execution_ids)) + ), + ).rowcount + or 0 + ) return node_executions_deleted, offloads_deleted diff --git a/api/tests/unit_tests/services/test_clear_free_plan_expired_workflow_run_logs.py b/api/tests/unit_tests/services/test_clear_free_plan_expired_workflow_run_logs.py index 913a8f7ff4..3728386e34 100644 --- a/api/tests/unit_tests/services/test_clear_free_plan_expired_workflow_run_logs.py +++ b/api/tests/unit_tests/services/test_clear_free_plan_expired_workflow_run_logs.py @@ -54,9 +54,9 @@ class FakeRepo: self.call_idx += 1 return batch - def delete_runs_with_related(self, runs: list[FakeRun], - delete_node_executions=None, - delete_trigger_logs=None) -> dict[str, int]: + def delete_runs_with_related( + self, runs: list[FakeRun], delete_node_executions=None, delete_trigger_logs=None + ) -> dict[str, int]: self.deleted.append([run.id for run in runs]) result = self.delete_result.copy() result["runs"] = len(runs)