blaxel 0.2.33__py3-none-any.whl → 0.2.35__py3-none-any.whl

blaxel/core/client/models/memory_allocation_by_name.py

@@ -1,70 +0,0 @@
-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="MemoryAllocationByName")
-
-
-@_attrs_define
-class MemoryAllocationByName:
-    """Memory allocation by service name
-
-    Attributes:
-        allocation (Union[Unset, float]): Memory allocation value
-        name (Union[Unset, str]): Name
-    """
-
-    allocation: Union[Unset, float] = UNSET
-    name: Union[Unset, str] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        allocation = self.allocation
-
-        name = self.name
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if allocation is not UNSET:
-            field_dict["allocation"] = allocation
-        if name is not UNSET:
-            field_dict["name"] = name
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        allocation = d.pop("allocation", UNSET)
-
-        name = d.pop("name", UNSET)
-
-        memory_allocation_by_name = cls(
-            allocation=allocation,
-            name=name,
-        )
-
-        memory_allocation_by_name.additional_properties = d
-        return memory_allocation_by_name
-
-    @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/core/client/models/memory_allocation_metric.py

@@ -1,61 +0,0 @@
-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="MemoryAllocationMetric")
-
-
-@_attrs_define
-class MemoryAllocationMetric:
-    """Metrics for memory allocation
-
-    Attributes:
-        total_allocation (Union[Unset, float]): Total memory allocation in GB-seconds
-    """
-
-    total_allocation: Union[Unset, float] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        total_allocation = self.total_allocation
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if total_allocation is not UNSET:
-            field_dict["totalAllocation"] = total_allocation
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        total_allocation = d.pop("totalAllocation", d.pop("total_allocation", UNSET))
-
-        memory_allocation_metric = cls(
-            total_allocation=total_allocation,
-        )
-
-        memory_allocation_metric.additional_properties = d
-        return memory_allocation_metric
-
-    @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/core/client/models/metric.py

@@ -1,79 +0,0 @@
-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="Metric")
-
-
-@_attrs_define
-class Metric:
-    """Metric
-
-    Attributes:
-        rate (Union[Unset, int]): Metric value
-        request_total (Union[Unset, int]): Metric value
-        timestamp (Union[Unset, str]): Metric timestamp
-    """
-
-    rate: Union[Unset, int] = UNSET
-    request_total: Union[Unset, int] = UNSET
-    timestamp: Union[Unset, str] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        rate = self.rate
-
-        request_total = self.request_total
-
-        timestamp = self.timestamp
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if rate is not UNSET:
-            field_dict["rate"] = rate
-        if request_total is not UNSET:
-            field_dict["requestTotal"] = request_total
-        if timestamp is not UNSET:
-            field_dict["timestamp"] = timestamp
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        rate = d.pop("rate", UNSET)
-
-        request_total = d.pop("requestTotal", d.pop("request_total", UNSET))
-
-        timestamp = d.pop("timestamp", UNSET)
-
-        metric = cls(
-            rate=rate,
-            request_total=request_total,
-            timestamp=timestamp,
-        )
-
-        metric.additional_properties = d
-        return metric
-
-    @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/core/client/models/metrics.py

