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

beamlit/models/runtime.py

@@ -6,7 +6,7 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.runtime_readiness_probe import RuntimeReadinessProbe
+    from ..models.runtime_readiness_probe_type_0 import RuntimeReadinessProbeType0
     from ..models.runtime_resources import RuntimeResources
 
 
@@ -24,7 +24,8 @@ class Runtime:
             Kubernetes EnvVar types
         image (Union[Unset, str]): The Docker image for the deployment
         model (Union[Unset, str]): The slug name of the origin model. Only used if the deployment is a ModelDeployment
-        readiness_probe (Union[Unset, RuntimeReadinessProbe]): The readiness probe. Should be a Kubernetes Probe type
+        readiness_probe (Union['RuntimeReadinessProbeType0', None, Unset]): The readiness probe. Should be a Kubernetes
+            Probe type
         resources (Union[Unset, RuntimeResources]): The resources for the deployment. Should be a Kubernetes
             ResourceRequirements type
         type (Union[Unset, str]): The type of origin for the deployment
@@ -35,12 +36,14 @@ class Runtime:
     envs: Union[Unset, List[Any]] = UNSET
     image: Union[Unset, str] = UNSET
     model: Union[Unset, str] = UNSET
-    readiness_probe: Union[Unset, "RuntimeReadinessProbe"] = UNSET
+    readiness_probe: Union["RuntimeReadinessProbeType0", None, Unset] = UNSET
     resources: Union[Unset, "RuntimeResources"] = UNSET
     type: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        from ..models.runtime_readiness_probe_type_0 import RuntimeReadinessProbeType0
+
         args: Union[Unset, List[Any]] = UNSET
         if not isinstance(self.args, Unset):
             args = self.args
@@ -57,9 +60,13 @@ class Runtime:
 
         model = self.model
 
-        readiness_probe: Union[Unset, Dict[str, Any]] = UNSET
-        if not isinstance(self.readiness_probe, Unset):
+        readiness_probe: Union[Dict[str, Any], None, Unset]
+        if isinstance(self.readiness_probe, Unset):
+            readiness_probe = UNSET
+        elif isinstance(self.readiness_probe, RuntimeReadinessProbeType0):
             readiness_probe = self.readiness_probe.to_dict()
+        else:
+            readiness_probe = self.readiness_probe
 
         resources: Union[Unset, Dict[str, Any]] = UNSET
         if not isinstance(self.resources, Unset):
@@ -91,7 +98,7 @@ class Runtime:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.runtime_readiness_probe import RuntimeReadinessProbe
+        from ..models.runtime_readiness_probe_type_0 import RuntimeReadinessProbeType0
         from ..models.runtime_resources import RuntimeResources
 
         d = src_dict.copy()
@@ -105,12 +112,22 @@ class Runtime:
 
         model = d.pop("model", UNSET)
 
-        _readiness_probe = d.pop("readiness_probe", UNSET)
-        readiness_probe: Union[Unset, RuntimeReadinessProbe]
-        if isinstance(_readiness_probe, Unset):
-            readiness_probe = UNSET
-        else:
-            readiness_probe = RuntimeReadinessProbe.from_dict(_readiness_probe)
+        def _parse_readiness_probe(data: object) -> Union["RuntimeReadinessProbeType0", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                readiness_probe_type_0 = RuntimeReadinessProbeType0.from_dict(data)
+
+                return readiness_probe_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["RuntimeReadinessProbeType0", None, Unset], data)
+
+        readiness_probe = _parse_readiness_probe(d.pop("readiness_probe", UNSET))
 
         _resources = d.pop("resources", UNSET)
         resources: Union[Unset, RuntimeResources]

beamlit/models/{runtime_readiness_probe.py → runtime_readiness_probe_type_0.py}

@@ -3,11 +3,11 @@ from typing import Any, Type, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="RuntimeReadinessProbe")
+T = TypeVar("T", bound="RuntimeReadinessProbeType0")
 
 
 @_attrs_define
-class RuntimeReadinessProbe:
+class RuntimeReadinessProbeType0:
     """The readiness probe. Should be a Kubernetes Probe type"""
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -21,10 +21,10 @@ class RuntimeReadinessProbe:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        runtime_readiness_probe = cls()
+        runtime_readiness_probe_type_0 = cls()
 
