beamlit 0.0.46__py3-none-any.whl → 0.0.47__py3-none-any.whl

beamlit/models/policy_spec.py

@@ -8,6 +8,7 @@ from ..types import UNSET, Unset
 if TYPE_CHECKING:
     from ..models.flavor import Flavor
     from ..models.policy_location import PolicyLocation
+    from ..models.policy_max_tokens import PolicyMaxTokens
 
 
 T = TypeVar("T", bound="PolicySpec")
@@ -20,14 +21,19 @@ class PolicySpec:
     Attributes:
         flavors (Union[Unset, list['Flavor']]): Types of hardware available for deployments
         locations (Union[Unset, list['PolicyLocation']]): PolicyLocations is a local type that wraps a slice of Location
+        max_tokens (Union[Unset, PolicyMaxTokens]): PolicyMaxTokens is a local type that wraps a slice of
+            PolicyMaxTokens
         resource_types (Union[Unset, list[str]]): PolicyResourceTypes is a local type that wraps a slice of
             PolicyResourceType
+        sandbox (Union[Unset, bool]): Sandbox mode
         type_ (Union[Unset, str]): Policy type, can be location or flavor
     """
 
     flavors: Union[Unset, list["Flavor"]] = UNSET
     locations: Union[Unset, list["PolicyLocation"]] = UNSET
+    max_tokens: Union[Unset, "PolicyMaxTokens"] = UNSET
     resource_types: Union[Unset, list[str]] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     type_: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -46,10 +52,18 @@ class PolicySpec:
                 componentsschemas_policy_locations_item = componentsschemas_policy_locations_item_data.to_dict()
                 locations.append(componentsschemas_policy_locations_item)
 
+        max_tokens: Union[Unset, dict[str, Any]] = UNSET
+        if self.max_tokens and not isinstance(self.max_tokens, Unset) and not isinstance(self.max_tokens, dict):
+            max_tokens = self.max_tokens.to_dict()
+        elif self.max_tokens and isinstance(self.max_tokens, dict):
+            max_tokens = self.max_tokens
+
         resource_types: Union[Unset, list[str]] = UNSET
         if not isinstance(self.resource_types, Unset):
             resource_types = self.resource_types
 
+        sandbox = self.sandbox
+
         type_ = self.type_
 
         field_dict: dict[str, Any] = {}
@@ -59,8 +73,12 @@ class PolicySpec:
             field_dict["flavors"] = flavors
         if locations is not UNSET:
             field_dict["locations"] = locations
+        if max_tokens is not UNSET:
+            field_dict["maxTokens"] = max_tokens
         if resource_types is not UNSET:
             field_dict["resourceTypes"] = resource_types
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if type_ is not UNSET:
             field_dict["type"] = type_
 
@@ -70,6 +88,7 @@ class PolicySpec:
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.flavor import Flavor
         from ..models.policy_location import PolicyLocation
+        from ..models.policy_max_tokens import PolicyMaxTokens
 
         if not src_dict:
             return None
@@ -90,14 +109,25 @@ class PolicySpec:
 
             locations.append(componentsschemas_policy_locations_item)
 
+        _max_tokens = d.pop("maxTokens", UNSET)
+        max_tokens: Union[Unset, PolicyMaxTokens]
+        if isinstance(_max_tokens, Unset):
+            max_tokens = UNSET
+        else:
+            max_tokens = PolicyMaxTokens.from_dict(_max_tokens)
+
         resource_types = cast(list[str], d.pop("resourceTypes", UNSET))
 
+        sandbox = d.pop("sandbox", UNSET)
+
         type_ = d.pop("type", UNSET)
 
         policy_spec = cls(
             flavors=flavors,
             locations=locations,
+            max_tokens=max_tokens,
             resource_types=resource_types,
+            sandbox=sandbox,
             type_=type_,
         )
 

beamlit/models/request_duration_over_time_metrics.py

@@ -25,8 +25,14 @@ class RequestDurationOverTimeMetrics:
 
     def to_dict(self) -> dict[str, Any]:
         request_duration_over_time: Union[Unset, dict[str, Any]] = UNSET
-        if self.request_duration_over_time and not isinstance(self.request_duration_over_time, Unset):
+        if (
+            self.request_duration_over_time
+            and not isinstance(self.request_duration_over_time, Unset)
+            and not isinstance(self.request_duration_over_time, dict)
+        ):
             request_duration_over_time = self.request_duration_over_time.to_dict()
+        elif self.request_duration_over_time and isinstance(self.request_duration_over_time, dict):
+            request_duration_over_time = self.request_duration_over_time
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/request_total_by_origin_metric.py

@@ -33,12 +33,24 @@ class RequestTotalByOriginMetric:
 
     def to_dict(self) -> dict[str, Any]:
         request_total_by_origin: Union[Unset, dict[str, Any]] = UNSET
-        if self.request_total_by_origin and not isinstance(self.request_total_by_origin, Unset):
+        if (
+            self.request_total_by_origin
+            and not isinstance(self.request_total_by_origin, Unset)
+            and not isinstance(self.request_total_by_origin, dict)
+        ):
             request_total_by_origin = self.request_total_by_origin.to_dict()
+        elif self.request_total_by_origin and isinstance(self.request_total_by_origin, dict):
+            request_total_by_origin = self.request_total_by_origin
 
         request_total_by_origin_and_code: Union[Unset, dict[str, Any]] = UNSET
-        if self.request_total_by_origin_and_code and not isinstance(self.request_total_by_origin_and_code, Unset):
+        if (
+            self.request_total_by_origin_and_code
+            and not isinstance(self.request_total_by_origin_and_code, Unset)
+            and not isinstance(self.request_total_by_origin_and_code, dict)
+        ):
             request_total_by_origin_and_code = self.request_total_by_origin_and_code.to_dict()
+        elif self.request_total_by_origin_and_code and isinstance(self.request_total_by_origin_and_code, dict):
+            request_total_by_origin_and_code = self.request_total_by_origin_and_code
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/request_total_metric.py

@@ -37,14 +37,22 @@ class RequestTotalMetric:
         request_total = self.request_total
 
         request_total_per_code: Union[Unset, dict[str, Any]] = UNSET
-        if self.request_total_per_code and not isinstance(self.request_total_per_code, Unset):
+        if (
+            self.request_total_per_code
+            and not isinstance(self.request_total_per_code, Unset)
+            and not isinstance(self.request_total_per_code, dict)
+        ):
             request_total_per_code = self.request_total_per_code.to_dict()
+        elif self.request_total_per_code and isinstance(self.request_total_per_code, dict):
+            request_total_per_code = self.request_total_per_code
 
         rps = self.rps
 
         rps_per_code: Union[Unset, dict[str, Any]] = UNSET
-        if self.rps_per_code and not isinstance(self.rps_per_code, Unset):
+        if self.rps_per_code and not isinstance(self.rps_per_code, Unset) and not isinstance(self.rps_per_code, dict):
             rps_per_code = self.rps_per_code.to_dict()
+        elif self.rps_per_code and isinstance(self.rps_per_code, dict):
+            rps_per_code = self.rps_per_code
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/resource_environment_metrics.py

@@ -72,36 +72,62 @@ class ResourceEnvironmentMetrics:
                 last_n_requests.append(componentsschemas_array_metric_item)
 
         latency: Union[Unset, dict[str, Any]] = UNSET
-        if self.latency and not isinstance(self.latency, Unset):
+        if self.latency and not isinstance(self.latency, Unset) and not isinstance(self.latency, dict):
             latency = self.latency.to_dict()
+        elif self.latency and isinstance(self.latency, dict):
+            latency = self.latency
 
         request_duration_over_time: Union[Unset, dict[str, Any]] = UNSET
-        if self.request_duration_over_time and not isinstance(self.request_duration_over_time, Unset):
+        if (
+            self.request_duration_over_time
+            and not isinstance(self.request_duration_over_time, Unset)
+            and not isinstance(self.request_duration_over_time, dict)
+        ):
             request_duration_over_time = self.request_duration_over_time.to_dict()
+        elif self.request_duration_over_time and isinstance(self.request_duration_over_time, dict):
+            request_duration_over_time = self.request_duration_over_time
 
         request_total = self.request_total
 
         request_total_by_origin: Union[Unset, dict[str, Any]] = UNSET
-        if self.request_total_by_origin and not isinstance(self.request_total_by_origin, Unset):
+        if (
+            self.request_total_by_origin
+            and not isinstance(self.request_total_by_origin, Unset)
+            and not isinstance(self.request_total_by_origin, dict)
+        ):
             request_total_by_origin = self.request_total_by_origin.to_dict()
+        elif self.request_total_by_origin and isinstance(self.request_total_by_origin, dict):
+            request_total_by_origin = self.request_total_by_origin
 
         request_total_per_code: Union[Unset, dict[str, Any]] = UNSET
-        if self.request_total_per_code and not isinstance(self.request_total_per_code, Unset):
+        if (
+            self.request_total_per_code
+            and not isinstance(self.request_total_per_code, Unset)
+            and not isinstance(self.request_total_per_code, dict)
+        ):
             request_total_per_code = self.request_total_per_code.to_dict()
+        elif self.request_total_per_code and isinstance(self.request_total_per_code, dict):
+            request_total_per_code = self.request_total_per_code
 
         rps = self.rps
 
         rps_per_code: Union[Unset, dict[str, Any]] = UNSET
-        if self.rps_per_code and not isinstance(self.rps_per_code, Unset):
+        if self.rps_per_code and not isinstance(self.rps_per_code, Unset) and not isinstance(self.rps_per_code, dict):
             rps_per_code = self.rps_per_code.to_dict()
+        elif self.rps_per_code and isinstance(self.rps_per_code, dict):
+            rps_per_code = self.rps_per_code
 
         token_rate: Union[Unset, dict[str, Any]] = UNSET
-        if self.token_rate and not isinstance(self.token_rate, Unset):
+        if self.token_rate and not isinstance(self.token_rate, Unset) and not isinstance(self.token_rate, dict):
             token_rate = self.token_rate.to_dict()
+        elif self.token_rate and isinstance(self.token_rate, dict):
+            token_rate = self.token_rate
 
         token_total: Union[Unset, dict[str, Any]] = UNSET
-        if self.token_total and not isinstance(self.token_total, Unset):
+        if self.token_total and not isinstance(self.token_total, Unset) and not isinstance(self.token_total, dict):
             token_total = self.token_total.to_dict()
+        elif self.token_total and isinstance(self.token_total, dict):
+            token_total = self.token_total
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/resource_log.py

@@ -14,16 +14,20 @@ class ResourceLog:
 
     Attributes:
         message (Union[Unset, str]): Content of the log
+        severity (Union[Unset, int]): Severity of the log
         timestamp (Union[Unset, str]): The timestamp of the log
     """
 
     message: Union[Unset, str] = UNSET
