From da6fa55eeda5b3063e35774eccd637931639af6e Mon Sep 17 00:00:00 2001
From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com>
Date: Fri, 12 Dec 2025 04:31:09 +0000
Subject: [PATCH] [autofix.ci] apply automated fixes

---
 ..._api_workflow_node_execution_repository.py | 34 +++++++++++--------
 ...ear_free_plan_expired_workflow_run_logs.py |  6 ++--
 2 files changed, 23 insertions(+), 17 deletions(-)

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)