-        runtime_readiness_probe.additional_properties = d
-        return runtime_readiness_probe
+        runtime_readiness_probe_type_0.additional_properties = d
+        return runtime_readiness_probe_type_0
 
     @property
     def additional_keys(self) -> list[str]:

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, Dict, List, Type, TypeVar, Union, cast
 
 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 import StoreAgentLabels
+    from ..models.store_agent_labels_type_0 import StoreAgentLabelsType0
     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[Unset, StoreAgentLabels]): Store agent labels
+        labels (Union['StoreAgentLabelsType0', None, Unset]): Store agent labels
         name (Union[Unset, str]): Store agent name
     """
 
@@ -38,11 +38,13 @@ class StoreAgent:
     description: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
     image: Union[Unset, str] = UNSET
-    labels: Union[Unset, "StoreAgentLabels"] = UNSET
+    labels: Union["StoreAgentLabelsType0", None, Unset] = 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
@@ -64,9 +66,13 @@ class StoreAgent:
 
         image = self.image
 
-        labels: Union[Unset, Dict[str, Any]] = UNSET
-        if not isinstance(self.labels, Unset):
+        labels: Union[Dict[str, Any], None, Unset]
+        if isinstance(self.labels, Unset):
+            labels = UNSET
+        elif isinstance(self.labels, StoreAgentLabelsType0):
             labels = self.labels.to_dict()
+        else:
+            labels = self.labels
 
         name = self.name
 
@@ -98,7 +104,7 @@ class StoreAgent:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.store_agent_labels import StoreAgentLabels
+        from ..models.store_agent_labels_type_0 import StoreAgentLabelsType0
         from ..models.store_configuration import StoreConfiguration
 
         d = src_dict.copy()
@@ -123,12 +129,22 @@ class StoreAgent:
 
         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)
+        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))
 
         name = d.pop("name", UNSET)
 

beamlit/models/{store_agent_labels.py → store_agent_labels_type_0.py}

@@ -3,11 +3,11 @@ 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")
+T = TypeVar("T", bound="StoreAgentLabelsType0")
 
 
 @_attrs_define
-class StoreAgentLabels:
+class StoreAgentLabelsType0:
     """Store agent labels"""
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -21,10 +21,10 @@ class StoreAgentLabels:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        store_agent_labels = cls()
+        store_agent_labels_type_0 = cls()
 
-        store_agent_labels.additional_properties = d
-        return store_agent_labels
+        store_agent_labels_type_0.additional_properties = d
+        return store_agent_labels_type_0
 
     @property
     def additional_keys(self) -> list[str]:

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[Unset, str]): Conditional rendering for the configuration, example: provider === 'openai'
+        if_ (Union[None, 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[Unset, str] = UNSET
+    if_: Union[None, Unset, str] = UNSET
     name: Union[Unset, str] = UNSET
     options: Union[Unset, List["StoreConfigurationOption"]] = UNSET
     required: Union[Unset, bool] = UNSET
@@ -48,7 +48,11 @@ class StoreConfiguration:
 
         display_name = self.display_name
 
-        if_ = self.if_
+        if_: Union[None, Unset, str]
+        if isinstance(self.if_, Unset):
+            if_ = UNSET
+        else:
+            if_ = self.if_
 
         name = self.name
 
@@ -100,7 +104,14 @@ class StoreConfiguration:
 
         display_name = d.pop("display_name", UNSET)
 
-        if_ = d.pop("if", 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))
 
         name = d.pop("name", UNSET)
 

beamlit/models/store_configuration_option.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, Type, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -13,18 +13,23 @@ class StoreConfigurationOption:
     """Store configuration options for a select type configuration
 
     Attributes:
-        if_ (Union[Unset, str]): Conditional rendering for the configuration option, example: provider === 'openai'
+        if_ (Union[None, 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
+    if_: Union[None, 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_
+        if_: Union[None, Unset, str]
+        if isinstance(self.if_, Unset):
+            if_ = UNSET
+        else:
+            if_ = self.if_
 
         label = self.label
 
@@ -45,7 +50,15 @@ class StoreConfigurationOption:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        if_ = d.pop("if", 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))
 
         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
+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
@@ -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 import StoreFunctionLabels
+    from ..models.store_function_labels_type_0 import StoreFunctionLabelsType0
     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[Unset, StoreFunctionLabels]): Store function labels
+        labels (Union['StoreFunctionLabelsType0', None, Unset]): Store function labels
         name (Union[Unset, str]): Store function name
         parameters (Union[Unset, List['StoreFunctionParameter']]): Store function parameters
     """
@@ -43,12 +43,14 @@ class StoreFunction:
     display_name: Union[Unset, str] = UNSET
     image: Union[Unset, str] = UNSET
     kit: Union[Unset, List["StoreFunctionKit"]] = UNSET
-    labels: Union[Unset, "StoreFunctionLabels"] = UNSET
+    labels: Union["StoreFunctionLabelsType0", None, Unset] = 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
@@ -77,9 +79,13 @@ class StoreFunction:
                 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: Union[Dict[str, Any], None, Unset]
+        if isinstance(self.labels, Unset):
+            labels = UNSET
+        elif isinstance(self.labels, StoreFunctionLabelsType0):
             labels = self.labels.to_dict()
+        else:
+            labels = self.labels
 
         name = self.name
 
@@ -124,7 +130,7 @@ 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 import StoreFunctionLabels
+        from ..models.store_function_labels_type_0 import StoreFunctionLabelsType0
         from ..models.store_function_parameter import StoreFunctionParameter
 
         d = src_dict.copy()
@@ -156,12 +162,22 @@ class StoreFunction:
 
             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)
+        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))
 
         name = d.pop("name", UNSET)
 

beamlit/models/{store_function_labels.py → store_function_labels_type_0.py}

@@ -3,11 +3,11 @@ 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")
+T = TypeVar("T", bound="StoreFunctionLabelsType0")
 
 
 @_attrs_define
-class StoreFunctionLabels:
+class StoreFunctionLabelsType0:
     """Store function labels"""
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -21,10 +21,10 @@ class StoreFunctionLabels:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        store_function_labels = cls()
+        store_function_labels_type_0 = cls()
 
-        store_function_labels.additional_properties = d
-        return store_function_labels
+        store_function_labels_type_0.additional_properties = d
+        return store_function_labels_type_0
 
     @property
     def additional_keys(self) -> list[str]:

beamlit/run.py

@@ -0,0 +1,49 @@
+import urllib.parse
+from contextlib import contextmanager
+from typing import Any, Callable, Dict, List, Optional
+
+import requests
+from beamlit.client import AuthenticatedClient
+from beamlit.common.settings import get_settings
+
+
+class RunClient:
+    def __init__(self, client: AuthenticatedClient):
+        self.client = client
+
+    def run(
+        self,
+        resource_type: str,
+        resource_name: str,
+        environment: str,
+        method: str,
+        path: str = "",
+        headers: Optional[Dict[str, str]] = None,
+        json: Optional[Dict[str, Any]] = None,
+        data: Optional[str] = None,
+        params: Optional[Dict[str, str]] = None,
+    ) -> requests.Response:
+        settings = get_settings()
+        headers = headers or {}
+        params = params or {}
+
+        # Build the path
+        if path:
+            path = f"{settings.workspace}/{resource_type}s/{resource_name}/{path}"
+        else:
+            path = f"{settings.workspace}/{resource_type}s/{resource_name}"
+
+        client = self.client.get_httpx_client()
+        url = urllib.parse.urljoin(settings.run_url, path)
+
+        kwargs = {
+            "headers": headers,
+            "params": {"environment": environment, **params},
+        }
+        if data:
+            kwargs["data"] = data
+        if json:
+            kwargs["json"] = json
+
+        response = client.request(method, url, **kwargs)
+        return response

{beamlit-0.0.19.dist-info → beamlit-0.0.20rc2.dist-info}/METADATA

@@ -1,13 +1,16 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.19
+Version: 0.0.20rc2
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
 Requires-Dist: attrs>=21.3.0
 Requires-Dist: httpx<0.28.0,>=0.20.0
+Requires-Dist: pydantic-settings<2.7.0,>=2.6.1
+Requires-Dist: pydantic<2.11.0,>=2.10.3
 Requires-Dist: python-dateutil>=2.8.0
-Requires-Dist: pyyaml==6.0.2
+Requires-Dist: pyyaml<6.1.0,>=6.0.2
+Requires-Dist: requests<2.33.0,>=2.32.3
 Description-Content-Type: text/markdown
 
 # beamlit