dayhoff-tools 1.9.0__py3-none-any.whl → 1.9.1__py3-none-any.whl

dayhoff_tools/cli/engine_commands.py

@@ -319,87 +319,8 @@ def format_status(state: str, ready: Optional[bool]) -> str:
         return state
 
 
-# --------------------------------------------------------------------------------
-# Audit helpers (Phase 1 observability)
-# --------------------------------------------------------------------------------
-
-
-def _get_engine_audit_bucket() -> Optional[str]:
-    """Return the engine audit bucket name from SSM Parameter Store, if configured."""
-    try:
-        ssm = boto3.client("ssm", region_name="us-east-1")
-        resp = ssm.get_parameter(Name="/dev/studio-manager/engine-audit-bucket")
-        return resp["Parameter"]["Value"]
-    except ClientError:
-        return None
-
-
-def _fetch_last_audit_via_ssm(instance_id: str) -> Optional[Dict]:
-    """Fetch last shutdown attempt audit from the engine via SSM (fast best-effort)."""
-    try:
-        ssm = boto3.client("ssm", region_name="us-east-1")
-        resp = ssm.send_command(
-            InstanceIds=[instance_id],
-            DocumentName="AWS-RunShellScript",
-            Parameters={
-                "commands": [
-                    "cat /var/log/idle-detector/last_shutdown_attempt.json 2>/dev/null || true",
-                ],
-                "executionTimeout": ["3"],
-            },
-        )
-        cid = resp["Command"]["CommandId"]
-        time.sleep(1)
-        inv = ssm.get_command_invocation(CommandId=cid, InstanceId=instance_id)
-        if inv["Status"] != "Success":
-            return None
-        content = inv["StandardOutputContent"].strip()
-        if not content:
-            return None
-        return json.loads(content)
-    except Exception:
-        return None
-
-
-def _fetch_last_audit_via_s3(instance_id: str) -> Optional[Dict]:
-    """Fetch the newest audit object from S3 if available."""
-    bucket = _get_engine_audit_bucket()
-    if not bucket:
-        return None
-    try:
-        s3 = boto3.client("s3", region_name="us-east-1")
-        paginator = s3.get_paginator("list_objects_v2")
-        newest = None
-        for page in paginator.paginate(
-            Bucket=bucket, Prefix=f"{instance_id}/", MaxKeys=1000
-        ):
-            for obj in page.get("Contents", []):
-                lm = obj.get("LastModified")
-                if newest is None or (lm and lm > newest["LastModified"]):
-                    newest = obj
-        if not newest:
-            return None
-        obj = s3.get_object(Bucket=bucket, Key=newest["Key"])
-        data = obj["Body"].read().decode("utf-8")
-        return json.loads(data)
-    except Exception:
-        return None
 
 
-def _summarize_audit(audit: Dict) -> str:
-    """Return a compact one-line summary for status output."""
-    ts = audit.get("ts", "?")
-    shutdown = audit.get("shutdown", {})
-    result = shutdown.get("result", "?")
-    detach = audit.get("detach", {})
-    num_detached = sum(
-        1 for r in detach.get("results", []) if r.get("status") == "success"
-    )
-    idle = audit.get("idle", {})
-    elapsed = int(idle.get("elapsed_sec", 0))
-    threshold = int(idle.get("threshold_sec", 0))
-    return f"Last shutdown attempt: {ts} result={result} detach={num_detached} idle={elapsed//60}/{threshold//60}m"
-
 
 def resolve_engine(name_or_id: str, engines: List[Dict]) -> Dict:
     """Resolve engine by name or ID with interactive selection."""
