blaxel 0.1.21__py3-none-any.whl → 0.1.22rc70__py3-none-any.whl

blaxel/client/models/resource_log_response.py

@@ -0,0 +1,83 @@
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ResourceLogResponse")
+
+
+@_attrs_define
+class ResourceLogResponse:
+    """Response for a resource log
+
+    Attributes:
+        chart (Union[Unset, list[Any]]): Chart
+        logs (Union[Unset, list[Any]]): Logs
+        total_count (Union[Unset, int]): Total count of logs
+    """
+
+    chart: Union[Unset, list[Any]] = UNSET
+    logs: Union[Unset, list[Any]] = UNSET
+    total_count: Union[Unset, int] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        chart: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.chart, Unset):
+            chart = self.chart
+
+        logs: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.logs, Unset):
+            logs = self.logs
+
+        total_count = self.total_count
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if chart is not UNSET:
+            field_dict["chart"] = chart
+        if logs is not UNSET:
+            field_dict["logs"] = logs
+        if total_count is not UNSET:
+            field_dict["totalCount"] = total_count
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        chart = cast(list[Any], d.pop("chart", UNSET))
+
+        logs = cast(list[Any], d.pop("logs", UNSET))
+
+        total_count = d.pop("totalCount", UNSET)
+
+        resource_log_response = cls(
+            chart=chart,
+            logs=logs,
+            total_count=total_count,
+        )
+
+        resource_log_response.additional_properties = d
+        return resource_log_response
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/client/models/resource_metrics.py

@@ -6,6 +6,7 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.billable_time_metric import BillableTimeMetric
     from ..models.latency_metric import LatencyMetric
     from ..models.memory_allocation_metric import MemoryAllocationMetric
     from ..models.metric import Metric
@@ -30,6 +31,7 @@ class ResourceMetrics:
     """Metrics for a single resource deployment (eg. model deployment, function deployment)
 
     Attributes:
+        billable_time (Union[Unset, BillableTimeMetric]): Billable time metric
         inference_errors_global (Union[Unset, list['Metric']]): Array of metrics
         inference_global (Union[Unset, list['Metric']]): Array of metrics
         last_n_requests (Union[Unset, list['Metric']]): Array of metrics
@@ -59,6 +61,7 @@ class ResourceMetrics:
         token_total (Union[Unset, TokenTotalMetric]): Token total metric
     """
 
+    billable_time: Union[Unset, "BillableTimeMetric"] = UNSET
     inference_errors_global: Union[Unset, list["Metric"]] = UNSET
     inference_global: Union[Unset, list["Metric"]] = UNSET
     last_n_requests: Union[Unset, list["Metric"]] = UNSET
@@ -83,6 +86,16 @@ class ResourceMetrics:
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        billable_time: Union[Unset, dict[str, Any]] = UNSET
+        if (
+            self.billable_time
+            and not isinstance(self.billable_time, Unset)
+            and not isinstance(self.billable_time, dict)
+        ):
+            billable_time = self.billable_time.to_dict()
+        elif self.billable_time and isinstance(self.billable_time, dict):
+            billable_time = self.billable_time
+
         inference_errors_global: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.inference_errors_global, Unset):
             inference_errors_global = []
@@ -244,6 +257,8 @@ class ResourceMetrics:
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
+        if billable_time is not UNSET:
+            field_dict["billableTime"] = billable_time
         if inference_errors_global is not UNSET:
             field_dict["inferenceErrorsGlobal"] = inference_errors_global
         if inference_global is not UNSET:
@@ -291,6 +306,7 @@ class ResourceMetrics:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.billable_time_metric import BillableTimeMetric
         from ..models.latency_metric import LatencyMetric
         from ..models.memory_allocation_metric import MemoryAllocationMetric
         from ..models.metric import Metric
@@ -313,6 +329,13 @@ class ResourceMetrics:
         if not src_dict:
             return None
         d = src_dict.copy()
+        _billable_time = d.pop("billableTime", UNSET)
+        billable_time: Union[Unset, BillableTimeMetric]
+        if isinstance(_billable_time, Unset):
+            billable_time = UNSET
+        else:
+            billable_time = BillableTimeMetric.from_dict(_billable_time)
+
         inference_errors_global = []
         _inference_errors_global = d.pop("inferenceErrorsGlobal", UNSET)
         for componentsschemas_array_metric_item_data in _inference_errors_global or []:
