beamlit 0.0.19__py3-none-any.whl → 0.0.20__py3-none-any.whl

beamlit/models/resource_environment_metrics_query_per_second_per_region_per_code.py

@@ -0,0 +1,73 @@
+from typing import TYPE_CHECKING, Any, Dict, Type, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.qps import QPS
+
+
+T = TypeVar("T", bound="ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode")
+
+
+@_attrs_define
+class ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode:
+    """RPS value (in last 24 hours) per response status code per location, for the model deployment
+
+    Attributes:
+        region (Union[Unset, QPS]): Query per second per element, can be per response status code (e.g. 200, 400) or per
+            location
+    """
+
+    region: Union[Unset, "QPS"] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        region: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.region, Unset):
+            region = self.region.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if region is not UNSET:
+            field_dict["region"] = region
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.qps import QPS
+
+        d = src_dict.copy()
+        _region = d.pop("region", UNSET)
+        region: Union[Unset, QPS]
+        if isinstance(_region, Unset):
+            region = UNSET
+        else:
+            region = QPS.from_dict(_region)
+
+        resource_environment_metrics_query_per_second_per_region_per_code = cls(
+            region=region,
+        )
+
+        resource_environment_metrics_query_per_second_per_region_per_code.additional_properties = d
+        return resource_environment_metrics_query_per_second_per_region_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

beamlit/models/resource_log.py

@@ -0,0 +1,68 @@
+from typing import Any, Type, 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="ResourceLog")
+
+
+@_attrs_define
+class ResourceLog:
+    """Log for a resource deployment (eg. model deployment, function deployment)
+
+    Attributes:
+        message (Union[Unset, str]): Content of the log
+        timestamp (Union[Unset, str]): The timestamp of the log
+    """
+
+    message: Union[Unset, str] = UNSET
+    timestamp: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        message = self.message
+
+        timestamp = self.timestamp
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if message is not UNSET:
+            field_dict["message"] = message
+        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:
+        d = src_dict.copy()
+        message = d.pop("message", UNSET)
+
+        timestamp = d.pop("timestamp", UNSET)
+
+        resource_log = cls(
+            message=message,
+            timestamp=timestamp,
+        )
+
+        resource_log.additional_properties = d
+        return resource_log
+
+    @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

beamlit/models/resource_metrics.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -18,43 +18,72 @@ class ResourceMetrics:
     """Metrics for a single resource (eg. model, function)
 
     Attributes:
-        inference_per_second_global (Union[Unset, List['Metric']]): Array of metrics
+        inference_global (Union[Unset, list['Metric']]): Array of metrics
+        inference_per_second_global (Union[Unset, list['Metric']]): Array of metrics
+        query_per_region (Union[Unset, QPS]): Query per second per element, can be per response status code (e.g. 200,
+            400) or per location
         query_per_second_per_region (Union[Unset, QPS]): Query per second per element, can be per response status code
             (e.g. 200, 400) or per location
     """
 
-    inference_per_second_global: Union[Unset, List["Metric"]] = UNSET
+    inference_global: Union[Unset, list["Metric"]] = UNSET
+    inference_per_second_global: Union[Unset, list["Metric"]] = UNSET
+    query_per_region: Union[Unset, "QPS"] = UNSET
     query_per_second_per_region: Union[Unset, "QPS"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        inference_per_second_global: Union[Unset, List[Dict[str, Any]]] = UNSET
+        inference_global: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.inference_global, Unset):
+            inference_global = []
+            for componentsschemas_array_metric_item_data in self.inference_global:
+                componentsschemas_array_metric_item = componentsschemas_array_metric_item_data.to_dict()
+                inference_global.append(componentsschemas_array_metric_item)
+
+        inference_per_second_global: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.inference_per_second_global, Unset):
             inference_per_second_global = []
             for componentsschemas_array_metric_item_data in self.inference_per_second_global:
                 componentsschemas_array_metric_item = componentsschemas_array_metric_item_data.to_dict()
                 inference_per_second_global.append(componentsschemas_array_metric_item)
 
-        query_per_second_per_region: Union[Unset, Dict[str, Any]] = UNSET
+        query_per_region: Union[Unset, dict[str, Any]] = UNSET
+        if not isinstance(self.query_per_region, Unset):
+            query_per_region = self.query_per_region.to_dict()
+
+        query_per_second_per_region: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.query_per_second_per_region, Unset):
             query_per_second_per_region = self.query_per_second_per_region.to_dict()
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
+        if inference_global is not UNSET:
+            field_dict["inference_global"] = inference_global
         if inference_per_second_global is not UNSET:
             field_dict["inference_per_second_global"] = inference_per_second_global
+        if query_per_region is not UNSET:
+            field_dict["query_per_region"] = query_per_region
         if query_per_second_per_region is not UNSET:
             field_dict["query_per_second_per_region"] = query_per_second_per_region
 
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.metric import Metric
         from ..models.qps import QPS
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
+        inference_global = []
+        _inference_global = d.pop("inference_global", UNSET)
+        for componentsschemas_array_metric_item_data in _inference_global or []:
+            componentsschemas_array_metric_item = Metric.from_dict(componentsschemas_array_metric_item_data)
+
+            inference_global.append(componentsschemas_array_metric_item)
+
         inference_per_second_global = []
         _inference_per_second_global = d.pop("inference_per_second_global", UNSET)
         for componentsschemas_array_metric_item_data in _inference_per_second_global or []:
@@ -62,6 +91,13 @@ class ResourceMetrics:
 
             inference_per_second_global.append(componentsschemas_array_metric_item)
 
+        _query_per_region = d.pop("query_per_region", UNSET)
+        query_per_region: Union[Unset, QPS]
+        if isinstance(_query_per_region, Unset):
+            query_per_region = UNSET
+        else:
+            query_per_region = QPS.from_dict(_query_per_region)
+
         _query_per_second_per_region = d.pop("query_per_second_per_region", UNSET)
         query_per_second_per_region: Union[Unset, QPS]
         if isinstance(_query_per_second_per_region, Unset):
@@ -70,7 +106,9 @@ class ResourceMetrics:
             query_per_second_per_region = QPS.from_dict(_query_per_second_per_region)
 
         resource_metrics = cls(
+            inference_global=inference_global,
             inference_per_second_global=inference_per_second_global,
+            query_per_region=query_per_region,
             query_per_second_per_region=query_per_second_per_region,
         )
 

beamlit/models/runtime.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, 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
@@ -18,38 +18,38 @@ class Runtime:
     """Set of configurations for a deployment
 
     Attributes:
-        args (Union[Unset, List[Any]]): The arguments to pass to the deployment runtime
-        command (Union[Unset, List[Any]]): The command to run the deployment
-        envs (Union[Unset, List[Any]]): The environment variables to set in the deployment. Should be a list of
+        args (Union[Unset, list[Any]]): The arguments to pass to the deployment runtime
+        command (Union[Unset, list[Any]]): The command to run the deployment
+        envs (Union[Unset, list[Any]]): The environment variables to set in the deployment. Should be a list of
             Kubernetes EnvVar types
         image (Union[Unset, str]): The Docker image for the deployment
         model (Union[Unset, str]): The slug name of the origin model. Only used if the deployment is a ModelDeployment
         readiness_probe (Union[Unset, RuntimeReadinessProbe]): The readiness probe. Should be a Kubernetes Probe type
         resources (Union[Unset, RuntimeResources]): The resources for the deployment. Should be a Kubernetes
             ResourceRequirements type
-        type (Union[Unset, str]): The type of origin for the deployment
+        type_ (Union[Unset, str]): The type of origin for the deployment
     """
 
-    args: Union[Unset, List[Any]] = UNSET
-    command: Union[Unset, List[Any]] = UNSET
-    envs: Union[Unset, List[Any]] = UNSET
+    args: Union[Unset, list[Any]] = UNSET
+    command: Union[Unset, list[Any]] = UNSET
+    envs: Union[Unset, list[Any]] = UNSET
     image: Union[Unset, str] = UNSET
     model: Union[Unset, str] = UNSET
     readiness_probe: Union[Unset, "RuntimeReadinessProbe"] = UNSET
     resources: Union[Unset, "RuntimeResources"] = UNSET
-    type: Union[Unset, str] = UNSET
+    type_: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        args: Union[Unset, List[Any]] = UNSET
+        args: Union[Unset, list[Any]] = UNSET
         if not isinstance(self.args, Unset):
             args = self.args
 
-        command: Union[Unset, List[Any]] = UNSET
+        command: Union[Unset, list[Any]] = UNSET
         if not isinstance(self.command, Unset):
             command = self.command
 
-        envs: Union[Unset, List[Any]] = UNSET
+        envs: Union[Unset, list[Any]] = UNSET
         if not isinstance(self.envs, Unset):
             envs = self.envs
 
@@ -57,15 +57,15 @@ class Runtime:
 
         model = self.model
 
-        readiness_probe: Union[Unset, Dict[str, Any]] = UNSET
+        readiness_probe: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.readiness_probe, Unset):
             readiness_probe = self.readiness_probe.to_dict()
 
-        resources: Union[Unset, Dict[str, Any]] = UNSET
+        resources: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.resources, Unset):
             resources = self.resources.to_dict()
 
-        type = self.type
+        type_ = self.type_
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -84,22 +84,24 @@ class Runtime:
             field_dict["readiness_probe"] = readiness_probe
         if resources is not UNSET:
             field_dict["resources"] = resources
-        if type is not UNSET:
-            field_dict["type"] = type
+        if type_ is not UNSET:
+            field_dict["type"] = type_
 
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.runtime_readiness_probe import RuntimeReadinessProbe
         from ..models.runtime_resources import RuntimeResources
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
-        args = cast(List[Any], d.pop("args", UNSET))
+        args = cast(list[Any], d.pop("args", UNSET))
 
-        command = cast(List[Any], d.pop("command", UNSET))
+        command = cast(list[Any], d.pop("command", UNSET))
 
-        envs = cast(List[Any], d.pop("envs", UNSET))
+        envs = cast(list[Any], d.pop("envs", UNSET))
 
         image = d.pop("image", UNSET)
 
