beamlit 0.1.0__py3-none-any.whl

beamlit/models/serverless_config.py

@@ -0,0 +1,122 @@
+from typing import Any, Type, 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="ServerlessConfig")
+
+
+@_attrs_define
+class ServerlessConfig:
+    """Configuration for the serverless model deployment
+
+    Attributes:
+        last_pod_retention_period (Union[Unset, str]): The last pod retention period
+        max_num_replicas (Union[Unset, int]): The maximum number of replicas
+        metric (Union[Unset, str]): Metric to scale on, can be "cpu" or "memory" or "rps" or "concurrency"
+        min_num_replicas (Union[Unset, int]): The minimum number of replicas
+        scale_down_delay (Union[Unset, str]): The scale down delay
+        scale_up_minimum (Union[Unset, int]): The scale up minimum
+        stable_window (Union[Unset, str]): The stable window
+        target (Union[Unset, str]): Target value for the metric
+    """
+
+    last_pod_retention_period: Union[Unset, str] = UNSET
+    max_num_replicas: Union[Unset, int] = UNSET
+    metric: Union[Unset, str] = UNSET
+    min_num_replicas: Union[Unset, int] = UNSET
+    scale_down_delay: Union[Unset, str] = UNSET
+    scale_up_minimum: Union[Unset, int] = UNSET
+    stable_window: Union[Unset, str] = UNSET
+    target: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        last_pod_retention_period = self.last_pod_retention_period
+
+        max_num_replicas = self.max_num_replicas
+
+        metric = self.metric
+
+        min_num_replicas = self.min_num_replicas
+
+        scale_down_delay = self.scale_down_delay
+
+        scale_up_minimum = self.scale_up_minimum
+
+        stable_window = self.stable_window
+
+        target = self.target
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if last_pod_retention_period is not UNSET:
+            field_dict["last_pod_retention_period"] = last_pod_retention_period
+        if max_num_replicas is not UNSET:
+            field_dict["max_num_replicas"] = max_num_replicas
+        if metric is not UNSET:
+            field_dict["metric"] = metric
+        if min_num_replicas is not UNSET:
+            field_dict["min_num_replicas"] = min_num_replicas
+        if scale_down_delay is not UNSET:
+            field_dict["scale_down_delay"] = scale_down_delay
+        if scale_up_minimum is not UNSET:
+            field_dict["scale_up_minimum"] = scale_up_minimum
+        if stable_window is not UNSET:
+            field_dict["stable_window"] = stable_window
+        if target is not UNSET:
+            field_dict["target"] = target
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        last_pod_retention_period = d.pop("last_pod_retention_period", UNSET)
+
+        max_num_replicas = d.pop("max_num_replicas", UNSET)
+
+        metric = d.pop("metric", UNSET)
+
+        min_num_replicas = d.pop("min_num_replicas", UNSET)
+
+        scale_down_delay = d.pop("scale_down_delay", UNSET)
+
+        scale_up_minimum = d.pop("scale_up_minimum", UNSET)
+
+        stable_window = d.pop("stable_window", UNSET)
+
+        target = d.pop("target", UNSET)
+
+        serverless_config = cls(
+            last_pod_retention_period=last_pod_retention_period,
+            max_num_replicas=max_num_replicas,
+            metric=metric,
+            min_num_replicas=min_num_replicas,
+            scale_down_delay=scale_down_delay,
+            scale_up_minimum=scale_up_minimum,
+            stable_window=stable_window,
+            target=target,
+        )
+
+        serverless_config.additional_properties = d
+        return serverless_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/standard_fields_dynamo_db.py

@@ -0,0 +1,86 @@
+from typing import Any, Type, 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="StandardFieldsDynamoDb")
+
+
+@_attrs_define
+class StandardFieldsDynamoDb:
+    """Standard fields for Persistance
+
+    Attributes:
+        created_at (Union[Unset, str]): The date and time when the resource was created
+        created_by (Union[Unset, str]): The user or service account who created the resource
+        updated_at (Union[Unset, str]): The date and time when the resource was updated
+        updated_by (Union[Unset, str]): The user or service account who updated the resource
+    """
+
+    created_at: Union[Unset, str] = UNSET
+    created_by: Union[Unset, str] = UNSET
+    updated_at: Union[Unset, str] = UNSET
+    updated_by: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        created_at = self.created_at
+
+        created_by = self.created_by
+
+        updated_at = self.updated_at
+
+        updated_by = self.updated_by
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if created_at is not UNSET:
+            field_dict["created_at"] = created_at
+        if created_by is not UNSET:
+            field_dict["created_by"] = created_by
+        if updated_at is not UNSET:
+            field_dict["updated_at"] = updated_at
+        if updated_by is not UNSET:
+            field_dict["updated_by"] = updated_by
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        created_at = d.pop("created_at", UNSET)
+
+        created_by = d.pop("created_by", UNSET)
+
+        updated_at = d.pop("updated_at", UNSET)
+
+        updated_by = d.pop("updated_by", UNSET)
+
+        standard_fields_dynamo_db = cls(
+            created_at=created_at,
+            created_by=created_by,
+            updated_at=updated_at,
+            updated_by=updated_by,
+        )
+
+        standard_fields_dynamo_db.additional_properties = d
+        return standard_fields_dynamo_db
+
+    @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/store_agent.py

@@ -0,0 +1,165 @@
+from typing import TYPE_CHECKING, Any, Dict, List, Type, 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.store_agent_labels import StoreAgentLabels
+    from ..models.store_configuration import StoreConfiguration
+
+
+T = TypeVar("T", bound="StoreAgent")
+
+
+@_attrs_define
+class StoreAgent:
+    """Store agent
+
+    Attributes:
+        created_at (Union[Unset, str]): The date and time when the resource was created
+        created_by (Union[Unset, str]): The user or service account who created the resource
+        updated_at (Union[Unset, str]): The date and time when the resource was updated
+        updated_by (Union[Unset, str]): The user or service account who updated the resource
+        configuration (Union[Unset, List['StoreConfiguration']]): Store agent configuration
+        description (Union[Unset, str]): Store agent description
+        display_name (Union[Unset, str]): Store agent display name
+        image (Union[Unset, str]): Store agent image
+        labels (Union[Unset, StoreAgentLabels]): Store agent labels
+        name (Union[Unset, str]): Store agent name
+    """
+
+    created_at: Union[Unset, str] = UNSET
+    created_by: Union[Unset, str] = UNSET
+    updated_at: Union[Unset, str] = UNSET
+    updated_by: Union[Unset, str] = UNSET
+    configuration: Union[Unset, List["StoreConfiguration"]] = UNSET
+    description: Union[Unset, str] = UNSET
+    display_name: Union[Unset, str] = UNSET
+    image: Union[Unset, str] = UNSET
+    labels: Union[Unset, "StoreAgentLabels"] = UNSET
+    name: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        created_at = self.created_at
+
+        created_by = self.created_by
+
+        updated_at = self.updated_at
+
+        updated_by = self.updated_by
+
+        configuration: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.configuration, Unset):
+            configuration = []
+            for configuration_item_data in self.configuration:
+                configuration_item = configuration_item_data.to_dict()
+                configuration.append(configuration_item)
+
+        description = self.description
+
+        display_name = self.display_name
+
+        image = self.image
+
+        labels: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.labels, Unset):
+            labels = self.labels.to_dict()
+
+        name = self.name
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if created_at is not UNSET:
+            field_dict["created_at"] = created_at
+        if created_by is not UNSET:
+            field_dict["created_by"] = created_by
+        if updated_at is not UNSET:
+            field_dict["updated_at"] = updated_at
+        if updated_by is not UNSET:
+            field_dict["updated_by"] = updated_by
+        if configuration is not UNSET:
+            field_dict["configuration"] = configuration
+        if description is not UNSET:
+            field_dict["description"] = description
+        if display_name is not UNSET:
+            field_dict["display_name"] = display_name
+        if image is not UNSET:
+            field_dict["image"] = image
+        if labels is not UNSET:
+            field_dict["labels"] = labels
+        if name is not UNSET:
+            field_dict["name"] = name
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.store_agent_labels import StoreAgentLabels
+        from ..models.store_configuration import StoreConfiguration
+
+        d = src_dict.copy()
+        created_at = d.pop("created_at", UNSET)
+
+        created_by = d.pop("created_by", UNSET)
+
+        updated_at = d.pop("updated_at", UNSET)
+
+        updated_by = d.pop("updated_by", UNSET)
+
+        configuration = []
+        _configuration = d.pop("configuration", UNSET)
+        for configuration_item_data in _configuration or []:
+            configuration_item = StoreConfiguration.from_dict(configuration_item_data)
+
+            configuration.append(configuration_item)
+
+        description = d.pop("description", UNSET)
+
+        display_name = d.pop("display_name", UNSET)
+
+        image = d.pop("image", UNSET)
+
+        _labels = d.pop("labels", UNSET)
+        labels: Union[Unset, StoreAgentLabels]
+        if isinstance(_labels, Unset):
+            labels = UNSET
+        else:
+            labels = StoreAgentLabels.from_dict(_labels)
+
+        name = d.pop("name", UNSET)
+
+        store_agent = cls(
+            created_at=created_at,
+            created_by=created_by,
+            updated_at=updated_at,
+            updated_by=updated_by,
+            configuration=configuration,
+            description=description,
+            display_name=display_name,
+            image=image,
+            labels=labels,
+            name=name,
+        )
+
+        store_agent.additional_properties = d
+        return store_agent
+
+    @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/store_agent_configuration.py

@@ -0,0 +1,95 @@
+from typing import Any, Type, 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="StoreAgentConfiguration")
+
+
+@_attrs_define
+class StoreAgentConfiguration:
+    """Store agent configuration used to configure your agent directly from beamlit interface
+
+    Attributes:
+        description (Union[Unset, str]): Store function configuration description
+        if_ (Union[Unset, str]): Conditional rendering for the configuration, example: provider === 'openai'
+        name (Union[Unset, str]): Store function configuration name
+        required (Union[Unset, bool]): Store function configuration required
+        secret (Union[Unset, bool]): Store function configuration secret
+    """
+
+    description: Union[Unset, str] = UNSET
+    if_: Union[Unset, str] = UNSET
+    name: Union[Unset, str] = UNSET
+    required: Union[Unset, bool] = UNSET
+    secret: Union[Unset, bool] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        description = self.description
+
+        if_ = self.if_
+
+        name = self.name
+
+        required = self.required
+
+        secret = self.secret
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if description is not UNSET:
+            field_dict["description"] = description
+        if if_ is not UNSET:
+            field_dict["if"] = if_
+        if name is not UNSET:
+            field_dict["name"] = name
+        if required is not UNSET:
+            field_dict["required"] = required
+        if secret is not UNSET:
+            field_dict["secret"] = secret
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        description = d.pop("description", UNSET)
+
+        if_ = d.pop("if", UNSET)
+
+        name = d.pop("name", UNSET)
+
+        required = d.pop("required", UNSET)
+
+        secret = d.pop("secret", UNSET)
+
+        store_agent_configuration = cls(
+            description=description,
+            if_=if_,
+            name=name,
+            required=required,
+            secret=secret,
+        )
+
+        store_agent_configuration.additional_properties = d
+        return store_agent_configuration
+
+    @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/store_agent_labels.py

@@ -0,0 +1,43 @@
+from typing import Any, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="StoreAgentLabels")
+
+
+@_attrs_define
+class StoreAgentLabels:
+    """Store agent labels"""
+
+    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:
+        d = src_dict.copy()
+        store_agent_labels = cls()
+
+        store_agent_labels.additional_properties = d
+        return store_agent_labels
+
+    @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/store_configuration.py

@@ -0,0 +1,149 @@
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.store_configuration_option import StoreConfigurationOption
+
+
+T = TypeVar("T", bound="StoreConfiguration")
+
+
+@_attrs_define
+class StoreConfiguration:
+    """Store configuration for resources (eg: agent, function, etc)
+
+    Attributes:
+        available_models (Union[Unset, List[str]]): Available models for the configuration
+        description (Union[Unset, str]): Store configuration description
+        display_name (Union[Unset, str]): Store configuration display name
+        if_ (Union[Unset, str]): Conditional rendering for the configuration, example: provider === 'openai'
+        name (Union[Unset, str]): Store configuration name
+        options (Union[Unset, List['StoreConfigurationOption']]):
+        required (Union[Unset, bool]): Store configuration required
+        secret (Union[Unset, bool]): Store configuration secret
+        type (Union[Unset, str]): Store configuration type
+    """
+
+    available_models: Union[Unset, List[str]] = UNSET
+    description: Union[Unset, str] = UNSET
+    display_name: Union[Unset, str] = UNSET
+    if_: Union[Unset, str] = UNSET
+    name: Union[Unset, str] = UNSET
+    options: Union[Unset, List["StoreConfigurationOption"]] = UNSET
+    required: Union[Unset, bool] = UNSET
+    secret: Union[Unset, bool] = UNSET
+    type: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        available_models: Union[Unset, List[str]] = UNSET
+        if not isinstance(self.available_models, Unset):
+            available_models = self.available_models
+
+        description = self.description
+
+        display_name = self.display_name
+
+        if_ = self.if_
+
+        name = self.name
+
+        options: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.options, Unset):
+            options = []
+            for options_item_data in self.options:
+                options_item = options_item_data.to_dict()
+                options.append(options_item)
+
+        required = self.required
+
+        secret = self.secret
+
+        type = self.type
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if available_models is not UNSET:
+            field_dict["available_models"] = available_models
+        if description is not UNSET:
+            field_dict["description"] = description
+        if display_name is not UNSET:
+            field_dict["display_name"] = display_name
+        if if_ is not UNSET:
+            field_dict["if"] = if_
+        if name is not UNSET:
+            field_dict["name"] = name
+        if options is not UNSET:
+            field_dict["options"] = options
+        if required is not UNSET:
+            field_dict["required"] = required
+        if secret is not UNSET:
+            field_dict["secret"] = secret
+        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:
+        from ..models.store_configuration_option import StoreConfigurationOption
+
+        d = src_dict.copy()
+        available_models = cast(List[str], d.pop("available_models", UNSET))
+
+        description = d.pop("description", UNSET)
+
+        display_name = d.pop("display_name", UNSET)
+
+        if_ = d.pop("if", UNSET)
+
+        name = d.pop("name", UNSET)
+
+        options = []
+        _options = d.pop("options", UNSET)
+        for options_item_data in _options or []:
+            options_item = StoreConfigurationOption.from_dict(options_item_data)
+
+            options.append(options_item)
+
+        required = d.pop("required", UNSET)
+
+        secret = d.pop("secret", UNSET)
+
+        type = d.pop("type", UNSET)
+
+        store_configuration = cls(
+            available_models=available_models,
+            description=description,
+            display_name=display_name,
+            if_=if_,
+            name=name,
+            options=options,
+            required=required,
+            secret=secret,
+            type=type,
+        )
+
+        store_configuration.additional_properties = d
+        return store_configuration
+
+    @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