@@ -1081,15 +1002,7 @@ def engine_status(
         status_lines.append(_sensor_line(" IDE   ", "IDEConnectionSensor", "🖥"))
         status_lines.append(_sensor_line("Docker", "DockerWorkloadSensor", "🐳"))
 
-    # Audit one-liner (best-effort SSM fetch)
-    try:
-        last_audit = _fetch_last_audit_via_ssm(engine["instance_id"])
-        if last_audit:
-            status_lines.append("")
-            status_lines.append("[bold]Shutdown Audit:[/bold]")
-            status_lines.append(f"  • {_summarize_audit(last_audit)}")
-    except Exception:
-        pass
+
 
     # Combine top summary and details
     all_lines = top_lines + status_lines
@@ -1128,65 +1041,7 @@ def engine_status(
             console.print(f"[red]❌ Error fetching log: {e}[/red]")
 
 
-@engine_app.command("why")
-def engine_why(
-    name_or_id: str = typer.Argument(help="Engine name or instance ID"),
-    raw: bool = typer.Option(False, "--raw", help="Print raw audit JSON"),
-):
-    """Explain the last idle-detector shutdown attempt for an engine.
-
-    Tries SSM (on-instance file) first, then falls back to S3 audit bucket.
-    """
-    check_aws_sso()
 
-    # Resolve engine
-    response = make_api_request("GET", "/engines")
-    if response.status_code != 200:
-        console.print("[red]❌ Failed to fetch engines[/red]")
-        raise typer.Exit(1)
-
-    engines = response.json().get("engines", [])
-    engine = resolve_engine(name_or_id, engines)
-
-    audit = _fetch_last_audit_via_ssm(
-        engine["instance_id"]
-    ) or _fetch_last_audit_via_s3(engine["instance_id"])
-    if not audit:
-        console.print("No audit found (engine may not have attempted shutdown yet).")
-        raise typer.Exit(0)
-
-    if raw:
-        console.print_json(data=audit)
-        return
-
-    # Pretty summary
-    status = _summarize_audit(audit)
-    panel_lines = [
-        f"[bold]{status}[/bold]",
-        "",
-        "[bold]Sensors:[/bold]",
-    ]
-    for r in audit.get("idle", {}).get("reason_snapshot", []):
-        active = "✓" if r.get("active") else "-"
-        reason = r.get("reason") or ""
-        sensor = r.get("sensor")
-        panel_lines.append(f"  {active} {sensor}: {reason}")
-    panel_lines.append("")
-    panel_lines.append("[bold]Detach results:[/bold]")
-    for res in audit.get("detach", {}).get("results", []):
-        panel_lines.append(
-            f"  - {res.get('studio_id')}: {res.get('status')} {res.get('error') or ''}"
-        )
-    s3_info = audit.get("s3", {})
-    if s3_info.get("uploaded"):
-        panel_lines.append("")
-        panel_lines.append(
-            f"[dim]S3: s3://{s3_info.get('bucket')}/{s3_info.get('key')}[/dim]"
-        )
-
-    console.print(
-        Panel("\n".join(panel_lines), title="Idle Shutdown Audit", border_style="blue")
-    )
 
 
 @engine_app.command("stop")

{dayhoff_tools-1.9.0.dist-info → dayhoff_tools-1.9.1.dist-info}/METADATA

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

{dayhoff_tools-1.9.0.dist-info → dayhoff_tools-1.9.1.dist-info}/RECORD

@@ -3,7 +3,7 @@ dayhoff_tools/chemistry/standardizer.py,sha256=uMn7VwHnx02nc404eO6fRuS4rsl4dvSPf
 dayhoff_tools/chemistry/utils.py,sha256=jt-7JgF-GeeVC421acX-bobKbLU_X94KNOW24p_P-_M,2257
 dayhoff_tools/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dayhoff_tools/cli/cloud_commands.py,sha256=33qcWLmq-FwEXMdL3F0OHm-5Stlh2r65CldyEZgQ1no,40904
-dayhoff_tools/cli/engine_commands.py,sha256=o_7d4YhHO5U_WncIowx91v6qCPnalFPqayx5p9CnSug,110214
+dayhoff_tools/cli/engine_commands.py,sha256=HcMjlfxHWGp8J9ai2pOel6RkT3W-BcaaNk1KBBDWmLg,104796
 dayhoff_tools/cli/main.py,sha256=LoFs3SI4fdCjP4pdxEAhri-_q0dmNYupmBCRE4KbBac,5933
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
 dayhoff_tools/cli/utility_commands.py,sha256=WQTHOh1MttuxaJjl2c6zMa4x7_JuaKMQgcyotYrU3GA,25883
@@ -27,7 +27,7 @@ dayhoff_tools/intake/uniprot.py,sha256=BZYJQF63OtPcBBnQ7_P9gulxzJtqyorgyuDiPeOJq
 dayhoff_tools/logs.py,sha256=DKdeP0k0kliRcilwvX0mUB2eipO5BdWUeHwh-VnsICs,838
 dayhoff_tools/sqlite.py,sha256=jV55ikF8VpTfeQqqlHSbY8OgfyfHj8zgHNpZjBLos_E,18672
 dayhoff_tools/warehouse.py,sha256=UETBtZD3r7WgvURqfGbyHlT7cxoiVq8isjzMuerKw8I,24475
-dayhoff_tools-1.9.0.dist-info/METADATA,sha256=3fx18V_-b31gFOH9QMYxAC_fCXPdfFjekTrcq2z0x-Y,2914
-dayhoff_tools-1.9.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.9.0.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.9.0.dist-info/RECORD,,
+dayhoff_tools-1.9.1.dist-info/METADATA,sha256=m7wy7HkaFJon6aQO127tsXQ-GPYiK21NtYFS1FEdh58,2914
+dayhoff_tools-1.9.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.9.1.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.9.1.dist-info/RECORD,,