dayhoff-tools 1.3.27__tar.gz → 1.3.28__tar.gz

{dayhoff_tools-1.3.27 → dayhoff_tools-1.3.28}/PKG-INFO

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

{dayhoff_tools-1.3.27 → dayhoff_tools-1.3.28}/dayhoff_tools/cli/engine_commands.py

@@ -15,7 +15,7 @@ from botocore.exceptions import ClientError, NoCredentialsError
 from rich import box
 from rich.console import Console
 from rich.panel import Panel
-from rich.progress import Progress, SpinnerColumn, TextColumn
+from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from rich.prompt import Confirm, IntPrompt, Prompt
 from rich.table import Table
 import re
@@ -1594,9 +1594,28 @@ def attach_studio(
                 raise typer.Exit(1)
             console.print("[green]✓ Engine started[/green]")
             console.print("Waiting for engine to be ready...")
-            import time
-
-            time.sleep(10)
+            # Wait until the engine is fully running & marked ready (max 5 min)
+            console.print("Waiting for engine to be ready (this can take a couple of minutes)…")
+            from rich.progress import Progress, SpinnerColumn, TimeElapsedColumn
+
+            with Progress(SpinnerColumn(), TimeElapsedColumn(), TextColumn("[progress.description]{task.description}"), transient=True) as prog:
+                prog.add_task("Awaiting engine readiness…", total=None)
+                max_attempts = 30  # 30 × 10 s  ≈ 5 min
+                for _ in range(max_attempts):
+                    time.sleep(10)
+
+                    status_resp = make_api_request("GET", f"/engines/{engine['instance_id']}")
+                    if status_resp.status_code != 200:
+                        continue
+
+                    detailed = status_resp.json().get("engine", {})
+                    state_ok = detailed.get("state", "").lower() == "running"
+                    ready_ok = detailed.get("ready") is True
+
+                    if state_ok and ready_ok:
+                        break
+                else:
+                    console.print("[yellow]Engine is still starting up. If attachment fails, wait a little longer and retry.[/yellow]")
         else:
             raise typer.Exit(1)
 

{dayhoff_tools-1.3.27 → dayhoff_tools-1.3.28}/pyproject.toml

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