beamlit 0.0.24rc21__py3-none-any.whl → 0.0.26__py3-none-any.whl

beamlit/models/agent.py

@@ -7,6 +7,7 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.agent_spec import AgentSpec
+    from ..models.core_status import CoreStatus
     from ..models.environment_metadata import EnvironmentMetadata
 
 
@@ -20,10 +21,12 @@ class Agent:
     Attributes:
         metadata (Union[Unset, EnvironmentMetadata]): Environment metadata
         spec (Union[Unset, AgentSpec]): Agent specification
+        status (Union[Unset, CoreStatus]): Core status
     """
 
     metadata: Union[Unset, "EnvironmentMetadata"] = UNSET
     spec: Union[Unset, "AgentSpec"] = UNSET
+    status: Union[Unset, "CoreStatus"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -35,6 +38,10 @@ class Agent:
         if 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()
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -42,12 +49,15 @@ class Agent:
             field_dict["metadata"] = metadata
         if spec is not UNSET:
             field_dict["spec"] = spec
+        if status is not UNSET:
+            field_dict["status"] = status
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.agent_spec import AgentSpec
+        from ..models.core_status import CoreStatus
         from ..models.environment_metadata import EnvironmentMetadata
 
         if not src_dict:
@@ -67,9 +77,17 @@ class Agent:
         else:
             spec = AgentSpec.from_dict(_spec)
 
+        _status = d.pop("status", UNSET)
+        status: Union[Unset, CoreStatus]
+        if isinstance(_status, Unset):
+            status = UNSET
+        else:
+            status = CoreStatus.from_dict(_status)
+
         agent = cls(
             metadata=metadata,
             spec=spec,
+            status=status,
         )
 
         agent.additional_properties = d

beamlit/models/agent_history_event.py

@@ -70,7 +70,7 @@ class AgentHistoryEvent:
         if status is not UNSET:
             field_dict["status"] = status
         if sub_function is not UNSET:
-            field_dict["sub_function"] = sub_function
+            field_dict["subFunction"] = sub_function
         if took is not UNSET:
             field_dict["took"] = took
         if type_ is not UNSET:
@@ -95,7 +95,7 @@ class AgentHistoryEvent:
 
         status = d.pop("status", UNSET)
 
-        sub_function = d.pop("sub_function", UNSET)
+        sub_function = d.pop("subFunction", UNSET)
 
         took = d.pop("took", UNSET)
 

beamlit/models/api_key.py

@@ -70,7 +70,7 @@ class ApiKey:
         if updated_by is not UNSET:
             field_dict["updatedBy"] = updated_by
         if api_key is not UNSET:
-            field_dict["api_key"] = api_key
+            field_dict["apiKey"] = api_key
         if expires_in is not UNSET:
             field_dict["expires_in"] = expires_in
         if id is not UNSET:
@@ -97,7 +97,7 @@ class ApiKey:
 
         updated_by = d.pop("updatedBy", UNSET)
 
-        api_key = d.pop("api_key", UNSET)
+        api_key = d.pop("apiKey", UNSET)
 
         expires_in = d.pop("expires_in", UNSET)
 

beamlit/models/continent.py

@@ -30,7 +30,7 @@ class Continent:
         field_dict.update(self.additional_properties)
         field_dict.update({})
         if display_name is not UNSET:
-            field_dict["display_name"] = display_name
+            field_dict["displayName"] = display_name
         if name is not UNSET:
             field_dict["name"] = name
 
@@ -41,7 +41,7 @@ class Continent:
         if not src_dict:
             return None
         d = src_dict.copy()
-        display_name = d.pop("display_name", UNSET)
+        display_name = d.pop("displayName", UNSET)
 
         name = d.pop("name", UNSET)
 

beamlit/models/core_status.py

@@ -0,0 +1,62 @@
+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="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
+    """
+
+    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
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        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.additional_properties = d
+        return core_status
+
+    @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/country.py

