beamlit 0.0.19__py3-none-any.whl → 0.0.20__py3-none-any.whl

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

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -73,7 +73,9 @@ class ServerlessConfig:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -49,7 +49,9 @@ class StandardFieldsDynamoDb:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -22,7 +22,7 @@ class StoreAgent:
         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
+        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
@@ -34,7 +34,7 @@ class StoreAgent:
     created_by: Union[Unset, str] = UNSET
     updated_at: Union[Unset, str] = UNSET
     updated_by: Union[Unset, str] = UNSET
-    configuration: Union[Unset, List["StoreConfiguration"]] = UNSET
+    configuration: Union[Unset, list["StoreConfiguration"]] = UNSET
     description: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
     image: Union[Unset, str] = UNSET
@@ -51,7 +51,7 @@ class StoreAgent:
 
         updated_by = self.updated_by
 
-        configuration: Union[Unset, List[Dict[str, Any]]] = UNSET
+        configuration: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.configuration, Unset):
             configuration = []
             for configuration_item_data in self.configuration:
@@ -64,7 +64,7 @@ class StoreAgent:
 
         image = self.image
 
-        labels: Union[Unset, Dict[str, Any]] = UNSET
+        labels: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.labels, Unset):
             labels = self.labels.to_dict()
 
@@ -97,10 +97,12 @@ class StoreAgent:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
         created_at = d.pop("created_at", UNSET)
 

beamlit/models/store_agent_configuration.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -55,7 +55,9 @@ class StoreAgentConfiguration:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -19,7 +19,9 @@ class StoreAgentLabels:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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()
 

beamlit/models/store_agent_labels_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="StoreAgentLabelsType0")
+
+
+@_attrs_define
+class StoreAgentLabelsType0:
+    """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_type_0 = cls()
+
+        store_agent_labels_type_0.additional_properties = d
+        return store_agent_labels_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/store_configuration.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -17,30 +17,30 @@ class StoreConfiguration:
     """Store configuration for resources (eg: agent, function, etc)
 
     Attributes:
-        available_models (Union[Unset, List[str]]): Available models for the configuration
+        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']]):
+        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
+        type_ (Union[Unset, str]): Store configuration type
     """
 
-    available_models: Union[Unset, List[str]] = UNSET
+    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
+    options: Union[Unset, list["StoreConfigurationOption"]] = UNSET
     required: Union[Unset, bool] = UNSET
     secret: Union[Unset, bool] = UNSET
-    type: 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]:
-        available_models: Union[Unset, List[str]] = UNSET
+        available_models: Union[Unset, list[str]] = UNSET
         if not isinstance(self.available_models, Unset):
             available_models = self.available_models
 
@@ -52,7 +52,7 @@ class StoreConfiguration:
 
         name = self.name
 
-        options: Union[Unset, List[Dict[str, Any]]] = UNSET
+        options: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.options, Unset):
             options = []
             for options_item_data in self.options:
@@ -63,7 +63,7 @@ class StoreConfiguration:
 
         secret = self.secret
 
-        type = self.type
+        type_ = self.type_
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -84,17 +84,19 @@ class StoreConfiguration:
             field_dict["required"] = required
         if secret is not UNSET:
             field_dict["secret"] = secret
-        if type is not UNSET:
-            field_dict["type"] = type
+        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:
+    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))
+        available_models = cast(list[str], d.pop("available_models", UNSET))
 
         description = d.pop("description", UNSET)
 
@@ -115,7 +117,7 @@ class StoreConfiguration:
 
         secret = d.pop("secret", UNSET)
 
-        type = d.pop("type", UNSET)
+        type_ = d.pop("type", UNSET)
 
         store_configuration = cls(
             available_models=available_models,
@@ -126,7 +128,7 @@ class StoreConfiguration:
             options=options,
             required=required,
             secret=secret,
-            type=type,
+            type_=type_,
         )
 
         store_configuration.additional_properties = d

beamlit/models/store_configuration_option.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -43,7 +43,9 @@ class StoreConfigurationOption:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
         d = src_dict.copy()
         if_ = d.pop("if", UNSET)
 

beamlit/models/store_function.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -24,28 +24,28 @@ class StoreFunction:
         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
+        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
+        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
+        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
+    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
+    kit: Union[Unset, list["StoreFunctionKit"]] = UNSET
     labels: Union[Unset, "StoreFunctionLabels"] = UNSET
     name: Union[Unset, str] = UNSET
-    parameters: Union[Unset, List["StoreFunctionParameter"]] = 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]:
@@ -57,7 +57,7 @@ class StoreFunction:
 
         updated_by = self.updated_by
 
-        configuration: Union[Unset, List[Dict[str, Any]]] = UNSET
+        configuration: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.configuration, Unset):
             configuration = []
             for configuration_item_data in self.configuration:
@@ -70,20 +70,20 @@ class StoreFunction:
 
         image = self.image
 
-        kit: Union[Unset, List[Dict[str, Any]]] = UNSET
+        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
+        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
+        parameters: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.parameters, Unset):
             parameters = []
             for parameters_item_data in self.parameters:
@@ -121,12 +121,14 @@ class StoreFunction:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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
 
+        if not src_dict:
+            return None
         d = src_dict.copy()
         created_at = d.pop("created_at", UNSET)
 

beamlit/models/store_function_configuration.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -55,7 +55,9 @@ class StoreFunctionConfiguration:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -20,13 +20,13 @@ class StoreFunctionKit:
         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
+        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
+    parameters: Union[Unset, list["StoreFunctionParameter"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -34,7 +34,7 @@ class StoreFunctionKit:
 
         name = self.name
 
-        parameters: Union[Unset, List[Dict[str, Any]]] = UNSET
+        parameters: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.parameters, Unset):
             parameters = []
             for parameters_item_data in self.parameters:
@@ -54,9 +54,11 @@ class StoreFunctionKit:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -19,7 +19,9 @@ class StoreFunctionLabels:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    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()
 

beamlit/models/store_function_labels_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="StoreFunctionLabelsType0")
+
+
+@_attrs_define
+class StoreFunctionLabelsType0:
+    """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_type_0 = cls()
+
+        store_function_labels_type_0.additional_properties = d
+        return store_function_labels_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