@@ -1,273 +0,0 @@
-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.last_n_requests_metric import LastNRequestsMetric
-    from ..models.metric import Metric
-    from ..models.metrics_models import MetricsModels
-    from ..models.metrics_request_total_per_code import MetricsRequestTotalPerCode
-    from ..models.metrics_rps_per_code import MetricsRpsPerCode
-    from ..models.request_total_response_data import RequestTotalResponseData
-
-
-T = TypeVar("T", bound="Metrics")
-
-
-@_attrs_define
-class Metrics:
-    """Metrics for resources
-
-    Attributes:
-        agents (Union[Unset, Any]): Metrics for agents
-        functions (Union[Unset, Any]): Metrics for functions
-        inference_error_global (Union[Unset, list['Metric']]): Array of metrics
-        inference_global (Union[Unset, list[Any]]): Historical requests for all resources globally
-        items (Union[Unset, list['RequestTotalResponseData']]): Historical requests for all resources globally
-        jobs (Union[Unset, Any]): Metrics for jobs
-        last_n_requests (Union[Unset, list['LastNRequestsMetric']]): Metric value
-        models (Union[Unset, MetricsModels]): Metrics for models
-        request_total (Union[Unset, float]): Number of requests for all resources globally
-        request_total_per_code (Union[Unset, MetricsRequestTotalPerCode]): Number of requests for all resources globally
-            per code
-        rps (Union[Unset, float]): Number of requests per second for all resources globally
-        rps_per_code (Union[Unset, MetricsRpsPerCode]): Number of requests per second for all resources globally per
-            code
-        sandboxes (Union[Unset, Any]): Metrics for sandboxes
-    """
-
-    agents: Union[Unset, Any] = UNSET
-    functions: Union[Unset, Any] = UNSET
-    inference_error_global: Union[Unset, list["Metric"]] = UNSET
-    inference_global: Union[Unset, list[Any]] = UNSET
-    items: Union[Unset, list["RequestTotalResponseData"]] = UNSET
-    jobs: Union[Unset, Any] = UNSET
-    last_n_requests: Union[Unset, list["LastNRequestsMetric"]] = UNSET
-    models: Union[Unset, "MetricsModels"] = UNSET
-    request_total: Union[Unset, float] = UNSET
-    request_total_per_code: Union[Unset, "MetricsRequestTotalPerCode"] = UNSET
-    rps: Union[Unset, float] = UNSET
-    rps_per_code: Union[Unset, "MetricsRpsPerCode"] = UNSET
-    sandboxes: Union[Unset, Any] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        agents = self.agents
-
-        functions = self.functions
-
-        inference_error_global: Union[Unset, list[dict[str, Any]]] = UNSET
-        if not isinstance(self.inference_error_global, Unset):
-            inference_error_global = []
-            for componentsschemas_array_metric_item_data in self.inference_error_global:
-                if type(componentsschemas_array_metric_item_data) is dict:
-                    componentsschemas_array_metric_item = componentsschemas_array_metric_item_data
-                else:
-                    componentsschemas_array_metric_item = (
-                        componentsschemas_array_metric_item_data.to_dict()
-                    )
-                inference_error_global.append(componentsschemas_array_metric_item)
-
-        inference_global: Union[Unset, list[Any]] = UNSET
-        if not isinstance(self.inference_global, Unset):
-            inference_global = self.inference_global
-
-        items: Union[Unset, list[dict[str, Any]]] = UNSET
-        if not isinstance(self.items, Unset):
-            items = []
-            for items_item_data in self.items:
-                if type(items_item_data) is dict:
-                    items_item = items_item_data
-                else:
-                    items_item = items_item_data.to_dict()
-                items.append(items_item)
-
-        jobs = self.jobs
-
-        last_n_requests: Union[Unset, list[dict[str, Any]]] = UNSET
-        if not isinstance(self.last_n_requests, Unset):
-            last_n_requests = []
-            for last_n_requests_item_data in self.last_n_requests:
-                if type(last_n_requests_item_data) is dict:
-                    last_n_requests_item = last_n_requests_item_data
-                else:
-                    last_n_requests_item = last_n_requests_item_data.to_dict()
-                last_n_requests.append(last_n_requests_item)
-
-        models: Union[Unset, dict[str, Any]] = UNSET
-        if self.models and not isinstance(self.models, Unset) and not isinstance(self.models, dict):
-            models = self.models.to_dict()
-        elif self.models and isinstance(self.models, dict):
-            models = self.models
-
-        request_total = self.request_total
-
-        request_total_per_code: Union[Unset, dict[str, Any]] = UNSET
-        if (
-            self.request_total_per_code
-            and not isinstance(self.request_total_per_code, Unset)
-            and not isinstance(self.request_total_per_code, dict)
-        ):
-            request_total_per_code = self.request_total_per_code.to_dict()
-        elif self.request_total_per_code and isinstance(self.request_total_per_code, dict):
-            request_total_per_code = self.request_total_per_code
-
-        rps = self.rps
-
-        rps_per_code: Union[Unset, dict[str, Any]] = UNSET
-        if (
-            self.rps_per_code
-            and not isinstance(self.rps_per_code, Unset)
-            and not isinstance(self.rps_per_code, dict)
-        ):
-            rps_per_code = self.rps_per_code.to_dict()
-        elif self.rps_per_code and isinstance(self.rps_per_code, dict):
-            rps_per_code = self.rps_per_code
-
-        sandboxes = self.sandboxes
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if agents is not UNSET:
-            field_dict["agents"] = agents
-        if functions is not UNSET:
-            field_dict["functions"] = functions
-        if inference_error_global is not UNSET:
-            field_dict["inferenceErrorGlobal"] = inference_error_global
-        if inference_global is not UNSET:
-            field_dict["inferenceGlobal"] = inference_global
-        if items is not UNSET:
-            field_dict["items"] = items
-        if jobs is not UNSET:
-            field_dict["jobs"] = jobs
-        if last_n_requests is not UNSET:
-            field_dict["lastNRequests"] = last_n_requests
-        if models is not UNSET:
-            field_dict["models"] = models
-        if request_total is not UNSET:
-            field_dict["requestTotal"] = request_total
-        if request_total_per_code is not UNSET:
-            field_dict["requestTotalPerCode"] = request_total_per_code
-        if rps is not UNSET:
-            field_dict["rps"] = rps
-        if rps_per_code is not UNSET:
-            field_dict["rpsPerCode"] = rps_per_code
-        if sandboxes is not UNSET:
-            field_dict["sandboxes"] = sandboxes
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
-        from ..models.last_n_requests_metric import LastNRequestsMetric
-        from ..models.metric import Metric
-        from ..models.metrics_models import MetricsModels
-        from ..models.metrics_request_total_per_code import MetricsRequestTotalPerCode
-        from ..models.metrics_rps_per_code import MetricsRpsPerCode
-        from ..models.request_total_response_data import RequestTotalResponseData
-
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        agents = d.pop("agents", UNSET)
-
-        functions = d.pop("functions", UNSET)
-
-        inference_error_global = []
-        _inference_error_global = d.pop(
-            "inferenceErrorGlobal", d.pop("inference_error_global", UNSET)
-        )
-        for componentsschemas_array_metric_item_data in _inference_error_global or []:
-            componentsschemas_array_metric_item = Metric.from_dict(
-                componentsschemas_array_metric_item_data
-            )
-
-            inference_error_global.append(componentsschemas_array_metric_item)
-
-        inference_global = cast(
-            list[Any], d.pop("inferenceGlobal", d.pop("inference_global", UNSET))
-        )
-
-        items = []
-        _items = d.pop("items", UNSET)
-        for items_item_data in _items or []:
-            items_item = RequestTotalResponseData.from_dict(items_item_data)
-
-            items.append(items_item)
-
-        jobs = d.pop("jobs", UNSET)
-
-        last_n_requests = []
-        _last_n_requests = d.pop("lastNRequests", d.pop("last_n_requests", UNSET))
-        for last_n_requests_item_data in _last_n_requests or []:
-            last_n_requests_item = LastNRequestsMetric.from_dict(last_n_requests_item_data)
-
-            last_n_requests.append(last_n_requests_item)
-
-        _models = d.pop("models", UNSET)
-        models: Union[Unset, MetricsModels]
-        if isinstance(_models, Unset):
-            models = UNSET
-        else:
-            models = MetricsModels.from_dict(_models)
-
-        request_total = d.pop("requestTotal", d.pop("request_total", UNSET))
-
-        _request_total_per_code = d.pop(
-            "requestTotalPerCode", d.pop("request_total_per_code", UNSET)
-        )
-        request_total_per_code: Union[Unset, MetricsRequestTotalPerCode]
-        if isinstance(_request_total_per_code, Unset):
-            request_total_per_code = UNSET
-        else:
-            request_total_per_code = MetricsRequestTotalPerCode.from_dict(_request_total_per_code)
-
-        rps = d.pop("rps", UNSET)
-
-        _rps_per_code = d.pop("rpsPerCode", d.pop("rps_per_code", UNSET))
-        rps_per_code: Union[Unset, MetricsRpsPerCode]
-        if isinstance(_rps_per_code, Unset):
-            rps_per_code = UNSET
-        else:
-            rps_per_code = MetricsRpsPerCode.from_dict(_rps_per_code)
-
-        sandboxes = d.pop("sandboxes", UNSET)
-
-        metrics = cls(
-            agents=agents,
-            functions=functions,
-            inference_error_global=inference_error_global,
-            inference_global=inference_global,
-            items=items,
-            jobs=jobs,
-            last_n_requests=last_n_requests,
-            models=models,
-            request_total=request_total,
-            request_total_per_code=request_total_per_code,
-            rps=rps,
-            rps_per_code=rps_per_code,
-            sandboxes=sandboxes,
-        )
-
-        metrics.additional_properties = d
-        return metrics
-
-    @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/core/client/models/metrics_models.py

