beamlit 0.0.20rc3__py3-none-any.whl → 0.0.20rc5__py3-none-any.whl

beamlit/models/resource_deployment_metrics_query_per_region_per_code.py

@@ -0,0 +1,75 @@
+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="ResourceDeploymentMetricsQueryPerRegionPerCode")
+
+
+@_attrs_define
+class ResourceDeploymentMetricsQueryPerRegionPerCode:
+    """Number of requests done on the resource 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
+
+        if not src_dict:
+            return None
+        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_deployment_metrics_query_per_region_per_code = cls(
+            region=region,
+        )
+
+        resource_deployment_metrics_query_per_region_per_code.additional_properties = d
+        return resource_deployment_metrics_query_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_deployment_metrics_query_per_second_per_region_per_code.py

@@ -0,0 +1,75 @@
+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="ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode")
+
+
+@_attrs_define
+class ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode:
+    """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
+
+        if not src_dict:
+            return None
+        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_deployment_metrics_query_per_second_per_region_per_code = cls(
+            region=region,
+        )
+
+        resource_deployment_metrics_query_per_second_per_region_per_code.additional_properties = d
+        return resource_deployment_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_environment_metrics.py

@@ -0,0 +1,172 @@
+from typing import TYPE_CHECKING, Any, Dict, List, 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.metric import Metric
+    from ..models.qps import QPS
+    from ..models.resource_environment_metrics_inference_per_second_per_region import (
+        ResourceEnvironmentMetricsInferencePerSecondPerRegion,
+    )
+    from ..models.resource_environment_metrics_query_per_second_per_region_per_code import (
+        ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode,
+    )
+
+
+T = TypeVar("T", bound="ResourceEnvironmentMetrics")
+
+
+@_attrs_define
+class ResourceEnvironmentMetrics:
+    """Metrics for a single resource deployment (eg. model deployment, function deployment)
+
+    Attributes:
+        inference_per_second_global (Union[Unset, List['Metric']]): Array of metrics
+        inference_per_second_per_region (Union[Unset, ResourceEnvironmentMetricsInferencePerSecondPerRegion]):
+            Historical requests per second (RPS) per location, for the model deployment
+        query_per_second_global (Union[Unset, float]): RPS value (in last 24 hours) for the model deployment globally
+        query_per_second_per_code_global (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
+        query_per_second_per_region_per_code (Union[Unset, ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode]):
+            RPS value (in last 24 hours) per response status code per location, for the model deployment
+    """
+
+    inference_per_second_global: Union[Unset, List["Metric"]] = UNSET
+    inference_per_second_per_region: Union[Unset, "ResourceEnvironmentMetricsInferencePerSecondPerRegion"] = UNSET
+    query_per_second_global: Union[Unset, float] = UNSET
+    query_per_second_per_code_global: Union[Unset, "QPS"] = UNSET
+    query_per_second_per_region: Union[Unset, "QPS"] = UNSET
+    query_per_second_per_region_per_code: Union[Unset, "ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode"] = (
+        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
+        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)
+
+        inference_per_second_per_region: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.inference_per_second_per_region, Unset):
+            inference_per_second_per_region = self.inference_per_second_per_region.to_dict()
+
+        query_per_second_global = self.query_per_second_global
+
+        query_per_second_per_code_global: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.query_per_second_per_code_global, Unset):
+            query_per_second_per_code_global = self.query_per_second_per_code_global.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()
+
+        query_per_second_per_region_per_code: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.query_per_second_per_region_per_code, Unset):
+            query_per_second_per_region_per_code = self.query_per_second_per_region_per_code.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if inference_per_second_global is not UNSET:
+            field_dict["inference_per_second_global"] = inference_per_second_global
+        if inference_per_second_per_region is not UNSET:
+            field_dict["inference_per_second_per_region"] = inference_per_second_per_region
+        if query_per_second_global is not UNSET:
+            field_dict["query_per_second_global"] = query_per_second_global
+        if query_per_second_per_code_global is not UNSET:
+            field_dict["query_per_second_per_code_global"] = query_per_second_per_code_global
+        if query_per_second_per_region is not UNSET:
+            field_dict["query_per_second_per_region"] = query_per_second_per_region
+        if query_per_second_per_region_per_code is not UNSET:
+            field_dict["query_per_second_per_region_per_code"] = query_per_second_per_region_per_code
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.metric import Metric
+        from ..models.qps import QPS
+        from ..models.resource_environment_metrics_inference_per_second_per_region import (
+            ResourceEnvironmentMetricsInferencePerSecondPerRegion,
+        )
+        from ..models.resource_environment_metrics_query_per_second_per_region_per_code import (
+            ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode,
+        )
+
+        d = src_dict.copy()
+        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 []:
+            componentsschemas_array_metric_item = Metric.from_dict(componentsschemas_array_metric_item_data)
+
+            inference_per_second_global.append(componentsschemas_array_metric_item)
+
+        _inference_per_second_per_region = d.pop("inference_per_second_per_region", UNSET)
+        inference_per_second_per_region: Union[Unset, ResourceEnvironmentMetricsInferencePerSecondPerRegion]
+        if isinstance(_inference_per_second_per_region, Unset):
+            inference_per_second_per_region = UNSET
+        else:
+            inference_per_second_per_region = ResourceEnvironmentMetricsInferencePerSecondPerRegion.from_dict(
+                _inference_per_second_per_region
+            )
+
+        query_per_second_global = d.pop("query_per_second_global", UNSET)
+
+        _query_per_second_per_code_global = d.pop("query_per_second_per_code_global", UNSET)
+        query_per_second_per_code_global: Union[Unset, QPS]
+        if isinstance(_query_per_second_per_code_global, Unset):
+            query_per_second_per_code_global = UNSET
+        else:
+            query_per_second_per_code_global = QPS.from_dict(_query_per_second_per_code_global)
+
+        _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):
+            query_per_second_per_region = UNSET
+        else:
+            query_per_second_per_region = QPS.from_dict(_query_per_second_per_region)
+
+        _query_per_second_per_region_per_code = d.pop("query_per_second_per_region_per_code", UNSET)
+        query_per_second_per_region_per_code: Union[Unset, ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode]
+        if isinstance(_query_per_second_per_region_per_code, Unset):
+            query_per_second_per_region_per_code = UNSET
+        else:
+            query_per_second_per_region_per_code = ResourceEnvironmentMetricsQueryPerSecondPerRegionPerCode.from_dict(
+                _query_per_second_per_region_per_code
+            )
+
+        resource_environment_metrics = cls(
+            inference_per_second_global=inference_per_second_global,
+            inference_per_second_per_region=inference_per_second_per_region,
+            query_per_second_global=query_per_second_global,
+            query_per_second_per_code_global=query_per_second_per_code_global,
+            query_per_second_per_region=query_per_second_per_region,
+            query_per_second_per_region_per_code=query_per_second_per_region_per_code,
+        )
+
+        resource_environment_metrics.additional_properties = d
+        return resource_environment_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

