mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-23 08:25:57 +00:00
fix run creation cancellation rollback
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user