blaxel 0.2.34__py3-none-any.whl → 0.2.35__py3-none-any.whl

blaxel/core/client/models/pending_invitation_workspace_details.py

@@ -1,10 +1,16 @@
-from typing import Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
+if TYPE_CHECKING:
+    from ..models.pending_invitation_workspace_details_emails_item import (
+        PendingInvitationWorkspaceDetailsEmailsItem,
+    )
+
+
 T = TypeVar("T", bound="PendingInvitationWorkspaceDetails")
 
 
@@ -13,18 +19,24 @@ class PendingInvitationWorkspaceDetails:
     """Workspace details
 
     Attributes:
-        emails (Union[Unset, list[Any]]): List of user emails in the workspace
+        emails (Union[Unset, list['PendingInvitationWorkspaceDetailsEmailsItem']]): List of user emails in the workspace
         user_number (Union[Unset, float]): Number of users in the workspace
     """
 
-    emails: Union[Unset, list[Any]] = UNSET
+    emails: Union[Unset, list["PendingInvitationWorkspaceDetailsEmailsItem"]] = UNSET
     user_number: Union[Unset, float] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        emails: Union[Unset, list[Any]] = UNSET
+        emails: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.emails, Unset):
-            emails = self.emails
+            emails = []
+            for emails_item_data in self.emails:
+                if type(emails_item_data) is dict:
+                    emails_item = emails_item_data
+                else:
+                    emails_item = emails_item_data.to_dict()
+                emails.append(emails_item)
 
         user_number = self.user_number
 
@@ -40,10 +52,19 @@ class PendingInvitationWorkspaceDetails:
 
     @classmethod
     def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T | None:
+        from ..models.pending_invitation_workspace_details_emails_item import (
+            PendingInvitationWorkspaceDetailsEmailsItem,
+        )
+
         if not src_dict:
             return None
         d = src_dict.copy()
-        emails = cast(list[Any], d.pop("emails", UNSET))
+        emails = []
+        _emails = d.pop("emails", UNSET)
+        for emails_item_data in _emails or []:
+            emails_item = PendingInvitationWorkspaceDetailsEmailsItem.from_dict(emails_item_data)
+
+            emails.append(emails_item)
 
         user_number = d.pop("user_number", UNSET)
 

blaxel/core/client/models/{metrics_request_total_per_code.py → pending_invitation_workspace_details_emails_item.py}

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

blaxel/core/client/models/policy.py

@@ -1,10 +1,8 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 if TYPE_CHECKING:
     from ..models.metadata import Metadata
     from ..models.policy_spec import PolicySpec
@@ -18,38 +16,34 @@ class Policy:
     """Rule that controls how a deployment is made and served (e.g. location restrictions)
 
     Attributes:
-        metadata (Union[Unset, Metadata]): Metadata
-        spec (Union[Unset, PolicySpec]): Policy specification
+        metadata (Metadata): Common metadata fields shared by all Blaxel resources including name, labels, timestamps,
+            and ownership information
+        spec (PolicySpec): Policy specification
     """
 
-    metadata: Union[Unset, "Metadata"] = UNSET
-    spec: Union[Unset, "PolicySpec"] = UNSET
+    metadata: "Metadata"
+    spec: "PolicySpec"
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        metadata: Union[Unset, dict[str, Any]] = UNSET
-        if (
-            self.metadata
-            and not isinstance(self.metadata, Unset)
-            and not isinstance(self.metadata, dict)
-        ):
-            metadata = self.metadata.to_dict()
-        elif self.metadata and isinstance(self.metadata, dict):
+        if type(self.metadata) is dict:
             metadata = self.metadata
+        else:
+            metadata = self.metadata.to_dict()
 
-        spec: Union[Unset, dict[str, Any]] = UNSET
-        if self.spec and not isinstance(self.spec, Unset) and not isinstance(self.spec, dict):
-            spec = self.spec.to_dict()
-        elif self.spec and isinstance(self.spec, dict):
+        if type(self.spec) is dict:
             spec = self.spec
+        else:
+            spec = self.spec.to_dict()
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if metadata is not UNSET:
-            field_dict["metadata"] = metadata
-        if spec is not UNSET:
-            field_dict["spec"] = spec
+        field_dict.update(
+            {
+                "metadata": metadata,
+                "spec": spec,
+            }
+        )
 
         return field_dict
 
