blaxel 0.1.9rc35__py3-none-any.whl → 0.1.9rc37__py3-none-any.whl

blaxel/client/models/model_spec.py

@@ -9,10 +9,8 @@ if TYPE_CHECKING:
     from ..models.core_spec_configurations import CoreSpecConfigurations
     from ..models.flavor import Flavor
     from ..models.model_private_cluster import ModelPrivateCluster
-    from ..models.pod_template_spec import PodTemplateSpec
     from ..models.revision_configuration import RevisionConfiguration
     from ..models.runtime import Runtime
-    from ..models.serverless_config import ServerlessConfig
 
 
 T = TypeVar("T", bound="ModelSpec")
@@ -24,29 +22,25 @@ class ModelSpec:
 
     Attributes:
         configurations (Union[Unset, CoreSpecConfigurations]): Optional configurations for the object
-        enabled (Union[Unset, bool]): Enable or disable the agent
+        enabled (Union[Unset, bool]): Enable or disable the resource
         flavors (Union[Unset, list['Flavor']]): Types of hardware available for deployments
         integration_connections (Union[Unset, list[str]]):
-        pod_template (Union[Unset, PodTemplateSpec]): Pod template specification
         policies (Union[Unset, list[str]]):
         private_clusters (Union[Unset, ModelPrivateCluster]): Private cluster where the model deployment is deployed
         revision (Union[Unset, RevisionConfiguration]): Revision configuration
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
         sandbox (Union[Unset, bool]): Sandbox mode
-        serverless_config (Union[Unset, ServerlessConfig]): Configuration for a serverless deployment
     """
 
     configurations: Union[Unset, "CoreSpecConfigurations"] = UNSET
     enabled: Union[Unset, bool] = UNSET
     flavors: Union[Unset, list["Flavor"]] = UNSET
     integration_connections: Union[Unset, list[str]] = UNSET
-    pod_template: Union[Unset, "PodTemplateSpec"] = UNSET
     policies: Union[Unset, list[str]] = UNSET
     private_clusters: Union[Unset, "ModelPrivateCluster"] = UNSET
     revision: Union[Unset, "RevisionConfiguration"] = UNSET
     runtime: Union[Unset, "Runtime"] = UNSET
     sandbox: Union[Unset, bool] = UNSET
-    serverless_config: Union[Unset, "ServerlessConfig"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -76,12 +70,6 @@ class ModelSpec:
         if not isinstance(self.integration_connections, Unset):
             integration_connections = self.integration_connections
 
-        pod_template: Union[Unset, dict[str, Any]] = UNSET
-        if self.pod_template and not isinstance(self.pod_template, Unset) and not isinstance(self.pod_template, dict):
-            pod_template = self.pod_template.to_dict()
-        elif self.pod_template and isinstance(self.pod_template, dict):
-            pod_template = self.pod_template
-
         policies: Union[Unset, list[str]] = UNSET
         if not isinstance(self.policies, Unset):
             policies = self.policies
@@ -110,16 +98,6 @@ class ModelSpec:
 
         sandbox = self.sandbox
 
-        serverless_config: Union[Unset, dict[str, Any]] = UNSET
-        if (
-            self.serverless_config
-            and not isinstance(self.serverless_config, Unset)
-            and not isinstance(self.serverless_config, dict)
-        ):
-            serverless_config = self.serverless_config.to_dict()
-        elif self.serverless_config and isinstance(self.serverless_config, dict):
-            serverless_config = self.serverless_config
-
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -131,8 +109,6 @@ class ModelSpec:
             field_dict["flavors"] = flavors
         if integration_connections is not UNSET:
             field_dict["integrationConnections"] = integration_connections
-        if pod_template is not UNSET:
-            field_dict["podTemplate"] = pod_template
         if policies is not UNSET:
             field_dict["policies"] = policies
         if private_clusters is not UNSET:
@@ -143,8 +119,6 @@ class ModelSpec:
             field_dict["runtime"] = runtime
         if sandbox is not UNSET:
             field_dict["sandbox"] = sandbox
-        if serverless_config is not UNSET:
-            field_dict["serverlessConfig"] = serverless_config
 
         return field_dict
 
@@ -153,10 +127,8 @@ class ModelSpec:
         from ..models.core_spec_configurations import CoreSpecConfigurations
         from ..models.flavor import Flavor
         from ..models.model_private_cluster import ModelPrivateCluster
-        from ..models.pod_template_spec import PodTemplateSpec
         from ..models.revision_configuration import RevisionConfiguration
         from ..models.runtime import Runtime
-        from ..models.serverless_config import ServerlessConfig
 
         if not src_dict:
             return None
@@ -179,13 +151,6 @@ class ModelSpec:
 
         integration_connections = cast(list[str], d.pop("integrationConnections", UNSET))
 
-        _pod_template = d.pop("podTemplate", UNSET)
-        pod_template: Union[Unset, PodTemplateSpec]
-        if isinstance(_pod_template, Unset):
-            pod_template = UNSET
-        else:
-            pod_template = PodTemplateSpec.from_dict(_pod_template)
-
         policies = cast(list[str], d.pop("policies", UNSET))
 
         _private_clusters = d.pop("privateClusters", UNSET)
@@ -211,25 +176,16 @@ class ModelSpec:
 
         sandbox = d.pop("sandbox", UNSET)
 
-        _serverless_config = d.pop("serverlessConfig", UNSET)
-        serverless_config: Union[Unset, ServerlessConfig]
-        if isinstance(_serverless_config, Unset):
-            serverless_config = UNSET
-        else:
-            serverless_config = ServerlessConfig.from_dict(_serverless_config)
-
         model_spec = cls(
             configurations=configurations,
             enabled=enabled,
             flavors=flavors,
             integration_connections=integration_connections,
-            pod_template=pod_template,
             policies=policies,
             private_clusters=private_clusters,
             revision=revision,
             runtime=runtime,
             sandbox=sandbox,
-            serverless_config=serverless_config,
         )
 
         model_spec.additional_properties = d

blaxel/client/models/{agent_chain.py → port.py}

@@ -5,46 +5,40 @@ from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
-T = TypeVar("T", bound="AgentChain")
+T = TypeVar("T", bound="Port")
 
 
 @_attrs_define
-class AgentChain:
-    """Agent chain configuration
+class Port:
+    """A port for a resource
 
     Attributes:
-        description (Union[Unset, str]): Description of the agent in case you want to override the default one
-        enabled (Union[Unset, bool]): Whether the agent chain is enabled
-        name (Union[Unset, str]): The name of the agent to chain to
-        prompt (Union[Unset, str]): Prompt of the agent in case you want to override the default one
+        name (Union[Unset, str]): The name of the port
+        protocol (Union[Unset, str]): The protocol of the port
+        target (Union[Unset, int]): The target port of the port
     """
 
