Merge pull request #80 from UiPath/feat/run-history-eviction

feat: evict old terminal runs to cap history at 50

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-dev"
-version = "0.0.54"
+version = "0.0.55"
 description = "UiPath Developer Console"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath/dev/infrastructure/logging_handlers.py

@@ -6,7 +6,7 @@
 import os
 import re
 import threading
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Callable, Pattern
 
 from uipath.runtime.logging import UiPathRuntimeExecutionLogHandler
@@ -35,7 +35,7 @@ def emit(self, record: logging.LogRecord):
                 run_id=self.run_id,
                 level=record.levelname,
                 message=self.format(record),
-                timestamp=datetime.fromtimestamp(record.created),
+                timestamp=datetime.fromtimestamp(record.created, tz=timezone.utc),
             )
             self.callback(log_data)
         except Exception:

src/uipath/dev/infrastructure/tracing_exporter.py

@@ -1,7 +1,7 @@
 """Custom OpenTelemetry trace exporter for CLI UI integration."""
 
 import logging
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Callable, Sequence
 
 from opentelemetry import trace
@@ -82,7 +82,7 @@ def _export_span(self, span: ReadableSpan):
             trace_id=trace_id,
             status=status,
             duration_ms=duration_ms,
-            timestamp=datetime.fromtimestamp(start_time),
+            timestamp=datetime.fromtimestamp(start_time, tz=timezone.utc),
             attributes=dict(span.attributes) if span.attributes else {},
         )
 
@@ -97,7 +97,9 @@ def _export_span(self, span: ReadableSpan):
                     run_id=run_id_val,
                     level=log_level,
                     message=event.name,
-                    timestamp=datetime.fromtimestamp(event.timestamp / 1_000_000_000),
+                    timestamp=datetime.fromtimestamp(
+                        event.timestamp / 1_000_000_000, tz=timezone.utc
+                    ),
                 )
                 self.on_log(log_data)
 

src/uipath/dev/models/chat.py

@@ -1,6 +1,6 @@
 """Aggregates conversation messages from conversation events."""
 