@@ -61,19 +55,9 @@ class Policy:
         if not src_dict:
             return None
         d = src_dict.copy()
-        _metadata = d.pop("metadata", UNSET)
-        metadata: Union[Unset, Metadata]
-        if isinstance(_metadata, Unset):
-            metadata = UNSET
-        else:
-            metadata = Metadata.from_dict(_metadata)
+        metadata = Metadata.from_dict(d.pop("metadata"))
 
-        _spec = d.pop("spec", UNSET)
-        spec: Union[Unset, PolicySpec]
-        if isinstance(_spec, Unset):
-            spec = UNSET
-        else:
-            spec = PolicySpec.from_dict(_spec)
+        spec = PolicySpec.from_dict(d.pop("spec"))
 
         policy = cls(
             metadata=metadata,

blaxel/core/client/models/policy_location.py

@@ -3,6 +3,7 @@ from typing import Any, TypeVar, Union
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.policy_location_type import PolicyLocationType
 from ..types import UNSET, Unset
 
 T = TypeVar("T", bound="PolicyLocation")
@@ -13,18 +14,20 @@ class PolicyLocation:
     """Policy location
 
     Attributes:
-        name (Union[Unset, str]): Policy location name
-        type_ (Union[Unset, str]): Policy location type
+        name (Union[Unset, str]): Policy location name Example: EU.
+        type_ (Union[Unset, PolicyLocationType]): Policy location type Example: continent.
     """
 
     name: Union[Unset, str] = UNSET
-    type_: Union[Unset, str] = UNSET
+    type_: Union[Unset, PolicyLocationType] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         name = self.name
 
-        type_ = self.type_
+        type_: Union[Unset, str] = UNSET
+        if not isinstance(self.type_, Unset):
+            type_ = self.type_.value
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -43,7 +46,12 @@ class PolicyLocation:
         d = src_dict.copy()
         name = d.pop("name", UNSET)
 
-        type_ = d.pop("type", d.pop("type_", UNSET))
+        _type_ = d.pop("type", d.pop("type_", UNSET))
+        type_: Union[Unset, PolicyLocationType]
+        if isinstance(_type_, Unset):
+            type_ = UNSET
+        else:
+            type_ = PolicyLocationType(_type_)
 
         policy_location = cls(
             name=name,

blaxel/core/client/models/policy_location_type.py

@@ -0,0 +1,19 @@
+from enum import Enum
+
+
+class PolicyLocationType(str, Enum):
+    CONTINENT = "continent"
+    COUNTRY = "country"
+    LOCATION = "location"
+
+    def __str__(self) -> str:
+        return str(self.value)
+
+    @classmethod
+    def _missing_(cls, value: object) -> "PolicyLocationType | None":
+        if isinstance(value, str):
+            upper_value = value.upper()
+            for member in cls:
+                if member.value.upper() == upper_value:
+                    return member
+        return None

blaxel/core/client/models/policy_max_tokens.py

@@ -13,12 +13,12 @@ class PolicyMaxTokens:
     """PolicyMaxTokens is a local type that wraps a slice of PolicyMaxTokens
 
     Attributes:
-        granularity (Union[Unset, str]): Granularity
-        input_ (Union[Unset, int]): Input
-        output (Union[Unset, int]): Output
-        ratio_input_over_output (Union[Unset, int]): RatioInputOverOutput
-        step (Union[Unset, int]): Step
-        total (Union[Unset, int]): Total
+        granularity (Union[Unset, str]): Granularity Example: minute.
+        input_ (Union[Unset, int]): Input Example: 10000.
+        output (Union[Unset, int]): Output Example: 5000.
+        ratio_input_over_output (Union[Unset, int]): RatioInputOverOutput Example: 2.
+        step (Union[Unset, int]): Step Example: 1.
+        total (Union[Unset, int]): Total Example: 15000.
     """
 
     granularity: Union[Unset, str] = UNSET

blaxel/core/client/models/policy_resource_type.py

@@ -0,0 +1,20 @@
+from enum import Enum
+
+
+class PolicyResourceType(str, Enum):
+    AGENT = "agent"
+    FUNCTION = "function"
+    MODEL = "model"
+    SANDBOX = "sandbox"
+
+    def __str__(self) -> str:
+        return str(self.value)
+
+    @classmethod
+    def _missing_(cls, value: object) -> "PolicyResourceType | None":
+        if isinstance(value, str):
+            upper_value = value.upper()
+            for member in cls:
+                if member.value.upper() == upper_value:
+                    return member
+        return None

blaxel/core/client/models/policy_spec.py

@@ -1,8 +1,10 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.policy_resource_type import PolicyResourceType
+from ..models.policy_spec_type import PolicySpecType
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
@@ -23,18 +25,18 @@ class PolicySpec:
         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
+        resource_types (Union[Unset, list[PolicyResourceType]]): 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
+        type_ (Union[Unset, PolicySpecType]): Policy type, can be location or flavor Example: location.
     """
 
     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
+    resource_types: Union[Unset, list[PolicyResourceType]] = UNSET
     sandbox: Union[Unset, bool] = UNSET
-    type_: Union[Unset, str] = UNSET
+    type_: Union[Unset, PolicySpecType] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -74,11 +76,18 @@ class PolicySpec:
 
         resource_types: Union[Unset, list[str]] = UNSET
         if not isinstance(self.resource_types, Unset):
-            resource_types = self.resource_types
+            resource_types = []
+            for componentsschemas_policy_resource_types_item_data in self.resource_types:
+                componentsschemas_policy_resource_types_item = (
+                    componentsschemas_policy_resource_types_item_data.value
+                )
+                resource_types.append(componentsschemas_policy_resource_types_item)
 
         sandbox = self.sandbox
 
-        type_ = self.type_
+        type_: Union[Unset, str] = UNSET
+        if not isinstance(self.type_, Unset):
+            type_ = self.type_.value
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -130,11 +139,23 @@ class PolicySpec:
         else:
             max_tokens = PolicyMaxTokens.from_dict(_max_tokens)
 
-        resource_types = cast(list[str], d.pop("resourceTypes", d.pop("resource_types", UNSET)))
+        resource_types = []
+        _resource_types = d.pop("resourceTypes", d.pop("resource_types", UNSET))
+        for componentsschemas_policy_resource_types_item_data in _resource_types or []:
+            componentsschemas_policy_resource_types_item = PolicyResourceType(
+                componentsschemas_policy_resource_types_item_data
+            )
+
+            resource_types.append(componentsschemas_policy_resource_types_item)
 
         sandbox = d.pop("sandbox", UNSET)
 
-        type_ = d.pop("type", d.pop("type_", UNSET))
+        _type_ = d.pop("type", d.pop("type_", UNSET))
+        type_: Union[Unset, PolicySpecType]
+        if isinstance(_type_, Unset):
+            type_ = UNSET
+        else:
+            type_ = PolicySpecType(_type_)
 
         policy_spec = cls(
             flavors=flavors,

blaxel/core/client/models/policy_spec_type.py

@@ -0,0 +1,19 @@
+from enum import Enum
+
+
+class PolicySpecType(str, Enum):
+    FLAVOR = "flavor"
+    LOCATION = "location"
+    MAXTOKEN = "maxToken"
+
+    def __str__(self) -> str:
+        return str(self.value)
+
+    @classmethod
+    def _missing_(cls, value: object) -> "PolicySpecType | None":
+        if isinstance(value, str):
+            upper_value = value.upper()
+            for member in cls:
+                if member.value.upper() == upper_value:
+                    return member
+        return None

blaxel/core/client/models/port.py

@@ -3,6 +3,7 @@ from typing import Any, TypeVar, Union
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.port_protocol import PortProtocol
 from ..types import UNSET, Unset
 
 T = TypeVar("T", bound="Port")
@@ -13,32 +14,36 @@ class Port:
     """A port for a resource
 
     Attributes:
-        name (Union[Unset, str]): The name of the port
-        protocol (Union[Unset, str]): The protocol of the port
-        target (Union[Unset, int]): The target port of the port
+        target (int): The target port of the port Example: 8080.
+        name (Union[Unset, str]): The name of the port Example: http.
+        protocol (Union[Unset, PortProtocol]): The protocol of the port Example: HTTP.
     """
 
+    target: int
     name: Union[Unset, str] = UNSET
-    protocol: Union[Unset, str] = UNSET
-    target: Union[Unset, int] = UNSET
+    protocol: Union[Unset, PortProtocol] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        target = self.target
+
         name = self.name
 
-        protocol = self.protocol
-
-        target = self.target
+        protocol: Union[Unset, str] = UNSET
+        if not isinstance(self.protocol, Unset):
+            protocol = self.protocol.value
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
+        field_dict.update(
+            {
+                "target": target,
+            }
+        )
         if name is not UNSET:
             field_dict["name"] = name
         if protocol is not UNSET:
             field_dict["protocol"] = protocol
-        if target is not UNSET:
-            field_dict["target"] = target
 
         return field_dict
 
@@ -47,16 +52,21 @@ class Port:
         if not src_dict:
             return None
         d = src_dict.copy()
-        name = d.pop("name", UNSET)
+        target = d.pop("target")
 
-        protocol = d.pop("protocol", UNSET)
+        name = d.pop("name", UNSET)
 
-        target = d.pop("target", UNSET)
+        _protocol = d.pop("protocol", UNSET)
+        protocol: Union[Unset, PortProtocol]
+        if isinstance(_protocol, Unset):
+            protocol = UNSET
+        else:
+            protocol = PortProtocol(_protocol)
 
         port = cls(
+            target=target,
             name=name,
             protocol=protocol,
-            target=target,
         )
 
         port.additional_properties = d

blaxel/core/client/models/port_protocol.py

@@ -0,0 +1,19 @@
+from enum import Enum
+
+
+class PortProtocol(str, Enum):
+    HTTP = "HTTP"
+    TCP = "TCP"
+    UDP = "UDP"
+
+    def __str__(self) -> str:
+        return str(self.value)
+
+    @classmethod
+    def _missing_(cls, value: object) -> "PortProtocol | None":
+        if isinstance(value, str):
+            upper_value = value.upper()
+            for member in cls:
+                if member.value.upper() == upper_value:
+                    return member
+        return None

blaxel/core/client/models/preview.py

@@ -1,10 +1,8 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..types import UNSET, Unset
-
 if TYPE_CHECKING:
     from ..models.preview_metadata import PreviewMetadata
     from ..models.preview_spec import PreviewSpec
@@ -18,38 +16,33 @@ class Preview:
     """Preview of a Resource
 
     Attributes:
-        metadata (Union[Unset, PreviewMetadata]): PreviewMetadata
-        spec (Union[Unset, PreviewSpec]): Preview of a Resource
+        metadata (PreviewMetadata): PreviewMetadata
+        spec (PreviewSpec): Preview of a Resource
     """
 
-    metadata: Union[Unset, "PreviewMetadata"] = UNSET
-    spec: Union[Unset, "PreviewSpec"] = UNSET
+    metadata: "PreviewMetadata"
+    spec: "PreviewSpec"
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        metadata: Union[Unset, dict[str, Any]] = UNSET
-        if (
-            self.metadata
-            and not isinstance(self.metadata, Unset)
-            and not isinstance(self.metadata, dict)
-        ):
-            metadata = self.metadata.to_dict()
-        elif self.metadata and isinstance(self.metadata, dict):
+        if type(self.metadata) is dict:
             metadata = self.metadata
+        else:
+            metadata = self.metadata.to_dict()
 
-        spec: Union[Unset, dict[str, Any]] = UNSET
-        if self.spec and not isinstance(self.spec, Unset) and not isinstance(self.spec, dict):
-            spec = self.spec.to_dict()
-        elif self.spec and isinstance(self.spec, dict):
+        if type(self.spec) is dict:
             spec = self.spec
+        else:
+            spec = self.spec.to_dict()
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if metadata is not UNSET:
-            field_dict["metadata"] = metadata
-        if spec is not UNSET:
-            field_dict["spec"] = spec
+        field_dict.update(
+            {
+                "metadata": metadata,
+                "spec": spec,
+            }
+        )
 
         return field_dict
 
@@ -61,19 +54,9 @@ class Preview:
         if not src_dict:
             return None
         d = src_dict.copy()
-        _metadata = d.pop("metadata", UNSET)
-        metadata: Union[Unset, PreviewMetadata]
-        if isinstance(_metadata, Unset):
-            metadata = UNSET
-        else:
-            metadata = PreviewMetadata.from_dict(_metadata)
+        metadata = PreviewMetadata.from_dict(d.pop("metadata"))
 
-        _spec = d.pop("spec", UNSET)
-        spec: Union[Unset, PreviewSpec]
-        if isinstance(_spec, Unset):
-            spec = UNSET
-        else:
-            spec = PreviewSpec.from_dict(_spec)
+        spec = PreviewSpec.from_dict(d.pop("spec"))
 
         preview = cls(
             metadata=metadata,