@@ -30,7 +30,7 @@ class Country:
         field_dict.update(self.additional_properties)
         field_dict.update({})
         if display_name is not UNSET:
-            field_dict["display_name"] = display_name
+            field_dict["displayName"] = display_name
         if name is not UNSET:
             field_dict["name"] = name
 
@@ -41,7 +41,7 @@ class Country:
         if not src_dict:
             return None
         d = src_dict.copy()
-        display_name = d.pop("display_name", UNSET)
+        display_name = d.pop("displayName", UNSET)
 
         name = d.pop("name", UNSET)
 

beamlit/models/environment_metrics.py

@@ -18,11 +18,9 @@ class EnvironmentMetrics:
 
     Attributes:
         inference_global (Union[Unset, list['Metric']]): Array of metrics
-        inference_per_second_global (Union[Unset, list['Metric']]): Array of metrics
     """
 
     inference_global: Union[Unset, list["Metric"]] = UNSET
-    inference_per_second_global: Union[Unset, list["Metric"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -33,20 +31,11 @@ class EnvironmentMetrics:
                 componentsschemas_array_metric_item = componentsschemas_array_metric_item_data.to_dict()
                 inference_global.append(componentsschemas_array_metric_item)
 
-        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)
-
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
         if inference_global is not UNSET:
-            field_dict["inference_global"] = inference_global
-        if inference_per_second_global is not UNSET:
-            field_dict["inference_per_second_global"] = inference_per_second_global
+            field_dict["inferenceGlobal"] = inference_global
 
         return field_dict
 
@@ -58,22 +47,14 @@ class EnvironmentMetrics:
             return None
         d = src_dict.copy()
         inference_global = []
-        _inference_global = d.pop("inference_global", UNSET)
+        _inference_global = d.pop("inferenceGlobal", UNSET)
         for componentsschemas_array_metric_item_data in _inference_global or []:
             componentsschemas_array_metric_item = Metric.from_dict(componentsschemas_array_metric_item_data)
 
             inference_global.append(componentsschemas_array_metric_item)
 
-        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)
-
         environment_metrics = cls(
             inference_global=inference_global,
-            inference_per_second_global=inference_per_second_global,
         )
 
         environment_metrics.additional_properties = d

beamlit/models/function.py

@@ -6,6 +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.environment_metadata import EnvironmentMetadata
     from ..models.function_spec import FunctionSpec
 
@@ -20,10 +21,12 @@ class Function:
     Attributes:
         metadata (Union[Unset, EnvironmentMetadata]): Environment metadata
         spec (Union[Unset, FunctionSpec]): Function specification
+        status (Union[Unset, CoreStatus]): Core status
     """
 
     metadata: Union[Unset, "EnvironmentMetadata"] = UNSET
     spec: Union[Unset, "FunctionSpec"] = UNSET
+    status: Union[Unset, "CoreStatus"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -35,6 +38,10 @@ class Function:
         if 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()
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -42,11 +49,14 @@ class Function:
             field_dict["metadata"] = metadata
         if spec is not UNSET:
             field_dict["spec"] = spec
+        if status is not UNSET:
+            field_dict["status"] = status
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.core_status import CoreStatus
         from ..models.environment_metadata import EnvironmentMetadata
         from ..models.function_spec import FunctionSpec
 
@@ -67,9 +77,17 @@ 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)
+
         function = cls(
             metadata=metadata,
             spec=spec,
+            status=status,
         )
 
         function.additional_properties = d

beamlit/models/increase_and_rate_metric.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.metric import Metric
-
-
 T = TypeVar("T", bound="IncreaseAndRateMetric")
 
 
@@ -18,70 +14,40 @@ class IncreaseAndRateMetric:
 
     Attributes:
         inference_global (Union[Unset, Any]): Historical requests for all resources globally
-        inference_per_second_global (Union[Unset, list['Metric']]): Array of metrics
         query (Union[Unset, Any]): Number of requests for all resources globally
