beamlit 0.0.56rc107__py3-none-any.whl → 0.0.57__py3-none-any.whl

beamlit/models/model.py

@@ -7,7 +7,7 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.core_event import CoreEvent
-    from ..models.environment_metadata import EnvironmentMetadata
+    from ..models.metadata import Metadata
     from ..models.model_spec import ModelSpec
 
 
@@ -16,17 +16,17 @@ T = TypeVar("T", bound="Model")
 
 @_attrs_define
 class Model:
-    """Logical object representing a model, that can be instantiated in multiple environments as model deployments
+    """Logical object representing a model
 
     Attributes:
         events (Union[Unset, list['CoreEvent']]): Core events
-        metadata (Union[Unset, EnvironmentMetadata]): Environment metadata
+        metadata (Union[Unset, Metadata]): Metadata
         spec (Union[Unset, ModelSpec]): Model specification
         status (Union[Unset, str]): Model status
     """
 
     events: Union[Unset, list["CoreEvent"]] = UNSET
-    metadata: Union[Unset, "EnvironmentMetadata"] = UNSET
+    metadata: Union[Unset, "Metadata"] = UNSET
     spec: Union[Unset, "ModelSpec"] = UNSET
     status: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -70,7 +70,7 @@ class Model:
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.core_event import CoreEvent
-        from ..models.environment_metadata import EnvironmentMetadata
+        from ..models.metadata import Metadata
         from ..models.model_spec import ModelSpec
 
         if not src_dict:
@@ -84,11 +84,11 @@ class Model:
             events.append(componentsschemas_core_events_item)
 
         _metadata = d.pop("metadata", UNSET)
-        metadata: Union[Unset, EnvironmentMetadata]
+        metadata: Union[Unset, Metadata]
         if isinstance(_metadata, Unset):
             metadata = UNSET
         else:
-            metadata = EnvironmentMetadata.from_dict(_metadata)
+            metadata = Metadata.from_dict(_metadata)
 
         _spec = d.pop("spec", UNSET)
         spec: Union[Unset, ModelSpec]

beamlit/models/{resource_environment_metrics.py → resource_metrics.py}

@@ -10,21 +10,17 @@ if TYPE_CHECKING:
     from ..models.metric import Metric
     from ..models.request_duration_over_time_metrics import RequestDurationOverTimeMetrics
     from ..models.request_total_by_origin_metric import RequestTotalByOriginMetric
-    from ..models.resource_environment_metrics_request_total_per_code import (
-        ResourceEnvironmentMetricsRequestTotalPerCode,
-    )
-    from ..models.resource_environment_metrics_rps_per_code import (
-        ResourceEnvironmentMetricsRpsPerCode,
-    )
+    from ..models.resource_metrics_request_total_per_code import ResourceMetricsRequestTotalPerCode
+    from ..models.resource_metrics_rps_per_code import ResourceMetricsRpsPerCode
     from ..models.token_rate_metrics import TokenRateMetrics
     from ..models.token_total_metric import TokenTotalMetric
 
 
-T = TypeVar("T", bound="ResourceEnvironmentMetrics")
+T = TypeVar("T", bound="ResourceMetrics")
 
 
 @_attrs_define
-class ResourceEnvironmentMetrics:
+class ResourceMetrics:
     """Metrics for a single resource deployment (eg. model deployment, function deployment)
 
     Attributes:
@@ -34,11 +30,11 @@ class ResourceEnvironmentMetrics:
         request_duration_over_time (Union[Unset, RequestDurationOverTimeMetrics]): Request duration over time metrics
         request_total (Union[Unset, float]): Number of requests for the resource globally
         request_total_by_origin (Union[Unset, RequestTotalByOriginMetric]): Request total by origin metric
-        request_total_per_code (Union[Unset, ResourceEnvironmentMetricsRequestTotalPerCode]): Number of requests for the
-            resource globally per code
+        request_total_per_code (Union[Unset, ResourceMetricsRequestTotalPerCode]): Number of requests for the resource
+            globally per code
         rps (Union[Unset, float]): Number of requests per second for the resource globally