@@ -443,6 +466,7 @@ class ResourceMetrics:
             token_total = TokenTotalMetric.from_dict(_token_total)
 
         resource_metrics = cls(
+            billable_time=billable_time,
             inference_errors_global=inference_errors_global,
             inference_global=inference_global,
             last_n_requests=last_n_requests,

blaxel/client/models/resource_trace.py

@@ -0,0 +1,97 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ResourceTrace")
+
+
+@_attrs_define
+class ResourceTrace:
+    """Log for a resource deployment (eg. model deployment, function deployment)
+
+    Attributes:
+        duration (Union[Unset, int]): Duration in nanoseconds
+        has_error (Union[Unset, bool]): Has error
+        start_time (Union[Unset, str]): The timestamp of the log
+        status_code (Union[Unset, int]): Status code
+        trace_id (Union[Unset, str]): Trace ID of the log
+    """
+
+    duration: Union[Unset, int] = UNSET
+    has_error: Union[Unset, bool] = UNSET
+    start_time: Union[Unset, str] = UNSET
+    status_code: Union[Unset, int] = UNSET
+    trace_id: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        duration = self.duration
+
+        has_error = self.has_error
+
+        start_time = self.start_time
+
+        status_code = self.status_code
+
+        trace_id = self.trace_id
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if duration is not UNSET:
+            field_dict["duration"] = duration
+        if has_error is not UNSET:
+            field_dict["hasError"] = has_error
+        if start_time is not UNSET:
+            field_dict["startTime"] = start_time
+        if status_code is not UNSET:
+            field_dict["statusCode"] = status_code
+        if trace_id is not UNSET:
+            field_dict["traceID"] = trace_id
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        duration = d.pop("duration", UNSET)
+
+        has_error = d.pop("hasError", UNSET)
+
+        start_time = d.pop("startTime", UNSET)
+
+        status_code = d.pop("statusCode", UNSET)
+
+        trace_id = d.pop("traceID", UNSET)
+
+        resource_trace = cls(
+            duration=duration,
+            has_error=has_error,
+            start_time=start_time,
+            status_code=status_code,
+            trace_id=trace_id,
+        )
+
+        resource_trace.additional_properties = d
+        return resource_trace
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/client/models/websocket_message.py

@@ -0,0 +1,106 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="WebsocketMessage")
+
+
+@_attrs_define
+class WebsocketMessage:
+    """WebSocket connection details
+
+    Attributes:
+        created_at (Union[Unset, str]): The date and time when the resource was created
+        updated_at (Union[Unset, str]): The date and time when the resource was updated
+        id (Union[Unset, str]): Unique message ID
+        message (Union[Unset, str]): Message
+        ttl (Union[Unset, int]): TTL timestamp for automatic deletion
+        workspace (Union[Unset, str]): Workspace the connection belongs to
+    """
+
+    created_at: Union[Unset, str] = UNSET
+    updated_at: Union[Unset, str] = UNSET
+    id: Union[Unset, str] = UNSET
+    message: Union[Unset, str] = UNSET
+    ttl: Union[Unset, int] = UNSET
+    workspace: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        created_at = self.created_at
+
+        updated_at = self.updated_at
+
+        id = self.id
+
+        message = self.message
+
+        ttl = self.ttl
+
+        workspace = self.workspace
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if created_at is not UNSET:
+            field_dict["createdAt"] = created_at
+        if updated_at is not UNSET:
+            field_dict["updatedAt"] = updated_at
+        if id is not UNSET:
+            field_dict["id"] = id
+        if message is not UNSET:
+            field_dict["message"] = message
+        if ttl is not UNSET:
+            field_dict["ttl"] = ttl
+        if workspace is not UNSET:
+            field_dict["workspace"] = workspace
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        created_at = d.pop("createdAt", UNSET)
+
+        updated_at = d.pop("updatedAt", UNSET)
+
+        id = d.pop("id", UNSET)
+
+        message = d.pop("message", UNSET)
+
+        ttl = d.pop("ttl", UNSET)
+
+        workspace = d.pop("workspace", UNSET)
+
+        websocket_message = cls(
+            created_at=created_at,
+            updated_at=updated_at,
+            id=id,
+            message=message,
+            ttl=ttl,
+            workspace=workspace,
+        )
+
+        websocket_message.additional_properties = d
+        return websocket_message
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/common/internal.py

