beamlit 0.0.24rc20__py3-none-any.whl → 0.0.24rc21__py3-none-any.whl

beamlit/models/resource_log.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
@@ -37,7 +37,9 @@ class ResourceLog:
         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()
         message = d.pop("message", UNSET)
 

beamlit/models/runtime.py

@@ -23,10 +23,12 @@ class Runtime:
         envs (Union[Unset, list[Any]]): The environment variables to set in the deployment. Should be a list of
             Kubernetes EnvVar types
         image (Union[Unset, str]): The Docker image for the deployment
+        metric_port (Union[Unset, int]): The port to serve the metrics on
         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
         resources (Union[Unset, RuntimeResources]): The resources for the deployment. Should be a Kubernetes
             ResourceRequirements type
+        serving_port (Union[Unset, int]): The port to serve the model on
         type_ (Union[Unset, str]): The type of origin for the deployment
     """
 
@@ -34,9 +36,11 @@ class Runtime:
     command: Union[Unset, list[Any]] = UNSET
     envs: Union[Unset, list[Any]] = UNSET
     image: Union[Unset, str] = UNSET
+    metric_port: Union[Unset, int] = UNSET
     model: Union[Unset, str] = UNSET
     readiness_probe: Union[Unset, "RuntimeReadinessProbe"] = UNSET
     resources: Union[Unset, "RuntimeResources"] = UNSET
+    serving_port: Union[Unset, int] = UNSET
     type_: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -55,6 +59,8 @@ class Runtime:
 
         image = self.image
 
+        metric_port = self.metric_port
+
         model = self.model
 
         readiness_probe: Union[Unset, dict[str, Any]] = UNSET
@@ -65,6 +71,8 @@ class Runtime:
         if not isinstance(self.resources, Unset):
             resources = self.resources.to_dict()
 
+        serving_port = self.serving_port
+
         type_ = self.type_
 
         field_dict: dict[str, Any] = {}
@@ -78,12 +86,16 @@ class Runtime:
             field_dict["envs"] = envs
         if image is not UNSET:
             field_dict["image"] = image
+        if metric_port is not UNSET:
+            field_dict["metricPort"] = metric_port
         if model is not UNSET:
             field_dict["model"] = model
         if readiness_probe is not UNSET:
-            field_dict["readiness_probe"] = readiness_probe
+            field_dict["readinessProbe"] = readiness_probe
         if resources is not UNSET:
             field_dict["resources"] = resources
+        if serving_port is not UNSET:
+            field_dict["servingPort"] = serving_port
         if type_ is not UNSET:
             field_dict["type"] = type_
 
@@ -105,9 +117,11 @@ class Runtime:
 
         image = d.pop("image", UNSET)
 
+        metric_port = d.pop("metricPort", UNSET)
+
         model = d.pop("model", UNSET)
 
-        _readiness_probe = d.pop("readiness_probe", UNSET)
+        _readiness_probe = d.pop("readinessProbe", UNSET)
         readiness_probe: Union[Unset, RuntimeReadinessProbe]
         if isinstance(_readiness_probe, Unset):
             readiness_probe = UNSET
@@ -121,6 +135,8 @@ class Runtime:
         else:
             resources = RuntimeResources.from_dict(_resources)
 
+        serving_port = d.pop("servingPort", UNSET)
+
         type_ = d.pop("type", UNSET)
 
         runtime = cls(
@@ -128,9 +144,11 @@ class Runtime:
             command=command,
             envs=envs,
             image=image,
+            metric_port=metric_port,
             model=model,
             readiness_probe=readiness_probe,
             resources=resources,
+            serving_port=serving_port,
             type_=type_,
         )
 

beamlit/models/serverless_config.py

@@ -10,17 +10,24 @@ T = TypeVar("T", bound="ServerlessConfig")
 
 @_attrs_define
 class ServerlessConfig:
-    """Configuration for the serverless model deployment
+    """Configuration for a serverless 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]): The minimum amount of time that the last replica will remain
+            active AFTER a scale-to-zero decision is made
+        max_num_replicas (Union[Unset, int]): The maximum number of replicas for the deployment.
+        metric (Union[Unset, str]): Metric watched to make scaling decisions. Can be "cpu" or "memory" or "rps" or
+            "concurrency"
+        min_num_replicas (Union[Unset, int]): The minimum number of replicas for the deployment. Can be 0 or 1 (in which
+            case the deployment is always running in at least one location).
+        scale_down_delay (Union[Unset, str]): The time window which must pass at reduced concurrency before a scale-down
+            decision is applied. This can be useful, for example, to keep containers around for a configurable duration to
+            avoid a cold start penalty if new requests come in.
+        scale_up_minimum (Union[Unset, int]): The minimum number of replicas that will be created when the deployment
+            scales up from zero.
+        stable_window (Union[Unset, str]): The sliding time window over which metrics are averaged to provide the input
+            for scaling decisions
+        target (Union[Unset, str]): Target value for the watched metric
     """
 
     last_pod_retention_period: Union[Unset, str] = UNSET
