beamlit 0.0.34rc71__py3-none-any.whl → 0.0.34rc73__py3-none-any.whl

beamlit/models/metrics.py

@@ -1,10 +1,16 @@
-from typing import 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
 
 from ..types import UNSET, Unset
 
+if TYPE_CHECKING:
+    from ..models.metrics_models import MetricsModels
+    from ..models.metrics_request_total_per_code import MetricsRequestTotalPerCode
+    from ..models.metrics_rps_per_code import MetricsRpsPerCode
+
+
 T = TypeVar("T", bound="Metrics")
 
 
@@ -15,22 +21,24 @@ class Metrics:
     Attributes:
         agents (Union[Unset, Any]): Metrics for agents
         functions (Union[Unset, Any]): Metrics for functions
-        inference_global (Union[Unset, Any]): Historical requests for all resources globally
-        models (Union[Unset, Any]): Metrics for models
-        request_total (Union[Unset, Any]): Number of requests for all resources globally
-        request_total_per_code (Union[Unset, Any]): Number of requests for all resources globally per code
-        rps (Union[Unset, Any]): Number of requests per second for all resources globally
-        rps_per_code (Union[Unset, Any]): Number of requests per second for all resources globally per code
+        inference_global (Union[Unset, list[Any]]): Historical requests for all resources globally
+        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
     """
 
     agents: Union[Unset, Any] = UNSET
     functions: Union[Unset, Any] = UNSET
-    inference_global: Union[Unset, Any] = UNSET
-    models: Union[Unset, Any] = UNSET
-    request_total: Union[Unset, Any] = UNSET
-    request_total_per_code: Union[Unset, Any] = UNSET
-    rps: Union[Unset, Any] = UNSET
-    rps_per_code: Union[Unset, Any] = UNSET
+    inference_global: Union[Unset, list[Any]] = 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
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -38,17 +46,25 @@ class Metrics:
 
         functions = self.functions
 
-        inference_global = self.inference_global
+        inference_global: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.inference_global, Unset):
+            inference_global = self.inference_global
 
-        models = self.models
+        models: Union[Unset, dict[str, Any]] = UNSET
+        if self.models and not isinstance(self.models, Unset):
+            models = self.models.to_dict()
 
         request_total = self.request_total
 
-        request_total_per_code = self.request_total_per_code
+        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):
+            request_total_per_code = self.request_total_per_code.to_dict()
 
         rps = self.rps
 
-        rps_per_code = self.rps_per_code
+        rps_per_code: Union[Unset, dict[str, Any]] = UNSET
+        if self.rps_per_code and not isinstance(self.rps_per_code, Unset):
+            rps_per_code = self.rps_per_code.to_dict()
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -74,6 +90,10 @@ class Metrics:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.metrics_models import MetricsModels
+        from ..models.metrics_request_total_per_code import MetricsRequestTotalPerCode
+        from ..models.metrics_rps_per_code import MetricsRpsPerCode
+
         if not src_dict:
             return None
         d = src_dict.copy()
@@ -81,17 +101,32 @@ class Metrics:
 
         functions = d.pop("functions", UNSET)
 
-        inference_global = d.pop("inferenceGlobal", UNSET)
+        inference_global = cast(list[Any], d.pop("inferenceGlobal", UNSET))
 
-        models = d.pop("models", UNSET)
+        _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", UNSET)
 
-        request_total_per_code = d.pop("requestTotalPerCode", UNSET)
+        _request_total_per_code = d.pop("requestTotalPerCode", 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", UNSET)
+        _rps_per_code = d.pop("rpsPerCode", 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)
 
         metrics = cls(
             agents=agents,

beamlit/models/metrics_models.py

@@ -0,0 +1,45 @@
+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:
+        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

beamlit/models/metrics_request_total_per_code.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="MetricsRequestTotalPerCode")
+
+
+@_attrs_define
+class MetricsRequestTotalPerCode:
+    """Number of requests 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:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        metrics_request_total_per_code = cls()
+
+        metrics_request_total_per_code.additional_properties = d
+        return metrics_request_total_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/metrics_rps_per_code.py

@@ -0,0 +1,45 @@
+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:
+        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

beamlit/models/model.py

@@ -7,7 +7,6 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.core_event import CoreEvent
-    from ..models.core_status import CoreStatus
     from ..models.environment_metadata import EnvironmentMetadata
     from ..models.model_spec import ModelSpec
 