+    severity: Union[Unset, int] = UNSET
     timestamp: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         message = self.message
 
+        severity = self.severity
+
         timestamp = self.timestamp
 
         field_dict: dict[str, Any] = {}
@@ -31,6 +35,8 @@ class ResourceLog:
         field_dict.update({})
         if message is not UNSET:
             field_dict["message"] = message
+        if severity is not UNSET:
+            field_dict["severity"] = severity
         if timestamp is not UNSET:
             field_dict["timestamp"] = timestamp
 
@@ -43,10 +49,13 @@ class ResourceLog:
         d = src_dict.copy()
         message = d.pop("message", UNSET)
 
+        severity = d.pop("severity", UNSET)
+
         timestamp = d.pop("timestamp", UNSET)
 
         resource_log = cls(
             message=message,
+            severity=severity,
             timestamp=timestamp,
         )
 

beamlit/models/runtime.py

@@ -64,12 +64,20 @@ class Runtime:
         model = self.model
 
         readiness_probe: Union[Unset, dict[str, Any]] = UNSET
-        if self.readiness_probe and not isinstance(self.readiness_probe, Unset):
+        if (
+            self.readiness_probe
+            and not isinstance(self.readiness_probe, Unset)
+            and not isinstance(self.readiness_probe, dict)
+        ):
             readiness_probe = self.readiness_probe.to_dict()
