feat: improve model rate limit tracking

2026-05-12 03:22:29 +08:00
parent 05632172d0
commit ba850a06c6
25 changed files with 1223 additions and 96 deletions
@@ -30,6 +30,17 @@ func main() {
 		os.Exit(1)
 	}
 	defer db.Close()
+	if recovery, err := db.RecoverInterruptedRuntimeState(ctx); err != nil {
+		logger.Error("recover interrupted runtime state failed", "error", err)
+		os.Exit(1)
+	} else if recovery.ReleasedConcurrencyLeases > 0 || recovery.ReleasedRateReservations > 0 || recovery.FailedAttempts > 0 || recovery.FailedTasks > 0 {
+		logger.Warn("interrupted runtime state recovered",
+			"releasedConcurrencyLeases", recovery.ReleasedConcurrencyLeases,
+			"releasedRateReservations", recovery.ReleasedRateReservations,
+			"failedAttempts", recovery.FailedAttempts,
+			"failedTasks", recovery.FailedTasks,
+		)
+	}

 	server := &http.Server{
 		Addr:              cfg.HTTPAddr,