beamlit/models/resource_environment_metrics_inference_per_second_per_region.py

@@ -0,0 +1,75 @@
+from typing import TYPE_CHECKING, Any, Dict, List, 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.metric import Metric
+
+
+T = TypeVar("T", bound="ResourceEnvironmentMetricsInferencePerSecondPerRegion")
+
+
+@_attrs_define
+class ResourceEnvironmentMetricsInferencePerSecondPerRegion:
+    """Historical requests per second (RPS) per location, for the model deployment
+
+    Attributes:
+        region (Union[Unset, List['Metric']]): Array of metrics
+    """
+
+    region: Union[Unset, List["Metric"]] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        region: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.region, Unset):
+            region = []
+            for componentsschemas_array_metric_item_data in self.region:
+                componentsschemas_array_metric_item = componentsschemas_array_metric_item_data.to_dict()
+                region.append(componentsschemas_array_metric_item)
+
+        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.metric import Metric
+
+        d = src_dict.copy()
+        region = []
+        _region = d.pop("region", UNSET)
+        for componentsschemas_array_metric_item_data in _region or []:
+            componentsschemas_array_metric_item = Metric.from_dict(componentsschemas_array_metric_item_data)
+
+            region.append(componentsschemas_array_metric_item)
+
+        resource_environment_metrics_inference_per_second_per_region = cls(
+            region=region,
+        )
+
+        resource_environment_metrics_inference_per_second_per_region.additional_properties = d
+        return resource_environment_metrics_inference_per_second_per_region
+
+    @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_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