-        rps_per_code (Union[Unset, ResourceEnvironmentMetricsRpsPerCode]): Number of requests per second for the
-            resource globally per code
+        rps_per_code (Union[Unset, ResourceMetricsRpsPerCode]): Number of requests per second for the resource globally
+            per code
         token_rate (Union[Unset, TokenRateMetrics]): Token rate metrics
         token_total (Union[Unset, TokenTotalMetric]): Token total metric
     """
@@ -49,9 +45,9 @@ class ResourceEnvironmentMetrics:
     request_duration_over_time: Union[Unset, "RequestDurationOverTimeMetrics"] = UNSET
     request_total: Union[Unset, float] = UNSET
     request_total_by_origin: Union[Unset, "RequestTotalByOriginMetric"] = UNSET
-    request_total_per_code: Union[Unset, "ResourceEnvironmentMetricsRequestTotalPerCode"] = UNSET
+    request_total_per_code: Union[Unset, "ResourceMetricsRequestTotalPerCode"] = UNSET
     rps: Union[Unset, float] = UNSET
-    rps_per_code: Union[Unset, "ResourceEnvironmentMetricsRpsPerCode"] = UNSET
+    rps_per_code: Union[Unset, "ResourceMetricsRpsPerCode"] = UNSET
     token_rate: Union[Unset, "TokenRateMetrics"] = UNSET
     token_total: Union[Unset, "TokenTotalMetric"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -163,12 +159,10 @@ class ResourceEnvironmentMetrics:
         from ..models.metric import Metric
         from ..models.request_duration_over_time_metrics import RequestDurationOverTimeMetrics
         from ..models.request_total_by_origin_metric import RequestTotalByOriginMetric
-        from ..models.resource_environment_metrics_request_total_per_code import (
-            ResourceEnvironmentMetricsRequestTotalPerCode,
-        )
-        from ..models.resource_environment_metrics_rps_per_code import (
-            ResourceEnvironmentMetricsRpsPerCode,
+        from ..models.resource_metrics_request_total_per_code import (
+            ResourceMetricsRequestTotalPerCode,
         )
+        from ..models.resource_metrics_rps_per_code import ResourceMetricsRpsPerCode
         from ..models.token_rate_metrics import TokenRateMetrics
         from ..models.token_total_metric import TokenTotalMetric
 
@@ -213,20 +207,20 @@ class ResourceEnvironmentMetrics:
             request_total_by_origin = RequestTotalByOriginMetric.from_dict(_request_total_by_origin)
 
         _request_total_per_code = d.pop("requestTotalPerCode", UNSET)
-        request_total_per_code: Union[Unset, ResourceEnvironmentMetricsRequestTotalPerCode]
+        request_total_per_code: Union[Unset, ResourceMetricsRequestTotalPerCode]
         if isinstance(_request_total_per_code, Unset):
             request_total_per_code = UNSET
         else:
-            request_total_per_code = ResourceEnvironmentMetricsRequestTotalPerCode.from_dict(_request_total_per_code)
+            request_total_per_code = ResourceMetricsRequestTotalPerCode.from_dict(_request_total_per_code)
 
         rps = d.pop("rps", UNSET)
 
         _rps_per_code = d.pop("rpsPerCode", UNSET)
-        rps_per_code: Union[Unset, ResourceEnvironmentMetricsRpsPerCode]
+        rps_per_code: Union[Unset, ResourceMetricsRpsPerCode]
         if isinstance(_rps_per_code, Unset):
             rps_per_code = UNSET
         else:
-            rps_per_code = ResourceEnvironmentMetricsRpsPerCode.from_dict(_rps_per_code)
+            rps_per_code = ResourceMetricsRpsPerCode.from_dict(_rps_per_code)
 
         _token_rate = d.pop("tokenRate", UNSET)
         token_rate: Union[Unset, TokenRateMetrics]
@@ -242,7 +236,7 @@ class ResourceEnvironmentMetrics:
         else:
             token_total = TokenTotalMetric.from_dict(_token_total)
 
-        resource_environment_metrics = cls(
+        resource_metrics = cls(
             inference_global=inference_global,
             last_n_requests=last_n_requests,
             latency=latency,
@@ -256,8 +250,8 @@ class ResourceEnvironmentMetrics:
             token_total=token_total,
         )
 
-        resource_environment_metrics.additional_properties = d
-        return resource_environment_metrics
+        resource_metrics.additional_properties = d
+        return resource_metrics
 
     @property
     def additional_keys(self) -> list[str]:

beamlit/models/{resource_environment_metrics_request_total_per_code.py → resource_metrics_request_total_per_code.py}

@@ -3,11 +3,11 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="ResourceEnvironmentMetricsRequestTotalPerCode")
+T = TypeVar("T", bound="ResourceMetricsRequestTotalPerCode")
 
 
 @_attrs_define
-class ResourceEnvironmentMetricsRequestTotalPerCode:
+class ResourceMetricsRequestTotalPerCode:
     """Number of requests for the resource globally per code"""
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -23,10 +23,10 @@ class ResourceEnvironmentMetricsRequestTotalPerCode:
         if not src_dict:
             return None
         d = src_dict.copy()
-        resource_environment_metrics_request_total_per_code = cls()
+        resource_metrics_request_total_per_code = cls()
 
-        resource_environment_metrics_request_total_per_code.additional_properties = d
-        return resource_environment_metrics_request_total_per_code
+        resource_metrics_request_total_per_code.additional_properties = d
+        return resource_metrics_request_total_per_code
 
     @property
     def additional_keys(self) -> list[str]:

beamlit/models/{resource_environment_metrics_rps_per_code.py → resource_metrics_rps_per_code.py}

@@ -3,11 +3,11 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="ResourceEnvironmentMetricsRpsPerCode")
+T = TypeVar("T", bound="ResourceMetricsRpsPerCode")
 
 
 @_attrs_define
-class ResourceEnvironmentMetricsRpsPerCode:
+class ResourceMetricsRpsPerCode:
     """Number of requests per second for the resource globally per code"""
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -23,10 +23,10 @@ class ResourceEnvironmentMetricsRpsPerCode:
         if not src_dict:
             return None
         d = src_dict.copy()
-        resource_environment_metrics_rps_per_code = cls()
+        resource_metrics_rps_per_code = cls()
 
-        resource_environment_metrics_rps_per_code.additional_properties = d
-        return resource_environment_metrics_rps_per_code
+        resource_metrics_rps_per_code.additional_properties = d
+        return resource_metrics_rps_per_code
 
     @property
     def additional_keys(self) -> list[str]:

beamlit/models/{model_release.py → revision_metadata.py}

@@ -5,34 +5,34 @@ from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
-T = TypeVar("T", bound="ModelRelease")
+T = TypeVar("T", bound="RevisionMetadata")
 
 
 @_attrs_define
-class ModelRelease:
-    """Model release, used to deploy a model from one environment to another
+class RevisionMetadata:
+    """Revision metadata
 
     Attributes:
-        from_ (Union[Unset, str]): Origin environment from which the model is released
-        to (Union[Unset, str]): Destination environment to which the model is released
+        created_at (Union[Unset, str]): Revision created at
+        id (Union[Unset, str]): Revision ID
     """
 