-    description: Union[Unset, str] = UNSET
-    enabled: Union[Unset, bool] = UNSET
     name: Union[Unset, str] = UNSET
-    prompt: Union[Unset, str] = UNSET
+    protocol: Union[Unset, str] = UNSET
+    target: Union[Unset, int] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        description = self.description
-
-        enabled = self.enabled
-
         name = self.name
 
-        prompt = self.prompt
+        protocol = self.protocol
+
+        target = self.target
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
-        if description is not UNSET:
-            field_dict["description"] = description
-        if enabled is not UNSET:
-            field_dict["enabled"] = enabled
         if name is not UNSET:
             field_dict["name"] = name
-        if prompt is not UNSET:
-            field_dict["prompt"] = prompt
+        if protocol is not UNSET:
+            field_dict["protocol"] = protocol
+        if target is not UNSET:
+            field_dict["target"] = target
 
         return field_dict
 
@@ -53,23 +47,20 @@ class AgentChain:
         if not src_dict:
             return None
         d = src_dict.copy()
-        description = d.pop("description", UNSET)
-
-        enabled = d.pop("enabled", UNSET)
-
         name = d.pop("name", UNSET)
 
-        prompt = d.pop("prompt", UNSET)
+        protocol = d.pop("protocol", UNSET)
+
+        target = d.pop("target", UNSET)
 
-        agent_chain = cls(
-            description=description,
-            enabled=enabled,
+        port = cls(
             name=name,
-            prompt=prompt,
+            protocol=protocol,
+            target=target,
         )
 
-        agent_chain.additional_properties = d
-        return agent_chain
+        port.additional_properties = d
+        return port
 
     @property
     def additional_keys(self) -> list[str]:

blaxel/client/models/request_total_metric.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -20,6 +20,7 @@ class RequestTotalMetric:
     """Metrics for request total
 
     Attributes:
+        items (Union[Unset, list[Any]]): Historical requests for all resources globally
         request_total (Union[Unset, float]): Number of requests for all resources globally
         request_total_per_code (Union[Unset, RequestTotalMetricRequestTotalPerCode]): Number of requests for all
             resources globally per code
@@ -27,6 +28,7 @@ class RequestTotalMetric:
         rps_per_code (Union[Unset, RequestTotalMetricRpsPerCode]): Number of requests for all resources globally
     """
 
+    items: Union[Unset, list[Any]] = UNSET
     request_total: Union[Unset, float] = UNSET
     request_total_per_code: Union[Unset, "RequestTotalMetricRequestTotalPerCode"] = UNSET
     rps: Union[Unset, float] = UNSET
@@ -34,6 +36,10 @@ class RequestTotalMetric:
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        items: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.items, Unset):
+            items = self.items
+
         request_total = self.request_total
 
         request_total_per_code: Union[Unset, dict[str, Any]] = UNSET