@@ -119,7 +121,7 @@ class Runtime:
         else:
             resources = RuntimeResources.from_dict(_resources)
 
-        type = d.pop("type", UNSET)
+        type_ = d.pop("type", UNSET)
 
         runtime = cls(
             args=args,
@@ -129,7 +131,7 @@ class Runtime:
             model=model,
             readiness_probe=readiness_probe,
             resources=resources,
-            type=type,
+            type_=type_,
         )
 
         runtime.additional_properties = d

beamlit/models/runtime_readiness_probe.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -19,7 +19,9 @@ class RuntimeReadinessProbe:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         runtime_readiness_probe = cls()
 

beamlit/models/runtime_readiness_probe_type_0.py

@@ -0,0 +1,43 @@
+from typing import Any, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RuntimeReadinessProbeType0")
+
+
+@_attrs_define
+class RuntimeReadinessProbeType0:
+    """The readiness probe. Should be a Kubernetes Probe type"""
+
+    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:
+        d = src_dict.copy()
+        runtime_readiness_probe_type_0 = cls()
+
+        runtime_readiness_probe_type_0.additional_properties = d
+        return runtime_readiness_probe_type_0
+
+    @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

beamlit/models/runtime_resources.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -19,7 +19,9 @@ class RuntimeResources:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         runtime_resources = cls()
 

beamlit/models/runtime_type_0.py

@@ -0,0 +1,111 @@
+from typing import Any, List, Type, 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="RuntimeType0")
+
+
+@_attrs_define
+class RuntimeType0:
+    """Set of configurations for a deployment
+
+    Attributes:
+        args (Union[Unset, List[Any]]): The arguments to pass to the deployment runtime
+        command (Union[Unset, List[Any]]): The command to run the deployment
+        envs (Union[Unset, List[Any]]): The environment variables to set in the deployment. Should be a list of
+            Kubernetes EnvVar types
+        image (Union[Unset, str]): The Docker image for the deployment
+        model (Union[Unset, str]): The slug name of the origin model. Only used if the deployment is a ModelDeployment
+        type (Union[Unset, str]): The type of origin for the deployment
+    """
+
+    args: Union[Unset, List[Any]] = UNSET
+    command: Union[Unset, List[Any]] = UNSET
+    envs: Union[Unset, List[Any]] = UNSET
+    image: Union[Unset, str] = UNSET
+    model: Union[Unset, str] = UNSET
+    type: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        args: Union[Unset, List[Any]] = UNSET
+        if not isinstance(self.args, Unset):
+            args = self.args
+
+        command: Union[Unset, List[Any]] = UNSET
+        if not isinstance(self.command, Unset):
+            command = self.command
+
+        envs: Union[Unset, List[Any]] = UNSET
+        if not isinstance(self.envs, Unset):
+            envs = self.envs
+
+        image = self.image
+
+        model = self.model
+
+        type = self.type
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if args is not UNSET:
+            field_dict["args"] = args
+        if command is not UNSET:
+            field_dict["command"] = command
+        if envs is not UNSET:
+            field_dict["envs"] = envs
+        if image is not UNSET:
+            field_dict["image"] = image
+        if model is not UNSET:
+            field_dict["model"] = model
+        if type is not UNSET:
+            field_dict["type"] = type
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        args = cast(List[Any], d.pop("args", UNSET))
+
+        command = cast(List[Any], d.pop("command", UNSET))
+
+        envs = cast(List[Any], d.pop("envs", UNSET))
+
+        image = d.pop("image", UNSET)
+
+        model = d.pop("model", UNSET)
+
+        type = d.pop("type", UNSET)
+
+        runtime_type_0 = cls(
+            args=args,
+            command=command,
+            envs=envs,
+            image=image,
+            model=model,
+            type=type,
+        )
+
+        runtime_type_0.additional_properties = d
+        return runtime_type_0
+
+    @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

beamlit/models/runtime_type_0_readiness_probe.py

@@ -0,0 +1,43 @@
+from typing import Any, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RuntimeType0ReadinessProbe")
+
+
+@_attrs_define
+class RuntimeType0ReadinessProbe:
+    """The readiness probe. Should be a Kubernetes Probe type"""
+
+    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:
+        d = src_dict.copy()
+        runtime_type_0_readiness_probe = cls()
+
+        runtime_type_0_readiness_probe.additional_properties = d
+        return runtime_type_0_readiness_probe
+
+    @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

beamlit/models/runtime_type_0_readiness_probe_type_0.py

@@ -0,0 +1,43 @@
+from typing import Any, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RuntimeType0ReadinessProbeType0")
+
+
+@_attrs_define
+class RuntimeType0ReadinessProbeType0:
+    """The readiness probe. Should be a Kubernetes Probe type"""
+
+    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:
+        d = src_dict.copy()
+        runtime_type_0_readiness_probe_type_0 = cls()
+
+        runtime_type_0_readiness_probe_type_0.additional_properties = d
+        return runtime_type_0_readiness_probe_type_0
+
+    @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