fix: keep agent alive across worker crashes
This commit is contained in:
@@ -5,6 +5,7 @@ import json
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
from collections import deque
|
||||
from concurrent.futures import ThreadPoolExecutor, as_completed
|
||||
from datetime import timedelta
|
||||
from pathlib import Path
|
||||
@@ -600,8 +601,10 @@ def run_poll_loop(
|
||||
else:
|
||||
architecture_bootstrap_summary["reason"] = "outcome_sync_disabled"
|
||||
|
||||
cycle_summaries: list[dict[str, Any]] = []
|
||||
queue_cleanup_runs: list[dict[str, Any]] = []
|
||||
retained_cycles = max(10, int(os.getenv("MODELHUB_AGENT_RETAINED_CYCLE_SUMMARIES", "50")))
|
||||
retained_cleanups = max(5, int(os.getenv("MODELHUB_AGENT_RETAINED_CLEANUP_SUMMARIES", "20")))
|
||||
cycle_summaries: deque[dict[str, Any]] = deque(maxlen=retained_cycles)
|
||||
queue_cleanup_runs: deque[dict[str, Any]] = deque(maxlen=retained_cleanups)
|
||||
submitted_total = 0
|
||||
cycles = 0
|
||||
stopped_reason = "max_cycles_reached"
|
||||
@@ -920,8 +923,12 @@ def run_poll_loop(
|
||||
"submittedTotal": submitted_total,
|
||||
"stoppedReason": stopped_reason,
|
||||
"pollRunDir": str(poll_run_dir),
|
||||
"cycleSummaries": cycle_summaries,
|
||||
"queueCleanupRuns": queue_cleanup_runs,
|
||||
"cycleSummaries": list(cycle_summaries),
|
||||
"cycleSummariesRetained": len(cycle_summaries),
|
||||
"cycleSummariesRetentionLimit": retained_cycles,
|
||||
"queueCleanupRuns": list(queue_cleanup_runs),
|
||||
"queueCleanupRunsRetained": len(queue_cleanup_runs),
|
||||
"queueCleanupRunsRetentionLimit": retained_cleanups,
|
||||
"architectureBootstrap": architecture_bootstrap_summary,
|
||||
"outcomeStats": stats_report,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user