@@ -74,6 +74,8 @@ class ResourceMetrics:
         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)

beamlit/models/runtime.py

@@ -6,7 +6,7 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.runtime_readiness_probe_type_0 import RuntimeReadinessProbeType0
+    from ..models.runtime_readiness_probe import RuntimeReadinessProbe
     from ..models.runtime_resources import RuntimeResources
 
 
@@ -24,8 +24,7 @@ class Runtime:
             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['RuntimeReadinessProbeType0', None, Unset]): The readiness probe. Should be a Kubernetes
-            Probe type
+        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
@@ -36,14 +35,12 @@ class Runtime:
     envs: Union[Unset, List[Any]] = UNSET
     image: Union[Unset, str] = UNSET
     model: Union[Unset, str] = UNSET
-    readiness_probe: Union["RuntimeReadinessProbeType0", None, Unset] = UNSET
+    readiness_probe: Union[Unset, "RuntimeReadinessProbe"] = UNSET
     resources: Union[Unset, "RuntimeResources"] = UNSET
     type: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        from ..models.runtime_readiness_probe_type_0 import RuntimeReadinessProbeType0
-
         args: Union[Unset, List[Any]] = UNSET
         if not isinstance(self.args, Unset):
             args = self.args
@@ -60,13 +57,9 @@ class Runtime:
 
         model = self.model
 
-        readiness_probe: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.readiness_probe, Unset):
-            readiness_probe = UNSET
-        elif isinstance(self.readiness_probe, RuntimeReadinessProbeType0):
+        readiness_probe: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.readiness_probe, Unset):
             readiness_probe = self.readiness_probe.to_dict()
-        else:
-            readiness_probe = self.readiness_probe
 
         resources: Union[Unset, Dict[str, Any]] = UNSET
         if not isinstance(self.resources, Unset):
@@ -98,9 +91,11 @@ class Runtime:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.runtime_readiness_probe_type_0 import RuntimeReadinessProbeType0
+        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))
 
@@ -112,22 +107,12 @@ class Runtime:
 
         model = d.pop("model", UNSET)
 
-        def _parse_readiness_probe(data: object) -> Union["RuntimeReadinessProbeType0", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                readiness_probe_type_0 = RuntimeReadinessProbeType0.from_dict(data)
-
-                return readiness_probe_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["RuntimeReadinessProbeType0", None, Unset], data)
-
-        readiness_probe = _parse_readiness_probe(d.pop("readiness_probe", UNSET))
+        _readiness_probe = d.pop("readiness_probe", UNSET)
+        readiness_probe: Union[Unset, RuntimeReadinessProbe]
+        if isinstance(_readiness_probe, Unset):
+            readiness_probe = UNSET
+        else:
+            readiness_probe = RuntimeReadinessProbe.from_dict(_readiness_probe)
 
         _resources = d.pop("resources", UNSET)
         resources: Union[Unset, RuntimeResources]

beamlit/models/runtime_readiness_probe.py

@@ -0,0 +1,45 @@
+from typing import Any, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RuntimeReadinessProbe")
+
+
+@_attrs_define
+class RuntimeReadinessProbe:
+    """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:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        runtime_readiness_probe = cls()
+
+        runtime_readiness_probe.additional_properties = d
+        return runtime_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