dayhoff-tools 1.3.27__py3-none-any.whl → 1.3.28__py3-none-any.whl

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.dist-info → dayhoff_tools-1.3.28.dist-info}/METADATA

@@ -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.dist-info → dayhoff_tools-1.3.28.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=KL_LVKBEnjf87v8Lq29iGhGuhBjO7zzh3BVwwcxE0lE,80501
+dayhoff_tools/cli/engine_commands.py,sha256=qnueTQBXrnNkEaPjICvNdFytDI-mO-RIc4REsGrFiiQ,81715
 dayhoff_tools/cli/main.py,sha256=rgeEHD9lJ8SBCR34BTLb7gVInHUUdmEBNXAJnq5yEU4,4795
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
 dayhoff_tools/cli/utility_commands.py,sha256=qs8vH9TBFHsOPC3X8cU3qZigM3dDn-2Ytq4o_F2WubU,27874
@@ -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=8YbnQ--usrEgDQGfvpV4MrMji55A0rq2hZaOgFGh6ag,15896
-dayhoff_tools-1.3.27.dist-info/METADATA,sha256=YSyFCovUECiZ3SXVVPxshzizo6aoqdiqJynCWQSGlTw,2825
-dayhoff_tools-1.3.27.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.3.27.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.3.27.dist-info/RECORD,,
+dayhoff_tools-1.3.28.dist-info/METADATA,sha256=PKfyhnreJSp0BFCQDAQ_6_7GWk44e_xs0FszGQOwJv0,2825
+dayhoff_tools-1.3.28.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.3.28.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.3.28.dist-info/RECORD,,