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

beamlit/models/integration_connection.py

@@ -28,11 +28,11 @@ class IntegrationConnection:
 
     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/integration_connection_spec.py

@@ -20,23 +20,27 @@ class IntegrationConnectionSpec:
     Attributes:
         config (Union[Unset, IntegrationConnectionConfig]): Integration config
         integration (Union[Unset, str]): Integration type
+        sandbox (Union[Unset, bool]): Sandbox mode
         secret (Union[Unset, IntegrationConnectionSecret]): Integration secret
     """
 
     config: Union[Unset, "IntegrationConnectionConfig"] = UNSET
     integration: Union[Unset, str] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     secret: Union[Unset, "IntegrationConnectionSecret"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         config: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.config, Unset):
+        if self.config and not isinstance(self.config, Unset):
             config = self.config.to_dict()
 
         integration = self.integration
 
+        sandbox = self.sandbox
+
         secret: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.secret, Unset):
+        if self.secret and not isinstance(self.secret, Unset):
             secret = self.secret.to_dict()
 
         field_dict: dict[str, Any] = {}
@@ -46,6 +50,8 @@ class IntegrationConnectionSpec:
             field_dict["config"] = config
         if integration is not UNSET:
             field_dict["integration"] = integration
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if secret is not UNSET:
             field_dict["secret"] = secret
 
@@ -68,6 +74,8 @@ class IntegrationConnectionSpec:
 
         integration = d.pop("integration", UNSET)
 
+        sandbox = d.pop("sandbox", UNSET)
+
         _secret = d.pop("secret", UNSET)
         secret: Union[Unset, IntegrationConnectionSecret]
         if isinstance(_secret, Unset):
@@ -78,6 +86,7 @@ class IntegrationConnectionSpec:
         integration_connection_spec = cls(
             config=config,
             integration=integration,
+            sandbox=sandbox,
             secret=secret,
         )
 

beamlit/models/integration_repository.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="IntegrationRepository")
+
+
+@_attrs_define
+class IntegrationRepository:
+    """Integration repository
+
+    Attributes:
+        id (Union[Unset, str]): Repository ID
+        name (Union[Unset, str]): Repository name
+        organization (Union[Unset, str]): Repository owner
+        url (Union[Unset, str]): Repository URL
+    """
+
+    id: Union[Unset, str] = UNSET
+    name: Union[Unset, str] = UNSET
+    organization: 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]:
+        id = self.id
+
+        name = self.name
+
+        organization = self.organization
+
+        url = self.url
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if id is not UNSET:
+            field_dict["id"] = id
+        if name is not UNSET:
+            field_dict["name"] = name
+        if organization is not UNSET:
+            field_dict["organization"] = organization
+        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()
+        id = d.pop("id", UNSET)
+
+        name = d.pop("name", UNSET)
+
+        organization = d.pop("organization", UNSET)
+
+        url = d.pop("url", UNSET)
+
+        integration_repository = cls(
+            id=id,
+            name=name,
+            organization=organization,
+            url=url,
+        )
+
+        integration_repository.additional_properties = d
+        return integration_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/last_n_requests_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="LastNRequestsMetric")
+
+
+@_attrs_define
+class LastNRequestsMetric:
+    """Last N requests
+
+    Attributes:
+        date (Union[Unset, str]): Timestamp
+        environment (Union[Unset, str]): Environment
+        workload_type (Union[Unset, str]): Workload type
+        workspace (Union[Unset, str]): Workspace
+    """
+
+    date: Union[Unset, str] = UNSET
+    environment: Union[Unset, str] = UNSET
+    workload_type: Union[Unset, str] = UNSET
+    workspace: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        date = self.date
+
+        environment = self.environment
+
+        workload_type = self.workload_type
+
+        workspace = self.workspace
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if date is not UNSET:
+            field_dict["date"] = date
+        if environment is not UNSET:
+            field_dict["environment"] = environment
+        if workload_type is not UNSET:
+            field_dict["workloadType"] = workload_type
+        if workspace is not UNSET:
+            field_dict["workspace"] = workspace
+
+        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()
+        date = d.pop("date", UNSET)
+
+        environment = d.pop("environment", UNSET)
+
+        workload_type = d.pop("workloadType", UNSET)
+
+        workspace = d.pop("workspace", UNSET)
+
+        last_n_requests_metric = cls(
+            date=date,
+            environment=environment,
+            workload_type=workload_type,
+            workspace=workspace,
+        )
+
+        last_n_requests_metric.additional_properties = d
+        return last_n_requests_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/latency_metric.py

@@ -0,0 +1,124 @@
+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.histogram_bucket import HistogramBucket
+    from ..models.histogram_stats import HistogramStats
+
+
+T = TypeVar("T", bound="LatencyMetric")
+
+
+@_attrs_define
+class LatencyMetric:
+    """Latency metrics
+
+    Attributes:
+        global_histogram (Union[Unset, HistogramBucket]): Histogram bucket
+        global_stats (Union[Unset, HistogramStats]): Histogram stats
+        histogram_per_code (Union[Unset, HistogramBucket]): Histogram bucket
+        stats_per_code (Union[Unset, HistogramStats]): Histogram stats
+    """
+
+    global_histogram: Union[Unset, "HistogramBucket"] = UNSET
+    global_stats: Union[Unset, "HistogramStats"] = UNSET
+    histogram_per_code: Union[Unset, "HistogramBucket"] = UNSET
+    stats_per_code: Union[Unset, "HistogramStats"] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        global_histogram: Union[Unset, dict[str, Any]] = UNSET
+        if self.global_histogram and not isinstance(self.global_histogram, Unset):
+            global_histogram = self.global_histogram.to_dict()
+
+        global_stats: Union[Unset, dict[str, Any]] = UNSET
+        if self.global_stats and not isinstance(self.global_stats, Unset):
+            global_stats = self.global_stats.to_dict()
+
+        histogram_per_code: Union[Unset, dict[str, Any]] = UNSET
+        if self.histogram_per_code and not isinstance(self.histogram_per_code, Unset):
+            histogram_per_code = self.histogram_per_code.to_dict()
+
+        stats_per_code: Union[Unset, dict[str, Any]] = UNSET
+        if self.stats_per_code and not isinstance(self.stats_per_code, Unset):
+            stats_per_code = self.stats_per_code.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if global_histogram is not UNSET:
+            field_dict["globalHistogram"] = global_histogram
+        if global_stats is not UNSET:
+            field_dict["globalStats"] = global_stats
+        if histogram_per_code is not UNSET:
+            field_dict["histogramPerCode"] = histogram_per_code
+        if stats_per_code is not UNSET:
+            field_dict["statsPerCode"] = stats_per_code
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.histogram_bucket import HistogramBucket
+        from ..models.histogram_stats import HistogramStats
+
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        _global_histogram = d.pop("globalHistogram", UNSET)
+        global_histogram: Union[Unset, HistogramBucket]
+        if isinstance(_global_histogram, Unset):
+            global_histogram = UNSET
+        else:
+            global_histogram = HistogramBucket.from_dict(_global_histogram)
+
+        _global_stats = d.pop("globalStats", UNSET)
+        global_stats: Union[Unset, HistogramStats]
+        if isinstance(_global_stats, Unset):
+            global_stats = UNSET
+        else:
+            global_stats = HistogramStats.from_dict(_global_stats)
+
+        _histogram_per_code = d.pop("histogramPerCode", UNSET)
+        histogram_per_code: Union[Unset, HistogramBucket]
+        if isinstance(_histogram_per_code, Unset):
+            histogram_per_code = UNSET
+        else:
+            histogram_per_code = HistogramBucket.from_dict(_histogram_per_code)
+
+        _stats_per_code = d.pop("statsPerCode", UNSET)
+        stats_per_code: Union[Unset, HistogramStats]
+        if isinstance(_stats_per_code, Unset):
+            stats_per_code = UNSET
+        else:
+            stats_per_code = HistogramStats.from_dict(_stats_per_code)
+
+        latency_metric = cls(
+            global_histogram=global_histogram,
+            global_stats=global_stats,
+            histogram_per_code=histogram_per_code,
+            stats_per_code=stats_per_code,
+        )
+
+        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/metadata.py

@@ -49,7 +49,7 @@ class Metadata:
         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/metric.py

@@ -13,26 +13,32 @@ class Metric:
     """Metric
 
     Attributes:
