blaxel 0.2.1rc73__py3-none-any.whl → 0.2.1rc75__py3-none-any.whl

blaxel/core/sandbox/client/models/directory.py

@@ -1,10 +1,8 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 if TYPE_CHECKING:
     from ..models.file import File
     from ..models.subdirectory import Subdirectory
@@ -17,54 +15,49 @@ T = TypeVar("T", bound="Directory")
 class Directory:
     """
     Attributes:
-        files (Union[Unset, list['File']]):
-        name (Union[Unset, str]):
-        path (Union[Unset, str]):
-        subdirectories (Union[Unset, list['Subdirectory']]): @name Subdirectories
+        files (list['File']):
+        name (str):
+        path (str):
+        subdirectories (list['Subdirectory']): @name Subdirectories
     """
 
-    files: Union[Unset, list["File"]] = UNSET
-    name: Union[Unset, str] = UNSET
-    path: Union[Unset, str] = UNSET
-    subdirectories: Union[Unset, list["Subdirectory"]] = UNSET
+    files: list["File"]
+    name: str
+    path: str
+    subdirectories: list["Subdirectory"]
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        files: Union[Unset, list[dict[str, Any]]] = UNSET
-        if not isinstance(self.files, Unset):
-            files = []
-            for files_item_data in self.files:
-                if type(files_item_data) is dict:
-                    files_item = files_item_data
-                else:
-                    files_item = files_item_data.to_dict()
-                files.append(files_item)
+        files = []
+        for files_item_data in self.files:
+            if type(files_item_data) is dict:
+                files_item = files_item_data
+            else:
+                files_item = files_item_data.to_dict()
+            files.append(files_item)
 
         name = self.name
 
         path = self.path
 
-        subdirectories: Union[Unset, list[dict[str, Any]]] = UNSET
-        if not isinstance(self.subdirectories, Unset):
-            subdirectories = []
-            for subdirectories_item_data in self.subdirectories:
-                if type(subdirectories_item_data) is dict:
-                    subdirectories_item = subdirectories_item_data
-                else:
-                    subdirectories_item = subdirectories_item_data.to_dict()
-                subdirectories.append(subdirectories_item)
+        subdirectories = []
+        for subdirectories_item_data in self.subdirectories:
+            if type(subdirectories_item_data) is dict:
+                subdirectories_item = subdirectories_item_data
+            else:
+                subdirectories_item = subdirectories_item_data.to_dict()
+            subdirectories.append(subdirectories_item)
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if files is not UNSET:
-            field_dict["files"] = files
-        if name is not UNSET:
-            field_dict["name"] = name
-        if path is not UNSET:
-            field_dict["path"] = path
-        if subdirectories is not UNSET:
-            field_dict["subdirectories"] = subdirectories
+        field_dict.update(
+            {
+                "files": files,
+                "name": name,
+                "path": path,
+                "subdirectories": subdirectories,
+            }
+        )
 
         return field_dict
 
@@ -77,19 +70,19 @@ class Directory:
             return None
         d = src_dict.copy()
         files = []
-        _files = d.pop("files", UNSET)
-        for files_item_data in _files or []:
+        _files = d.pop("files")
+        for files_item_data in _files:
             files_item = File.from_dict(files_item_data)
 
             files.append(files_item)
 
-        name = d.pop("name", UNSET)
+        name = d.pop("name")
 
-        path = d.pop("path", UNSET)
+        path = d.pop("path")
 
         subdirectories = []
-        _subdirectories = d.pop("subdirectories", UNSET)
-        for subdirectories_item_data in _subdirectories or []:
+        _subdirectories = d.pop("subdirectories")
+        for subdirectories_item_data in _subdirectories:
             subdirectories_item = Subdirectory.from_dict(subdirectories_item_data)
 
             subdirectories.append(subdirectories_item)

blaxel/core/sandbox/client/models/error_response.py

@@ -1,10 +1,8 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="ErrorResponse")
 
 
@@ -12,10 +10,10 @@ T = TypeVar("T", bound="ErrorResponse")
 class ErrorResponse:
     """
     Attributes:
-        error (Union[Unset, str]):  Example: Error message.
+        error (str):  Example: Error message.
     """
 