@@ -57,6 +63,8 @@ class RequestTotalMetric:
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
+        if items is not UNSET:
+            field_dict["items"] = items
         if request_total is not UNSET:
             field_dict["requestTotal"] = request_total
         if request_total_per_code is not UNSET:
@@ -78,6 +86,8 @@ class RequestTotalMetric:
         if not src_dict:
             return None
         d = src_dict.copy()
+        items = cast(list[Any], d.pop("items", UNSET))
+
         request_total = d.pop("requestTotal", UNSET)
 
         _request_total_per_code = d.pop("requestTotalPerCode", UNSET)
@@ -97,6 +107,7 @@ class RequestTotalMetric:
             rps_per_code = RequestTotalMetricRpsPerCode.from_dict(_rps_per_code)
 
         request_total_metric = cls(
+            items=items,
             request_total=request_total,
             request_total_per_code=request_total_per_code,
             rps=rps,

blaxel/client/models/request_total_response_data.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="RequestTotalResponseData")
+
+
+@_attrs_define
+class RequestTotalResponseData:
+    """Request total response data
+
+    Attributes:
+        request_total (Union[Unset, float]): Request total
+        status_code (Union[Unset, str]): Status code
+        workload_id (Union[Unset, str]): Workload ID
+        workload_type (Union[Unset, str]): Workload type
+        workspace (Union[Unset, str]): Workspace
+    """
+
+    request_total: Union[Unset, float] = UNSET
+    status_code: Union[Unset, str] = UNSET
+    workload_id: Union[Unset, str] = UNSET
+    workload_type: Union[Unset, str] = UNSET
+    workspace: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        request_total = self.request_total
+
+        status_code = self.status_code
+
+        workload_id = self.workload_id
+
+        workload_type = self.workload_type
+
+        workspace = self.workspace
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if request_total is not UNSET:
+            field_dict["requestTotal"] = request_total
+        if status_code is not UNSET:
+            field_dict["statusCode"] = status_code
+        if workload_id is not UNSET:
+            field_dict["workloadId"] = workload_id
+        if workload_type is not UNSET:
+            field_dict["workloadType"] = workload_type
+        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()
+        request_total = d.pop("requestTotal", UNSET)
+
+        status_code = d.pop("statusCode", UNSET)
+
+        workload_id = d.pop("workloadId", UNSET)
+
+        workload_type = d.pop("workloadType", UNSET)
+
+        workspace = d.pop("workspace", UNSET)
+
+        request_total_response_data = cls(
+            request_total=request_total,
+            status_code=status_code,
+            workload_id=workload_id,
+            workload_type=workload_type,
+            workspace=workspace,
+        )
+
+        request_total_response_data.additional_properties = d
+        return request_total_response_data
+
+    @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_log.py

@@ -16,11 +16,13 @@ class ResourceLog:
         message (Union[Unset, str]): Content of the log
         severity (Union[Unset, int]): Severity of the log
         timestamp (Union[Unset, str]): The timestamp of the log
+        trace_id (Union[Unset, str]): Trace ID of the log
     """
 
     message: Union[Unset, str] = UNSET
     severity: Union[Unset, int] = UNSET
     timestamp: Union[Unset, str] = 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]:
@@ -30,6 +32,8 @@ class ResourceLog:
 
         timestamp = self.timestamp
 
+        trace_id = self.trace_id
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -39,6 +43,8 @@ class ResourceLog:
             field_dict["severity"] = severity
         if timestamp is not UNSET:
             field_dict["timestamp"] = timestamp
+        if trace_id is not UNSET:
+            field_dict["trace_id"] = trace_id
 
         return field_dict
 
@@ -53,10 +59,13 @@ class ResourceLog:
 
         timestamp = d.pop("timestamp", UNSET)
 
+        trace_id = d.pop("trace_id", UNSET)
+
         resource_log = cls(
             message=message,
             severity=severity,
             timestamp=timestamp,
+            trace_id=trace_id,
         )
 
         resource_log.additional_properties = d