@@ -23,13 +22,13 @@ class Model:
         events (Union[Unset, list['CoreEvent']]): Core events
         metadata (Union[Unset, EnvironmentMetadata]): Environment metadata
         spec (Union[Unset, ModelSpec]): Model specification
-        status (Union[Unset, CoreStatus]): Core status
+        status (Union[Unset, str]): Model status
     """
 
     events: Union[Unset, list["CoreEvent"]] = UNSET
     metadata: Union[Unset, "EnvironmentMetadata"] = UNSET
     spec: Union[Unset, "ModelSpec"] = UNSET
-    status: Union[Unset, "CoreStatus"] = UNSET
+    status: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -48,9 +47,7 @@ class Model:
         if self.spec and not isinstance(self.spec, Unset):
             spec = self.spec.to_dict()
 
-        status: Union[Unset, dict[str, Any]] = UNSET
-        if self.status and not isinstance(self.status, Unset):
-            status = self.status.to_dict()
+        status = self.status
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -69,7 +66,6 @@ class Model:
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.core_event import CoreEvent
-        from ..models.core_status import CoreStatus
         from ..models.environment_metadata import EnvironmentMetadata
         from ..models.model_spec import ModelSpec
 
@@ -97,12 +93,7 @@ class Model:
         else:
             spec = ModelSpec.from_dict(_spec)
 
-        _status = d.pop("status", UNSET)
-        status: Union[Unset, CoreStatus]
-        if isinstance(_status, Unset):
-            status = UNSET
-        else:
-            status = CoreStatus.from_dict(_status)
+        status = d.pop("status", UNSET)
 
         model = cls(
             events=events,

beamlit/models/request_duration_over_time_metric.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="RequestDurationOverTimeMetric")
+
+
+@_attrs_define
+class RequestDurationOverTimeMetric:
+    """Request duration over time metric
+
+    Attributes:
+        average (Union[Unset, float]): Average request duration
+        p50 (Union[Unset, float]): P50 request duration
+        p90 (Union[Unset, float]): P90 request duration
+        p99 (Union[Unset, float]): P99 request duration
+        timestamp (Union[Unset, str]): Timestamp
+    """
+
+    average: Union[Unset, float] = UNSET
+    p50: Union[Unset, float] = UNSET
+    p90: Union[Unset, float] = UNSET
+    p99: Union[Unset, float] = UNSET
+    timestamp: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        average = self.average
+
+        p50 = self.p50
+
+        p90 = self.p90
+
+        p99 = self.p99
+
+        timestamp = self.timestamp
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if average is not UNSET:
+            field_dict["average"] = average
+        if p50 is not UNSET:
+            field_dict["p50"] = p50
+        if p90 is not UNSET:
+            field_dict["p90"] = p90
+        if p99 is not UNSET:
+            field_dict["p99"] = p99
+        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:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        average = d.pop("average", UNSET)
+
+        p50 = d.pop("p50", UNSET)
+
+        p90 = d.pop("p90", UNSET)
+
+        p99 = d.pop("p99", UNSET)
+
+        timestamp = d.pop("timestamp", UNSET)
+
+        request_duration_over_time_metric = cls(
+            average=average,
+            p50=p50,
+            p90=p90,
+            p99=p99,
+            timestamp=timestamp,
+        )
+
+        request_duration_over_time_metric.additional_properties = d
+        return request_duration_over_time_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

beamlit/models/request_duration_over_time_metrics.py

@@ -0,0 +1,74 @@
+from typing import TYPE_CHECKING, Any, 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.request_duration_over_time_metric import RequestDurationOverTimeMetric
+
+
+T = TypeVar("T", bound="RequestDurationOverTimeMetrics")
+
+
+@_attrs_define
+class RequestDurationOverTimeMetrics:
+    """Request duration over time metrics
+
+    Attributes:
+        request_duration_over_time (Union[Unset, RequestDurationOverTimeMetric]): Request duration over time metric
+    """
+
+    request_duration_over_time: Union[Unset, "RequestDurationOverTimeMetric"] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        request_duration_over_time: Union[Unset, dict[str, Any]] = UNSET
+        if self.request_duration_over_time and not isinstance(self.request_duration_over_time, Unset):
+            request_duration_over_time = self.request_duration_over_time.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if request_duration_over_time is not UNSET:
+            field_dict["requestDurationOverTime"] = request_duration_over_time
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.request_duration_over_time_metric import RequestDurationOverTimeMetric
+
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        _request_duration_over_time = d.pop("requestDurationOverTime", UNSET)
+        request_duration_over_time: Union[Unset, RequestDurationOverTimeMetric]
+        if isinstance(_request_duration_over_time, Unset):
+            request_duration_over_time = UNSET
+        else:
+            request_duration_over_time = RequestDurationOverTimeMetric.from_dict(_request_duration_over_time)
+
+        request_duration_over_time_metrics = cls(
+            request_duration_over_time=request_duration_over_time,
+        )
+
+        request_duration_over_time_metrics.additional_properties = d
+        return request_duration_over_time_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/request_total_by_origin_metric.py

@@ -0,0 +1,103 @@
+from typing import TYPE_CHECKING, Any, 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.request_total_by_origin_metric_request_total_by_origin import (
+        RequestTotalByOriginMetricRequestTotalByOrigin,
+    )
+    from ..models.request_total_by_origin_metric_request_total_by_origin_and_code import (
+        RequestTotalByOriginMetricRequestTotalByOriginAndCode,
+    )
+
+
+T = TypeVar("T", bound="RequestTotalByOriginMetric")
+
+
+@_attrs_define
+class RequestTotalByOriginMetric:
+    """Request total by origin metric
+
+    Attributes:
+        request_total_by_origin (Union[Unset, RequestTotalByOriginMetricRequestTotalByOrigin]): Request total by origin
+        request_total_by_origin_and_code (Union[Unset, RequestTotalByOriginMetricRequestTotalByOriginAndCode]): Request
+            total by origin and code
+    """
+
+    request_total_by_origin: Union[Unset, "RequestTotalByOriginMetricRequestTotalByOrigin"] = UNSET
+    request_total_by_origin_and_code: Union[Unset, "RequestTotalByOriginMetricRequestTotalByOriginAndCode"] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        request_total_by_origin: Union[Unset, dict[str, Any]] = UNSET
+        if self.request_total_by_origin and not isinstance(self.request_total_by_origin, Unset):
+            request_total_by_origin = self.request_total_by_origin.to_dict()
+
+        request_total_by_origin_and_code: Union[Unset, dict[str, Any]] = UNSET
+        if self.request_total_by_origin_and_code and not isinstance(self.request_total_by_origin_and_code, Unset):
+            request_total_by_origin_and_code = self.request_total_by_origin_and_code.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if request_total_by_origin is not UNSET:
+            field_dict["requestTotalByOrigin"] = request_total_by_origin
+        if request_total_by_origin_and_code is not UNSET:
+            field_dict["requestTotalByOriginAndCode"] = request_total_by_origin_and_code
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.request_total_by_origin_metric_request_total_by_origin import (
+            RequestTotalByOriginMetricRequestTotalByOrigin,
+        )
+        from ..models.request_total_by_origin_metric_request_total_by_origin_and_code import (
+            RequestTotalByOriginMetricRequestTotalByOriginAndCode,
+        )
+
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        _request_total_by_origin = d.pop("requestTotalByOrigin", UNSET)
+        request_total_by_origin: Union[Unset, RequestTotalByOriginMetricRequestTotalByOrigin]
+        if isinstance(_request_total_by_origin, Unset):
+            request_total_by_origin = UNSET
+        else:
+            request_total_by_origin = RequestTotalByOriginMetricRequestTotalByOrigin.from_dict(_request_total_by_origin)
+
+        _request_total_by_origin_and_code = d.pop("requestTotalByOriginAndCode", UNSET)
+        request_total_by_origin_and_code: Union[Unset, RequestTotalByOriginMetricRequestTotalByOriginAndCode]
+        if isinstance(_request_total_by_origin_and_code, Unset):
+            request_total_by_origin_and_code = UNSET
+        else:
+            request_total_by_origin_and_code = RequestTotalByOriginMetricRequestTotalByOriginAndCode.from_dict(
+                _request_total_by_origin_and_code
+            )
+
+        request_total_by_origin_metric = cls(
+            request_total_by_origin=request_total_by_origin,
+            request_total_by_origin_and_code=request_total_by_origin_and_code,
+        )
+
+        request_total_by_origin_metric.additional_properties = d
+        return request_total_by_origin_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

beamlit/models/request_total_by_origin_metric_request_total_by_origin.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RequestTotalByOriginMetricRequestTotalByOrigin")
+
+
+@_attrs_define
+class RequestTotalByOriginMetricRequestTotalByOrigin:
+    """Request total by origin"""
+
+    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()
+        request_total_by_origin_metric_request_total_by_origin = cls()
+
+        request_total_by_origin_metric_request_total_by_origin.additional_properties = d
+        return request_total_by_origin_metric_request_total_by_origin
+
+    @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