@@ -1,45 +0,0 @@
-from typing import Any, TypeVar
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="MetricsModels")
-
-
-@_attrs_define
-class MetricsModels:
-    """Metrics for models"""
-
-    additional_properties: dict[str, Any] = _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 | None:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        metrics_models = cls()
-
-        metrics_models.additional_properties = d
-        return metrics_models
-
-    @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/core/client/models/metrics_rps_per_code.py

@@ -1,45 +0,0 @@
-from typing import Any, TypeVar
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="MetricsRpsPerCode")
-
-
-@_attrs_define
-class MetricsRpsPerCode:
-    """Number of requests per second for all resources globally per code"""
-
-    additional_properties: dict[str, Any] = _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 | None:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        metrics_rps_per_code = cls()
-
-        metrics_rps_per_code.additional_properties = d
-        return metrics_rps_per_code
-
-    @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/core/client/models/pod_template_spec.py

@@ -1,45 +0,0 @@
-from typing import Any, TypeVar
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="PodTemplateSpec")
-
-
-@_attrs_define
-class PodTemplateSpec:
-    """Pod template specification"""
-
-    additional_properties: dict[str, Any] = _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 | None:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        pod_template_spec = cls()
-
-        pod_template_spec.additional_properties = d
-        return pod_template_spec
-
-    @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