blaxel 0.2.7rc84__py3-none-any.whl → 0.2.7rc85__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

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Any, Callable, Dict, Literal, Optional, Union
+from typing import Callable, Dict, Literal, Optional, Union
 
 import httpx
 
@@ -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):
@@ -76,25 +76,14 @@ class SandboxProcess(SandboxAction):
         return {"close": close}
 
     async def exec(
-        self, process: Union[ProcessRequest, ProcessRequestWithLog, Dict[str, Any]]
-    ) -> ProcessResponse:
-        on_log = None
-        if isinstance(process, ProcessRequestWithLog):
+        self, process: Union[ProcessRequest, ProcessRequestWithLog]
+    ) -> 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 = process.on_log
-            process = process.to_dict()
-
-        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
+        should_wait_for_completion = getattr(process, "wait_for_completion", True)
 
         async with self.get_client() as client_instance:
             response = await client_instance.post("/process", json=process.to_dict())
@@ -118,7 +107,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 +115,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/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.7rc84.dist-info → blaxel-0.2.7rc85.dist-info}/METADATA

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

{blaxel-0.2.7rc84.dist-info → blaxel-0.2.7rc85.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/process.py,sha256=L4ArSZhDmcKUl3V4llaqtJq_C3QzDTwsuOcydRmEpxY,8382
 blaxel/core/sandbox/sandbox.py,sha256=y6mV2-i6TsSZGQsvBBGptR514OfgvhV0hXBMEsOBcsY,8648
 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.7rc84.dist-info/METADATA,sha256=CMNjVjNUuFFCW_4BCEtY7cNHgol4ozfTNZDKTGDTQ1o,9879
-blaxel-0.2.7rc84.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.7rc84.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.7rc84.dist-info/RECORD,,
+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,,