blaxel 0.2.0rc6__py3-none-any.whl → 0.2.1__py3-none-any.whl

blaxel/core/sandbox/__init__.py

@@ -8,6 +8,7 @@ from .sandbox import (
 from .types import (
     CopyResponse,
     SandboxConfiguration,
+    SandboxCreateConfiguration,
     SandboxFilesystemFile,
     SessionCreateOptions,
     SessionWithToken,
@@ -19,6 +20,7 @@ __all__ = [
     "SessionCreateOptions",
     "SessionWithToken",
     "SandboxConfiguration",
+    "SandboxCreateConfiguration",
     "WatchEvent",
     "SandboxFilesystemFile",
     "CopyResponse",

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

@@ -75,16 +75,54 @@ class SandboxProcess(SandboxAction):
 
         return {"close": close}
 
-    async def exec(self, process: Union[ProcessRequest, Dict[str, Any]]) -> ProcessResponse:
+    async def exec(
+        self,
+        process: Union[ProcessRequest, Dict[str, Any]],
+        on_log: Optional[Callable[[str], None]] = None,
+    ) -> 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())
-            self.handle_response_error(
-                response, response.json() if response.content else None, None
-            )
-            return ProcessResponse.from_dict(response.json())
+            # Parse JSON response only once, with better error handling
+            response_data = None
+            if response.content:
+                try:
+                    response_data = response.json()
+                except Exception:
+                    # If JSON parsing fails, check the response first
+                    self.handle_response_error(response, None, None)
+                    raise
+
+            self.handle_response_error(response, response_data, None)
+            result = ProcessResponse.from_dict(response_data)
+
+            # 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 completion
+                    result = await self.wait(result.pid, interval=50)
+                finally:
+                    # Clean up log streaming
+                    if stream_control:
+                        stream_control["close"]()
+            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/core/sandbox/sandbox.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging
 import time
+import uuid
 from typing import Any, Dict, List, Union
 
 from ..client.api.compute.create_sandbox import asyncio as create_sandbox
@@ -8,13 +9,13 @@ from ..client.api.compute.delete_sandbox import asyncio as delete_sandbox
 from ..client.api.compute.get_sandbox import asyncio as get_sandbox
 from ..client.api.compute.list_sandboxes import asyncio as list_sandboxes
 from ..client.client import client
-from ..client.models import Metadata, Sandbox
+from ..client.models import Metadata, Runtime, Sandbox, SandboxSpec
 from .filesystem import SandboxFileSystem
 from .network import SandboxNetwork
 from .preview import SandboxPreviews
 from .process import SandboxProcess
 from .session import SandboxSessions
-from .types import SandboxConfiguration, SessionWithToken
+from .types import SandboxConfiguration, SandboxCreateConfiguration, SessionWithToken
 
 logger = logging.getLogger(__name__)
 
@@ -67,15 +68,54 @@ class SandboxInstance:
                 raise Exception("Sandbox did not deploy in time")
 
     @classmethod
-    async def create(cls, sandbox: Union[Sandbox, Dict[str, Any]]) -> "SandboxInstance":
-        if isinstance(sandbox, dict):
-            sandbox = Sandbox.from_dict(sandbox)
-
-        if not sandbox.spec:
-            raise Exception("Sandbox spec is required")
-        if not sandbox.spec.runtime:
-            raise Exception("Sandbox runtime is required")
-        sandbox.spec.runtime.generation = sandbox.spec.runtime.generation or "mk3"
+    async def create(
+        cls, sandbox: Union[Sandbox, SandboxCreateConfiguration, Dict[str, Any], None] = None
+    ) -> "SandboxInstance":
+        # Generate default values
+        default_name = f"sandbox-{uuid.uuid4().hex[:8]}"
+        default_image = "blaxel/prod-base:latest"
+        default_memory = 4096
+
+        # Handle SandboxCreateConfiguration or simple dict with name/image/memory keys
+        if sandbox is None or isinstance(sandbox, (SandboxCreateConfiguration, dict)) and (
+            not isinstance(sandbox, Sandbox) and (
+                sandbox is None or
+                'name' in (sandbox if isinstance(sandbox, dict) else sandbox.__dict__) or
+                'image' in (sandbox if isinstance(sandbox, dict) else sandbox.__dict__) or
+                'memory' in (sandbox if isinstance(sandbox, dict) else sandbox.__dict__)
+            )
+        ):
+            if sandbox is None:
+                sandbox = SandboxCreateConfiguration()
+            elif isinstance(sandbox, dict) and not isinstance(sandbox, Sandbox):
+                sandbox = SandboxCreateConfiguration.from_dict(sandbox)
+
+            # Set defaults if not provided
+            name = sandbox.name or default_name
+            image = sandbox.image or default_image
+            memory = sandbox.memory or default_memory
+
+            # Create full Sandbox object
+            sandbox = Sandbox(
+                metadata=Metadata(name=name),
+                spec=SandboxSpec(runtime=Runtime(image=image, memory=memory, generation="mk3"))
+            )
+        else:
+            # Handle existing Sandbox object or dict conversion
+            if isinstance(sandbox, dict):
+                sandbox = Sandbox.from_dict(sandbox)
+
+            # Set defaults for missing fields
+            if not sandbox.metadata:
+                sandbox.metadata = Metadata(name=uuid.uuid4().hex.replace('-', ''))
+            if not sandbox.spec:
+                sandbox.spec = SandboxSpec(runtime=Runtime(image=default_image))
+            if not sandbox.spec.runtime:
+                sandbox.spec.runtime = Runtime(image=default_image, memory=default_memory)
+
+            sandbox.spec.runtime.image = sandbox.spec.runtime.image or default_image
+            sandbox.spec.runtime.memory = sandbox.spec.runtime.memory or default_memory
+            sandbox.spec.runtime.generation = sandbox.spec.runtime.generation or "mk3"
 
         response = await create_sandbox(
             client=client,
@@ -106,14 +146,30 @@ class SandboxInstance:
 
     @classmethod
     async def create_if_not_exists(
-        cls, sandbox: Union[Sandbox, Dict[str, Any]]
+        cls, sandbox: Union[Sandbox, SandboxCreateConfiguration, Dict[str, Any]]
     ) -> "SandboxInstance":
         """Create a sandbox if it doesn't exist, otherwise return existing."""
-        if isinstance(sandbox, dict):
-            sandbox = Sandbox.from_dict(sandbox)
-
         try:
-            sandbox_instance = await cls.get(sandbox.metadata.name)
+            # Extract name from different configuration types
+            if isinstance(sandbox, SandboxCreateConfiguration):
+                name = sandbox.name
+            elif isinstance(sandbox, dict):
+                if 'name' in sandbox:
+                    name = sandbox['name']
+                elif 'metadata' in sandbox and isinstance(sandbox['metadata'], dict):
+                    name = sandbox['metadata'].get('name')
+                else:
+                    # If no name provided, we can't check if it exists, so create new
+                    return await cls.create(sandbox)
+            elif isinstance(sandbox, Sandbox):
+                name = sandbox.metadata.name if sandbox.metadata else None
+            else:
+                name = None
+
+            if not name:
+                raise ValueError("Sandbox name is required")
+
+            sandbox_instance = await cls.get(name)
             return sandbox_instance
         except Exception as e:
             # Check if it's a 404 error (sandbox not found)

blaxel/core/sandbox/types.py

@@ -101,3 +101,24 @@ class CopyResponse:
         self.message = message
         self.source = source
         self.destination = destination
+
+
+class SandboxCreateConfiguration:
+    """Simplified configuration for creating sandboxes with default values."""
+    def __init__(
+        self,
+        name: Optional[str] = None,
+        image: Optional[str] = None,
+        memory: Optional[int] = None,
+    ):
+        self.name = name
+        self.image = image
+        self.memory = memory
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "SandboxCreateConfiguration":
+        return cls(
+            name=data.get("name"),
+            image=data.get("image"),
+            memory=data.get("memory"),
+        )

blaxel/llamaindex/custom/cohere.py

@@ -47,7 +47,6 @@ class Cohere(FunctionCallingLLM):
 
         llm = Cohere(model="command", api_key=api_key)
         resp = llm.complete("Paul Graham is ")
-        print(resp)
         ```
     """
 
@@ -55,9 +54,7 @@ class Cohere(FunctionCallingLLM):
     temperature: Optional[float] = Field(
         description="The temperature to use for sampling.", default=None
     )
-    max_retries: int = Field(
-        default=10, description="The maximum number of API retries."
-    )
+    max_retries: int = Field(default=10, description="The maximum number of API retries.")
     additional_kwargs: Dict[str, Any] = Field(
         default_factory=dict, description="Additional kwargs for the Cohere API."
     )
@@ -167,9 +164,7 @@ class Cohere(FunctionCallingLLM):
         error_on_no_tool_call: bool = False,
     ) -> List[ToolSelection]:
         """Predict and call the tool."""
-        tool_calls: List[ToolCall] = (
-            response.message.additional_kwargs.get("tool_calls", []) or []
-        )
+        tool_calls: List[ToolCall] = response.message.additional_kwargs.get("tool_calls", []) or []
 
         if len(tool_calls) < 1 and error_on_no_tool_call:
             raise ValueError(
@@ -218,10 +213,8 @@ class Cohere(FunctionCallingLLM):
 
         messages, documents = remove_documents_from_messages(messages)
 
-        tool_results: Optional[
-            List[Dict[str, Any]]
-        ] = _messages_to_cohere_tool_results_curr_chat_turn(messages) or kwargs.get(
-            "tool_results"
+        tool_results: Optional[List[Dict[str, Any]]] = (
+            _messages_to_cohere_tool_results_curr_chat_turn(messages) or kwargs.get("tool_results")
         )
         if not tool_results:
             tool_results = None
@@ -235,12 +228,8 @@ class Cohere(FunctionCallingLLM):
                 if message.role == MessageRole.TOOL:
                     temp_tool_results += _message_to_cohere_tool_results(messages, i)
 
-                    if (i == len(messages) - 1) or messages[
-                        i + 1
-                    ].role != MessageRole.TOOL:
-                        cohere_message = _get_message_cohere_format(
-                            message, temp_tool_results
-                        )
+                    if (i == len(messages) - 1) or messages[i + 1].role != MessageRole.TOOL:
+                        cohere_message = _get_message_cohere_format(message, temp_tool_results)
                         chat_history.append(cohere_message)
                         temp_tool_results = []
                 else:
@@ -262,12 +251,8 @@ class Cohere(FunctionCallingLLM):
                 if message.role == MessageRole.TOOL:
                     temp_tool_results += _message_to_cohere_tool_results(messages, i)
 
-                    if (i == len(messages) - 1) or messages[
-                        i + 1
-                    ].role != MessageRole.TOOL:
-                        cohere_message = _get_message_cohere_format(
-                            message, temp_tool_results
-                        )
+                    if (i == len(messages) - 1) or messages[i + 1].role != MessageRole.TOOL:
+                        cohere_message = _get_message_cohere_format(message, temp_tool_results)
                         chat_history.append(cohere_message)
                         temp_tool_results = []
                 else:
@@ -327,9 +312,7 @@ class Cohere(FunctionCallingLLM):
         )
 
     @llm_completion_callback()
-    def complete(
-        self, prompt: str, formatted: bool = False, **kwargs: Any
-    ) -> CompletionResponse:
+    def complete(self, prompt: str, formatted: bool = False, **kwargs: Any) -> CompletionResponse:
         all_kwargs = self._get_all_kwargs(**kwargs)
         if "stream" in all_kwargs:
             warnings.warn(
@@ -351,9 +334,7 @@ class Cohere(FunctionCallingLLM):
         )
 
     @llm_chat_callback()
-    def stream_chat(
-        self, messages: Sequence[ChatMessage], **kwargs: Any
-    ) -> ChatResponseGen:
+    def stream_chat(self, messages: Sequence[ChatMessage], **kwargs: Any) -> ChatResponseGen:
         all_kwargs = self._get_all_kwargs(**kwargs)
         all_kwargs["stream"] = True
         if all_kwargs["model"] not in CHAT_MODELS:
@@ -402,16 +383,12 @@ class Cohere(FunctionCallingLLM):
             for r in response:
                 content_delta = r.text
                 content += content_delta
-                yield CompletionResponse(
-                    text=content, delta=content_delta, raw=r._asdict()
-                )
+                yield CompletionResponse(text=content, delta=content_delta, raw=r._asdict())
 
         return gen()
 
     @llm_chat_callback()
-    async def achat(
-        self, messages: Sequence[ChatMessage], **kwargs: Any
-    ) -> ChatResponse:
+    async def achat(self, messages: Sequence[ChatMessage], **kwargs: Any) -> ChatResponse:
         all_kwargs = self._get_all_kwargs(**kwargs)
         if all_kwargs["model"] not in CHAT_MODELS:
             raise ValueError(f"{all_kwargs['model']} not supported for chat")
@@ -535,8 +512,6 @@ class Cohere(FunctionCallingLLM):
             async for r in response:
                 content_delta = r.text
                 content += content_delta
-                yield CompletionResponse(
-                    text=content, delta=content_delta, raw=r._asdict()
-                )
+                yield CompletionResponse(text=content, delta=content_delta, raw=r._asdict())
 
         return gen()

blaxel/telemetry/instrumentation/blaxel_core.py

@@ -10,9 +10,10 @@ from blaxel.telemetry.span import SpanManager
 
 logger = logging.getLogger(__name__)
 
+
 class BlaxelCoreInstrumentor(BaseInstrumentor):
     def instrumentation_dependencies(self):
-        return ["blaxel.core"]
+        return []
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")
@@ -121,4 +122,4 @@ class BlaxelCoreInstrumentor(BaseInstrumentor):
             Tool.sync_coroutine = traced_sync_coroutine
             return Tool
 
-        blaxel.core.tools.convert_mcp_tool_to_blaxel_tool = traced_convert_mcp_tool_to_blaxel_tool
+        blaxel.core.tools.convert_mcp_tool_to_blaxel_tool = traced_convert_mcp_tool_to_blaxel_tool

blaxel/telemetry/instrumentation/blaxel_langgraph.py

@@ -11,7 +11,7 @@ logger = logging.getLogger(__name__)
 
 class BlaxelLanggraphInstrumentor(BaseInstrumentor):
     def instrumentation_dependencies(self):
-        return ["blaxel.langgraph", "blaxel.core"]
+        return ["langgraph"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")

blaxel/telemetry/instrumentation/blaxel_langgraph_gemini.py

@@ -316,7 +316,7 @@ class BlaxelLanggraphGeminiInstrumentor(BaseInstrumentor):
         Config.exception_logger = exception_logger
 
     def instrumentation_dependencies(self) -> Collection[str]:
-        return ["blaxel.langgraph"]
+        return ["langgraph"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")

blaxel/telemetry/instrumentation/blaxel_llamaindex.py

@@ -30,7 +30,7 @@ TO_INSTRUMENT = [
 
 class BlaxelLlamaIndexInstrumentor(BaseInstrumentor):
     def instrumentation_dependencies(self):
-        return ["blaxel.llamaindex", "blaxel.core"]
+        return ["llama_index"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")

blaxel/telemetry/instrumentation/map.py

@@ -35,28 +35,28 @@ MAPPINGS: Dict[str, InstrumentationMapping] = {
         required_packages=["google-generativeai"],
         ignore_if_packages=[],
     ),
-    "blaxel.core": InstrumentationMapping(
+    "blaxel_core": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_core",
         class_name="BlaxelCoreInstrumentor",
-        required_packages=["blaxel.core"],
+        required_packages=[],
         ignore_if_packages=[],
     ),
-    "blaxel.langgraph": InstrumentationMapping(
+    "blaxel_langgraph": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_langgraph",
         class_name="BlaxelLanggraphInstrumentor",
-        required_packages=["blaxel.langgraph"],
+        required_packages=["langgraph"],
         ignore_if_packages=[],
     ),
-    "blaxel.langgraph_gemini": InstrumentationMapping(
+    "blaxel_langgraph_gemini": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_langgraph_gemini",
         class_name="BlaxelLanggraphGeminiInstrumentor",
-        required_packages=["blaxel.langgraph"],
+        required_packages=["langgraph"],
         ignore_if_packages=[],
     ),
-    "blaxel.llamaindex": InstrumentationMapping(
+    "blaxel_llamaindex": InstrumentationMapping(
         module_path="blaxel.telemetry.instrumentation.blaxel_llamaindex",
         class_name="BlaxelLlamaIndexInstrumentor",
-        required_packages=["blaxel_llamaindex"],
+        required_packages=["llama_index"],
         ignore_if_packages=[],
     ),
 }

{blaxel-0.2.0rc6.dist-info → blaxel-0.2.1.dist-info}/METADATA

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

{blaxel-0.2.0rc6.dist-info → blaxel-0.2.1.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=eWTH7PWNQZe9efqParwgXzltc7L4OyPtT_REqCdQybU,551
+blaxel/core/sandbox/__init__.py,sha256=oF3sX5MbwSqfwhOtF5ODYWwapHffbkp2UI78jPBn78U,617
 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=M6FulOxPghUBpb5fLxu1Rd3ekLeCbZ_dgt4s1X2Cneo,5354
-blaxel/core/sandbox/process.py,sha256=FahDx-FOqXPV3ajVI6aKdXqx4Oi4fK3egojHjTGy8ro,6705
-blaxel/core/sandbox/sandbox.py,sha256=an06JrTl2kfS5guYo0ZT2A-Uu5Zr7BQAu-fWXcCZnhY,5368
+blaxel/core/sandbox/process.py,sha256=7zEngDTs2XiNsMm9TZ4lEDUiRpS3af8dw60pEvRuHDY,8357
+blaxel/core/sandbox/sandbox.py,sha256=4sG_i_VcuGx_cKEGMY-2eUUuGmtQBYMNIToL7YDuluM,8151
 blaxel/core/sandbox/session.py,sha256=4SH1tyXcQ9UqJx4lMwxAlp7x9Te_anDrdSEG6AlNkvU,4496
-blaxel/core/sandbox/types.py,sha256=gmNAt8x7PrJZHpw2_2aWWDE5uG6H-uEgdIKTYIw4G0g,2981
+blaxel/core/sandbox/types.py,sha256=yaKkp5Sn0b4Ri2_ZbdEjXuA-rjo2_DfDIHYz-cueZCA,3564
 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
@@ -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
@@ -379,7 +379,7 @@ blaxel/llamaindex/__init__.py,sha256=iZ3QbZhlwKvP91ChcqSXVkpRrzurMxJoQfKdZFzE2AA
 blaxel/llamaindex/model.py,sha256=FYUCTA80LKC4-UfESh43htoHrsoZjwwVaxXD51T0IuE,2557
 blaxel/llamaindex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 blaxel/llamaindex/tools.py,sha256=LgrKe-o0I8_JE4DxcM2YYlHN-15LKmXuBKm08NYLV9w,907
-blaxel/llamaindex/custom/cohere.py,sha256=Igj5Y1ozf1V4feIXfBHDdaTFU7od_wuOhm0yChZNxMY,19109
+blaxel/llamaindex/custom/cohere.py,sha256=zoUv4NWwMZLZEynYTC3JuOvtinguRgtuRwS31wIm3rI,18713
 blaxel/openai/__init__.py,sha256=YkizVtcYL2m9v-z5B1EReYVu9n9V-DCxJhSB2mvqOs0,123
 blaxel/openai/model.py,sha256=lGz4zrV4sURPb6aLtroMRV5-CgfQUq15PeTgjc7QkTI,600
 blaxel/openai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -394,15 +394,15 @@ blaxel/telemetry/exporters.py,sha256=EoX3uaBVku1Rg49pSNXKFyHhgY5OV3Ih6UlqgjF5epw
 blaxel/telemetry/manager.py,sha256=3yYBxxqQKl1rCKrn0GVz9jR5jouC1nsElbAaH8tTtgA,9075
 blaxel/telemetry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 blaxel/telemetry/span.py,sha256=b8lpWe5nqek4w9YrBLJCIYz4Z3n9Z_49H_AkqpAGOiM,3660
-blaxel/telemetry/instrumentation/blaxel_core.py,sha256=gY2QoAr2UgnMabLyoxR1X79SVUaFM_hQJIzDdy80SuY,4892
-blaxel/telemetry/instrumentation/blaxel_langgraph.py,sha256=5C6mM0aKa8rbaAmhe7Z3HWHhZ-A_BZ0iio9Hn-GvIVM,4257
-blaxel/telemetry/instrumentation/blaxel_langgraph_gemini.py,sha256=2_5vcaNU7fS7ZisSLadNpFfiuy4zeRUDONJXhlQ9L1M,11960
-blaxel/telemetry/instrumentation/blaxel_llamaindex.py,sha256=EvgFhLUv5QViz7rAMJs5w_Oks-5NOe1v7A3B9b0NxeM,3122
-blaxel/telemetry/instrumentation/map.py,sha256=D7htA88-H8r7jHutBWnr_KEL8OZqLvvnRbU9jWiAA8g,2207
+blaxel/telemetry/instrumentation/blaxel_core.py,sha256=7PNseKq7bOQ4Z6a8jW7AGm0ojcdcusjoiH_5iQsR5ic,4881
+blaxel/telemetry/instrumentation/blaxel_langgraph.py,sha256=sBxt8kzRjsFepm86vaCBbGvpYI4mWPK_PB3wiLxSuOk,4235
+blaxel/telemetry/instrumentation/blaxel_langgraph_gemini.py,sha256=hQbOaaMsHte_Igq_vhPgNXYd40IwuqPB0TUP-5VAtME,11953
+blaxel/telemetry/instrumentation/blaxel_llamaindex.py,sha256=HA4YUXrRYvG5U1J8MAJnymmmj4ZCDydv-pXXhrftHOA,3101
+blaxel/telemetry/instrumentation/map.py,sha256=PCzZJj39yiYVYJrxLBNP-NW-tjjYyTijwEDeI9njuDY,2174
 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.0rc6.dist-info/METADATA,sha256=aI1uHOODyBfwDgJAbMDwmZlI__mZnGgzX9u-tY1PO8g,9878
-blaxel-0.2.0rc6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.0rc6.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.0rc6.dist-info/RECORD,,
+blaxel-0.2.1.dist-info/METADATA,sha256=Fz0hc9Jnt47pb6WAgAlNHECYHluqJoPtRefCqjr6tdQ,9875
+blaxel-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.1.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.1.dist-info/RECORD,,