blaxel 0.2.4__py3-none-any.whl → 0.2.5__py3-none-any.whl

blaxel/core/sandbox/__init__.py

@@ -7,6 +7,7 @@ from .sandbox import (
 )
 from .types import (
     CopyResponse,
+    ProcessRequestWithLog,
     SandboxConfiguration,
     SandboxCreateConfiguration,
     SandboxFilesystemFile,
@@ -28,4 +29,5 @@ __all__ = [
     "SandboxFileSystem",
     "SandboxPreviews",
     "SandboxProcess",
+    "ProcessRequestWithLog",
 ]

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 SandboxConfiguration
+from .types import ProcessRequestWithLog, SandboxConfiguration
 
 
 class SandboxProcess(SandboxAction):
@@ -76,11 +76,17 @@ class SandboxProcess(SandboxAction):
         return {"close": close}
 
     async def exec(
-        self,
-        process: Union[ProcessRequest, Dict[str, Any]],
-        on_log: Optional[Callable[[str], None]] = None,
+        self, process: Union[ProcessRequest, ProcessRequestWithLog, Dict[str, Any]]
     ) -> ProcessResponse:
+        on_log = None
+        if isinstance(process, ProcessRequestWithLog):
+            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

blaxel/core/sandbox/types.py

@@ -1,8 +1,11 @@
 from datetime import datetime
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Callable, Dict, List, Optional, Union
+
+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
 
 
 class SessionCreateOptions:
@@ -164,9 +167,18 @@ class SandboxCreateConfiguration:
             if isinstance(env, dict):
                 # Validate that the dict has the required keys
                 if "name" not in env or "value" not in env:
-                    raise ValueError(f"Environment variable dict must have 'name' and 'value' keys: {env}")
+                    raise ValueError(
+                        f"Environment variable dict must have 'name' and 'value' keys: {env}"
+                    )
                 env_objects.append({"name": env["name"], "value": env["value"]})
             else:
-                raise ValueError(f"Invalid env type: {type(env)}. Expected dict with 'name' and 'value' keys.")
+                raise ValueError(
+                    f"Invalid env type: {type(env)}. Expected dict with 'name' and 'value' keys."
+                )
 
         return env_objects
+
+
+@_attrs_define
+class ProcessRequestWithLog(ProcessRequest):
+    on_log: Callable[[str], None] = None

{blaxel-0.2.4.dist-info → blaxel-0.2.5.dist-info}/METADATA

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

{blaxel-0.2.4.dist-info → blaxel-0.2.5.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=oF3sX5MbwSqfwhOtF5ODYWwapHffbkp2UI78jPBn78U,617
+blaxel/core/sandbox/__init__.py,sha256=fyxxdPhVIkI1FwSI4tHG1DH0W4sYuNaWvVxO7_due2U,673
 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=7zEngDTs2XiNsMm9TZ4lEDUiRpS3af8dw60pEvRuHDY,8357
+blaxel/core/sandbox/process.py,sha256=g13yig8pvIwHtABFXwxWKSd5K4KYhEQQcGumME9I-GA,8609
 blaxel/core/sandbox/sandbox.py,sha256=y6mV2-i6TsSZGQsvBBGptR514OfgvhV0hXBMEsOBcsY,8648
 blaxel/core/sandbox/session.py,sha256=3PfoekfdVzLYttsmKeK3MePhuprjqv_FDyVQTQME0OE,5277
-blaxel/core/sandbox/types.py,sha256=nsbvZLUxvDZkg2oExE3HdXx2qIpPaaoa7_lGYBzQ8Xk,5638
+blaxel/core/sandbox/types.py,sha256=E-IjEetAMhFDz-mDizFlifCrXoIeJnkbxolQS3p8TKQ,5936
 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.4.dist-info/METADATA,sha256=oh2nI5vCd2gRNp-9jWiQigKLjSeuVwz7ODk1yny-PMM,9875
-blaxel-0.2.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.4.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.4.dist-info/RECORD,,
+blaxel-0.2.5.dist-info/METADATA,sha256=t_ALDJA5oFDqRr8BtWRKJ5Q94y1AfBo3oMEKy7QXb-w,9875
+blaxel-0.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.5.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.5.dist-info/RECORD,,