+        elif self.readiness_probe and isinstance(self.readiness_probe, dict):
+            readiness_probe = self.readiness_probe
 
         resources: Union[Unset, dict[str, Any]] = UNSET
-        if self.resources and not isinstance(self.resources, Unset):
+        if self.resources and not isinstance(self.resources, Unset) and not isinstance(self.resources, dict):
             resources = self.resources.to_dict()
+        elif self.resources and isinstance(self.resources, dict):
+            resources = self.resources
 
         serving_port = self.serving_port
 

beamlit/models/store_agent.py

@@ -65,8 +65,10 @@ class StoreAgent:
         image = self.image
 
         labels: Union[Unset, dict[str, Any]] = UNSET
-        if self.labels and not isinstance(self.labels, Unset):
+        if self.labels and not isinstance(self.labels, Unset) and not isinstance(self.labels, dict):
             labels = self.labels.to_dict()
+        elif self.labels and isinstance(self.labels, dict):
+            labels = self.labels
 
         name = self.name
 

beamlit/models/store_function.py

@@ -78,8 +78,10 @@ class StoreFunction:
                 kit.append(kit_item)
 
         labels: Union[Unset, dict[str, Any]] = UNSET
-        if self.labels and not isinstance(self.labels, Unset):
+        if self.labels and not isinstance(self.labels, Unset) and not isinstance(self.labels, dict):
             labels = self.labels.to_dict()