-    error: Union[Unset, str] = UNSET
+    error: str
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -23,9 +21,11 @@ class ErrorResponse:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if error is not UNSET:
-            field_dict["error"] = error
+        field_dict.update(
+            {
+                "error": error,
+            }
+        )
 
         return field_dict
 
@@ -34,7 +34,7 @@ class ErrorResponse:
         if not src_dict:
             return None
         d = src_dict.copy()
-        error = d.pop("error", UNSET)
+        error = d.pop("error")
 
         error_response = cls(
             error=error,

blaxel/core/sandbox/client/models/file.py

@@ -1,10 +1,8 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="File")
 
 
@@ -12,22 +10,22 @@ T = TypeVar("T", bound="File")
 class File:
     """
     Attributes:
-        group (Union[Unset, str]):
-        last_modified (Union[Unset, str]):
-        name (Union[Unset, str]):
-        owner (Union[Unset, str]):
-        path (Union[Unset, str]):
-        permissions (Union[Unset, str]):
-        size (Union[Unset, int]):
+        group (str):
+        last_modified (str):
+        name (str):
+        owner (str):
+        path (str):
+        permissions (str):
+        size (int):
     """
 
-    group: Union[Unset, str] = UNSET
-    last_modified: Union[Unset, str] = UNSET
-    name: Union[Unset, str] = UNSET
-    owner: Union[Unset, str] = UNSET
-    path: Union[Unset, str] = UNSET
-    permissions: Union[Unset, str] = UNSET
-    size: Union[Unset, int] = UNSET
+    group: str
+    last_modified: str
+    name: str
+    owner: str
+    path: str
+    permissions: str
+    size: int
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -47,21 +45,17 @@ class File:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if group is not UNSET:
-            field_dict["group"] = group
-        if last_modified is not UNSET:
-            field_dict["lastModified"] = last_modified
-        if name is not UNSET:
-            field_dict["name"] = name
-        if owner is not UNSET:
-            field_dict["owner"] = owner
-        if path is not UNSET:
-            field_dict["path"] = path
-        if permissions is not UNSET:
-            field_dict["permissions"] = permissions
-        if size is not UNSET:
-            field_dict["size"] = size
+        field_dict.update(
+            {
+                "group": group,
+                "lastModified": last_modified,
+                "name": name,
+                "owner": owner,
+                "path": path,
+                "permissions": permissions,
+                "size": size,
+            }
+        )
 
         return field_dict
 
@@ -70,19 +64,19 @@ class File:
         if not src_dict:
             return None
         d = src_dict.copy()
-        group = d.pop("group", UNSET)
+        group = d.pop("group")
 
-        last_modified = d.pop("lastModified", UNSET)
+        last_modified = d.pop("lastModified")
 
-        name = d.pop("name", UNSET)
+        name = d.pop("name")
 
-        owner = d.pop("owner", UNSET)
+        owner = d.pop("owner")
 
-        path = d.pop("path", UNSET)
+        path = d.pop("path")
 
-        permissions = d.pop("permissions", UNSET)
+        permissions = d.pop("permissions")
 
-        size = d.pop("size", UNSET)
+        size = d.pop("size")
 
         file = cls(
             group=group,

blaxel/core/sandbox/client/models/file_with_content.py

@@ -1,10 +1,8 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="FileWithContent")
 
 
@@ -12,24 +10,24 @@ T = TypeVar("T", bound="FileWithContent")
 class FileWithContent:
     """
     Attributes:
-        content (Union[Unset, str]):
-        group (Union[Unset, str]):
-        last_modified (Union[Unset, str]):
-        name (Union[Unset, str]):
-        owner (Union[Unset, str]):
-        path (Union[Unset, str]):
-        permissions (Union[Unset, str]):
-        size (Union[Unset, int]):
+        content (str):
+        group (str):
+        last_modified (str):
+        name (str):
+        owner (str):
+        path (str):
+        permissions (str):
+        size (int):
     """
 
