dayhoff-tools 1.9.2__tar.gz → 1.9.4__tar.gz

{dayhoff_tools-1.9.2 → dayhoff_tools-1.9.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.9.2
+Version: 1.9.4
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com

{dayhoff_tools-1.9.2 → dayhoff_tools-1.9.4}/dayhoff_tools/cli/engine_commands.py

@@ -773,23 +773,22 @@ def engine_status(
         run_disp = engine["state"].capitalize()
 
     # Compose Active/Idle header with extra detail when idle
-    if idle_detector.get("status") == "active":
-        active_disp = "[green]Active[/green]"
-    else:
-        # show Idle <elapsed>/<threshold> if available; else just N/A for stopped states
+    def _compute_active_disp(idle_info: Dict[str, Any]) -> str:
+        if idle_info.get("status") == "active":
+            return "[green]Active[/green]"
         if running_state in ("stopped", "stopping"):
-            active_disp = "[dim]N/A[/dim]"
-        else:
-            idle_seconds = idle_detector.get("idle_seconds")
-            thresh = idle_detector.get("idle_threshold")
-            if isinstance(idle_seconds, (int, float)) and isinstance(
-                thresh, (int, float)
-            ):
-                active_disp = (
-                    f"[yellow]Idle {int(idle_seconds)//60}m/{int(thresh)//60}m[/yellow]"
-                )
-            else:
-                active_disp = "[yellow]Idle[/yellow]"
+            return "[dim]N/A[/dim]"
+        idle_seconds_v = idle_info.get("idle_seconds")
+        thresh_v = idle_info.get("idle_threshold")
+        if isinstance(idle_seconds_v, (int, float)) and isinstance(
+            thresh_v, (int, float)
+        ):
+            return (
+                f"[yellow]Idle {int(idle_seconds_v)//60}m/{int(thresh_v)//60}m[/yellow]"
+            )
+        return "[yellow]Idle[/yellow]"
+
+    active_disp = _compute_active_disp(idle_detector)
 
     top_lines = [
         f"[blue]{engine['name']}[/blue]  {run_disp}  {active_disp}\n",
@@ -895,9 +894,10 @@ def engine_status(
                 status_lines.append(
                     f"  • GPU Drivers: {'OK' if health.get('drivers_ok') else 'MISSING'}"
                 )
-                status_lines.append(
-                    f"  • Idle Detector: {health.get('idle_detector_timer', 'unknown')}"
+                idle_stat = health.get("idle_detector_service") or health.get(
+                    "idle_detector_timer", "unknown"
                 )
+                status_lines.append(f"  • Idle Detector: {idle_stat}")
         except Exception:
             pass
 
@@ -1005,6 +1005,13 @@ def engine_status(
             for k, v in overlay.items():
                 idle_detector.setdefault(k, v)
 
+        # Recompute header display using enriched overlay values
+        try:
+            active_disp = _compute_active_disp(idle_detector)
+            top_lines[0] = f"[blue]{engine['name']}[/blue]  {run_disp}  {active_disp}\n"
+        except Exception:
+            pass
+
     # Activity Sensors (show all with YES/no)
     if idle_detector.get("available"):
         status_lines.append("")
@@ -2681,7 +2688,7 @@ def idle_timeout_cmd(
     cmd = (
         "sudo sed -i '/^IDLE_TIMEOUT_SECONDS=/d' /etc/engine.env && "
         f"echo 'IDLE_TIMEOUT_SECONDS={seconds}' | sudo tee -a /etc/engine.env >/dev/null && "
-        "sudo systemctl restart engine-idle-detector.timer"
+        "sudo systemctl restart engine-idle-detector.service"
     )
 
     resp = ssm.send_command(
@@ -2820,8 +2827,7 @@ def repair_engine(
         "echo 'finished' | sudo tee /opt/dayhoff/state/engine-init.stage > /dev/null",
         # Ensure SSM agent is running
         "sudo systemctl restart amazon-ssm-agent 2>/dev/null || true",
-        # Restart idle detector
-        "sudo systemctl restart engine-idle-detector.timer 2>/dev/null || true",
+        # Restart idle detector (service only)
         "sudo systemctl restart engine-idle-detector.service 2>/dev/null || true",
         # Report status
         "echo '=== Repair Complete ===' && echo 'Sentinel: ' && ls -la /opt/dayhoff/first_boot_complete.sentinel",

{dayhoff_tools-1.9.2 → dayhoff_tools-1.9.4}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.9.2"
+version = "1.9.4"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}