blaxel 0.2.8__py3-none-any.whl → 0.2.9__py3-none-any.whl

blaxel/core/sandbox/__init__.py

@@ -8,6 +8,7 @@ from .sandbox import (
 from .types import (
     CopyResponse,
     ProcessRequestWithLog,
+    ProcessResponseWithLog,
     SandboxConfiguration,
     SandboxCreateConfiguration,
     SandboxFilesystemFile,
@@ -30,4 +31,5 @@ __all__ = [
     "SandboxPreviews",
     "SandboxProcess",
     "ProcessRequestWithLog",
+    "ProcessResponseWithLog",
 ]

blaxel/core/sandbox/process.py

@@ -7,7 +7,7 @@ from ..common.settings import settings
 from .action import SandboxAction
 from .client.models import ProcessResponse, SuccessResponse
 from .client.models.process_request import ProcessRequest
-from .types import ProcessRequestWithLog, SandboxConfiguration
+from .types import ProcessRequestWithLog, ProcessResponseWithLog, SandboxConfiguration
 
 
 class SandboxProcess(SandboxAction):
@@ -77,7 +77,8 @@ class SandboxProcess(SandboxAction):
 
     async def exec(
         self, process: Union[ProcessRequest, ProcessRequestWithLog, Dict[str, Any]]
-    ) -> ProcessResponse:
+    ) -> Union[ProcessResponse, ProcessResponseWithLog]:
+        """Execute a process in the sandbox."""
         on_log = None
         if isinstance(process, ProcessRequestWithLog):
             on_log = process.on_log
@@ -95,7 +96,6 @@ class SandboxProcess(SandboxAction):
         # 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
@@ -118,7 +118,7 @@ class SandboxProcess(SandboxAction):
                     stream_control = self.stream_logs(result.pid, {"on_log": on_log})
                 try:
                     # Wait for process completion
-                    result = await self.wait(result.pid, interval=50)
+                    result = await self.wait(result.pid, interval=500, max_wait=1000 * 60 * 60)
                 finally:
                     # Clean up log streaming
                     if stream_control:
@@ -126,8 +126,11 @@ class SandboxProcess(SandboxAction):
             else:
                 # For non-blocking execution, set up log streaming immediately if requested
                 if on_log is not None:
-                    stream = self.stream_logs(result.pid, {"on_log": on_log})
-                    result.additional_properties["close"] = stream["close"]
+                    stream_control = self.stream_logs(result.pid, {"on_log": on_log})
+                    return ProcessResponseWithLog(
+                        result, lambda: stream_control["close"]() if stream_control else None
+                    )
+
             return result
 
     async def wait(

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/core/sandbox/types.py

@@ -6,6 +6,7 @@ from attrs import define as _attrs_define
 from ..client.models import Port, Sandbox
 from ..client.types import UNSET
 from .client.models.process_request import ProcessRequest
+from .client.models.process_response import ProcessResponse
 
 
 class SessionCreateOptions:
@@ -182,3 +183,26 @@ class SandboxCreateConfiguration:
 @_attrs_define
 class ProcessRequestWithLog(ProcessRequest):
     on_log: Callable[[str], None] = None
+
+
+class ProcessResponseWithLog:
+    """A process response with additional close functionality for stream management."""
+
+    def __init__(self, process_response: ProcessResponse, close_func: Callable[[], None]):
+        self._process_response = process_response
+        self._close_func = close_func
+
+    def close(self) -> None:
+        """Close the log stream without terminating the process."""
+        self._close_func()
+
+    def __getattr__(self, name: str) -> Any:
+        """Delegate attribute access to the underlying ProcessResponse."""
+        return getattr(self._process_response, name)
+
+    def __setattr__(self, name: str, value: Any) -> None:
+        """Handle setting attributes, preserving special attributes."""
+        if name.startswith("_") or name == "close":
+            super().__setattr__(name, value)
+        else:
+            setattr(self._process_response, name, value)

{blaxel-0.2.8.dist-info → blaxel-0.2.9.dist-info}/METADATA

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

{blaxel-0.2.8.dist-info → blaxel-0.2.9.dist-info}/RECORD

@@ -304,15 +304,15 @@ blaxel/core/mcp/__init__.py,sha256=5VjkiQFb1QWW5QKRgwPHARlxZJ9Xqaz0diJTpM8LLF0,1
 blaxel/core/mcp/client.py,sha256=aK3wSnsO8DmT1BZqw4eiCMF71Jwvni6Qga0DhPP806Y,5437
 blaxel/core/mcp/server.py,sha256=tXySGZKgK3IllYOzYOecp58BixKBkmAIvQp_4nSM_Ww,5919
 blaxel/core/models/__init__.py,sha256=HbRDsMnUFHkPC-MMkzPXh4mUqkVjqO6p3j7m00N_XSo,1722
-blaxel/core/sandbox/__init__.py,sha256=fyxxdPhVIkI1FwSI4tHG1DH0W4sYuNaWvVxO7_due2U,673
+blaxel/core/sandbox/__init__.py,sha256=sDsT9d6_bQvB-7dR6ZxODYrkzqQdQznhtXTfdqcSRLI,731
 blaxel/core/sandbox/action.py,sha256=9Zjkco7YkLzBThD3N2Hr5SpeEiqU_-Ktk8HlKpkpiAg,2802
 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=g13yig8pvIwHtABFXwxWKSd5K4KYhEQQcGumME9I-GA,8609
-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=E-IjEetAMhFDz-mDizFlifCrXoIeJnkbxolQS3p8TKQ,5936
+blaxel/core/sandbox/types.py,sha256=hf3WCk8P2e87jT4cJGuz3aZCihffun8bVb2k91YfVBU,6894
 blaxel/core/sandbox/client/__init__.py,sha256=N26bD5o1jsTb48oExow6Rgivd8ylaU9jaWZfZsVilP8,128
 blaxel/core/sandbox/client/client.py,sha256=tcP8cJ4Q3dV9aB3yQ01dDXO-ekfsa3WGGFz4DQAEf8I,7079
 blaxel/core/sandbox/client/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
@@ -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.8.dist-info/METADATA,sha256=rrITjLt4KlYbQ4-wpWGd1hhxIvwkYjLyusPvuqflfgA,9875
-blaxel-0.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.8.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.8.dist-info/RECORD,,
+blaxel-0.2.9.dist-info/METADATA,sha256=9bQ896L9Q_2VbaSu0vqkvYf_vJ8KAtnn4RKyOFFGlm4,9875
+blaxel-0.2.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.9.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.9.dist-info/RECORD,,