+        rate (Union[Unset, int]): Metric value
+        request_total (Union[Unset, int]): Metric value
         timestamp (Union[Unset, str]): Metric timestamp
-        value (Union[Unset, str]): Metric value
     """
 
+    rate: Union[Unset, int] = UNSET
+    request_total: Union[Unset, int] = 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,4 +1,4 @@
-from typing import TYPE_CHECKING, 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
@@ -6,7 +6,9 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.increase_and_rate_metric import IncreaseAndRateMetric
+    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")
@@ -17,91 +19,124 @@ class Metrics:
     """Metrics for resources
 
     Attributes:
-        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
+        agents (Union[Unset, Any]): Metrics for agents
+        functions (Union[Unset, Any]): Metrics for functions
+        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
     """
 
-    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
+    agents: Union[Unset, Any] = UNSET
+    functions: 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]:
-        inference_global = self.inference_global
+        agents = self.agents
 
-        query = self.query
+        functions = self.functions
 
-        agents: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.agents, Unset):
-            agents = self.agents.to_dict()
-
-        functions: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.functions, Unset):
-            functions = self.functions.to_dict()
+        inference_global: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.inference_global, Unset):
+            inference_global = self.inference_global
 
         models: Union[Unset, dict[str, Any]] = UNSET
-        if not isinstance(self.models, Unset):
+        if self.models and not isinstance(self.models, Unset):
             models = self.models.to_dict()
 
+        request_total = self.request_total
+
+        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: 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)
         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
+        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()
-        inference_global = d.pop("inferenceGlobal", UNSET)
-
-        query = d.pop("query", UNSET)
+        agents = d.pop("agents", 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 = d.pop("functions", UNSET)
-        functions: Union[Unset, IncreaseAndRateMetric]
-        if isinstance(_functions, Unset):
-            functions = UNSET
-        else:
-            functions = IncreaseAndRateMetric.from_dict(_functions)
+        inference_global = cast(list[Any], d.pop("inferenceGlobal", UNSET))
 
         _models = d.pop("models", UNSET)
-        models: Union[Unset, IncreaseAndRateMetric]
+        models: Union[Unset, MetricsModels]
         if isinstance(_models, Unset):
             models = UNSET
         else:
-            models = IncreaseAndRateMetric.from_dict(_models)
+            models = MetricsModels.from_dict(_models)
+
+        request_total = d.pop("requestTotal", 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: 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(
-            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/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