dayhoff-tools 1.6.11__py3-none-any.whl → 1.6.13__py3-none-any.whl

dayhoff_tools/cli/engine_commands.py

@@ -1867,14 +1867,22 @@ def _attempt_studio_attach(studio, engine, target_user, public_key):
     # 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"
-        console.print(f"[dim]DEBUG: Got 202 - polling operation {op_id}[/dim]")
-        poll_ok, poll_err = _poll_operation(op_id)
-        console.print(f"[dim]DEBUG: Poll result: ok={poll_ok}, err={poll_err}[/dim]")
-        if poll_ok:
-            return True, None
-        return False, poll_err
+        console.print(f"[dim]DEBUG: Got 202 - operation {op_id} started[/dim]")
+        
+        # Instead of polling the operation status (which seems broken),
+        # just wait a bit and check if the studio is actually attached
+        # This is what dh studio status does and it works reliably
+        time.sleep(5)  # Give the async operation a moment to start
+        
+        # Now check periodically if the studio is attached
+        for check in range(20):  # Check for up to 60 seconds
+            if _is_studio_attached(studio["studio_id"], engine["instance_id"]):
+                console.print("[dim]DEBUG: Studio attachment detected via status check[/dim]")
+                return True, None
+            time.sleep(3)
+        
+        # If we get here, attachment didn't complete in reasonable time
+        return False, None  # Return None to trigger retry
 
     # --- determine if we should retry ---
     recoverable = False
@@ -1937,16 +1945,20 @@ def _poll_operation(operation_id: str, timeout: int = 600) -> Tuple[bool, str]:
         # ---- fetch status ---------------------------------------------------
         op_resp = make_api_request("GET", f"/operations/{operation_id}")
         if op_resp.status_code != 200:
+            console.print(f"[dim]DEBUG: Operation poll failed: HTTP {op_resp.status_code}[/dim]")
             return False, (
                 f"Failed to fetch operation status (HTTP {op_resp.status_code})"
             )
         data = op_resp.json()
         op_status = data.get("status")
+        console.print(f"[dim]DEBUG: Operation status: {op_status}[/dim]")
 
         # ---- interpret state -----------------------------------------------
         if op_status == "COMPLETED":
+            console.print("[dim]DEBUG: Operation completed successfully[/dim]")
             return True, ""
         if op_status == "FAILED":
+            console.print(f"[dim]DEBUG: Operation failed: {data.get('error', 'Operation failed')}[/dim]")
             return False, data.get("error", "Operation failed")
 
         # Not finished yet – wait a little and poll again (no live spinner)

{dayhoff_tools-1.6.11.dist-info → dayhoff_tools-1.6.13.dist-info}/METADATA

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

{dayhoff_tools-1.6.11.dist-info → dayhoff_tools-1.6.13.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=ssaYgeQDa3vWbPBlHxZdDPtJpfArZZ7POb6eJBKHong,97240
+dayhoff_tools/cli/engine_commands.py,sha256=Loc-ppwuuuAZbj5j2s0aNT5_R1u7oiOx7bDjjWuw3y8,98069
 dayhoff_tools/cli/main.py,sha256=tRN7WCBHg6uyNp6rA54pKTCoVmBntta2i0Yas3bUpZ4,4853
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
 dayhoff_tools/cli/utility_commands.py,sha256=FRZTPrjsG_qmIIqoNxd1Q1vVkS_5w8aY33IrVYVNCLg,18131
@@ -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=heaYc64qplgN3_1WVPFmqj53goStioWwY5NqlWc4c0s,24453
-dayhoff_tools-1.6.11.dist-info/METADATA,sha256=puTF_Rt1kbAL-R8hz0BpxBjNacbGcJlACatfMSsSwgU,2915
-dayhoff_tools-1.6.11.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.6.11.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.6.11.dist-info/RECORD,,
+dayhoff_tools-1.6.13.dist-info/METADATA,sha256=Vj0JcsRhUcJE8witxYBmGTiCWLdunubI_01WecoV3Xw,2915
+dayhoff_tools-1.6.13.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.6.13.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.6.13.dist-info/RECORD,,