-    content: Union[Unset, str] = UNSET
-    group: Union[Unset, str] = UNSET
-    last_modified: Union[Unset, str] = UNSET
-    name: Union[Unset, str] = UNSET
-    owner: Union[Unset, str] = UNSET
-    path: Union[Unset, str] = UNSET
-    permissions: Union[Unset, str] = UNSET
-    size: Union[Unset, int] = UNSET
+    content: str
+    group: str
+    last_modified: str
+    name: str
+    owner: str
+    path: str
+    permissions: str
+    size: int
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -51,23 +49,18 @@ class FileWithContent:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if content is not UNSET:
-            field_dict["content"] = content
-        if group is not UNSET:
-            field_dict["group"] = group
-        if last_modified is not UNSET:
-            field_dict["lastModified"] = last_modified
-        if name is not UNSET:
-            field_dict["name"] = name
-        if owner is not UNSET:
-            field_dict["owner"] = owner
-        if path is not UNSET:
-            field_dict["path"] = path
-        if permissions is not UNSET:
-            field_dict["permissions"] = permissions
-        if size is not UNSET:
-            field_dict["size"] = size
+        field_dict.update(
+            {
+                "content": content,
+                "group": group,
+                "lastModified": last_modified,
+                "name": name,
+                "owner": owner,
+                "path": path,
+                "permissions": permissions,
+                "size": size,
+            }
+        )
 
         return field_dict
 
@@ -76,21 +69,21 @@ class FileWithContent:
         if not src_dict:
             return None
         d = src_dict.copy()
-        content = d.pop("content", UNSET)
+        content = d.pop("content")
 
-        group = d.pop("group", UNSET)
+        group = d.pop("group")
 
-        last_modified = d.pop("lastModified", UNSET)
+        last_modified = d.pop("lastModified")
 
-        name = d.pop("name", UNSET)
+        name = d.pop("name")
 
-        owner = d.pop("owner", UNSET)
+        owner = d.pop("owner")
 
-        path = d.pop("path", UNSET)
+        path = d.pop("path")
 
-        permissions = d.pop("permissions", UNSET)
+        permissions = d.pop("permissions")
 
-        size = d.pop("size", UNSET)
+        size = d.pop("size")
 
         file_with_content = cls(
             content=content,

blaxel/core/sandbox/client/models/process_logs.py

@@ -1,10 +1,8 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="ProcessLogs")
 
 
@@ -12,14 +10,14 @@ T = TypeVar("T", bound="ProcessLogs")
 class ProcessLogs:
     """
     Attributes:
-        logs (Union[Unset, str]):  Example: logs output.
-        stderr (Union[Unset, str]):  Example: stderr output.
-        stdout (Union[Unset, str]):  Example: stdout output.
+        logs (str):  Example: logs output.
+        stderr (str):  Example: stderr output.
+        stdout (str):  Example: stdout output.
     """
 
-    logs: Union[Unset, str] = UNSET
-    stderr: Union[Unset, str] = UNSET
-    stdout: Union[Unset, str] = UNSET
+    logs: str
+    stderr: str
+    stdout: str
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -31,13 +29,13 @@ class ProcessLogs:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if logs is not UNSET:
-            field_dict["logs"] = logs
-        if stderr is not UNSET:
-            field_dict["stderr"] = stderr
-        if stdout is not UNSET:
-            field_dict["stdout"] = stdout
+        field_dict.update(
+            {
+                "logs": logs,
+                "stderr": stderr,
+                "stdout": stdout,
+            }
+        )
 
         return field_dict
 
@@ -46,11 +44,11 @@ class ProcessLogs:
         if not src_dict:
             return None
         d = src_dict.copy()
-        logs = d.pop("logs", UNSET)
+        logs = d.pop("logs")
 
-        stderr = d.pop("stderr", UNSET)
+        stderr = d.pop("stderr")
 