@@ -2,6 +2,7 @@ import hashlib
 import os
 from logging import getLogger
 from typing import Optional
+
 from .env import env
 
 logger = getLogger(__name__)

blaxel/jobs/__init__.py

@@ -8,7 +8,7 @@ from typing import Any, Awaitable, Callable, Dict
 import requests
 
 from ..client import client
-from ..common.internal import get_global_unique_hash, get_forced_url
+from ..common.internal import get_forced_url, get_global_unique_hash
 from ..common.settings import settings
 from ..instrumentation.span import SpanManager
 

blaxel/sandbox/base.py

@@ -1,9 +1,8 @@
-import os
 
 from httpx import Response
 
 from ..client.models import Sandbox
-from ..common.internal import get_global_unique_hash, get_forced_url
+from ..common.internal import get_forced_url, get_global_unique_hash
 from ..common.settings import settings
 from .client.client import client
 from .client.models import ErrorResponse

blaxel/sandbox/client/api/process/delete_process_identifier_kill.py

@@ -6,32 +6,18 @@ import httpx
 from ... import errors
 from ...client import Client
 from ...models.error_response import ErrorResponse
-from ...models.process_kill_request import ProcessKillRequest
 from ...models.success_response import SuccessResponse
 from ...types import Response
 
 
 def _get_kwargs(
     identifier: str,
-    *,
-    body: ProcessKillRequest,
 ) -> dict[str, Any]:
-    headers: dict[str, Any] = {}
-
     _kwargs: dict[str, Any] = {
         "method": "delete",
         "url": f"/process/{identifier}/kill",
     }
 
-    if type(body) == dict:
-        _body = body
-    else:
-        _body = body.to_dict()
-
-    _kwargs["json"] = _body
-    headers["Content-Type"] = "application/json"
-
-    _kwargs["headers"] = headers
     return _kwargs
 
 
