Update backend/workers/control_tick.py

backend/workers/control_tick.py

@@ -60,6 +60,17 @@ def main():
         safe = json.loads(json.dumps(result_dict, default=str))
         redis.set_json("control:last_tick", safe, ttl=1200)  # 20 min TTL
         log.info("Tick result saved to Redis")
+
+        # Also persist budget state for the /control/budget endpoint
+        budget_info = {
+            "energy_cost_kwh": result_dict.get("energy_cost_kwh", 0),
+            "budget_spent_kwh": result_dict.get("budget_spent_kwh", 0),
+            "budget_remaining_kwh": result_dict.get("budget_remaining_kwh", 0),
+            "daily_budget_kwh": 25.0 * 5.0 / 100.0,  # MAX_ENERGY_REDUCTION_PCT
+            "stage": result_dict.get("stage_id", "unknown"),
+            "last_updated": datetime.now(timezone.utc).isoformat(),
+        }
+        redis.set_json("control:budget", budget_info, ttl=1800)  # 30 min TTL
     else:
         log.warning("Redis not available — tick result not persisted")