-        stdout = d.pop("stdout", UNSET)
+        stdout = d.pop("stdout")
 
         process_logs = cls(
             logs=logs,

blaxel/core/sandbox/client/models/process_response.py

@@ -1,10 +1,9 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
 from ..models.process_response_status import ProcessResponseStatus
-from ..types import UNSET, Unset
 
 T = TypeVar("T", bound="ProcessResponse")
 
@@ -13,24 +12,26 @@ T = TypeVar("T", bound="ProcessResponse")
 class ProcessResponse:
     """
     Attributes:
-        command (Union[Unset, str]):  Example: ls -la.
-        completed_at (Union[Unset, str]):  Example: Wed, 01 Jan 2023 12:01:00 GMT.
-        exit_code (Union[Unset, int]):
-        name (Union[Unset, str]):  Example: my-process.
-        pid (Union[Unset, str]):  Example: 1234.
-        started_at (Union[Unset, str]):  Example: Wed, 01 Jan 2023 12:00:00 GMT.
-        status (Union[Unset, ProcessResponseStatus]):  Example: running.
-        working_dir (Union[Unset, str]):  Example: /home/user.
+        command (str):  Example: ls -la.
+        completed_at (str):  Example: Wed, 01 Jan 2023 12:01:00 GMT.
+        exit_code (int):
+        logs (str):  Example: logs output.
+        name (str):  Example: my-process.
+        pid (str):  Example: 1234.
+        started_at (str):  Example: Wed, 01 Jan 2023 12:00:00 GMT.
+        status (ProcessResponseStatus):  Example: running.
+        working_dir (str):  Example: /home/user.
     """
 
-    command: Union[Unset, str] = UNSET
-    completed_at: Union[Unset, str] = UNSET
-    exit_code: Union[Unset, int] = UNSET
-    name: Union[Unset, str] = UNSET
-    pid: Union[Unset, str] = UNSET
-    started_at: Union[Unset, str] = UNSET
-    status: Union[Unset, ProcessResponseStatus] = UNSET
-    working_dir: Union[Unset, str] = UNSET
+    command: str
+    completed_at: str
+    exit_code: int
+    logs: str
+    name: str
+    pid: str
+    started_at: str
+    status: ProcessResponseStatus
+    working_dir: str
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -40,37 +41,33 @@ class ProcessResponse:
 
         exit_code = self.exit_code
 
+        logs = self.logs
+
         name = self.name
 
         pid = self.pid
 
         started_at = self.started_at
 
-        status: Union[Unset, str] = UNSET
-        if not isinstance(self.status, Unset):
-            status = self.status.value
+        status = self.status.value
 
         working_dir = self.working_dir
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if command is not UNSET:
-            field_dict["command"] = command
-        if completed_at is not UNSET:
-            field_dict["completedAt"] = completed_at
-        if exit_code is not UNSET:
-            field_dict["exitCode"] = exit_code
-        if name is not UNSET:
-            field_dict["name"] = name
-        if pid is not UNSET:
-            field_dict["pid"] = pid
-        if started_at is not UNSET:
-            field_dict["startedAt"] = started_at
-        if status is not UNSET:
-            field_dict["status"] = status
-        if working_dir is not UNSET:
-            field_dict["workingDir"] = working_dir
+        field_dict.update(
+            {
+                "command": command,
+                "completedAt": completed_at,
+                "exitCode": exit_code,
+                "logs": logs,
+                "name": name,
+                "pid": pid,
+                "startedAt": started_at,
+                "status": status,
+                "workingDir": working_dir,
+            }
+        )
 
         return field_dict
 
@@ -79,31 +76,29 @@ class ProcessResponse:
         if not src_dict:
             return None
         d = src_dict.copy()
-        command = d.pop("command", UNSET)
+        command = d.pop("command")
+
+        completed_at = d.pop("completedAt")
 
-        completed_at = d.pop("completedAt", UNSET)
+        exit_code = d.pop("exitCode")
 
-        exit_code = d.pop("exitCode", UNSET)
+        logs = d.pop("logs")
 
-        name = d.pop("name", UNSET)
+        name = d.pop("name")
 
-        pid = d.pop("pid", UNSET)
+        pid = d.pop("pid")
 
-        started_at = d.pop("startedAt", UNSET)
+        started_at = d.pop("startedAt")
 
-        _status = d.pop("status", UNSET)
-        status: Union[Unset, ProcessResponseStatus]
-        if isinstance(_status, Unset):
-            status = UNSET
-        else:
-            status = ProcessResponseStatus(_status)
+        status = ProcessResponseStatus(d.pop("status"))
 
-        working_dir = d.pop("workingDir", UNSET)
+        working_dir = d.pop("workingDir")
 
         process_response = cls(
             command=command,
             completed_at=completed_at,
             exit_code=exit_code,
+            logs=logs,
             name=name,
             pid=pid,
             started_at=started_at,

blaxel/core/sandbox/client/models/subdirectory.py

@@ -1,10 +1,8 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="Subdirectory")
 
 
@@ -12,12 +10,12 @@ T = TypeVar("T", bound="Subdirectory")
 class Subdirectory:
     """
     Attributes:
-        name (Union[Unset, str]):
-        path (Union[Unset, str]):
+        name (str):
+        path (str):
     """
 
-    name: Union[Unset, str] = UNSET
-    path: Union[Unset, str] = UNSET
+    name: str
+    path: str
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -27,11 +25,12 @@ class Subdirectory:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if name is not UNSET:
-            field_dict["name"] = name
-        if path is not UNSET:
-            field_dict["path"] = path
+        field_dict.update(
+            {
+                "name": name,
+                "path": path,
+            }
+        )
 
         return field_dict
 
@@ -40,9 +39,9 @@ class Subdirectory:
         if not src_dict:
             return None
         d = src_dict.copy()
-        name = d.pop("name", UNSET)
+        name = d.pop("name")
 
-        path = d.pop("path", UNSET)
+        path = d.pop("path")
 
         subdirectory = cls(
             name=name,

blaxel/core/sandbox/client/models/success_response.py

@@ -1,10 +1,8 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="SuccessResponse")
 
 
@@ -12,12 +10,12 @@ T = TypeVar("T", bound="SuccessResponse")
 class SuccessResponse:
     """
     Attributes:
-        message (Union[Unset, str]):  Example: File created successfully.
-        path (Union[Unset, str]):  Example: /path/to/file.
+        message (str):  Example: File created successfully.
+        path (str):  Example: /path/to/file.
     """
 
-    message: Union[Unset, str] = UNSET
-    path: Union[Unset, str] = UNSET
+    message: str
+    path: str
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -27,11 +25,12 @@ class SuccessResponse:
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if message is not UNSET:
-            field_dict["message"] = message
-        if path is not UNSET:
-            field_dict["path"] = path
+        field_dict.update(
+            {
+                "message": message,
+                "path": path,
+            }
+        )
 
         return field_dict
 
@@ -40,9 +39,9 @@ class SuccessResponse:
         if not src_dict:
             return None
         d = src_dict.copy()
-        message = d.pop("message", UNSET)
+        message = d.pop("message")
 
-        path = d.pop("path", UNSET)
+        path = d.pop("path")
 
         success_response = cls(
             message=message,

blaxel/core/sandbox/process.py

@@ -10,21 +10,6 @@ from .client.models.process_request import ProcessRequest
 from .types import SandboxConfiguration
 
 
-class ProcessResponseWithLogs(ProcessResponse):
-    """Extended ProcessResponse that adds a logs property."""
-
-    logs: Optional[str] = None
-
-    @classmethod
-    def from_process_response(cls, process_response: ProcessResponse, logs: Optional[str] = None):
-        """Create ProcessResponseWithLogs from an existing ProcessResponse."""
-        # Create a new instance from the dict representation
-        instance = cls.from_dict(process_response.to_dict())
-        # Add the logs
-        instance.logs = logs
-        return instance
-
-
 class SandboxProcess(SandboxAction):
     def __init__(self, sandbox_config: SandboxConfiguration):
         super().__init__(sandbox_config)
@@ -94,13 +79,19 @@ class SandboxProcess(SandboxAction):
         self,
         process: Union[ProcessRequest, Dict[str, Any]],
         on_log: Optional[Callable[[str], None]] = None,
-    ) -> ProcessResponseWithLogs:
+    ) -> ProcessResponse:
         if isinstance(process, dict):
             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
             response_data = None
             if response.content:
@@ -114,32 +105,24 @@ class SandboxProcess(SandboxAction):
             self.handle_response_error(response, response_data, None)
             result = ProcessResponse.from_dict(response_data)
 
-            # Setup log streaming if on_log is provided
-            stream_control = None
-            if on_log is not None:
-                stream_control = self.stream_logs(result.pid, {"on_log": on_log})
-
-            # Wait for completion if requested
-            logs = None
-            if process.wait_for_completion:
+            # Handle wait_for_completion with parallel log streaming
+            if should_wait_for_completion:
+                stream_control = None
+                if on_log is not None:
+                    stream_control = self.stream_logs(result.pid, {"on_log": on_log})
                 try:
-                    # Wait for process to complete
-                    result = await self.wait(result.pid)
-
-                    # Get the logs
-                    logs = await self.logs(result.pid, "all")
-                except Exception as e:
-                    # If waiting fails, still try to return what we have
+                    # Wait for process completion
+                    result = await self.wait(result.pid, interval=50)
+                finally:
+                    # Clean up log streaming
                     if stream_control:
                         stream_control["close"]()
-                    raise e
-
-            # Close stream if it was opened and we're done
-            if stream_control and process.wait_for_completion:
-                stream_control["close"]()
-
-            # Return wrapped response with logs
-            return ProcessResponseWithLogs.from_process_response(result, logs)
+            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"]
+            return result
 
     async def wait(
         self, identifier: str, max_wait: int = 60000, interval: int = 1000

{blaxel-0.2.1rc73.dist-info → blaxel-0.2.1rc75.dist-info}/METADATA

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

{blaxel-0.2.1rc73.dist-info → blaxel-0.2.1rc75.dist-info}/RECORD

@@ -309,7 +309,7 @@ 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=M6FulOxPghUBpb5fLxu1Rd3ekLeCbZ_dgt4s1X2Cneo,5354
-blaxel/core/sandbox/process.py,sha256=jgOPL5yLzhd98CLO7nBUABIAPFK5S5z34h6L_TyXsuU,8753
+blaxel/core/sandbox/process.py,sha256=7zEngDTs2XiNsMm9TZ4lEDUiRpS3af8dw60pEvRuHDY,8357
 blaxel/core/sandbox/sandbox.py,sha256=tsRGvkb2dnGM5pmOiIxWvOmT42P9BDXWtW3h6_CIffI,5802
 blaxel/core/sandbox/session.py,sha256=4SH1tyXcQ9UqJx4lMwxAlp7x9Te_anDrdSEG6AlNkvU,4496
 blaxel/core/sandbox/types.py,sha256=gmNAt8x7PrJZHpw2_2aWWDE5uG6H-uEgdIKTYIw4G0g,2981
@@ -340,21 +340,21 @@ blaxel/core/sandbox/client/api/process/get_ws_process_identifier_logs_stream.py,
 blaxel/core/sandbox/client/api/process/post_process.py,sha256=CgA1Q0vLAfeOT5nMd2ESrO8d6_0kGsrcBs1LSEVZrjo,4656
 blaxel/core/sandbox/client/models/__init__.py,sha256=eOax5cMg8O7Skpi0vApItrajTY0gx-wRHVe3YC5wBt8,1356
 blaxel/core/sandbox/client/models/delete_network_process_pid_monitor_response_200.py,sha256=9cQgKDjG98sMridjXKgeR2oZzFKcQ0G9QIojhwYFosI,1376
-blaxel/core/sandbox/client/models/directory.py,sha256=TlL3WlWhtdfuaqw4dzNVqnyPcApmwlOoImW25tgvIvg,3821
-blaxel/core/sandbox/client/models/error_response.py,sha256=lI15zKBoD2X9yHzSiEaYGUn5TPTxWM7j1Tu5crtd23M,1581
-blaxel/core/sandbox/client/models/file.py,sha256=LXjXjnY2p3frh5utiptpVNQGlRSh3gCKhmDxlHAIN6w,3037
+blaxel/core/sandbox/client/models/directory.py,sha256=Cufag2aJuBHXfxDnfiXL4oEiQE5GSb5FWH7Tu_hOhtc,3186
+blaxel/core/sandbox/client/models/error_response.py,sha256=4UZjR8ej0uFPl4CmkMplYebqS3U9EDodqNBltQtGvxI,1493
+blaxel/core/sandbox/client/models/file.py,sha256=MkYNSh9Rj1Gt0txfBeFEcEVg1Z-Z1Tbp31nv3C_WZaA,2446
 blaxel/core/sandbox/client/models/file_request.py,sha256=xOZSru-fae-En-_2YBgkHa_6iGbqbJsG3RLqBuajVY0,2227
-blaxel/core/sandbox/client/models/file_with_content.py,sha256=55vccuRO-n613Ej7W5DXAd7cifHqOUeSSSD27Vf7RP8,3355
+blaxel/core/sandbox/client/models/file_with_content.py,sha256=iDMYClAJBJK-l1Kr0yKuNoiHNG3zJVt44XLsE1ax3IA,2680
 blaxel/core/sandbox/client/models/get_network_process_pid_ports_response_200.py,sha256=x4uv80kK0GVroWO98l5sE84a6uwZ8pnUKTpGg81ipWA,1351
 blaxel/core/sandbox/client/models/port_monitor_request.py,sha256=LK7sjAK1TF1ojgG4vGytaKLVtV6-SNXxfZ3sxew1cRE,1698
 blaxel/core/sandbox/client/models/post_network_process_pid_monitor_response_200.py,sha256=Y8BvNGKU8SlzTGqhaQZk_WWIrmFpNU0LVcmLFjNvqhA,1366
-blaxel/core/sandbox/client/models/process_logs.py,sha256=2yMnJcEDBFX3nWqv-msVqGXLMQ_pfIJF-UVdO03a0DY,2100
+blaxel/core/sandbox/client/models/process_logs.py,sha256=WVBgPsvUzPlBIuJzahGor-mXq7RF4qXhi5e9XcUPyOM,1847
 blaxel/core/sandbox/client/models/process_request.py,sha256=b5WoVKQtKibRw1a1xN6nv7Gj4wiAWGPL_Kj9WmjGKsE,4134
 blaxel/core/sandbox/client/models/process_request_env.py,sha256=hzVPY4mk4g4bKj_S3uXRyaXvp1V6WB2hlQM0XhB7Xgw,1294
-blaxel/core/sandbox/client/models/process_response.py,sha256=13UN4aTg0OZJnABCD-bzsHNzVRqkeJtsuqDaK0DsyBI,4011
+blaxel/core/sandbox/client/models/process_response.py,sha256=vXHK6NBnuf0KYVzpkaApC29KOKwDE-7jBxy_WszApXs,3239
 blaxel/core/sandbox/client/models/process_response_status.py,sha256=hCE1gCtheV83T6PYURG3k-rPZSauvyTFhsxLEtudgYE,246
-blaxel/core/sandbox/client/models/subdirectory.py,sha256=j1ORkqO74vAdPQ-oUMer85P8Gb2mAwky3P2gBCnqyGQ,1761
-blaxel/core/sandbox/client/models/success_response.py,sha256=JQbCUIdVJy_TJ3mp8IuvCGbKgCm_iZQMMrqn8uZkxCk,1874
+blaxel/core/sandbox/client/models/subdirectory.py,sha256=obztIgi_XjT4pPoT1-dyOX6O8k0e4ERZUTiI9qgkhJM,1593
+blaxel/core/sandbox/client/models/success_response.py,sha256=ksioQ0qNV7yhXlySY3_3EycsJ1ec7LSxBMwIq4HnfVQ,1703
 blaxel/core/tools/__init__.py,sha256=9_pgKEU5GbPfqfGN4aZTkthnaV4AVYf63jIZw44peBQ,10167
 blaxel/core/tools/common.py,sha256=JGK052v_fvwWBFYnIArlBnFFViYyFrqdDn3gdVf53EU,1332
 blaxel/core/tools/types.py,sha256=YPCGJ4vZDhqR0X2H_TWtc5chQScsC32nGTQdRKJlO8Y,707
@@ -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.1rc73.dist-info/METADATA,sha256=7M9mgqu97Tyk403KOchTBU9BzUlFHr8VXZ6JkuC0riM,9879
-blaxel-0.2.1rc73.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.1rc73.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.1rc73.dist-info/RECORD,,
+blaxel-0.2.1rc75.dist-info/METADATA,sha256=y5tYQNgSQuYwy1K_GYhy_h_oxz5iKupOB3-U_zVRN9I,9879
+blaxel-0.2.1rc75.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.1rc75.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.1rc75.dist-info/RECORD,,