@@ -54,19 +61,19 @@ class ServerlessConfig:
         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
+            field_dict["lastPodRetentionPeriod"] = last_pod_retention_period
         if max_num_replicas is not UNSET:
-            field_dict["max_num_replicas"] = max_num_replicas
+            field_dict["maxNumReplicas"] = 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
+            field_dict["minNumReplicas"] = min_num_replicas
         if scale_down_delay is not UNSET:
-            field_dict["scale_down_delay"] = scale_down_delay
+            field_dict["scaleDownDelay"] = scale_down_delay
         if scale_up_minimum is not UNSET:
-            field_dict["scale_up_minimum"] = scale_up_minimum
+            field_dict["scaleUpMinimum"] = scale_up_minimum
         if stable_window is not UNSET:
-            field_dict["stable_window"] = stable_window
+            field_dict["stableWindow"] = stable_window
         if target is not UNSET:
             field_dict["target"] = target
 
@@ -77,19 +84,19 @@ class ServerlessConfig:
         if not src_dict:
             return None
         d = src_dict.copy()
-        last_pod_retention_period = d.pop("last_pod_retention_period", UNSET)
+        last_pod_retention_period = d.pop("lastPodRetentionPeriod", UNSET)
 
-        max_num_replicas = d.pop("max_num_replicas", UNSET)
+        max_num_replicas = d.pop("maxNumReplicas", UNSET)
 
         metric = d.pop("metric", UNSET)
 
-        min_num_replicas = d.pop("min_num_replicas", UNSET)
+        min_num_replicas = d.pop("minNumReplicas", UNSET)
 
-        scale_down_delay = d.pop("scale_down_delay", UNSET)
+        scale_down_delay = d.pop("scaleDownDelay", UNSET)
 
-        scale_up_minimum = d.pop("scale_up_minimum", UNSET)
+        scale_up_minimum = d.pop("scaleUpMinimum", UNSET)
 
-        stable_window = d.pop("stable_window", UNSET)
+        stable_window = d.pop("stableWindow", UNSET)
 
         target = d.pop("target", UNSET)
 

beamlit/models/spec_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
@@ -10,11 +10,11 @@ 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]
+    """Configuration, this is a key value storage. In your object 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]): ACconfiguration secret
+        value (Union[Unset, str]): Configuration value
     """
 
     secret: Union[Unset, bool] = UNSET
@@ -37,7 +37,9 @@ class SpecConfiguration:
         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()
         secret = d.pop("secret", UNSET)
 

beamlit/models/store_agent.py

@@ -19,8 +19,8 @@ class StoreAgent:
 
     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
+        created_by (Union[Unset, str]): The user or service account who created the resource
         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
