Persist engine heartbeats during wait loops

This commit is contained in:
Thigazhezhilan J 2026-04-01 23:08:17 +05:30
parent 302edc5b79
commit f5bd36df64

View File

@ -130,6 +130,7 @@ def sleep_with_heartbeat(
while remaining > 0 and not stop_event.is_set(): while remaining > 0 and not stop_event.is_set():
time.sleep(min(step_seconds, remaining)) time.sleep(min(step_seconds, remaining))
_set_state(user_id, run_id, last_heartbeat_ts=datetime.utcnow().isoformat() + "Z") _set_state(user_id, run_id, last_heartbeat_ts=datetime.utcnow().isoformat() + "Z")
_update_engine_status(user_id, run_id, "RUNNING")
remaining -= step_seconds remaining -= step_seconds
def _clear_runner(user_id: str, run_id: str): def _clear_runner(user_id: str, run_id: str):