dayhoff-tools 1.9.8__tar.gz → 1.9.9__tar.gz

{dayhoff_tools-1.9.8 → dayhoff_tools-1.9.9}/PKG-INFO

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

{dayhoff_tools-1.9.8 → dayhoff_tools-1.9.9}/dayhoff_tools/cli/engine_commands.py

@@ -709,11 +709,7 @@ def engine_status(
     engine = resolve_engine(name_or_id, engines)
     
     # Fast status display (default)
-    if not detailed:
-        # Skip the API call for studios - use basic info we already have
-        attached_studios = []
-        studio_user = engine.get("user")  # Use the engine's user as studio owner
-        
+    if not detailed:       
         # Fetch idle status via SSM with longer timeout
         ssm = boto3.client("ssm", region_name="us-east-1")
         idle_data = None  # Use None to indicate no data received
@@ -794,7 +790,7 @@ def engine_status(
         
         # Build status lines - minimal info for fast view
         status_lines = [
-            f"[blue]{engine['name']}[/blue]  {run_disp}{idle_disp}\n",
+            f"[blue]{engine['name']}[/blue]  {run_disp}{idle_disp}",
         ]
         
         # Add activity sensors if we have idle data
@@ -888,14 +884,14 @@ def engine_status(
 
     # Compose Active/Idle header with extra detail when idle
     def _compute_active_disp(idle_info: Dict[str, Any]) -> str:
+        # If we don't have idle info or it's explicitly unavailable, show N/A
+        if not idle_info or idle_info.get("available") == False:
+            return "[dim]N/A[/dim]"
+            
         if idle_info.get("status") == "active":
             return "[green]Active[/green]"
         if running_state in ("stopped", "stopping"):
             return "[dim]N/A[/dim]"
-        
-        # If we don't have idle info at all, show N/A
-        if not idle_info.get("available"):
-            return "[dim]N/A[/dim]"
             
         # If idle, show time/threshold with time remaining if available
         if idle_info.get("status") == "idle":
@@ -952,47 +948,51 @@ def engine_status(
         disk_usage = get_disk_usage_via_ssm(engine["instance_id"]) or "-"
         status_lines.append(f"Disk:       {disk_usage}")
 
-    # Idle timeout (show even when not idle)
-    idle_threshold_secs: Optional[int] = None
-    # Prefer value from idle detector overlay if present
-    try:
-        if isinstance(idle_detector.get("idle_threshold"), (int, float)):
-            idle_threshold_secs = int(idle_detector["idle_threshold"])
-    except Exception:
-        idle_threshold_secs = None
-
-    if idle_threshold_secs is None and engine["state"].lower() == "running":
-        # Fallback: read /etc/engine.env via SSM
+    # Idle timeout (show even when not idle) - but only if we have data
+    if idle_detector.get("available"):
+        idle_threshold_secs: Optional[int] = None
+        # Prefer value from idle detector overlay if present
         try:
-            ssm = boto3.client("ssm", region_name="us-east-1")
-            resp = ssm.send_command(
-                InstanceIds=[engine["instance_id"]],
-                DocumentName="AWS-RunShellScript",
-                Parameters={
-                    "commands": [
-                        "grep -E '^IDLE_TIMEOUT_SECONDS=' /etc/engine.env | cut -d'=' -f2 || echo 1800",
-                    ],
-                    "executionTimeout": ["5"],
-                },
-            )
-            cid = resp["Command"]["CommandId"]
-            time.sleep(1)
-            inv = ssm.get_command_invocation(
-                CommandId=cid, InstanceId=engine["instance_id"]
-            )
-            if inv.get("Status") == "Success":
-                out = (inv.get("StandardOutputContent") or "").strip()
-                if out:
-                    idle_threshold_secs = int(out.splitlines()[0].strip())
+            if isinstance(idle_detector.get("idle_threshold"), (int, float)):
+                idle_threshold_secs = int(idle_detector["idle_threshold"])
         except Exception:
             idle_threshold_secs = None
 
-    if idle_threshold_secs is None:
-        idle_threshold_secs = 1800
+        if idle_threshold_secs is None and engine["state"].lower() == "running":
+            # Fallback: read /etc/engine.env via SSM
+            try:
+                ssm = boto3.client("ssm", region_name="us-east-1")
+                resp = ssm.send_command(
+                    InstanceIds=[engine["instance_id"]],
+                    DocumentName="AWS-RunShellScript",
+                    Parameters={
+                        "commands": [
+                            "grep -E '^IDLE_TIMEOUT_SECONDS=' /etc/engine.env | cut -d'=' -f2 || echo '?'",
+                        ],
+                        "executionTimeout": ["5"],
+                    },
+                )
+                cid = resp["Command"]["CommandId"]
+                time.sleep(1)
+                inv = ssm.get_command_invocation(
+                    CommandId=cid, InstanceId=engine["instance_id"]
+                )
+                if inv.get("Status") == "Success":
+                    out = (inv.get("StandardOutputContent") or "").strip()
+                    if out and out != "?" and out.isdigit():
+                        idle_threshold_secs = int(out)
+            except Exception:
+                idle_threshold_secs = None
 
-    status_lines.append(
-        f"Idle timeout: {idle_threshold_secs//60}m ({idle_threshold_secs}s)"
-    )
+        if idle_threshold_secs is not None:
+            status_lines.append(
+                f"Idle timeout: {idle_threshold_secs//60}m ({idle_threshold_secs}s)"
+            )
+        else:
+            status_lines.append("Idle timeout: unknown")
+    else:
+        # No idle detector data available
+        status_lines.append("Idle timeout: N/A")
 
     # Health report (only if bootstrap finished)
     if stage_val == "finished":
@@ -1136,13 +1136,14 @@ def engine_status(
         else:
             for k, v in overlay.items():
                 idle_detector.setdefault(k, v)
+    else:
+        # SSM failed - mark as unavailable if we don't have good data
+        if not idle_detector.get("available"):
+            idle_detector = {"available": False}  # Mark as unavailable
 
-        # 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
+    # Recompute header display with latest data
+    active_disp = _compute_active_disp(idle_detector)
+    top_lines[0] = f"[blue]{engine['name']}[/blue]  {run_disp}  {active_disp}\n"
 
     # Activity Sensors (show all with YES/no)
     if idle_detector.get("available"):

{dayhoff_tools-1.9.8 → dayhoff_tools-1.9.9}/pyproject.toml

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