feat: evict old terminal runs to cap history at 50

cristipufu · claude · cristipufu · commit 54cb9eca4986 · 2026-02-21T11:32:11.000+02:00
RunService.runs grows indefinitely, leaking memory on long-running
servers. Add eviction logic that removes oldest completed/failed runs
when total exceeds MAX_RUNS (50), preserving active runs. Clean up
WebSocket subscriptions for evicted runs via on_run_removed callback.

Co-Authored-By: Claude Opus 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/src/uipath/dev/server/__init__.py b/src/uipath/dev/server/__init__.py
@@ -83,6 +83,7 @@ def __init__(
             on_state=self._on_state,
             on_interrupt=self._on_interrupt,
             debug_bridge_factory=lambda mode: WebDebugBridge(mode=mode),
+            on_run_removed=self.connection_manager.remove_run_subscriptions,
         )
 
     def create_app(self) -> Any:
diff --git a/src/uipath/dev/server/ws/manager.py b/src/uipath/dev/server/ws/manager.py
@@ -123,6 +123,10 @@ def unsubscribe(self, websocket: WebSocket, run_id: str) -> None:
             if not self._subscriptions[run_id]:
                 del self._subscriptions[run_id]
 
+    def remove_run_subscriptions(self, run_id: str) -> None:
+        """Remove all subscriptions for a run."""
+        self._subscriptions.pop(run_id, None)
+
     def broadcast_run_updated(self, run: ExecutionRun) -> None:
         """Broadcast a run update to all subscribers (safe from sync context)."""
         msg = server_message(ServerEvent.RUN_UPDATED, serialize_run(run))
diff --git a/src/uipath/dev/services/run_service.py b/src/uipath/dev/services/run_service.py
@@ -45,6 +45,8 @@
 from uipath.dev.models.execution import ExecutionMode, ExecutionRun
 from uipath.dev.services.chat_bridge import WebChatBridge
 
+MAX_RUNS = 50
+
 RunUpdatedCallback = Callable[[ExecutionRun], None]
 LogCallback = Callable[[LogData], None]
 TraceCallback = Callable[[TraceData], None]
@@ -98,6 +100,7 @@ def __init__(
         on_state: StateCallback | None = None,
         on_interrupt: InterruptCallback | None = None,
         debug_bridge_factory: DebugBridgeFactory | None = None,
+        on_run_removed: Callable[[str], None] | None = None,
     ) -> None:
         """Initialize RunService with runtime factory and trace manager."""
         self.runtime_factory = runtime_factory
@@ -111,6 +114,7 @@ def __init__(
         self.on_state = on_state
         self.on_interrupt = on_interrupt
         self._debug_bridge_factory = debug_bridge_factory
+        self._on_run_removed = on_run_removed
 
         self._exporter = RunContextExporter(
             on_trace=self.handle_trace,
@@ -125,6 +129,36 @@ def register_run(self, run: ExecutionRun) -> None:
         """Register a new run and emit an initial update."""
         self.runs[run.id] = run
         self._emit_run_updated(run)
+        self._evict_old_runs()
+
+    def _evict_old_runs(self) -> None:
+        """Remove oldest terminal runs when total exceeds MAX_RUNS."""
+        if len(self.runs) <= MAX_RUNS:
+            return
+
+        active = []
+        terminal = []
+        for run in self.runs.values():
+            if run.status in ("pending", "running", "suspended"):
+                active.append(run)
+            else:
+                terminal.append(run)
+
+        keep_terminal = MAX_RUNS - len(active)
+        if keep_terminal < 0:
+            keep_terminal = 0
+
+        if len(terminal) <= keep_terminal:
+            return
+
+        terminal.sort(
+            key=lambda r: r.end_time or datetime.min.replace(tzinfo=timezone.utc),
+            reverse=True,
+        )
+        for run in terminal[keep_terminal:]:
+            del self.runs[run.id]
+            if self._on_run_removed is not None:
+                self._on_run_removed(run.id)
 
     def get_run(self, run_id: str) -> ExecutionRun | None:
         """Get a registered run."""

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,7 @@ def __init__(`
`83`	`83`	`on_state=self._on_state,`
`84`	`84`	`on_interrupt=self._on_interrupt,`
`85`	`85`	`debug_bridge_factory=lambda mode: WebDebugBridge(mode=mode),`
	`86`	`+ on_run_removed=self.connection_manager.remove_run_subscriptions,`
`86`	`87`	`)`
`87`	`88`
`88`	`89`	`def create_app(self) -> Any:`