-        query_per_second (Union[Unset, float]): RPS value (in last 24 hours) per location, for all resources globally
     """
 
     inference_global: Union[Unset, Any] = UNSET
-    inference_per_second_global: Union[Unset, list["Metric"]] = UNSET
     query: Union[Unset, Any] = UNSET
-    query_per_second: Union[Unset, float] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         inference_global = self.inference_global
 
-        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)
-
         query = self.query
 
-        query_per_second = self.query_per_second
-
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
         if inference_global is not UNSET:
-            field_dict["inference_global"] = inference_global
-        if inference_per_second_global is not UNSET:
-            field_dict["inference_per_second_global"] = inference_per_second_global
+            field_dict["inferenceGlobal"] = inference_global
         if query is not UNSET:
             field_dict["query"] = query
-        if query_per_second is not UNSET:
-            field_dict["query_per_second"] = query_per_second
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.metric import Metric
-
         if not src_dict:
             return None
         d = src_dict.copy()
-        inference_global = d.pop("inference_global", UNSET)
-
-        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_global = d.pop("inferenceGlobal", UNSET)
 
         query = d.pop("query", UNSET)
 
-        query_per_second = d.pop("query_per_second", UNSET)
-
         increase_and_rate_metric = cls(
             inference_global=inference_global,
-            inference_per_second_global=inference_per_second_global,
             query=query,
-            query_per_second=query_per_second,
         )
 
         increase_and_rate_metric.additional_properties = d

beamlit/models/integration_config.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="IntegrationConfig")
+
+
+@_attrs_define
+class IntegrationConfig:
+    """Integration config"""
+
+    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()
+        integration_config = cls()
+
+        integration_config.additional_properties = d
+        return integration_config
+
+    @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/integration_connection_secret.py

@@ -26,7 +26,7 @@ class IntegrationConnectionSecret:
         field_dict.update(self.additional_properties)
         field_dict.update({})
         if api_key is not UNSET:
-            field_dict["api_key"] = api_key
+            field_dict["apiKey"] = api_key
 
         return field_dict
 
@@ -35,7 +35,7 @@ class IntegrationConnectionSecret:
         if not src_dict:
             return None
         d = src_dict.copy()
-        api_key = d.pop("api_key", UNSET)
+        api_key = d.pop("apiKey", UNSET)
 
         integration_connection_secret = cls(
             api_key=api_key,

beamlit/models/metrics.py

@@ -7,7 +7,6 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.increase_and_rate_metric import IncreaseAndRateMetric
-    from ..models.metric import Metric
 
 
 T = TypeVar("T", bound="Metrics")
@@ -19,18 +18,14 @@ class Metrics:
 
     Attributes:
         inference_global (Union[Unset, Any]): Historical requests for all resources globally
-        inference_per_second_global (Union[Unset, list['Metric']]): Array of metrics
         query (Union[Unset, Any]): Number of requests for all resources globally
-        query_per_second (Union[Unset, float]): RPS value (in last 24 hours) per location, 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
     """
 
     inference_global: Union[Unset, Any] = UNSET
-    inference_per_second_global: Union[Unset, list["Metric"]] = UNSET
     query: Union[Unset, Any] = UNSET
-    query_per_second: Union[Unset, float] = UNSET
     agents: Union[Unset, "IncreaseAndRateMetric"] = UNSET
     functions: Union[Unset, "IncreaseAndRateMetric"] = UNSET
     models: Union[Unset, "IncreaseAndRateMetric"] = UNSET
@@ -39,17 +34,8 @@ class Metrics:
     def to_dict(self) -> dict[str, Any]:
         inference_global = self.inference_global
 
-        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)
-
         query = self.query
 
-        query_per_second = self.query_per_second
-
         agents: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.agents, Unset):
             agents = self.agents.to_dict()
