beamlit 0.0.33rc49__py3-none-any.whl → 0.0.34__py3-none-any.whl

beamlit/models/core_event.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="CoreEvent")
+
+
+@_attrs_define
+class CoreEvent:
+    """Core event
+
+    Attributes:
+        message (Union[Unset, str]): Event message
+        status (Union[Unset, str]): Event status
+        time (Union[Unset, str]): Event time
+        type_ (Union[Unset, str]): Event type
+    """
+
+    message: Union[Unset, str] = UNSET
+    status: Union[Unset, str] = UNSET
+    time: 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]:
+        message = self.message
+
+        status = self.status
+
+        time = self.time
+
+        type_ = self.type_
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if message is not UNSET:
+            field_dict["message"] = message
+        if status is not UNSET:
+            field_dict["status"] = status
+        if time is not UNSET:
+            field_dict["time"] = time
+        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:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        message = d.pop("message", UNSET)
+
+        status = d.pop("status", UNSET)
+
+        time = d.pop("time", UNSET)
+
+        type_ = d.pop("type", UNSET)
+
+        core_event = cls(
+            message=message,
+            status=status,
+            time=time,
+            type_=type_,
+        )
+
+        core_event.additional_properties = d
+        return core_event
+
+    @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/core_spec.py

@@ -30,6 +30,7 @@ class CoreSpec:
         policies (Union[Unset, list[str]]):
         private_clusters (Union[Unset, ModelPrivateCluster]): Private cluster where the model deployment is deployed
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
+        sandbox (Union[Unset, bool]): Sandbox mode
         serverless_config (Union[Unset, ServerlessConfig]): Configuration for a serverless deployment
     """
 
@@ -41,12 +42,13 @@ class CoreSpec:
     policies: Union[Unset, list[str]] = UNSET
     private_clusters: Union[Unset, "ModelPrivateCluster"] = UNSET
     runtime: Union[Unset, "Runtime"] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     serverless_config: Union[Unset, "ServerlessConfig"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         configurations: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.configurations, Unset):
+        if self.configurations and not isinstance(self.configurations, Unset):
             configurations = self.configurations.to_dict()
 
         enabled = self.enabled
@@ -63,7 +65,7 @@ class CoreSpec:
             integration_connections = self.integration_connections
 
         pod_template: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.pod_template, Unset):
+        if self.pod_template and not isinstance(self.pod_template, Unset):
             pod_template = self.pod_template.to_dict()
 
         policies: Union[Unset, list[str]] = UNSET
@@ -71,15 +73,17 @@ class CoreSpec:
             policies = self.policies
 
         private_clusters: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.private_clusters, Unset):
+        if self.private_clusters and not isinstance(self.private_clusters, Unset):
             private_clusters = self.private_clusters.to_dict()
 
         runtime: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.runtime, Unset):
+        if self.runtime and not isinstance(self.runtime, Unset):
             runtime = self.runtime.to_dict()
 
+        sandbox = self.sandbox
+
         serverless_config: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.serverless_config, Unset):
+        if self.serverless_config and not isinstance(self.serverless_config, Unset):
             serverless_config = self.serverless_config.to_dict()
 
         field_dict: dict[str, Any] = {}
@@ -101,6 +105,8 @@ class CoreSpec:
             field_dict["privateClusters"] = private_clusters
         if runtime is not UNSET:
             field_dict["runtime"] = runtime
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if serverless_config is not UNSET:
             field_dict["serverlessConfig"] = serverless_config
 
@@ -159,6 +165,8 @@ class CoreSpec:
         else:
             runtime = Runtime.from_dict(_runtime)
 
+        sandbox = d.pop("sandbox", UNSET)
+
         _serverless_config = d.pop("serverlessConfig", UNSET)
         serverless_config: Union[Unset, ServerlessConfig]
         if isinstance(_serverless_config, Unset):
@@ -175,6 +183,7 @@ class CoreSpec:
             policies=policies,
             private_clusters=private_clusters,
             runtime=runtime,
+            sandbox=sandbox,
             serverless_config=serverless_config,
         )
 

beamlit/models/core_spec_configurations.py

@@ -26,7 +26,7 @@ class CoreSpecConfigurations:
 
     def to_dict(self) -> dict[str, Any]:
         key: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.key, Unset):
+        if self.key and not isinstance(self.key, Unset):
             key = self.key.to_dict()
 
         field_dict: dict[str, Any] = {}

beamlit/models/core_status.py

@@ -1,33 +1,20 @@
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="CoreStatus")
 
 
 @_attrs_define
 class CoreStatus:
-    """Core status
-
-    Attributes:
-        deployment_status (Union[Unset, str]): The status of the core, can be CREATED, UPDATED, DELETED, DEPLOYED,
-            DISABLED, or FAILED
-    """
+    """Core status"""
 
-    deployment_status: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        deployment_status = self.deployment_status
-
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if deployment_status is not UNSET:
-            field_dict["deploymentStatus"] = deployment_status
 
         return field_dict
 
@@ -36,11 +23,7 @@ class CoreStatus:
         if not src_dict:
             return None
         d = src_dict.copy()
-        deployment_status = d.pop("deploymentStatus", UNSET)
-
-        core_status = cls(
-            deployment_status=deployment_status,
-        )
+        core_status = cls()
 
         core_status.additional_properties = d
         return core_status

beamlit/models/environment.py

@@ -28,11 +28,11 @@ class Environment:
 
     def to_dict(self) -> dict[str, Any]:
         metadata: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.metadata, Unset):
+        if self.metadata and not isinstance(self.metadata, Unset):
             metadata = self.metadata.to_dict()
 
         spec: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.spec, Unset):
+        if self.spec and not isinstance(self.spec, Unset):
             spec = self.spec.to_dict()
 
         field_dict: dict[str, Any] = {}

beamlit/models/environment_metadata.py

@@ -51,7 +51,7 @@ class EnvironmentMetadata:
         display_name = self.display_name
 
         labels: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.labels, Unset):
+        if self.labels and not isinstance(self.labels, Unset):
             labels = self.labels.to_dict()
 
         name = self.name

beamlit/models/function.py

@@ -6,7 +6,7 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.core_status import CoreStatus
+    from ..models.core_event import CoreEvent
     from ..models.environment_metadata import EnvironmentMetadata
     from ..models.function_spec import FunctionSpec
 
@@ -19,32 +19,41 @@ class Function:
     """Function
 
     Attributes:
