fix run creation cancellation rollback

This commit is contained in:
rayhpeng
2026-05-22 17:58:28 +08:00
parent 0fae7c9cbb
commit f3e3a350ce
2 changed files with 57 additions and 2 deletions
@@ -144,6 +144,9 @@ class RunManager:
self._runs[run_id] = record
try:
await self._persist_new_run_to_store(record)
except asyncio.CancelledError:
self._runs.pop(run_id, None)
raise
except Exception:
self._runs.pop(run_id, None)
logger.warning("Failed to persist run %s; rolled back in-memory record", run_id, exc_info=True)
@@ -348,6 +351,9 @@ class RunManager:
self._runs[run_id] = record
try:
await self._persist_new_run_to_store(record)
except asyncio.CancelledError:
self._runs.pop(run_id, None)
raise
except Exception:
self._runs.pop(run_id, None)
logger.warning("Failed to persist run %s; rolled back in-memory record", run_id, exc_info=True)