Refactor session runtime persistence

src/run_state_store.py

@@ -275,6 +275,11 @@ class RunStateStore:
                 continue
             if isinstance(event, dict):
                 events.append(event)
+        elapsed_ms = row['elapsed_ms']
+        if row['status'] in ACTIVE_RUN_STATUSES:
+            started_at = row['started_at']
+            if isinstance(started_at, (int, float)):
+                elapsed_ms = max(0, int((time.time() - float(started_at)) * 1000))
         return {
             'run_id': row['run_id'],
             'session_id': row['session_id'],
@@ -283,7 +288,7 @@ class RunStateStore:
             'started_at': row['started_at'],
             'updated_at': row['updated_at'],
             'finished_at': row['finished_at'],
-            'elapsed_ms': row['elapsed_ms'],
+            'elapsed_ms': elapsed_ms,
             'pending_prompt': row['pending_prompt'] or '',
             'error': row['error'] or '',
             'cancellable': bool(row['cancellable']),