dreadnode 2.0.36__py3-none-any.whl → 2.0.37__py3-none-any.whl

dreadnode/core/environment.py

@@ -215,16 +215,32 @@ class TaskEnvironment(Environment):
         wait bounded by ``deadline_sec`` (caller's ``timeout_sec`` when set,
         otherwise the module default). A ``failed`` status raises with the
         server-provided error message.
+
+        A just-created environment's status can 404 for the first few polls: the
+        platform tracks the pending env in per-replica memory until the sandbox
+        row is persisted, so a status poll routed to a different API replica than
+        the one that handled the POST sees neither the tracker entry nor the row
+        yet. Treat that 404 as "still provisioning" and keep polling until the
+        deadline instead of aborting a provision that will succeed.
         """
+        from dreadnode.app.api.client import NotFoundError
+
         interval = 1.0
         deadline = asyncio.get_event_loop().time() + deadline_sec
         while True:
-            status = await asyncio.to_thread(
-                self.api_client.get_environment_status,
-                self.org,
-                self.workspace,
-                environment_id,
-            )
+            try:
+                status = await asyncio.to_thread(
+                    self.api_client.get_environment_status,
+                    self.org,
+                    self.workspace,
+                    environment_id,
+                )
+            except NotFoundError:
+                if asyncio.get_event_loop().time() >= deadline:
+                    raise
+                await asyncio.sleep(interval)
+                interval = min(interval * 1.5, _POLL_MAX_INTERVAL_SEC)
+                continue
             state = status.get("state")
             if state == "ready":
                 return status

{dreadnode-2.0.36.dist-info → dreadnode-2.0.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dreadnode
-Version: 2.0.36
+Version: 2.0.37
 Summary: Dreadnode SDK
 Project-URL: Homepage, https://dreadnode.io
 Project-URL: Documentation, https://docs.dreadnode.io

{dreadnode-2.0.36.dist-info → dreadnode-2.0.37.dist-info}/RECORD

@@ -419,7 +419,7 @@ dreadnode/capabilities/worker_runner.py,sha256=_qqWT26JQBWBXu_PgX94fLkL5ykMLllE9
 dreadnode/core/__init__.py,sha256=hJuoWE9VXUZrmnpRN0DDyataXKKnLSuDeoDBQvcNhQA,2751
 dreadnode/core/conditions.py,sha256=KDswJ4vVYW3xHFQ7WsPg7MyR38ZzF6u-givZGm4tNxs,37219
 dreadnode/core/discovery.py,sha256=vD_1ETS4RCL97rHyfoRF69rr3EeVCwdrAPy2OmPyEx8,6663
-dreadnode/core/environment.py,sha256=97ULse4O3bKpW6LdnRVGvO-LzOnfP8nMPoMm4x9oouw,19913
+dreadnode/core/environment.py,sha256=-BGKgIpy9YAEyVnxRzoX2Mi0IIUZrJJOknIjj-bcU6k,20752
 dreadnode/core/exceptions.py,sha256=zO8F-9HV4ReD3slFAA_vOul2-52uHLp745Rkc0UEsK4,7134
 dreadnode/core/execution.py,sha256=72ejZaj8iZQ0_ELqaRpYQysjRgIpWjavURtxixvOr_c,6745
 dreadnode/core/hook.py,sha256=fcenhloxGmubPk5gTifq60NigJepsGqWgNiju6gTqoM,7988
@@ -704,8 +704,8 @@ dreadnode/transforms/system_prompt_extraction.py,sha256=rtWKibIipr1tmFH3P95UE4W4
 dreadnode/transforms/text.py,sha256=8qEldTVeXblxxCCU31Q2MGx3dccMxESOoSga-Y87Mnk,19923
 dreadnode/transforms/video.py,sha256=pWPSAN3YokvWpYsAOvOVH1d2Zd5JPldl17ICw1532a8,33136
 dreadnode/transforms/xml_tools.py,sha256=Zj2sdvmyCoVLua4XkWnKYFRmS8ZFSWrQ5nunspCqpM4,15057
-dreadnode-2.0.36.dist-info/METADATA,sha256=QEEe27u0XJwyvzW1uYYZN4trx78HGL4bmQVPPoI8ICc,13481
-dreadnode-2.0.36.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-dreadnode-2.0.36.dist-info/entry_points.txt,sha256=LvV-eGt7GlPEHy1YDiClWhr4qbsX_D9l0_QNo0j_XPI,81
-dreadnode-2.0.36.dist-info/licenses/LICENSE,sha256=UMJAkT3OxyPBvOmQ74h_1v8DYFRYIfMfFiaP2lFZCmc,11375
-dreadnode-2.0.36.dist-info/RECORD,,
+dreadnode-2.0.37.dist-info/METADATA,sha256=k344TKwfQpUqwCdShD2dpMz7FApjK9iGH5UI2suKsAY,13481
+dreadnode-2.0.37.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+dreadnode-2.0.37.dist-info/entry_points.txt,sha256=LvV-eGt7GlPEHy1YDiClWhr4qbsX_D9l0_QNo0j_XPI,81
+dreadnode-2.0.37.dist-info/licenses/LICENSE,sha256=UMJAkT3OxyPBvOmQ74h_1v8DYFRYIfMfFiaP2lFZCmc,11375
+dreadnode-2.0.37.dist-info/RECORD,,