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

beamlit/api/service_accounts/create_workspace_service_account.py

@@ -6,9 +6,7 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.create_workspace_service_account_body import CreateWorkspaceServiceAccountBody
-from ...models.create_workspace_service_account_response_200 import (
-    CreateWorkspaceServiceAccountResponse200,
-)
+from ...models.create_workspace_service_account_response_200 import CreateWorkspaceServiceAccountResponse200
 from ...types import Response
 
 

beamlit/api/service_accounts/delete_workspace_service_account.py

@@ -5,9 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.delete_workspace_service_account_response_200 import (
-    DeleteWorkspaceServiceAccountResponse200,
-)
+from ...models.delete_workspace_service_account_response_200 import DeleteWorkspaceServiceAccountResponse200
 from ...types import Response
 
 

beamlit/api/service_accounts/get_workspace_service_accounts.py

@@ -5,9 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.get_workspace_service_accounts_response_200_item import (
-    GetWorkspaceServiceAccountsResponse200Item,
-)
+from ...models.get_workspace_service_accounts_response_200_item import GetWorkspaceServiceAccountsResponse200Item
 from ...types import Response
 
 

beamlit/api/service_accounts/update_workspace_service_account.py

@@ -6,9 +6,7 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.update_workspace_service_account_body import UpdateWorkspaceServiceAccountBody
-from ...models.update_workspace_service_account_response_200 import (
-    UpdateWorkspaceServiceAccountResponse200,
-)
+from ...models.update_workspace_service_account_response_200 import UpdateWorkspaceServiceAccountResponse200
 from ...types import Response
 
 

beamlit/models/__init__.py

@@ -11,6 +11,7 @@ from .agent_spec import AgentSpec
 from .api_key import ApiKey
 from .configuration import Configuration
 from .continent import Continent
+from .core_event import CoreEvent
 from .core_spec import CoreSpec
 from .core_spec_configurations import CoreSpecConfigurations
 from .core_status import CoreStatus
@@ -32,9 +33,7 @@ from .function_spec import FunctionSpec
 from .get_trace_ids_response_200 import GetTraceIdsResponse200
 from .get_trace_logs_response_200 import GetTraceLogsResponse200
 from .get_trace_response_200 import GetTraceResponse200
-from .get_workspace_service_accounts_response_200_item import (
-    GetWorkspaceServiceAccountsResponse200Item,
-)
+from .get_workspace_service_accounts_response_200_item import GetWorkspaceServiceAccountsResponse200Item
 from .increase_and_rate_metric import IncreaseAndRateMetric
 from .integration_config import IntegrationConfig
 from .integration_connection import IntegrationConnection
@@ -42,7 +41,10 @@ from .integration_connection_config import IntegrationConnectionConfig
 from .integration_connection_secret import IntegrationConnectionSecret
 from .integration_connection_spec import IntegrationConnectionSpec
 from .integration_model import IntegrationModel
+from .integration_repository import IntegrationRepository
 from .invite_workspace_user_body import InviteWorkspaceUserBody
+from .last_n_requests_metric import LastNRequestsMetric
+from .latency_metric import LatencyMetric
 from .location_response import LocationResponse
 from .metadata import Metadata
 from .metadata_labels import MetadataLabels
@@ -69,6 +71,8 @@ from .private_cluster import PrivateCluster
 from .private_location import PrivateLocation
 from .provider_config import ProviderConfig
 from .qps import QPS
+from .repository import Repository
+from .request_total_metric import RequestTotalMetric
 from .resource_deployment_metrics import ResourceDeploymentMetrics
 from .resource_deployment_metrics_inference_per_second_per_region import (
     ResourceDeploymentMetricsInferencePerSecondPerRegion,
@@ -77,12 +81,6 @@ from .resource_deployment_metrics_query_per_second_per_region_per_code import (
     ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode,
 )
 from .resource_environment_metrics import ResourceEnvironmentMetrics
-from .resource_environment_metrics_inference_per_region import (
-    ResourceEnvironmentMetricsInferencePerRegion,
-)
-from .resource_environment_metrics_query_per_region_per_code import (
-    ResourceEnvironmentMetricsQueryPerRegionPerCode,
-)
 from .resource_log import ResourceLog
 from .resource_metrics import ResourceMetrics
 from .runtime import Runtime
@@ -120,6 +118,7 @@ __all__ = (
     "ApiKey",
     "Configuration",
     "Continent",
+    "CoreEvent",
     "CoreSpec",
     "CoreSpecConfigurations",
     "CoreStatus",
@@ -149,7 +148,10 @@ __all__ = (
     "IntegrationConnectionSecret",
     "IntegrationConnectionSpec",
     "IntegrationModel",
+    "IntegrationRepository",
     "InviteWorkspaceUserBody",
+    "LastNRequestsMetric",
+    "LatencyMetric",
     "LocationResponse",
     "Metadata",
     "MetadataLabels",
@@ -176,12 +178,12 @@ __all__ = (
     "PrivateLocation",
     "ProviderConfig",
     "QPS",
+    "Repository",
+    "RequestTotalMetric",
     "ResourceDeploymentMetrics",
     "ResourceDeploymentMetricsInferencePerSecondPerRegion",
     "ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode",
     "ResourceEnvironmentMetrics",
-    "ResourceEnvironmentMetricsInferencePerRegion",
-    "ResourceEnvironmentMetricsQueryPerRegionPerCode",
     "ResourceLog",
     "ResourceMetrics",
     "Runtime",

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_event import CoreEvent
     from ..models.core_status import CoreStatus
     from ..models.environment_metadata import EnvironmentMetadata
 
@@ -19,17 +20,26 @@ class Agent:
     """Agent
 
     Attributes:
+        events (Union[Unset, list['CoreEvent']]): Core events
         metadata (Union[Unset, EnvironmentMetadata]): Environment metadata
         spec (Union[Unset, AgentSpec]): Agent specification
         status (Union[Unset, CoreStatus]): Core status
     """
 
+    events: Union[Unset, list["CoreEvent"]] = UNSET
     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]:
+        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 Agent:
         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:
@@ -57,12 +69,20 @@ class Agent:
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.agent_spec import AgentSpec
+        from ..models.core_event import CoreEvent
         from ..models.core_status import CoreStatus
         from ..models.environment_metadata import EnvironmentMetadata
 
         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 Agent:
             status = CoreStatus.from_dict(_status)
 
         agent = cls(
+            events=events,
             metadata=metadata,
             spec=spec,
             status=status,

beamlit/models/agent_spec.py

@@ -11,6 +11,7 @@ if TYPE_CHECKING:
     from ..models.flavor import Flavor
     from ..models.model_private_cluster import ModelPrivateCluster
     from ..models.pod_template_spec import PodTemplateSpec
+    from ..models.repository import Repository
     from ..models.runtime import Runtime
     from ..models.serverless_config import ServerlessConfig
 
@@ -36,6 +37,7 @@ class AgentSpec:
         description (Union[Unset, str]): Agent description
         functions (Union[Unset, list[str]]):
         model (Union[Unset, str]): Model name
+        repository (Union[Unset, Repository]): Repository
         store_id (Union[Unset, str]): Store id
     """
 
@@ -52,6 +54,7 @@ class AgentSpec:
     description: Union[Unset, str] = UNSET
     functions: Union[Unset, list[str]] = UNSET
     model: Union[Unset, str] = UNSET
+    repository: Union[Unset, "Repository"] = UNSET
     store_id: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -108,6 +111,10 @@ class AgentSpec:
 
         model = self.model
 
+        repository: Union[Unset, dict[str, Any]] = UNSET
+        if not isinstance(self.repository, Unset):
+            repository = self.repository.to_dict()
+
         store_id = self.store_id
 
         field_dict: dict[str, Any] = {}
@@ -139,6 +146,8 @@ class AgentSpec:
             field_dict["functions"] = functions
         if model is not UNSET:
             field_dict["model"] = model
+        if repository is not UNSET:
+            field_dict["repository"] = repository
         if store_id is not UNSET:
             field_dict["storeId"] = store_id
 
@@ -151,6 +160,7 @@ class AgentSpec:
         from ..models.flavor import Flavor
         from ..models.model_private_cluster import ModelPrivateCluster
         from ..models.pod_template_spec import PodTemplateSpec
+        from ..models.repository import Repository
         from ..models.runtime import Runtime
         from ..models.serverless_config import ServerlessConfig
 
@@ -218,6 +228,13 @@ class AgentSpec:
 
         model = d.pop("model", UNSET)
 
+        _repository = d.pop("repository", UNSET)
+        repository: Union[Unset, Repository]
+        if isinstance(_repository, Unset):
+            repository = UNSET
+        else:
+            repository = Repository.from_dict(_repository)
+
         store_id = d.pop("storeId", UNSET)
 
         agent_spec = cls(
@@ -234,6 +251,7 @@ class AgentSpec:
             description=description,
             functions=functions,
             model=model,
+            repository=repository,
             store_id=store_id,
         )
 

beamlit/models/core_event.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="CoreEvent")
+
+
+@_attrs_define
+class CoreEvent:
+    """Core event
+
+    Attributes:
+        message (Union[Unset, str]): Event message
+        time (Union[Unset, str]): Event time
+        type_ (Union[Unset, str]): Event type
+    """
+
+    message: 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
+
+        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 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)
+
+        time = d.pop("time", UNSET)
+
+        type_ = d.pop("type", UNSET)
+
+        core_event = cls(
+            message=message,
+            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/function.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.function_spec import FunctionSpec
@@ -19,17 +20,26 @@ 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
     """
 
+    events: Union[Unset, list["CoreEvent"]] = UNSET
     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]:
+        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 Function:
         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 Function:
 
     @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.function_spec import FunctionSpec
@@ -63,6 +76,13 @@ class Function:
         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 Function:
             status = CoreStatus.from_dict(_status)
 
         function = cls(
+            events=events,
             metadata=metadata,
             spec=spec,
             status=status,

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

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,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