beamlit 0.1.0__py3-none-any.whl

beamlit/models/store_configuration_option.py

@@ -0,0 +1,77 @@
+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="StoreConfigurationOption")
+
+
+@_attrs_define
+class StoreConfigurationOption:
+    """Store configuration options for a select type configuration
+
+    Attributes:
+        if_ (Union[Unset, str]): Conditional rendering for the configuration option, example: provider === 'openai'
+        label (Union[Unset, str]): Store configuration option label
+        value (Union[Unset, str]): Store configuration option value
+    """
+
+    if_: Union[Unset, str] = UNSET
+    label: 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]:
+        if_ = self.if_
+
+        label = self.label
+
+        value = self.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if if_ is not UNSET:
+            field_dict["if"] = if_
+        if label is not UNSET:
+            field_dict["label"] = label
+        if value is not UNSET:
+            field_dict["value"] = value
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        if_ = d.pop("if", UNSET)
+
+        label = d.pop("label", UNSET)
+
+        value = d.pop("value", UNSET)
+
+        store_configuration_option = cls(
+            if_=if_,
+            label=label,
+            value=value,
+        )
+
+        store_configuration_option.additional_properties = d
+        return store_configuration_option
+
+    @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_function.py

@@ -0,0 +1,207 @@
+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_configuration import StoreConfiguration
+    from ..models.store_function_kit import StoreFunctionKit
+    from ..models.store_function_labels import StoreFunctionLabels
+    from ..models.store_function_parameter import StoreFunctionParameter
+
+
+T = TypeVar("T", bound="StoreFunction")
+
+
+@_attrs_define
+class StoreFunction:
+    """Store function
+
+    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 function configuration
+        description (Union[Unset, str]): Store function description
+        display_name (Union[Unset, str]): Store function display name
+        image (Union[Unset, str]): Store function image
+        kit (Union[Unset, List['StoreFunctionKit']]): Store function kit
+        labels (Union[Unset, StoreFunctionLabels]): Store function labels
+        name (Union[Unset, str]): Store function name
+        parameters (Union[Unset, List['StoreFunctionParameter']]): Store function parameters
+    """
+
+    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
+    kit: Union[Unset, List["StoreFunctionKit"]] = UNSET
+    labels: Union[Unset, "StoreFunctionLabels"] = UNSET
+    name: Union[Unset, str] = UNSET
+    parameters: Union[Unset, List["StoreFunctionParameter"]] = 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
+
+        kit: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.kit, Unset):
+            kit = []
+            for kit_item_data in self.kit:
+                kit_item = kit_item_data.to_dict()
+                kit.append(kit_item)
+
+        labels: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.labels, Unset):
+            labels = self.labels.to_dict()
+
+        name = self.name
+
+        parameters: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.parameters, Unset):
+            parameters = []
+            for parameters_item_data in self.parameters:
+                parameters_item = parameters_item_data.to_dict()
+                parameters.append(parameters_item)
+
+        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 kit is not UNSET:
+            field_dict["kit"] = kit
+        if labels is not UNSET:
+            field_dict["labels"] = labels
+        if name is not UNSET:
+            field_dict["name"] = name
+        if parameters is not UNSET:
+            field_dict["parameters"] = parameters
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.store_configuration import StoreConfiguration
+        from ..models.store_function_kit import StoreFunctionKit
+        from ..models.store_function_labels import StoreFunctionLabels
+        from ..models.store_function_parameter import StoreFunctionParameter
+
+        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)
+
+        kit = []
+        _kit = d.pop("kit", UNSET)
+        for kit_item_data in _kit or []:
+            kit_item = StoreFunctionKit.from_dict(kit_item_data)
+
+            kit.append(kit_item)
+
+        _labels = d.pop("labels", UNSET)
+        labels: Union[Unset, StoreFunctionLabels]
+        if isinstance(_labels, Unset):
+            labels = UNSET
+        else:
+            labels = StoreFunctionLabels.from_dict(_labels)
+
+        name = d.pop("name", UNSET)
+
+        parameters = []
+        _parameters = d.pop("parameters", UNSET)
+        for parameters_item_data in _parameters or []:
+            parameters_item = StoreFunctionParameter.from_dict(parameters_item_data)
+
+            parameters.append(parameters_item)
+
+        store_function = 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,
+            kit=kit,
+            labels=labels,
+            name=name,
+            parameters=parameters,
+        )
+
+        store_function.additional_properties = d
+        return store_function
+
+    @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_function_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="StoreFunctionConfiguration")
+
+
+@_attrs_define
+class StoreFunctionConfiguration:
+    """Store function configuration used to configure your function 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_function_configuration = cls(
+            description=description,
+            if_=if_,
+            name=name,
+            required=required,
+            secret=secret,
+        )
+
+        store_function_configuration.additional_properties = d
+        return store_function_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_function_kit.py

@@ -0,0 +1,95 @@
+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_function_parameter import StoreFunctionParameter
+
+
+T = TypeVar("T", bound="StoreFunctionKit")
+
+
+@_attrs_define
+class StoreFunctionKit:
+    """Store function kit
+
+    Attributes:
+        description (Union[Unset, str]): Description of the function kit, very important for the agent to work with your
+            kit
+        name (Union[Unset, str]): The kit name, very important for the agent to work with your kit
+        parameters (Union[Unset, List['StoreFunctionParameter']]): Kit parameters, for your kit to be callable with an
+            Agent
+    """
+
+    description: Union[Unset, str] = UNSET
+    name: Union[Unset, str] = UNSET
+    parameters: Union[Unset, List["StoreFunctionParameter"]] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        description = self.description
+
+        name = self.name
+
+        parameters: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.parameters, Unset):
+            parameters = []
+            for parameters_item_data in self.parameters:
+                parameters_item = parameters_item_data.to_dict()
+                parameters.append(parameters_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if description is not UNSET:
+            field_dict["description"] = description
+        if name is not UNSET:
+            field_dict["name"] = name
+        if parameters is not UNSET:
+            field_dict["parameters"] = parameters
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.store_function_parameter import StoreFunctionParameter
+
+        d = src_dict.copy()
+        description = d.pop("description", UNSET)
+
+        name = d.pop("name", UNSET)
+
+        parameters = []
+        _parameters = d.pop("parameters", UNSET)
+        for parameters_item_data in _parameters or []:
+            parameters_item = StoreFunctionParameter.from_dict(parameters_item_data)
+
+            parameters.append(parameters_item)
+
+        store_function_kit = cls(
+            description=description,
+            name=name,
+            parameters=parameters,
+        )
+
+        store_function_kit.additional_properties = d
+        return store_function_kit
+
+    @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_function_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="StoreFunctionLabels")
+
+
+@_attrs_define
+class StoreFunctionLabels:
+    """Store function 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_function_labels = cls()
+
+        store_function_labels.additional_properties = d
+        return store_function_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_function_parameter.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="StoreFunctionParameter")
+
+
+@_attrs_define
+class StoreFunctionParameter:
+    """Store function parameter
+
+    Attributes:
+        description (Union[Unset, str]): Store function parameter description
+        name (Union[Unset, str]): Store function parameter name
+        required (Union[Unset, bool]): Store function parameter required
+        type (Union[Unset, str]): Store function parameter type
+    """
+
+    description: Union[Unset, str] = UNSET
+    name: Union[Unset, str] = UNSET
+    required: 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]:
+        description = self.description
+
+        name = self.name
+
+        required = self.required
+
+        type = self.type
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if description is not UNSET:
+            field_dict["description"] = description
+        if name is not UNSET:
+            field_dict["name"] = name
+        if required is not UNSET:
+            field_dict["required"] = required
+        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:
+        d = src_dict.copy()
+        description = d.pop("description", UNSET)
+
+        name = d.pop("name", UNSET)
+
+        required = d.pop("required", UNSET)
+
+        type = d.pop("type", UNSET)
+
+        store_function_parameter = cls(
+            description=description,
+            name=name,
+            required=required,
+            type=type,
+        )
+
+        store_function_parameter.additional_properties = d
+        return store_function_parameter
+
+    @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/update_workspace_service_account_body.py

@@ -0,0 +1,67 @@
+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="UpdateWorkspaceServiceAccountBody")
+
+
+@_attrs_define
+class UpdateWorkspaceServiceAccountBody:
+    """
+    Attributes:
+        description (Union[Unset, str]): Service account description
+        name (Union[Unset, str]): Service account name
+    """
+
+    description: Union[Unset, str] = UNSET
+    name: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        description = self.description
+
+        name = self.name
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if description is not UNSET:
+            field_dict["description"] = description
+        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:
+        d = src_dict.copy()
+        description = d.pop("description", UNSET)
+
+        name = d.pop("name", UNSET)
+
+        update_workspace_service_account_body = cls(
+            description=description,
+            name=name,
+        )
+
+        update_workspace_service_account_body.additional_properties = d
+        return update_workspace_service_account_body
+
+    @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