@@ -31,8 +31,8 @@ class StoreAgent:
     """
 
     created_at: Union[Unset, str] = UNSET
-    created_by: Union[Unset, str] = UNSET
     updated_at: Union[Unset, str] = UNSET
+    created_by: Union[Unset, str] = UNSET
     updated_by: Union[Unset, str] = UNSET
     configuration: Union[Unset, list["StoreConfiguration"]] = UNSET
     description: Union[Unset, str] = UNSET
@@ -45,10 +45,10 @@ class StoreAgent:
     def to_dict(self) -> dict[str, Any]:
         created_at = self.created_at
 
-        created_by = self.created_by
-
         updated_at = self.updated_at
 
+        created_by = self.created_by
+
         updated_by = self.updated_by
 
         configuration: Union[Unset, list[dict[str, Any]]] = UNSET
@@ -74,13 +74,13 @@ class StoreAgent:
         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
+            field_dict["createdAt"] = created_at
         if updated_at is not UNSET:
-            field_dict["updated_at"] = updated_at
+            field_dict["updatedAt"] = updated_at
+        if created_by is not UNSET:
+            field_dict["createdBy"] = created_by
         if updated_by is not UNSET:
-            field_dict["updated_by"] = updated_by
+            field_dict["updatedBy"] = updated_by
         if configuration is not UNSET:
             field_dict["configuration"] = configuration
         if description is not UNSET:
@@ -104,13 +104,13 @@ class StoreAgent:
         if not src_dict:
             return None
         d = src_dict.copy()
-        created_at = d.pop("created_at", UNSET)
+        created_at = d.pop("createdAt", UNSET)
 
-        created_by = d.pop("created_by", UNSET)
+        updated_at = d.pop("updatedAt", UNSET)
 
-        updated_at = d.pop("updated_at", UNSET)
+        created_by = d.pop("createdBy", UNSET)
 
-        updated_by = d.pop("updated_by", UNSET)
+        updated_by = d.pop("updatedBy", UNSET)
 
         configuration = []
         _configuration = d.pop("configuration", UNSET)
@@ -136,8 +136,8 @@ class StoreAgent:
 
         store_agent = cls(
             created_at=created_at,
-            created_by=created_by,
             updated_at=updated_at,
+            created_by=created_by,
             updated_by=updated_by,
             configuration=configuration,
             description=description,

beamlit/models/store_function.py

@@ -21,8 +21,8 @@ class StoreFunction:
 
     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
+        created_by (Union[Unset, str]): The user or service account who created the resource
         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
@@ -35,8 +35,8 @@ class StoreFunction:
     """
 
     created_at: Union[Unset, str] = UNSET
-    created_by: Union[Unset, str] = UNSET
     updated_at: Union[Unset, str] = UNSET
+    created_by: Union[Unset, str] = UNSET
     updated_by: Union[Unset, str] = UNSET
     configuration: Union[Unset, list["StoreConfiguration"]] = UNSET
     description: Union[Unset, str] = UNSET
@@ -51,10 +51,10 @@ class StoreFunction:
     def to_dict(self) -> dict[str, Any]:
         created_at = self.created_at
 
-        created_by = self.created_by
-
         updated_at = self.updated_at
 
+        created_by = self.created_by
+
         updated_by = self.updated_by
 
         configuration: Union[Unset, list[dict[str, Any]]] = UNSET
@@ -94,13 +94,13 @@ class StoreFunction:
         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
+            field_dict["createdAt"] = created_at
         if updated_at is not UNSET:
-            field_dict["updated_at"] = updated_at
+            field_dict["updatedAt"] = updated_at
+        if created_by is not UNSET:
+            field_dict["createdBy"] = created_by
         if updated_by is not UNSET:
-            field_dict["updated_by"] = updated_by
+            field_dict["updatedBy"] = updated_by
         if configuration is not UNSET:
             field_dict["configuration"] = configuration
         if description is not UNSET:
@@ -130,13 +130,13 @@ class StoreFunction:
         if not src_dict:
             return None
         d = src_dict.copy()
-        created_at = d.pop("created_at", UNSET)
+        created_at = d.pop("createdAt", UNSET)
 
-        created_by = d.pop("created_by", UNSET)
+        updated_at = d.pop("updatedAt", UNSET)
 
-        updated_at = d.pop("updated_at", UNSET)
+        created_by = d.pop("createdBy", UNSET)
 
-        updated_by = d.pop("updated_by", UNSET)
+        updated_by = d.pop("updatedBy", UNSET)
 
         configuration = []
         _configuration = d.pop("configuration", UNSET)
