blaxel 0.2.7rc85__py3-none-any.whl → 0.2.7rc86__py3-none-any.whl

blaxel/core/sandbox/process.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Callable, Dict, Literal, Optional, Union
+from typing import Any, Callable, Dict, Literal, Optional, Union
 
 import httpx
 
@@ -76,15 +76,26 @@ class SandboxProcess(SandboxAction):
         return {"close": close}
 
     async def exec(
-        self, process: Union[ProcessRequest, ProcessRequestWithLog]
+        self, process: Union[ProcessRequest, ProcessRequestWithLog, Dict[str, Any]]
     ) -> Union[ProcessResponse, ProcessResponseWithLog]:
         """Execute a process in the sandbox."""
-        on_log: Optional[Callable[[str], None]] = None
-        if isinstance(process, ProcessRequestWithLog) and process.on_log:
+        on_log = None
+        if isinstance(process, ProcessRequestWithLog):
             on_log = process.on_log
+            process = process.to_dict()
 
-        should_wait_for_completion = getattr(process, "wait_for_completion", True)
+        if isinstance(process, dict):
+            if "on_log" in process:
+                on_log = process["on_log"]
+                del process["on_log"]
+            process = ProcessRequest.from_dict(process)
 
+        # Store original wait_for_completion setting
+        should_wait_for_completion = process.wait_for_completion
+
+        # Always start process without wait_for_completion to avoid server-side blocking
+        if should_wait_for_completion and on_log is not None:
+            process.wait_for_completion = False
         async with self.get_client() as client_instance:
             response = await client_instance.post("/process", json=process.to_dict())
             # Parse JSON response only once, with better error handling

blaxel/core/sandbox/sandbox.py

@@ -47,7 +47,7 @@ class SandboxInstance:
     def spec(self):
         return self.sandbox.spec
 
-    async def wait(self, max_wait: int = 60000, interval: int = 1000) -> None:
+    async def wait(self, max_wait: int = 60000, interval: int = 1000) -> "SandboxInstance":
         start_time = time.time() * 1000  # Convert to milliseconds
         while self.sandbox.status != "DEPLOYED":
             await asyncio.sleep(interval / 1000)  # Convert to seconds
@@ -68,6 +68,17 @@ class SandboxInstance:
             if (time.time() * 1000) - start_time > max_wait:
                 raise Exception("Sandbox did not deploy in time")
 
+        if self.sandbox.status == "DEPLOYED":
+            try:
+                # This is a hack for sometime receiving a 502,
+                # need to remove this once we have a better way to handle this
+                await self.fs.ls("/")
+            except:
+                # pass
+                pass
+
+        return self
+
     @classmethod
     async def create(
         cls, sandbox: Union[Sandbox, SandboxCreateConfiguration, Dict[str, Any], None] = None
@@ -109,7 +120,9 @@ class SandboxInstance:
             sandbox = Sandbox(
                 metadata=Metadata(name=name),
                 spec=SandboxSpec(
-                    runtime=Runtime(image=image, memory=memory, ports=ports, envs=envs, generation="mk3")
+                    runtime=Runtime(
+                        image=image, memory=memory, ports=ports, envs=envs, generation="mk3"
+                    )
                 ),
             )
         else:

{blaxel-0.2.7rc85.dist-info → blaxel-0.2.7rc86.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.7rc85
+Version: 0.2.7rc86
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai

{blaxel-0.2.7rc85.dist-info → blaxel-0.2.7rc86.dist-info}/RECORD

@@ -309,8 +309,8 @@ blaxel/core/sandbox/action.py,sha256=9Zjkco7YkLzBThD3N2Hr5SpeEiqU_-Ktk8HlKpkpiAg
 blaxel/core/sandbox/filesystem.py,sha256=dyIvDdlPZO0ijD6mXXX8Yl0t75VijQ6_uMz_9rJd-_4,11317
 blaxel/core/sandbox/network.py,sha256=P5jLd4AAg1zgyIK4qGWvZaDZ5BzIcxRx2ffz_JLsLMI,357
 blaxel/core/sandbox/preview.py,sha256=g0uVbMsIi8gRXmmyOfSyqm1qO4Cv6rsq92fs_k884dY,6120
-blaxel/core/sandbox/process.py,sha256=L4ArSZhDmcKUl3V4llaqtJq_C3QzDTwsuOcydRmEpxY,8382
-blaxel/core/sandbox/sandbox.py,sha256=y6mV2-i6TsSZGQsvBBGptR514OfgvhV0hXBMEsOBcsY,8648
+blaxel/core/sandbox/process.py,sha256=95fu9WU-CCdJcnHV8tYErkKTv5YoRcOqO68BH9NopvQ,8837
+blaxel/core/sandbox/sandbox.py,sha256=J7wGqtelXZFk5XV99qTWSWS0TMHjo2C5qqZ7EYrD8j8,9036
 blaxel/core/sandbox/session.py,sha256=3PfoekfdVzLYttsmKeK3MePhuprjqv_FDyVQTQME0OE,5277
 blaxel/core/sandbox/types.py,sha256=hf3WCk8P2e87jT4cJGuz3aZCihffun8bVb2k91YfVBU,6894
 blaxel/core/sandbox/client/__init__.py,sha256=N26bD5o1jsTb48oExow6Rgivd8ylaU9jaWZfZsVilP8,128
@@ -402,7 +402,7 @@ blaxel/telemetry/instrumentation/map.py,sha256=PCzZJj39yiYVYJrxLBNP-NW-tjjYyTijw
 blaxel/telemetry/instrumentation/utils.py,sha256=KInMYZH-mu9_wvetmf0EmgrfN3Sw8IWk2Y95v2u90_U,1901
 blaxel/telemetry/log/log.py,sha256=RvQByRjZMoP_dRaAZu8oK6DTegsHs-xV4W-UIqis6CA,2461
 blaxel/telemetry/log/logger.py,sha256=NPAS3g82ryROjvc_DEZaTIfrcehoLEZoP-JkLxADxc0,4113
-blaxel-0.2.7rc85.dist-info/METADATA,sha256=31C827_vOZ8eKeaBRuS0i7jkp23F6n8YxzZVpmc1yto,9879
-blaxel-0.2.7rc85.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.7rc85.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.7rc85.dist-info/RECORD,,
+blaxel-0.2.7rc86.dist-info/METADATA,sha256=LVVDBCFGGEIuh90oSVc7EJtRtfGVHFLClRT4y5HhWbc,9879
+blaxel-0.2.7rc86.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.7rc86.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.7rc86.dist-info/RECORD,,