beamlit 0.0.34rc57__py3-none-any.whl → 0.0.34rc59__py3-none-any.whl

beamlit/models/latency_metric.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="LatencyMetric")
+
+
+@_attrs_define
+class LatencyMetric:
+    """Latency metrics"""
+
+    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()
+        latency_metric = cls()
+
+        latency_metric.additional_properties = d
+        return latency_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/metric.py

@@ -13,26 +13,32 @@ class Metric:
     """Metric
 
     Attributes:
+        rate (Union[Unset, str]): Metric value
+        request_total (Union[Unset, str]): Metric value
         timestamp (Union[Unset, str]): Metric timestamp
-        value (Union[Unset, str]): Metric value
     """
 
+    rate: Union[Unset, str] = UNSET
+    request_total: Union[Unset, str] = UNSET
     timestamp: Union[Unset, str] = UNSET
-    value: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        timestamp = self.timestamp
+        rate = self.rate
+
+        request_total = self.request_total
 
-        value = self.value
+        timestamp = self.timestamp
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
+        if rate is not UNSET:
+            field_dict["rate"] = rate
+        if request_total is not UNSET:
+            field_dict["requestTotal"] = request_total
         if timestamp is not UNSET:
             field_dict["timestamp"] = timestamp
-        if value is not UNSET:
-            field_dict["value"] = value
 
         return field_dict
 
@@ -41,13 +47,16 @@ class Metric:
         if not src_dict:
             return None
         d = src_dict.copy()
-        timestamp = d.pop("timestamp", UNSET)
+        rate = d.pop("rate", UNSET)
+
+        request_total = d.pop("requestTotal", UNSET)
 
-        value = d.pop("value", UNSET)
+        timestamp = d.pop("timestamp", UNSET)
 
         metric = cls(
+            rate=rate,
+            request_total=request_total,
             timestamp=timestamp,
-            value=value,
         )
 
         metric.additional_properties = d

beamlit/models/metrics.py

@@ -1,14 +1,10 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
+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
 
-if TYPE_CHECKING:
-    from ..models.increase_and_rate_metric import IncreaseAndRateMetric
-
-
 T = TypeVar("T", bound="Metrics")
 
 
@@ -17,91 +13,95 @@ class Metrics:
     """Metrics for resources
 
     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
-        query (Union[Unset, Any]): Number of requests for all resources globally
-        agents (Union[Unset, IncreaseAndRateMetric]): Metrics for resources
-        functions (Union[Unset, IncreaseAndRateMetric]): Metrics for resources
-        models (Union[Unset, IncreaseAndRateMetric]): Metrics for resources
+        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
     """
 
+    agents: Union[Unset, Any] = UNSET
+    functions: Union[Unset, Any] = UNSET
     inference_global: Union[Unset, Any] = UNSET
-    query: Union[Unset, Any] = UNSET
-    agents: Union[Unset, "IncreaseAndRateMetric"] = UNSET
-    functions: Union[Unset, "IncreaseAndRateMetric"] = UNSET
-    models: Union[Unset, "IncreaseAndRateMetric"] = 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
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        agents = self.agents
+
+        functions = self.functions
+
         inference_global = self.inference_global
 
-        query = self.query
+        models = self.models
+
+        request_total = self.request_total
 
-        agents: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.agents, Unset):
-            agents = self.agents.to_dict()
+        request_total_per_code = self.request_total_per_code
 
-        functions: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.functions, Unset):
-            functions = self.functions.to_dict()
+        rps = self.rps
 
-        models: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.models, Unset):
-            models = self.models.to_dict()
+        rps_per_code = self.rps_per_code
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
-        if inference_global is not UNSET:
-            field_dict["inferenceGlobal"] = inference_global
-        if query is not UNSET:
-            field_dict["query"] = query
         if agents is not UNSET:
             field_dict["agents"] = agents
         if functions is not UNSET:
             field_dict["functions"] = functions
+        if inference_global is not UNSET:
+            field_dict["inferenceGlobal"] = inference_global
         if models is not UNSET:
             field_dict["models"] = models
+        if request_total is not UNSET:
+            field_dict["requestTotal"] = request_total
+        if request_total_per_code is not UNSET:
+            field_dict["requestTotalPerCode"] = request_total_per_code
+        if rps is not UNSET:
+            field_dict["rps"] = rps
+        if rps_per_code is not UNSET:
+            field_dict["rpsPerCode"] = rps_per_code
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.increase_and_rate_metric import IncreaseAndRateMetric
-
         if not src_dict:
             return None
         d = src_dict.copy()
+        agents = d.pop("agents", UNSET)
+
+        functions = d.pop("functions", UNSET)
+
         inference_global = d.pop("inferenceGlobal", UNSET)
 