+        events (Union[Unset, list['CoreEvent']]): Core events
         metadata (Union[Unset, EnvironmentMetadata]): Environment metadata
         spec (Union[Unset, FunctionSpec]): Function specification
-        status (Union[Unset, CoreStatus]): Core status
+        status (Union[Unset, str]): Function status
     """
 
+    events: Union[Unset, list["CoreEvent"]] = UNSET
     metadata: Union[Unset, "EnvironmentMetadata"] = UNSET
     spec: Union[Unset, "FunctionSpec"] = 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]:
+        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):
+        if self.metadata and not isinstance(self.metadata, Unset):
             metadata = self.metadata.to_dict()
 
         spec: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.spec, Unset):
+        if self.spec and not isinstance(self.spec, Unset):
             spec = self.spec.to_dict()
 
-        status: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.status, Unset):
-            status = self.status.to_dict()
+        status = self.status
 
         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,13 +65,20 @@ class Function:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.core_status import CoreStatus
+        from ..models.core_event import CoreEvent
         from ..models.environment_metadata import EnvironmentMetadata
         from ..models.function_spec import FunctionSpec
 
         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):
@@ -77,14 +93,10 @@ class Function:
         else:
             spec = FunctionSpec.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)
 
         function = cls(
+            events=events,
             metadata=metadata,
             spec=spec,
             status=status,

beamlit/models/function_metadata.py

@@ -51,7 +51,7 @@ class FunctionMetadata:
         display_name = self.display_name
 
         labels: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.labels, Unset):
+        if self.labels and not isinstance(self.labels, Unset):
             labels = self.labels.to_dict()
 
         name = self.name

beamlit/models/function_render.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="FunctionRender")
+
+
+@_attrs_define
+class FunctionRender:
+    """FunctionRender"""
+
+    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()
+        function_render = cls()
+
+        function_render.additional_properties = d
+        return function_render
+
+    @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/function_spec.py

@@ -32,6 +32,7 @@ class FunctionSpec:
         policies (Union[Unset, list[str]]):
         private_clusters (Union[Unset, ModelPrivateCluster]): Private cluster where the model deployment is deployed
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
+        sandbox (Union[Unset, bool]): Sandbox mode
         serverless_config (Union[Unset, ServerlessConfig]): Configuration for a serverless deployment
         description (Union[Unset, str]): Function description, very important for the agent function to work with an LLM
         kit (Union[Unset, list['FunctionKit']]): The kit of the function deployment
@@ -48,6 +49,7 @@ class FunctionSpec:
     policies: Union[Unset, list[str]] = UNSET
     private_clusters: Union[Unset, "ModelPrivateCluster"] = UNSET
     runtime: Union[Unset, "Runtime"] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     serverless_config: Union[Unset, "ServerlessConfig"] = UNSET
     description: Union[Unset, str] = UNSET
     kit: Union[Unset, list["FunctionKit"]] = UNSET
@@ -57,7 +59,7 @@ class FunctionSpec:
 
     def to_dict(self) -> dict[str, Any]:
         configurations: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.configurations, Unset):
+        if self.configurations and not isinstance(self.configurations, Unset):
             configurations = self.configurations.to_dict()
 
         enabled = self.enabled
@@ -74,7 +76,7 @@ class FunctionSpec:
             integration_connections = self.integration_connections
 
         pod_template: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.pod_template, Unset):
+        if self.pod_template and not isinstance(self.pod_template, Unset):
             pod_template = self.pod_template.to_dict()
 
         policies: Union[Unset, list[str]] = UNSET
@@ -82,15 +84,17 @@ class FunctionSpec:
             policies = self.policies
 
         private_clusters: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.private_clusters, Unset):
+        if self.private_clusters and not isinstance(self.private_clusters, Unset):
             private_clusters = self.private_clusters.to_dict()
 
         runtime: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.runtime, Unset):
+        if self.runtime and not isinstance(self.runtime, Unset):
             runtime = self.runtime.to_dict()
 
+        sandbox = self.sandbox
+
         serverless_config: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.serverless_config, Unset):
+        if self.serverless_config and not isinstance(self.serverless_config, Unset):
             serverless_config = self.serverless_config.to_dict()
 
         description = self.description
@@ -130,6 +134,8 @@ class FunctionSpec:
             field_dict["privateClusters"] = private_clusters
         if runtime is not UNSET:
             field_dict["runtime"] = runtime
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if serverless_config is not UNSET:
             field_dict["serverlessConfig"] = serverless_config
         if description is not UNSET:
@@ -198,6 +204,8 @@ class FunctionSpec:
         else:
             runtime = Runtime.from_dict(_runtime)
 
+        sandbox = d.pop("sandbox", UNSET)
+
         _serverless_config = d.pop("serverlessConfig", UNSET)
         serverless_config: Union[Unset, ServerlessConfig]
         if isinstance(_serverless_config, Unset):
@@ -232,6 +240,7 @@ class FunctionSpec:
             policies=policies,
             private_clusters=private_clusters,
             runtime=runtime,
+            sandbox=sandbox,
             serverless_config=serverless_config,
             description=description,
             kit=kit,

beamlit/models/histogram_bucket.py

@@ -0,0 +1,79 @@
+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="HistogramBucket")
+
+
+@_attrs_define
+class HistogramBucket:
+    """Histogram bucket
+
+    Attributes:
+        count (Union[Unset, int]): Count
+        end (Union[Unset, float]): End
+        start (Union[Unset, float]): Start
+    """
+
+    count: Union[Unset, int] = UNSET
+    end: Union[Unset, float] = UNSET
+    start: Union[Unset, float] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        count = self.count
+
+        end = self.end
+
+        start = self.start
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if count is not UNSET:
+            field_dict["count"] = count
+        if end is not UNSET:
+            field_dict["end"] = end
+        if start is not UNSET:
+            field_dict["start"] = start
+
+        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()
+        count = d.pop("count", UNSET)
+
+        end = d.pop("end", UNSET)
+
+        start = d.pop("start", UNSET)
+
+        histogram_bucket = cls(
+            count=count,
+            end=end,
+            start=start,
+        )
+
+        histogram_bucket.additional_properties = d
+        return histogram_bucket
+
+    @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/histogram_stats.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="HistogramStats")
+
+
+@_attrs_define
+class HistogramStats:
+    """Histogram stats
+
+    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
+    """
+
+    average: Union[Unset, float] = UNSET
+    p50: Union[Unset, float] = UNSET
+    p90: Union[Unset, float] = UNSET
+    p99: Union[Unset, float] = 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
+
+        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
+
+        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)
+
+        histogram_stats = cls(
+            average=average,
+            p50=p50,
+            p90=p90,
+            p99=p99,
+        )
+
+        histogram_stats.additional_properties = d
+        return histogram_stats
+
+    @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/increase_and_rate_metric.py

@@ -14,25 +14,19 @@ class IncreaseAndRateMetric:
 
     Attributes:
         inference_global (Union[Unset, Any]): Historical requests for all resources globally
-        query (Union[Unset, Any]): Number of requests for all resources globally
     """
 
     inference_global: Union[Unset, Any] = UNSET
-    query: Union[Unset, Any] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         inference_global = self.inference_global
 
-        query = self.query
-
         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
 
         return field_dict
 
@@ -43,11 +37,8 @@ class IncreaseAndRateMetric:
         d = src_dict.copy()
         inference_global = d.pop("inferenceGlobal", UNSET)
 
-        query = d.pop("query", UNSET)
-
         increase_and_rate_metric = cls(
             inference_global=inference_global,
-            query=query,
         )
 
         increase_and_rate_metric.additional_properties = d