-    from_: Union[Unset, str] = UNSET
-    to: Union[Unset, str] = UNSET
+    created_at: Union[Unset, str] = UNSET
+    id: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        from_ = self.from_
+        created_at = self.created_at
 
-        to = self.to
+        id = self.id
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
-        if from_ is not UNSET:
-            field_dict["from"] = from_
-        if to is not UNSET:
-            field_dict["to"] = to
+        if created_at is not UNSET:
+            field_dict["createdAt"] = created_at
+        if id is not UNSET:
+            field_dict["id"] = id
 
         return field_dict
 
@@ -41,17 +41,17 @@ class ModelRelease:
         if not src_dict:
             return None
         d = src_dict.copy()
-        from_ = d.pop("from", UNSET)
+        created_at = d.pop("createdAt", UNSET)
 
-        to = d.pop("to", UNSET)
+        id = d.pop("id", UNSET)
 
-        model_release = cls(
-            from_=from_,
-            to=to,
+        revision_metadata = cls(
+            created_at=created_at,
+            id=id,
         )
 
-        model_release.additional_properties = d
-        return model_release
+        revision_metadata.additional_properties = d
+        return revision_metadata
 
     @property
     def additional_keys(self) -> list[str]:

beamlit/models/runtime.py

@@ -22,8 +22,8 @@ class Runtime:
         command (Union[Unset, list[Any]]): The command to run the deployment
         endpoint_name (Union[Unset, str]): Endpoint Name of the model. In case of hf_private_endpoint, it is the
             endpoint name. In case of hf_public_endpoint, it is not used.
-        envs (Union[Unset, list[Any]]): The environment variables to set in the deployment. Should be a list of
-            Kubernetes EnvVar types
+        envs (Union[Unset, list[Any]]): The env 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 at HuggingFace.

beamlit/models/workspace.py

@@ -22,7 +22,6 @@ class Workspace:
         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
         account_id (Union[Unset, str]): Workspace account id
-        default_environment (Union[Unset, str]): Workspace environment
         display_name (Union[Unset, str]): Workspace display name
         labels (Union[Unset, WorkspaceLabels]): Workspace labels
         name (Union[Unset, str]): Workspace name
@@ -34,7 +33,6 @@ class Workspace:
     created_by: Union[Unset, str] = UNSET
     updated_by: Union[Unset, str] = UNSET
     account_id: Union[Unset, str] = UNSET
-    default_environment: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
     labels: Union[Unset, "WorkspaceLabels"] = UNSET
     name: Union[Unset, str] = UNSET
@@ -52,8 +50,6 @@ class Workspace:
 
         account_id = self.account_id
 
-        default_environment = self.default_environment
-
         display_name = self.display_name
 
         labels: Union[Unset, dict[str, Any]] = UNSET
@@ -79,8 +75,6 @@ class Workspace:
             field_dict["updatedBy"] = updated_by
         if account_id is not UNSET:
             field_dict["accountId"] = account_id
-        if default_environment is not UNSET:
-            field_dict["defaultEnvironment"] = default_environment
         if display_name is not UNSET:
             field_dict["displayName"] = display_name
         if labels is not UNSET:
@@ -109,8 +103,6 @@ class Workspace:
 
         account_id = d.pop("accountId", UNSET)
 
-        default_environment = d.pop("defaultEnvironment", UNSET)
-
         display_name = d.pop("displayName", UNSET)
 
         _labels = d.pop("labels", UNSET)
@@ -130,7 +122,6 @@ class Workspace:
             created_by=created_by,
             updated_by=updated_by,
             account_id=account_id,
-            default_environment=default_environment,
             display_name=display_name,
             labels=labels,
             name=name,

beamlit/run.py

@@ -14,8 +14,7 @@ class RunClient:
     """Provides functionality for executing HTTP requests against Beamlit resources.
 
     This module contains the RunClient class which handles authenticated HTTP requests to Beamlit
-    resources. It allows users to interact with different resource types (like functions or services)
-    in specific environments, supporting various HTTP methods and request parameters.
+    resources. It allows users to interact with different resource types (like functions or services), supporting various HTTP methods and request parameters.
 
     Example:
         ```python
@@ -24,7 +23,6 @@ class RunClient:
         response = run_client.run(
             resource_type="function",
             resource_name="my-function",
-            environment="prod",
             method="POST",
             json={"key": "value"}
         )
@@ -41,7 +39,6 @@ class RunClient:
         self,
         resource_type: str,
         resource_name: str,
-        environment: str,
         method: str,
         path: str = "",
         headers: dict[str, str] | None = None,
@@ -56,7 +53,6 @@ class RunClient:
         Args:
             resource_type (str): The type of resource to interact with (e.g., 'function', 'service').
             resource_name (str): The name of the specific resource.
-            environment (str): The environment to execute the request in.
             method (str): The HTTP method to use (e.g., 'GET', 'POST', 'PUT', 'DELETE').
             path (str, optional): Additional path segments to append to the resource URL. Defaults to "".
             headers (dict[str, str] | None, optional): HTTP headers to include in the request. Defaults to None.
@@ -78,7 +74,7 @@ class RunClient:
 
         if cloud and path and service_name:
             url = f"https://{service_name}.{settings.run_internal_hostname}/{path}"
-        
+
         if cloud and not path and service_name:
             url = f"https://{service_name}.{settings.run_internal_hostname}"
 
@@ -90,7 +86,7 @@ class RunClient:
 
         kwargs = {
             "headers": headers,
-            "params": {"environment": environment, **params},
+            "params": {**params},
         }
         if data:
             kwargs["data"] = data

beamlit/serve/app.py

@@ -44,7 +44,7 @@ logger = getLogger(__name__)
 logger.info(f"Importing server module: {settings.server.module}")
 func = import_module()
 logger.info(
-    f"Running server with environment {settings.environment}"
+    f"Running server"
     f" on {settings.server.host}:{settings.server.port}"
 )
 func_params = inspect.signature(func).parameters
@@ -96,7 +96,6 @@ if websocket_detected:
 else:
     @app.post("/")
     async def root(request: Request):
-        settings = get_settings()
         logger = getLogger(__name__)
         try:
             original_func = getattr(func, "__wrapped__", func)
@@ -117,19 +116,16 @@ else:
             return JSONResponse(status_code=200, content=response)
         except ValueError as e:
             content = {"error": str(e)}
-            if settings.environment == "development":
-                content["traceback"] = str(traceback.format_exc())
+            content["traceback"] = str(traceback.format_exc())
             logger.error(str(traceback.format_exc()))
             return JSONResponse(status_code=400, content=content)
         except HTTPError as e:
             content = {"error": e.message, "status_code": e.status_code}
-            if settings.environment == "development":
-                content["traceback"] = str(traceback.format_exc())
+            content["traceback"] = str(traceback.format_exc())
             logger.error(f"{e.status_code} {str(traceback.format_exc())}")
             return JSONResponse(status_code=e.status_code, content=content)
         except Exception as e:
             content = {"error": f"Internal server error, {e}"}
-            if settings.environment == "development":
-                content["traceback"] = str(traceback.format_exc())
+            content["traceback"] = str(traceback.format_exc())
             logger.error(str(traceback.format_exc()))
             return JSONResponse(status_code=500, content=content)

{beamlit-0.0.56rc107.dist-info → beamlit-0.0.57.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.56rc107
+Version: 0.0.57
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 License-File: LICENSE
@@ -43,6 +43,7 @@ Requires-Dist: pyjwt>=2.10.1
 Requires-Dist: python-dateutil>=2.8.0
 Requires-Dist: pyyaml<6.1.0,>=6.0.2
 Requires-Dist: requests<2.33.0,>=2.32.3
+Requires-Dist: websocket>=0.2.1
 Requires-Dist: websockets>=14.1
 Description-Content-Type: text/markdown