refactor(api): type workflow run related counts with RelatedCountsDict TypedDict

Author: YB0y

api/services/retention/workflow_run/clear_free_plan_expired_workflow_run_logs.py

@@ -3,7 +3,7 @@
 import random
 import time
 from collections.abc import Iterable, Sequence
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, TypedDict
 
 import click
 from sqlalchemy.orm import Session, sessionmaker
@@ -145,7 +145,7 @@ def record_batch(
         targeted_runs: int,
         skipped_runs: int,
         deleted_runs: int,
-        related_counts: dict[str, int] | None,
+        related_counts: RelatedCountsDict | None,
         related_action: str | None,
         batch_duration_seconds: float,
     ) -> None:
@@ -173,6 +173,15 @@ def record_completion(self, *, status: str, job_duration_seconds: float) -> None
         self._record(self._job_duration_seconds, job_duration_seconds, attributes)
 
 
+class RelatedCountsDict(TypedDict):
+    node_executions: int
+    offloads: int
+    app_logs: int
+    trigger_logs: int
+    pauses: int
+    pause_reasons: int
+
+
 class WorkflowRunCleanup:
     def __init__(
         self,
@@ -506,7 +515,7 @@ def _count_trigger_logs(self, session: Session, run_ids: Sequence[str]) -> int:
         return trigger_repo.count_by_run_ids(run_ids)
 
     @staticmethod
-    def _empty_related_counts() -> dict[str, int]:
+    def _empty_related_counts() -> RelatedCountsDict:
         return {
             "node_executions": 0,
             "offloads": 0,
@@ -517,7 +526,7 @@ def _empty_related_counts() -> dict[str, int]:
         }
 
     @staticmethod
-    def _format_related_counts(counts: dict[str, int]) -> str:
+    def _format_related_counts(counts: RelatedCountsDict) -> str:
         return (
             f"node_executions {counts['node_executions']}, "
             f"offloads {counts['offloads']}, "