@@ -71,7 +57,6 @@ def sync_detailed(
     identifier: str,
     *,
     client: Union[Client],
-    body: ProcessKillRequest,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Kill a process
 
@@ -79,7 +64,6 @@ def sync_detailed(
 
     Args:
         identifier (str):
-        body (ProcessKillRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -91,7 +75,6 @@ def sync_detailed(
 
     kwargs = _get_kwargs(
         identifier=identifier,
-        body=body,
     )
 
     response = client.get_httpx_client().request(
@@ -105,7 +88,6 @@ def sync(
     identifier: str,
     *,
     client: Union[Client],
-    body: ProcessKillRequest,
 ) -> Optional[Union[ErrorResponse, SuccessResponse]]:
     """Kill a process
 
@@ -113,7 +95,6 @@ def sync(
 
     Args:
         identifier (str):
-        body (ProcessKillRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -126,7 +107,6 @@ def sync(
     return sync_detailed(
         identifier=identifier,
         client=client,
-        body=body,
     ).parsed
 
 
@@ -134,7 +114,6 @@ async def asyncio_detailed(
     identifier: str,
     *,
     client: Union[Client],
-    body: ProcessKillRequest,
 ) -> Response[Union[ErrorResponse, SuccessResponse]]:
     """Kill a process
 
@@ -142,7 +121,6 @@ async def asyncio_detailed(
 
     Args:
         identifier (str):
-        body (ProcessKillRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -154,7 +132,6 @@ async def asyncio_detailed(
 
     kwargs = _get_kwargs(
         identifier=identifier,
-        body=body,
     )
 
     response = await client.get_async_httpx_client().request(**kwargs)
@@ -166,7 +143,6 @@ async def asyncio(
     identifier: str,
     *,
     client: Union[Client],
-    body: ProcessKillRequest,
 ) -> Optional[Union[ErrorResponse, SuccessResponse]]:
     """Kill a process
 
@@ -174,7 +150,6 @@ async def asyncio(
 
     Args:
         identifier (str):
-        body (ProcessKillRequest):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -188,6 +163,5 @@ async def asyncio(
         await asyncio_detailed(
             identifier=identifier,
             client=client,
-            body=body,
         )
     ).parsed

blaxel/sandbox/client/models/__init__.py

@@ -11,9 +11,9 @@ from .file_with_content import FileWithContent
 from .get_network_process_pid_ports_response_200 import GetNetworkProcessPidPortsResponse200
 from .port_monitor_request import PortMonitorRequest
 from .post_network_process_pid_monitor_response_200 import PostNetworkProcessPidMonitorResponse200
-from .process_kill_request import ProcessKillRequest
 from .process_logs import ProcessLogs
 from .process_request import ProcessRequest
+from .process_request_env import ProcessRequestEnv
 from .process_response import ProcessResponse
 from .process_response_status import ProcessResponseStatus
 from .subdirectory import Subdirectory
@@ -29,9 +29,9 @@ __all__ = (
     "GetNetworkProcessPidPortsResponse200",
     "PortMonitorRequest",
     "PostNetworkProcessPidMonitorResponse200",
-    "ProcessKillRequest",
     "ProcessLogs",
     "ProcessRequest",
+    "ProcessRequestEnv",
     "ProcessResponse",
     "ProcessResponseStatus",
     "Subdirectory",

blaxel/sandbox/client/models/process_request.py

@@ -1,10 +1,14 @@
-from typing import Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
+if TYPE_CHECKING:
+    from ..models.process_request_env import ProcessRequestEnv
+
+
 T = TypeVar("T", bound="ProcessRequest")
 
 
@@ -13,6 +17,7 @@ class ProcessRequest:
     """
     Attributes:
         command (str):  Example: ls -la.
+        env (Union[Unset, ProcessRequestEnv]):  Example: {'{"PORT"': ' "3000"}'}.
         name (Union[Unset, str]):  Example: my-process.
         timeout (Union[Unset, int]):  Example: 30.
         wait_for_completion (Union[Unset, bool]):
@@ -21,6 +26,7 @@ class ProcessRequest:
     """
 
     command: str
+    env: Union[Unset, "ProcessRequestEnv"] = UNSET
     name: Union[Unset, str] = UNSET
     timeout: Union[Unset, int] = UNSET
     wait_for_completion: Union[Unset, bool] = UNSET
@@ -31,6 +37,12 @@ class ProcessRequest:
     def to_dict(self) -> dict[str, Any]:
         command = self.command
 
+        env: Union[Unset, dict[str, Any]] = UNSET
+        if self.env and not isinstance(self.env, Unset) and not isinstance(self.env, dict):
+            env = self.env.to_dict()
+        elif self.env and isinstance(self.env, dict):
+            env = self.env
+
         name = self.name
 
         timeout = self.timeout
@@ -50,6 +62,8 @@ class ProcessRequest:
                 "command": command,
             }
         )
+        if env is not UNSET:
+            field_dict["env"] = env
         if name is not UNSET:
             field_dict["name"] = name
         if timeout is not UNSET:
@@ -65,11 +79,20 @@ class ProcessRequest:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.process_request_env import ProcessRequestEnv
+
         if not src_dict:
             return None
         d = src_dict.copy()
         command = d.pop("command")
 
+        _env = d.pop("env", UNSET)
+        env: Union[Unset, ProcessRequestEnv]
+        if isinstance(_env, Unset):
+            env = UNSET
+        else:
+            env = ProcessRequestEnv.from_dict(_env)
+
         name = d.pop("name", UNSET)
 
         timeout = d.pop("timeout", UNSET)
@@ -82,6 +105,7 @@ class ProcessRequest:
 
         process_request = cls(
             command=command,
+            env=env,
             name=name,
             timeout=timeout,
             wait_for_completion=wait_for_completion,

blaxel/sandbox/client/models/process_request_env.py

@@ -0,0 +1,49 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ProcessRequestEnv")
+
+
+@_attrs_define
+class ProcessRequestEnv:
+    """
+    Example:
+        {'{"PORT"': ' "3000"}'}
+
+    """
+
+    additional_properties: dict[str, str] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        process_request_env = cls()
+
+        process_request_env.additional_properties = d
+        return process_request_env
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> str:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: str) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/tools/__init__.py

@@ -10,7 +10,7 @@ from mcp import ClientSession
 from mcp.types import CallToolResult
 from mcp.types import Tool as MCPTool
 
-from ..common.internal import get_global_unique_hash, get_forced_url
+from ..common.internal import get_forced_url, get_global_unique_hash
 from ..common.settings import settings
 from ..instrumentation.span import SpanManager
 from ..mcp.client import websocket_client

{blaxel-0.1.21.dist-info → blaxel-0.1.22rc70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.1.21
+Version: 0.1.22rc70
 Summary: Add your description here
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai