beamlit 0.0.20rc2__py3-none-any.whl → 0.0.20rc4__py3-none-any.whl

beamlit/models/runtime_type_0_readiness_probe.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="RuntimeType0ReadinessProbe")
+
+
+@_attrs_define
+class RuntimeType0ReadinessProbe:
+    """The readiness probe. Should be a Kubernetes Probe type"""
+
+    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()
+        runtime_type_0_readiness_probe = cls()
+
+        runtime_type_0_readiness_probe.additional_properties = d
+        return runtime_type_0_readiness_probe
+
+    @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/runtime_type_0_readiness_probe_type_0.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="RuntimeType0ReadinessProbeType0")
+
+
+@_attrs_define
+class RuntimeType0ReadinessProbeType0:
+    """The readiness probe. Should be a Kubernetes Probe type"""
+
+    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()
+        runtime_type_0_readiness_probe_type_0 = cls()
+
+        runtime_type_0_readiness_probe_type_0.additional_properties = d
+        return runtime_type_0_readiness_probe_type_0
+
+    @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/runtime_type_0_resources.py

@@ -0,0 +1,59 @@
+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="RuntimeType0Resources")
+
+
+@_attrs_define
+class RuntimeType0Resources:
+    """The resources for the deployment. Should be a Kubernetes ResourceRequirements type
+
+    Attributes:
+        requests (Union[Unset, str]):
+    """
+
+    requests: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        requests = self.requests
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if requests is not UNSET:
+            field_dict["requests"] = requests
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        requests = d.pop("requests", UNSET)
+
+        runtime_type_0_resources = cls(
+            requests=requests,
+        )
+
+        runtime_type_0_resources.additional_properties = d
+        return runtime_type_0_resources
+
+    @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/serverless_config.py

@@ -74,6 +74,8 @@ class ServerlessConfig:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         last_pod_retention_period = d.pop("last_pod_retention_period", UNSET)
 

beamlit/models/spec_configuration.py

@@ -0,0 +1,68 @@
+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="SpecConfiguration")
+
+
+@_attrs_define
+class SpecConfiguration:
+    """Agent configuration, this is a key value storage. In your agent you can retrieve the value with config[key]
+
+    Attributes:
+        secret (Union[Unset, bool]): Agent configuration secret
+        value (Union[Unset, str]): Agent configuration value
+    """
+
+    secret: Union[Unset, bool] = UNSET
+    value: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        secret = self.secret
+
+        value = self.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if secret is not UNSET:
+            field_dict["secret"] = secret
+        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()
+        secret = d.pop("secret", UNSET)
+
+        value = d.pop("value", UNSET)
+
+        spec_configuration = cls(
+            secret=secret,
+            value=value,
+        )
+
+        spec_configuration.additional_properties = d
+        return spec_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/standard_fields_dynamo_db.py

@@ -50,6 +50,8 @@ class StandardFieldsDynamoDb:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         created_at = d.pop("created_at", UNSET)
 

beamlit/models/store_agent.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
+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
@@ -6,7 +6,7 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.store_agent_labels_type_0 import StoreAgentLabelsType0
+    from ..models.store_agent_labels import StoreAgentLabels
     from ..models.store_configuration import StoreConfiguration
 
 
@@ -26,7 +26,7 @@ class StoreAgent:
         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['StoreAgentLabelsType0', None, Unset]): Store agent labels
+        labels (Union[Unset, StoreAgentLabels]): Store agent labels
         name (Union[Unset, str]): Store agent name
     """
 
@@ -38,13 +38,11 @@ class StoreAgent:
     description: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
     image: Union[Unset, str] = UNSET
-    labels: Union["StoreAgentLabelsType0", None, Unset] = 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]:
-        from ..models.store_agent_labels_type_0 import StoreAgentLabelsType0
-
         created_at = self.created_at
 
         created_by = self.created_by
@@ -66,13 +64,9 @@ class StoreAgent:
 
         image = self.image
 
-        labels: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.labels, Unset):
-            labels = UNSET
-        elif isinstance(self.labels, StoreAgentLabelsType0):
+        labels: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.labels, Unset):
             labels = self.labels.to_dict()
-        else:
-            labels = self.labels
 
         name = self.name
 
@@ -104,9 +98,11 @@ class StoreAgent:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.store_agent_labels_type_0 import StoreAgentLabelsType0
+        from ..models.store_agent_labels import StoreAgentLabels
         from ..models.store_configuration import StoreConfiguration
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
         created_at = d.pop("created_at", UNSET)
 
@@ -129,22 +125,12 @@ class StoreAgent:
 
         image = d.pop("image", UNSET)
 
-        def _parse_labels(data: object) -> Union["StoreAgentLabelsType0", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                labels_type_0 = StoreAgentLabelsType0.from_dict(data)
-
-                return labels_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["StoreAgentLabelsType0", None, Unset], data)
-
-        labels = _parse_labels(d.pop("labels", 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)
 

beamlit/models/store_agent_configuration.py

@@ -56,6 +56,8 @@ class StoreAgentConfiguration:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         description = d.pop("description", UNSET)
 

beamlit/models/store_agent_labels.py

@@ -0,0 +1,45 @@
+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:
+        if not src_dict:
+            return None
+        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

@@ -20,7 +20,7 @@ class StoreConfiguration:
         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[None, Unset, str]): Conditional rendering for the configuration, example: provider === 'openai'
+        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
@@ -31,7 +31,7 @@ class StoreConfiguration:
     available_models: Union[Unset, List[str]] = UNSET
     description: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
-    if_: Union[None, Unset, str] = UNSET
+    if_: Union[Unset, str] = UNSET
     name: Union[Unset, str] = UNSET
     options: Union[Unset, List["StoreConfigurationOption"]] = UNSET
     required: Union[Unset, bool] = UNSET
@@ -48,11 +48,7 @@ class StoreConfiguration:
 
         display_name = self.display_name
 
-        if_: Union[None, Unset, str]
-        if isinstance(self.if_, Unset):
-            if_ = UNSET
-        else:
-            if_ = self.if_
+        if_ = self.if_
 
         name = self.name
 
@@ -97,6 +93,8 @@ class StoreConfiguration:
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         from ..models.store_configuration_option import StoreConfigurationOption
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
         available_models = cast(List[str], d.pop("available_models", UNSET))
 
@@ -104,14 +102,7 @@ class StoreConfiguration:
 
         display_name = d.pop("display_name", UNSET)
 
-        def _parse_if_(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        if_ = _parse_if_(d.pop("if", UNSET))
+        if_ = d.pop("if", UNSET)
 
         name = d.pop("name", UNSET)
 

beamlit/models/store_configuration_option.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union, cast
+from typing import Any, Type, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -13,23 +13,18 @@ class StoreConfigurationOption:
     """Store configuration options for a select type configuration
 
     Attributes:
-        if_ (Union[None, Unset, str]): Conditional rendering for the configuration option, example: provider ===
-            'openai'
+        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[None, Unset, str] = UNSET
+    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_: Union[None, Unset, str]
-        if isinstance(self.if_, Unset):
-            if_ = UNSET
-        else:
-            if_ = self.if_
+        if_ = self.if_
 
         label = self.label
 
@@ -49,16 +44,10 @@ class StoreConfigurationOption:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
-
-        def _parse_if_(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        if_ = _parse_if_(d.pop("if", UNSET))
+        if_ = d.pop("if", UNSET)
 
         label = d.pop("label", UNSET)
 

beamlit/models/store_function.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
+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
@@ -8,7 +8,7 @@ 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_type_0 import StoreFunctionLabelsType0
+    from ..models.store_function_labels import StoreFunctionLabels
     from ..models.store_function_parameter import StoreFunctionParameter
 
 
@@ -29,7 +29,7 @@ class StoreFunction:
         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['StoreFunctionLabelsType0', None, Unset]): Store function labels
+        labels (Union[Unset, StoreFunctionLabels]): Store function labels
         name (Union[Unset, str]): Store function name
         parameters (Union[Unset, List['StoreFunctionParameter']]): Store function parameters
     """
@@ -43,14 +43,12 @@ class StoreFunction:
     display_name: Union[Unset, str] = UNSET
     image: Union[Unset, str] = UNSET
     kit: Union[Unset, List["StoreFunctionKit"]] = UNSET
-    labels: Union["StoreFunctionLabelsType0", None, Unset] = 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]:
-        from ..models.store_function_labels_type_0 import StoreFunctionLabelsType0
-
         created_at = self.created_at
 
         created_by = self.created_by
@@ -79,13 +77,9 @@ class StoreFunction:
                 kit_item = kit_item_data.to_dict()
                 kit.append(kit_item)
 
-        labels: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.labels, Unset):
-            labels = UNSET
-        elif isinstance(self.labels, StoreFunctionLabelsType0):
+        labels: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.labels, Unset):
             labels = self.labels.to_dict()
-        else:
-            labels = self.labels
 
         name = self.name
 
@@ -130,9 +124,11 @@ class StoreFunction:
     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_type_0 import StoreFunctionLabelsType0
+        from ..models.store_function_labels import StoreFunctionLabels
         from ..models.store_function_parameter import StoreFunctionParameter
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
         created_at = d.pop("created_at", UNSET)
 
@@ -162,22 +158,12 @@ class StoreFunction:
 
             kit.append(kit_item)
 
-        def _parse_labels(data: object) -> Union["StoreFunctionLabelsType0", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                labels_type_0 = StoreFunctionLabelsType0.from_dict(data)
-
-                return labels_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["StoreFunctionLabelsType0", None, Unset], data)
-
-        labels = _parse_labels(d.pop("labels", UNSET))
+        _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)
 

beamlit/models/store_function_configuration.py

@@ -56,6 +56,8 @@ class StoreFunctionConfiguration:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         description = d.pop("description", UNSET)
 

beamlit/models/store_function_kit.py

@@ -57,6 +57,8 @@ class StoreFunctionKit:
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         from ..models.store_function_parameter import StoreFunctionParameter
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
         description = d.pop("description", UNSET)
 

beamlit/models/store_function_labels.py

@@ -0,0 +1,45 @@
+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:
+        if not src_dict:
+            return None
+        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

@@ -50,6 +50,8 @@ class StoreFunctionParameter:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         description = d.pop("description", UNSET)