+        elif self.labels and isinstance(self.labels, dict):
+            labels = self.labels
 
         name = self.name
 

beamlit/models/token_rate_metrics.py

@@ -29,16 +29,30 @@ class TokenRateMetrics:
 
     def to_dict(self) -> dict[str, Any]:
         token_rate: Union[Unset, dict[str, Any]] = UNSET
-        if self.token_rate and not isinstance(self.token_rate, Unset):
+        if self.token_rate and not isinstance(self.token_rate, Unset) and not isinstance(self.token_rate, dict):
             token_rate = self.token_rate.to_dict()
+        elif self.token_rate and isinstance(self.token_rate, dict):
+            token_rate = self.token_rate
 
         token_rate_input: Union[Unset, dict[str, Any]] = UNSET
-        if self.token_rate_input and not isinstance(self.token_rate_input, Unset):
+        if (
+            self.token_rate_input
+            and not isinstance(self.token_rate_input, Unset)
+            and not isinstance(self.token_rate_input, dict)
+        ):
             token_rate_input = self.token_rate_input.to_dict()
+        elif self.token_rate_input and isinstance(self.token_rate_input, dict):
+            token_rate_input = self.token_rate_input
 
         token_rate_output: Union[Unset, dict[str, Any]] = UNSET
-        if self.token_rate_output and not isinstance(self.token_rate_output, Unset):
+        if (
+            self.token_rate_output
+            and not isinstance(self.token_rate_output, Unset)
+            and not isinstance(self.token_rate_output, dict)
+        ):
             token_rate_output = self.token_rate_output.to_dict()
+        elif self.token_rate_output and isinstance(self.token_rate_output, dict):
+            token_rate_output = self.token_rate_output
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/trace_ids_response.py

@@ -1,34 +1,20 @@
-from typing import Any, TypeVar, Union, cast
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 T = TypeVar("T", bound="TraceIdsResponse")
 
 
 @_attrs_define
 class TraceIdsResponse:
-    """Response containing trace IDs
-
-    Attributes:
-        trace_ids (Union[Unset, list[str]]): List of trace IDs
-    """
+    """Trace IDs response"""
 
-    trace_ids: Union[Unset, list[str]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        trace_ids: Union[Unset, list[str]] = UNSET
-        if not isinstance(self.trace_ids, Unset):
-            trace_ids = self.trace_ids
-
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if trace_ids is not UNSET:
-            field_dict["trace_ids"] = trace_ids
 
         return field_dict
 
@@ -37,11 +23,7 @@ class TraceIdsResponse:
         if not src_dict:
             return None
         d = src_dict.copy()
-        trace_ids = cast(list[str], d.pop("trace_ids", UNSET))
-
-        trace_ids_response = cls(
-            trace_ids=trace_ids,
-        )
+        trace_ids_response = cls()
 
         trace_ids_response.additional_properties = d
         return trace_ids_response

beamlit/models/workspace.py

@@ -51,8 +51,10 @@ class Workspace:
         display_name = self.display_name
 
         labels: Union[Unset, dict[str, Any]] = UNSET
-        if self.labels and not isinstance(self.labels, Unset):
+        if self.labels and not isinstance(self.labels, Unset) and not isinstance(self.labels, dict):
             labels = self.labels.to_dict()
+        elif self.labels and isinstance(self.labels, dict):
+            labels = self.labels
 
         name = self.name
 

{beamlit-0.0.46.dist-info → beamlit-0.0.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.46
+Version: 0.0.47
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 License-File: LICENSE
@@ -9,14 +9,8 @@ Requires-Dist: asgi-correlation-id<5.0.0,>=4.3.4
 Requires-Dist: attrs>=21.3.0
 Requires-Dist: fastapi[standard]<0.116.0,>=0.115.4
 Requires-Dist: httpx<0.28.0,>=0.20.0
-Requires-Dist: langchain-anthropic>=0.3.4
-Requires-Dist: langchain-cohere>=0.4.2
 Requires-Dist: langchain-community<0.4.0,>=0.3.3
 Requires-Dist: langchain-core<0.4.0,>=0.3.13
-Requires-Dist: langchain-deepseek-official>=0.1.0
-Requires-Dist: langchain-mistralai>=0.2.5
-Requires-Dist: langchain-openai<0.4.0,>=0.3.0
-Requires-Dist: langchain-xai>=0.2.0
 Requires-Dist: langgraph<0.3.0,>=0.2.40
 Requires-Dist: mcp>=1.2.1
 Requires-Dist: opentelemetry-api>=1.28.2