@@ -176,8 +176,8 @@ class StoreFunction:
 
         store_function = cls(
             created_at=created_at,
-            created_by=created_by,
             updated_at=updated_at,
+            created_by=created_by,
             updated_by=updated_by,
             configuration=configuration,
             description=description,

beamlit/models/time_fields.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
@@ -37,7 +37,9 @@ class TimeFields:
         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("createdAt", UNSET)
 

beamlit/models/websocket_channel.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 WebsocketChannel:
         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("createdAt", UNSET)
 

beamlit/models/workspace.py

@@ -18,30 +18,32 @@ class Workspace:
 
     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
+        created_by (Union[Unset, str]): The user or service account who created the resource
         updated_by (Union[Unset, str]): The user or service account who updated the resource
         display_name (Union[Unset, str]): Workspace display name
         labels (Union[Unset, WorkspaceLabels]): Workspace labels
         name (Union[Unset, str]): Workspace name
+        write_region (Union[Unset, str]): Workspace write region
     """
 
     created_at: Union[Unset, str] = UNSET
-    created_by: Union[Unset, str] = UNSET
     updated_at: Union[Unset, str] = UNSET
+    created_by: Union[Unset, str] = UNSET
     updated_by: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
     labels: Union[Unset, "WorkspaceLabels"] = UNSET
     name: Union[Unset, str] = UNSET
+    write_region: 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
 
+        created_by = self.created_by
+
         updated_by = self.updated_by
 
         display_name = self.display_name
@@ -52,23 +54,27 @@ class Workspace:
 
         name = self.name
 
+        write_region = self.write_region
+
         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
+            field_dict["createdAt"] = created_at
         if updated_at is not UNSET:
-            field_dict["updated_at"] = updated_at
+            field_dict["updatedAt"] = updated_at
+        if created_by is not UNSET:
+            field_dict["createdBy"] = created_by
         if updated_by is not UNSET:
-            field_dict["updated_by"] = updated_by
+            field_dict["updatedBy"] = updated_by
         if display_name is not UNSET:
             field_dict["display_name"] = display_name
         if labels is not UNSET:
             field_dict["labels"] = labels
         if name is not UNSET:
             field_dict["name"] = name
+        if write_region is not UNSET:
+            field_dict["writeRegion"] = write_region
 
         return field_dict
 
@@ -79,13 +85,13 @@ class Workspace:
         if not src_dict:
             return None
         d = src_dict.copy()
-        created_at = d.pop("created_at", UNSET)
+        created_at = d.pop("createdAt", UNSET)
 
-        created_by = d.pop("created_by", UNSET)
+        updated_at = d.pop("updatedAt", UNSET)
 
-        updated_at = d.pop("updated_at", UNSET)
+        created_by = d.pop("createdBy", UNSET)
 
-        updated_by = d.pop("updated_by", UNSET)
+        updated_by = d.pop("updatedBy", UNSET)
 
         display_name = d.pop("display_name", UNSET)
 
@@ -98,14 +104,17 @@ class Workspace:
 
         name = d.pop("name", UNSET)
 
+        write_region = d.pop("writeRegion", UNSET)
+
         workspace = cls(
             created_at=created_at,
-            created_by=created_by,
             updated_at=updated_at,
+            created_by=created_by,
             updated_by=updated_by,
             display_name=display_name,
             labels=labels,
             name=name,
+            write_region=write_region,
         )
 
         workspace.additional_properties = d

beamlit/run.py

@@ -2,7 +2,6 @@ import urllib.parse
 from typing import Any
 
 import requests
-
 from beamlit.client import AuthenticatedClient
 from beamlit.common.settings import get_settings
 

beamlit/serve/app.py

@@ -16,6 +16,7 @@ from .middlewares import AccessLogMiddleware, AddProcessTimeHeader
 sys.path.insert(0, os.getcwd())
 sys.path.insert(0, os.path.join(os.getcwd(), "src"))
 
+
 def import_module():
     settings = get_settings()
     main_module = importlib.import_module(".".join(settings.server.module.split(".")[0:-1]))

beamlit/serve/middlewares/accesslog.py

@@ -10,5 +10,7 @@ class AccessLogMiddleware(BaseHTTPMiddleware):
         process_time = response.headers.get("X-Process-Time")
         rid_header = response.headers.get("X-Request-Id")
         request_id = rid_header or response.headers.get("X-Beamlit-Request-Id")
-        logger.info(f"{request.method} {request.url.path} {response.status_code} {process_time}ms rid={request_id}")
+        logger.info(
+            f"{request.method} {request.url.path} {response.status_code} {process_time}ms rid={request_id}"
+        )
         return response

{beamlit-0.0.24rc20.dist-info → beamlit-0.0.24rc21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.24rc20
+Version: 0.0.24rc21
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12