dayhoff-tools 1.6.2__tar.gz → 1.6.4__tar.gz

{dayhoff_tools-1.6.2 → dayhoff_tools-1.6.4}/PKG-INFO

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

{dayhoff_tools-1.6.2 → dayhoff_tools-1.6.4}/dayhoff_tools/cli/engine_commands.py

@@ -1810,9 +1810,20 @@ def _attempt_studio_attach(studio, engine, target_user, public_key):
         },
     )
 
+    # Fast-path success
     if response.status_code == 200:
         return True, None
 
+    # Asynchronous path – API returned 202 Accepted and operation tracking ID
+    if response.status_code == 202:
+        op_id = response.json().get("operation_id")
+        if not op_id:
+            return False, "Async operation started but no operation_id returned"
+        poll_ok, poll_err = _poll_operation(op_id)
+        if poll_ok:
+            return True, None
+        return False, poll_err
+
     # --- determine if we should retry ---
     recoverable = False
     if response.status_code in (409, 503):
@@ -1848,6 +1859,31 @@ def _attempt_studio_attach(studio, engine, target_user, public_key):
     return False, None
 
 
+def _poll_operation(operation_id: str, timeout: int = 600) -> Tuple[bool, str]:
+    """Poll /operations/<id> until COMPLETED or FAILED.
+
+    Returns (True, "") on success or (False, error_message) on failure / timeout.
+    Uses a simple status spinner to avoid nested Live displays.
+    """
+    start_ts = time.time()
+    with console.status("[cyan]Waiting for studio attach to complete…", spinner="dots") as status:
+        while True:
+            if time.time() - start_ts > timeout:
+                return False, "Timed out waiting for attach operation to complete"
+            op_resp = make_api_request("GET", f"/operations/{operation_id}")
+            if op_resp.status_code != 200:
+                return False, f"Failed to fetch operation status (HTTP {op_resp.status_code})"
+            data = op_resp.json()
+            op_status = data.get("status")
+            if op_status == "COMPLETED":
+                return True, ""
+            if op_status == "FAILED":
+                return False, data.get("error", "Operation failed")
+            # update spinner text occasionally
+            status.update(f"[cyan]Waiting… ({op_status.lower()})")
+            time.sleep(3)
+    
+
 @studio_app.command("detach")
 def detach_studio(
     user: Optional[str] = typer.Option(

{dayhoff_tools-1.6.2 → dayhoff_tools-1.6.4}/pyproject.toml

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