@@ -66,13 +52,9 @@ class Metrics:
         field_dict.update(self.additional_properties)
         field_dict.update({})
         if inference_global is not UNSET:
-            field_dict["inference_global"] = inference_global
-        if inference_per_second_global is not UNSET:
-            field_dict["inference_per_second_global"] = inference_per_second_global
+            field_dict["inferenceGlobal"] = inference_global
         if query is not UNSET:
             field_dict["query"] = query
-        if query_per_second is not UNSET:
-            field_dict["query_per_second"] = query_per_second
         if agents is not UNSET:
             field_dict["agents"] = agents
         if functions is not UNSET:
@@ -85,24 +67,14 @@ class Metrics:
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.increase_and_rate_metric import IncreaseAndRateMetric
-        from ..models.metric import Metric
 
         if not src_dict:
             return None
         d = src_dict.copy()
-        inference_global = d.pop("inference_global", UNSET)
-
-        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_global = d.pop("inferenceGlobal", UNSET)
 
         query = d.pop("query", UNSET)
 
-        query_per_second = d.pop("query_per_second", UNSET)
-
         _agents = d.pop("agents", UNSET)
         agents: Union[Unset, IncreaseAndRateMetric]
         if isinstance(_agents, Unset):
@@ -126,9 +98,7 @@ class Metrics:
 
         metrics = cls(
             inference_global=inference_global,
-            inference_per_second_global=inference_per_second_global,
             query=query,
-            query_per_second=query_per_second,
             agents=agents,
             functions=functions,
             models=models,

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_status import CoreStatus
     from ..models.environment_metadata import EnvironmentMetadata
     from ..models.model_spec import ModelSpec
 
@@ -20,10 +21,12 @@ class Model:
     Attributes:
         metadata (Union[Unset, EnvironmentMetadata]): Environment metadata
         spec (Union[Unset, ModelSpec]): Model specification
+        status (Union[Unset, CoreStatus]): Core status
     """
 
     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]:
@@ -35,6 +38,10 @@ class Model:
         if 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()
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -42,11 +49,14 @@ class Model:
             field_dict["metadata"] = metadata
         if spec is not UNSET:
             field_dict["spec"] = spec
+        if status is not UNSET:
+            field_dict["status"] = status
 
         return field_dict
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.core_status import CoreStatus
         from ..models.environment_metadata import EnvironmentMetadata
         from ..models.model_spec import ModelSpec
 
@@ -67,9 +77,17 @@ 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)
+
         model = cls(
             metadata=metadata,
             spec=spec,
+            status=status,
         )
 
         model.additional_properties = d

beamlit/models/model_provider.py

@@ -87,7 +87,7 @@ class ModelProvider:
         if config is not UNSET:
             field_dict["config"] = config
         if display_name is not UNSET:
-            field_dict["display_name"] = display_name
+            field_dict["displayName"] = display_name
         if labels is not UNSET:
             field_dict["labels"] = labels
         if name is not UNSET:
@@ -124,7 +124,7 @@ class ModelProvider:
         else:
             config = ProviderConfig.from_dict(_config)
 
-        display_name = d.pop("display_name", UNSET)
+        display_name = d.pop("displayName", UNSET)
 
         _labels = d.pop("labels", UNSET)
         labels: Union[Unset, MetadataLabels]

beamlit/models/pending_invitation.py

@@ -64,7 +64,7 @@ class PendingInvitation:
         if email is not UNSET:
             field_dict["email"] = email
         if invited_by is not UNSET:
-            field_dict["invited_by"] = invited_by
+            field_dict["invitedBy"] = invited_by
         if role is not UNSET:
             field_dict["role"] = role
         if workspace is not UNSET:
@@ -87,7 +87,7 @@ class PendingInvitation:
 
         email = d.pop("email", UNSET)
 
-        invited_by = d.pop("invited_by", UNSET)
+        invited_by = d.pop("invitedBy", UNSET)
 
         role = d.pop("role", UNSET)