-from datetime import datetime
+from datetime import datetime, timezone
 from uuid import uuid4
 
 from uipath.core.chat import (
@@ -178,7 +178,7 @@ def get_timestamp(self, ev: UiPathConversationMessageEvent) -> str:
         """Choose timestamp from event if available, else fallback."""
         if ev.start and ev.start.timestamp:
             return ev.start.timestamp
-        return datetime.now().isoformat()
+        return datetime.now(timezone.utc).isoformat()
 
     def get_role(self, ev: UiPathConversationMessageEvent) -> str:
         """Infer the role of the message from the event."""
@@ -189,7 +189,7 @@ def get_role(self, ev: UiPathConversationMessageEvent) -> str:
 
 def get_user_message(user_text: str) -> UiPathConversationMessage:
     """Build a user message from text input."""
-    timestamp = datetime.now().isoformat()
+    timestamp = datetime.now(timezone.utc).isoformat()
     return UiPathConversationMessage(
         message_id=str(uuid4()),
         created_at=timestamp,
@@ -216,7 +216,7 @@ def get_user_message_event(
     """Build a conversation event representing a user message from text input."""
     message_id = str(uuid4())
     content_part_id = str(uuid4())
-    timestamp = datetime.now().isoformat()
+    timestamp = datetime.now(timezone.utc).isoformat()
 
     msg_start = UiPathConversationMessageStartEvent(
         role=role,

src/uipath/dev/models/data.py

@@ -1,7 +1,7 @@
 """Plain data classes for inter-component communication (no Textual dependency)."""
 
 from dataclasses import dataclass, field
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Any
 
 from uipath.core.chat import UiPathConversationMessage, UiPathConversationMessageEvent
@@ -28,7 +28,7 @@ class TraceData:
     trace_id: str | None = None
     status: str = "running"
     duration_ms: float | None = None
-    timestamp: datetime = field(default_factory=datetime.now)
+    timestamp: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
     attributes: dict[str, Any] = field(default_factory=dict)
 
 
@@ -41,7 +41,7 @@ class StateData:
     qualified_node_name: str | None = None
     phase: str | None = None
     payload: dict[str, Any] | None = None
-    timestamp: datetime = field(default_factory=datetime.now)
+    timestamp: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
 
 
 @dataclass

src/uipath/dev/models/execution.py

@@ -1,7 +1,7 @@
 """Models for representing execution runs and their data."""
 
 import os
-from datetime import datetime
+from datetime import datetime, timezone
 from enum import Enum
 from typing import Any, cast
 from uuid import uuid4
@@ -38,7 +38,7 @@ def __init__(
         self.mode = mode
         self.resume_data: Any | None = None
         self.output_data: dict[str, Any] | str | None = None
-        self.start_time = datetime.now()
+        self.start_time = datetime.now(timezone.utc)
         self.end_time: datetime | None = None
         self.status = "pending"  # pending, running, completed, failed, suspended
         self.traces: list[TraceData] = []
@@ -58,7 +58,7 @@ def duration(self) -> str:
             delta = self.end_time - self.start_time
             return f"{delta.total_seconds():.1f}s"
         elif self.start_time:
-            delta = datetime.now() - self.start_time
+            delta = datetime.now(timezone.utc) - self.start_time
             return f"{delta.total_seconds():.1f}s"
         return "0.0s"
 

src/uipath/dev/models/messages.py

@@ -1,6 +1,6 @@
 """Messages used for inter-component communication in the UiPath Developer Console."""
 
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Any
 
 from rich.console import RenderableType
@@ -24,7 +24,7 @@ def __init__(
         self.run_id = run_id
         self.level = level
         self.message = message
-        self.timestamp = timestamp or datetime.now()
+        self.timestamp = timestamp or datetime.now(timezone.utc)
         super().__init__()
 
     @classmethod
@@ -61,7 +61,7 @@ def __init__(
         self.trace_id = trace_id
         self.status = status
         self.duration_ms = duration_ms
-        self.timestamp = timestamp or datetime.now()
+        self.timestamp = timestamp or datetime.now(timezone.utc)
         self.attributes = attributes or {}
         super().__init__()
 

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:

src/uipath/dev/server/frontend/src/App.tsx

@@ -28,6 +28,7 @@ export default function App() {
     setStateEvents,
     setGraphCache,
     setActiveNode,
+    removeActiveNode,
   } = useRunStore();
   const { view, runId: routeRunId, setupEntrypoint, setupMode, navigate } = useHashRoute();
 
@@ -97,15 +98,18 @@ export default function App() {
           payload: s.payload,
         })),
       );
-      // Seed activeNodes from historical events so the next WS event has proper prev context
+      // Seed activeNodes from historical events (replay all to get correct prev + executing)
       if (detail.status !== "completed" && detail.status !== "failed") {
-        const lastStarted = [...detail.states].reverse().find((s) => s.phase === "started");
-        if (lastStarted) {
-          setActiveNode(runId, lastStarted.node_name, lastStarted.qualified_node_name);
+        for (const s of detail.states) {
+          if (s.phase === "started") {
+            setActiveNode(runId, s.node_name, s.qualified_node_name);
+          } else if (s.phase === "completed") {
+            removeActiveNode(runId, s.node_name);
+          }
         }
       }
     }
-  }, [upsertRun, setTraces, setLogs, setChatMessages, setStateEvents, setGraphCache, setActiveNode]);
+  }, [upsertRun, setTraces, setLogs, setChatMessages, setStateEvents, setGraphCache, setActiveNode, removeActiveNode]);
 
   // Subscribe to selected run
   useEffect(() => {

src/uipath/dev/server/frontend/src/components/graph/GraphPanel.tsx

@@ -448,18 +448,32 @@ export default function GraphPanel({ entrypoint, runId, breakpointNode, breakpoi
     );
   }, [breakpointNode, layoutSeq, setNodes]);
 
+  const stateEvents = useRunStore((s) => s.stateEvents[runId]);
+
   // Highlight edges + nodes during execution
   // - Paused at breakpoint: edges INTO breakpoint node + edges to next_nodes
-  // - Running: edges OUT of completed node, target nodes of those edges
+  // - Running: edges OUT of executing nodes, target nodes of those edges
   // - __start__: highlighted on first state event; __end__: highlighted when run completes
   useEffect(() => {
     const isPaused = !!breakpointNode;
-    let matchIds = new Set<string>(); // Full React Flow node IDs of the "current" node
+    let matchIds = new Set<string>(); // Full React Flow node IDs of the "current" node(s)
     const prevNodeIds = new Set<string>(); // Full RF IDs of the previous node (for edge filtering when paused)
     const nextNodeIds = new Set<string>(); // Full RF IDs of breakpoint next_nodes
     const activeTargetIds = new Set<string>(); // Full RF IDs for isActiveNode
     const nodeTypeById = new Map<string, string>();
 
+    // Derive currently-executing nodes from the full event log (always consistent)
+    const executingNodes = new Map<string, string | null>(); // nodeName → qualifiedNodeName
+    if (stateEvents) {
+      for (const evt of stateEvents) {
+        if (evt.phase === "started") {
+          executingNodes.set(evt.node_name, evt.qualified_node_name ?? null);
+        } else if (evt.phase === "completed") {
+          executingNodes.delete(evt.node_name);
+        }
+      }
+    }
+
     // 1) Build matchIds, nextNodeIds, node type map
     setNodes((nds) => {
       for (const n of nds) {
@@ -494,33 +508,38 @@ export default function GraphPanel({ entrypoint, runId, breakpointNode, breakpoi
         if (activeNode?.prev) {
           findNodeIds(activeNode.prev).forEach((id) => prevNodeIds.add(id));
         }
-      } else if (activeNode) {
-        // Try qualified name first (exact match via "subgraph:node" → "subgraph/node")
-        const qualifiedName = activeNode.qualifiedNodeName;
-        if (qualifiedName) {
-          const qualifiedId = qualifiedName.replace(/:/g, "/");
-          for (const n of nds) {
-            if (n.id === qualifiedId) {
-              matchIds.add(n.id);
-            }
+      } else if (executingNodes.size > 0) {
+        // Build label → RF ID lookup once
+        const labelToIds = new Map<string, Set<string>>();
+        for (const n of nds) {
+          const label = n.data?.label as string | undefined;
+          if (!label) continue;
+          const plainId = n.id.includes("/") ? n.id.split("/").pop()! : n.id;
+          for (const key of [plainId, label]) {
+            let s = labelToIds.get(key);
+            if (!s) { s = new Set(); labelToIds.set(key, s); }
+            s.add(n.id);
           }
         }
-        // Fallback: label/plainId matching
-        if (matchIds.size === 0) {
-          const labelToIds = new Map<string, Set<string>>();
-          for (const n of nds) {
-            const label = n.data?.label as string | undefined;
-            if (!label) continue;
-            const plainId = n.id.includes("/") ? n.id.split("/").pop()! : n.id;
-            for (const key of [plainId, label]) {
-              let s = labelToIds.get(key);
-              if (!s) { s = new Set(); labelToIds.set(key, s); }
-              s.add(n.id);
+
+        for (const [nodeName, qualifiedNodeName] of executingNodes) {
+          let found = false;
+          // Try qualified name first (exact match via "subgraph:node" → "subgraph/node")
+          if (qualifiedNodeName) {
+            const qualifiedId = qualifiedNodeName.replace(/:/g, "/");
+            for (const n of nds) {
+              if (n.id === qualifiedId) {
+                matchIds.add(n.id);
+                found = true;
+              }
             }
           }
-          matchIds = labelToIds.get(activeNode.current) ?? new Set<string>();
+          // Fallback: label/plainId matching
+          if (!found) {
+            const ids = labelToIds.get(nodeName);
+            if (ids) ids.forEach((id) => matchIds.add(id));
+          }
         }
-
       }
 
       return nds;
@@ -542,7 +561,7 @@ export default function GraphPanel({ entrypoint, runId, breakpointNode, breakpoi
           isActive = intoBreakpoint
             || (matchIds.has(e.source) && nextNodeIds.has(e.target));
         } else {
-          // Running: edges OUT of completed node
+          // Running: edges OUT of executing nodes
           isActive = matchIds.has(e.source);
           // For __end__: also highlight edges INTO it
           if (!isActive && nodeTypeById.get(e.target) === "endNode" && matchIds.has(e.target)) {
@@ -596,9 +615,7 @@ export default function GraphPanel({ entrypoint, runId, breakpointNode, breakpoi
           : n;
       }),
     );
-  }, [activeNode, breakpointNode, breakpointNextNodes, runStatus, layoutSeq, setNodes, setEdges]);
-
-  const stateEvents = useRunStore((s) => s.stateEvents[runId]);
+  }, [stateEvents, activeNode, breakpointNode, breakpointNextNodes, runStatus, layoutSeq, setNodes, setEdges]);
 
   // Subscribe to cached graph reactively (populated async from run detail)
   const cachedGraph = useRunStore((s) => s.graphCache[runId]);