-        query = d.pop("query", UNSET)
-
-        _agents = d.pop("agents", UNSET)
-        agents: Union[Unset, IncreaseAndRateMetric]
-        if isinstance(_agents, Unset):
-            agents = UNSET
-        else:
-            agents = IncreaseAndRateMetric.from_dict(_agents)
-
-        _functions = d.pop("functions", UNSET)
-        functions: Union[Unset, IncreaseAndRateMetric]
-        if isinstance(_functions, Unset):
-            functions = UNSET
-        else:
-            functions = IncreaseAndRateMetric.from_dict(_functions)
-
-        _models = d.pop("models", UNSET)
-        models: Union[Unset, IncreaseAndRateMetric]
-        if isinstance(_models, Unset):
-            models = UNSET
-        else:
-            models = IncreaseAndRateMetric.from_dict(_models)
+        models = d.pop("models", UNSET)
+
+        request_total = d.pop("requestTotal", UNSET)
+
+        request_total_per_code = d.pop("requestTotalPerCode", UNSET)
+
+        rps = d.pop("rps", UNSET)
+
+        rps_per_code = d.pop("rpsPerCode", UNSET)
 
         metrics = cls(
-            inference_global=inference_global,
-            query=query,
             agents=agents,
             functions=functions,
+            inference_global=inference_global,
             models=models,
+            request_total=request_total,
+            request_total_per_code=request_total_per_code,
+            rps=rps,
+            rps_per_code=rps_per_code,
         )
 
         metrics.additional_properties = d

beamlit/models/model.py

@@ -6,6 +6,7 @@ from attrs import field as _attrs_field
 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
@@ -19,17 +20,26 @@ class Model:
     """Logical object representing a model, that can be instantiated in multiple environments as model deployments
 
     Attributes:
+        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
     """
 
+    events: Union[Unset, list["CoreEvent"]] = UNSET
     metadata: Union[Unset, "EnvironmentMetadata"] = UNSET
     spec: Union[Unset, "ModelSpec"] = UNSET
     status: Union[Unset, "CoreStatus"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        events: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.events, Unset):
+            events = []
+            for componentsschemas_core_events_item_data in self.events:
+                componentsschemas_core_events_item = componentsschemas_core_events_item_data.to_dict()
+                events.append(componentsschemas_core_events_item)
+
         metadata: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.metadata, Unset):
             metadata = self.metadata.to_dict()
@@ -45,6 +55,8 @@ class Model:
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
+        if events is not UNSET:
+            field_dict["events"] = events
         if metadata is not UNSET:
             field_dict["metadata"] = metadata
         if spec is not UNSET:
@@ -56,6 +68,7 @@ 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
@@ -63,6 +76,13 @@ class Model:
         if not src_dict:
             return None
         d = src_dict.copy()
+        events = []
+        _events = d.pop("events", UNSET)
+        for componentsschemas_core_events_item_data in _events or []:
+            componentsschemas_core_events_item = CoreEvent.from_dict(componentsschemas_core_events_item_data)
+
+            events.append(componentsschemas_core_events_item)
+
         _metadata = d.pop("metadata", UNSET)
         metadata: Union[Unset, EnvironmentMetadata]
         if isinstance(_metadata, Unset):
@@ -85,6 +105,7 @@ class Model:
             status = CoreStatus.from_dict(_status)
 
         model = cls(
+            events=events,
             metadata=metadata,
             spec=spec,
             status=status,

beamlit/models/repository.py

@@ -0,0 +1,70 @@
+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="Repository")
+
+
+@_attrs_define
+class Repository:
+    """Repository
+
+    Attributes:
+        type_ (Union[Unset, str]): Repository type
+        url (Union[Unset, str]): Repository URL
+    """
+
+    type_: Union[Unset, str] = UNSET
+    url: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        type_ = self.type_
+
+        url = self.url
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if type_ is not UNSET:
+            field_dict["type"] = type_
+        if url is not UNSET:
+            field_dict["url"] = url
+
+        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()
+        type_ = d.pop("type", UNSET)
+
+        url = d.pop("url", UNSET)
+
+        repository = cls(
+            type_=type_,
+            url=url,
+        )
+
+        repository.additional_properties = d
+        return repository
+
+    @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_metric.py

@@ -0,0 +1,88 @@
+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="RequestTotalMetric")
+
+
+@_attrs_define
+class RequestTotalMetric:
+    """Metrics for request total
+
+    Attributes:
+        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
+        rps (Union[Unset, Any]): Number of requests for all resources globally
+        rps_per_code (Union[Unset, Any]): Number of requests for all resources globally
+    """
+
+    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
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        request_total = self.request_total
+
+        request_total_per_code = self.request_total_per_code
+
+        rps = self.rps
+
+        rps_per_code = self.rps_per_code
+
+        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 request_total_per_code is not UNSET:
+            field_dict["requestTotalPerCode"] = request_total_per_code
+        if rps is not UNSET:
+            field_dict["rps"] = rps
+        if rps_per_code is not UNSET:
+            field_dict["rpsPerCode"] = rps_per_code
+
+        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)
+
+        request_total_per_code = d.pop("requestTotalPerCode", UNSET)
+
+        rps = d.pop("rps", UNSET)
+
+        rps_per_code = d.pop("rpsPerCode", UNSET)
+
+        request_total_metric = cls(
+            request_total=request_total,
+            request_total_per_code=request_total_per_code,
+            rps=rps,
+            rps_per_code=rps_per_code,
+        )
+
+        request_total_metric.additional_properties = d
+        return request_total_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