blaxel 0.64.0__py3-none-any.whl

blaxel/models/runtime.py

@@ -0,0 +1,196 @@
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.runtime_startup_probe import RuntimeStartupProbe
+
+
+T = TypeVar("T", bound="Runtime")
+
+
+@_attrs_define
+class Runtime:
+    """Set of configurations for a deployment
+
+    Attributes:
+        args (Union[Unset, list[Any]]): The arguments to pass to the deployment runtime
+        command (Union[Unset, list[Any]]): The command to run the deployment
+        cpu (Union[Unset, int]): The CPU for the deployment in cores, only available for private cluster
+        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 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
+        memory (Union[Unset, int]): The memory for the deployment in MB
+        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.
+        organization (Union[Unset, str]): The organization of the model
+        serving_port (Union[Unset, int]): The port to serve the model on
+        startup_probe (Union[Unset, RuntimeStartupProbe]): The readiness probe. Should be a Kubernetes Probe type
+        type_ (Union[Unset, str]): The type of origin for the deployment (hf_private_endpoint, hf_public_endpoint)
+    """
+
+    args: Union[Unset, list[Any]] = UNSET
+    command: Union[Unset, list[Any]] = UNSET
+    cpu: Union[Unset, int] = UNSET
+    endpoint_name: Union[Unset, str] = UNSET
+    envs: Union[Unset, list[Any]] = UNSET
+    image: Union[Unset, str] = UNSET
+    memory: Union[Unset, int] = UNSET
+    metric_port: Union[Unset, int] = UNSET
+    model: Union[Unset, str] = UNSET
+    organization: Union[Unset, str] = UNSET
+    serving_port: Union[Unset, int] = UNSET
+    startup_probe: Union[Unset, "RuntimeStartupProbe"] = UNSET
+    type_: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        args: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.args, Unset):
+            args = self.args
+
+        command: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.command, Unset):
+            command = self.command
+
+        cpu = self.cpu
+
+        endpoint_name = self.endpoint_name
+
+        envs: Union[Unset, list[Any]] = UNSET
+        if not isinstance(self.envs, Unset):
+            envs = self.envs
+
+        image = self.image
+
+        memory = self.memory
+
+        metric_port = self.metric_port
+
+        model = self.model
+
+        organization = self.organization
+
+        serving_port = self.serving_port
+
+        startup_probe: Union[Unset, dict[str, Any]] = UNSET
+        if (
+            self.startup_probe
+            and not isinstance(self.startup_probe, Unset)
+            and not isinstance(self.startup_probe, dict)
+        ):
+            startup_probe = self.startup_probe.to_dict()
+        elif self.startup_probe and isinstance(self.startup_probe, dict):
+            startup_probe = self.startup_probe
+
+        type_ = self.type_
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if args is not UNSET:
+            field_dict["args"] = args
+        if command is not UNSET:
+            field_dict["command"] = command
+        if cpu is not UNSET:
+            field_dict["cpu"] = cpu
+        if endpoint_name is not UNSET:
+            field_dict["endpointName"] = endpoint_name
+        if envs is not UNSET:
+            field_dict["envs"] = envs
+        if image is not UNSET:
+            field_dict["image"] = image
+        if memory is not UNSET:
+            field_dict["memory"] = memory
+        if metric_port is not UNSET:
+            field_dict["metricPort"] = metric_port
+        if model is not UNSET:
+            field_dict["model"] = model
+        if organization is not UNSET:
+            field_dict["organization"] = organization
+        if serving_port is not UNSET:
+            field_dict["servingPort"] = serving_port
+        if startup_probe is not UNSET:
+            field_dict["startupProbe"] = startup_probe
+        if type_ is not UNSET:
+            field_dict["type"] = type_
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.runtime_startup_probe import RuntimeStartupProbe
+
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        args = cast(list[Any], d.pop("args", UNSET))
+
+        command = cast(list[Any], d.pop("command", UNSET))
+
+        cpu = d.pop("cpu", UNSET)
+
+        endpoint_name = d.pop("endpointName", UNSET)
+
+        envs = cast(list[Any], d.pop("envs", UNSET))
+
+        image = d.pop("image", UNSET)
+
+        memory = d.pop("memory", UNSET)
+
+        metric_port = d.pop("metricPort", UNSET)
+
+        model = d.pop("model", UNSET)
+
+        organization = d.pop("organization", UNSET)
+
+        serving_port = d.pop("servingPort", UNSET)
+
+        _startup_probe = d.pop("startupProbe", UNSET)
+        startup_probe: Union[Unset, RuntimeStartupProbe]
+        if isinstance(_startup_probe, Unset):
+            startup_probe = UNSET
+        else:
+            startup_probe = RuntimeStartupProbe.from_dict(_startup_probe)
+
+        type_ = d.pop("type", UNSET)
+
+        runtime = cls(
+            args=args,
+            command=command,
+            cpu=cpu,
+            endpoint_name=endpoint_name,
+            envs=envs,
+            image=image,
+            memory=memory,
+            metric_port=metric_port,
+            model=model,
+            organization=organization,
+            serving_port=serving_port,
+            startup_probe=startup_probe,
+            type_=type_,
+        )
+
+        runtime.additional_properties = d
+        return runtime
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/models/runtime_startup_probe.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RuntimeStartupProbe")
+
+
+@_attrs_define
+class RuntimeStartupProbe:
+    """The readiness probe. Should be a Kubernetes Probe type"""
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        runtime_startup_probe = cls()
+
+        runtime_startup_probe.additional_properties = d
+        return runtime_startup_probe
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/models/serverless_config.py

@@ -0,0 +1,80 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ServerlessConfig")
+
+
+@_attrs_define
+class ServerlessConfig:
+    """Configuration for a serverless deployment
+
+    Attributes:
+        max_scale (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).
+        min_scale (Union[Unset, int]): The maximum number of replicas for the deployment.
+        timeout (Union[Unset, int]): The timeout for the deployment in seconds
+    """
+
+    max_scale: Union[Unset, int] = UNSET
+    min_scale: Union[Unset, int] = UNSET
+    timeout: Union[Unset, int] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        max_scale = self.max_scale
+
+        min_scale = self.min_scale
+
+        timeout = self.timeout
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if max_scale is not UNSET:
+            field_dict["maxScale"] = max_scale
+        if min_scale is not UNSET:
+            field_dict["minScale"] = min_scale
+        if timeout is not UNSET:
+            field_dict["timeout"] = timeout
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        max_scale = d.pop("maxScale", UNSET)
+
+        min_scale = d.pop("minScale", UNSET)
+
+        timeout = d.pop("timeout", UNSET)
+
+        serverless_config = cls(
+            max_scale=max_scale,
+            min_scale=min_scale,
+            timeout=timeout,
+        )
+
+        serverless_config.additional_properties = d
+        return serverless_config
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/models/spec_configuration.py

@@ -0,0 +1,70 @@
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="SpecConfiguration")
+
+
+@_attrs_define
+class SpecConfiguration:
+    """Configuration, this is a key value storage. In your object you can retrieve the value with config[key]
+
+    Attributes:
+        secret (Union[Unset, bool]): ACconfiguration secret
+        value (Union[Unset, str]): Configuration value
+    """
+
+    secret: Union[Unset, bool] = UNSET
+    value: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        secret = self.secret
+
+        value = self.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if secret is not UNSET:
+            field_dict["secret"] = secret
+        if value is not UNSET:
+            field_dict["value"] = value
+
+        return field_dict
+
+    @classmethod
+    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)
+
+        value = d.pop("value", UNSET)
+
+        spec_configuration = cls(
+            secret=secret,
+            value=value,
+        )
+
+        spec_configuration.additional_properties = d
+        return spec_configuration
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/models/store_agent.py

@@ -0,0 +1,178 @@
+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.store_agent_labels import StoreAgentLabels
+    from ..models.store_configuration import StoreConfiguration
+
+
+T = TypeVar("T", bound="StoreAgent")
+
+
+@_attrs_define
+class StoreAgent:
+    """Store agent
+
+    Attributes:
+        created_at (Union[Unset, str]): The date and time when the resource was created
+        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
+        display_name (Union[Unset, str]): Store agent display name
+        image (Union[Unset, str]): Store agent image
+        labels (Union[Unset, StoreAgentLabels]): Store agent labels
+        name (Union[Unset, str]): Store agent name
+        prompt (Union[Unset, str]): Store agent prompt, this is to define what the agent does
+    """
+
+    created_at: 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
+    display_name: Union[Unset, str] = UNSET
+    image: Union[Unset, str] = UNSET
+    labels: Union[Unset, "StoreAgentLabels"] = UNSET
+    name: Union[Unset, str] = UNSET
+    prompt: 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
+
+        updated_at = self.updated_at
+
+        created_by = self.created_by
+
+        updated_by = self.updated_by
+
+        configuration: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.configuration, Unset):
+            configuration = []
+            for configuration_item_data in self.configuration:
+                configuration_item = configuration_item_data.to_dict()
+                configuration.append(configuration_item)
+
+        description = self.description
+
+        display_name = self.display_name
+
+        image = self.image
+
+        labels: Union[Unset, dict[str, Any]] = 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
+
+        prompt = self.prompt
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if created_at is not UNSET:
+            field_dict["createdAt"] = created_at
+        if updated_at is not UNSET:
+            field_dict["updatedAt"] = updated_at
+        if created_by is not UNSET:
+            field_dict["createdBy"] = created_by
+        if updated_by is not UNSET:
+            field_dict["updatedBy"] = updated_by
+        if configuration is not UNSET:
+            field_dict["configuration"] = configuration
+        if description is not UNSET:
+            field_dict["description"] = description
+        if display_name is not UNSET:
+            field_dict["displayName"] = display_name
+        if image is not UNSET:
+            field_dict["image"] = image
+        if labels is not UNSET:
+            field_dict["labels"] = labels
+        if name is not UNSET:
+            field_dict["name"] = name
+        if prompt is not UNSET:
+            field_dict["prompt"] = prompt
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.store_agent_labels import StoreAgentLabels
+        from ..models.store_configuration import StoreConfiguration
+
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        created_at = d.pop("createdAt", UNSET)
+
+        updated_at = d.pop("updatedAt", UNSET)
+
+        created_by = d.pop("createdBy", UNSET)
+
+        updated_by = d.pop("updatedBy", UNSET)
+
+        configuration = []
+        _configuration = d.pop("configuration", UNSET)
+        for configuration_item_data in _configuration or []:
+            configuration_item = StoreConfiguration.from_dict(configuration_item_data)
+
+            configuration.append(configuration_item)
+
+        description = d.pop("description", UNSET)
+
+        display_name = d.pop("displayName", UNSET)
+
+        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)
+
+        name = d.pop("name", UNSET)
+
+        prompt = d.pop("prompt", UNSET)
+
+        store_agent = cls(
+            created_at=created_at,
+            updated_at=updated_at,
+            created_by=created_by,
+            updated_by=updated_by,
+            configuration=configuration,
+            description=description,
+            display_name=display_name,
+            image=image,
+            labels=labels,
+            name=name,
+            prompt=prompt,
+        )
+
+        store_agent.additional_properties = d
+        return store_agent
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

blaxel/models/store_agent_labels.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="StoreAgentLabels")
+
+
+@_attrs_define
+class StoreAgentLabels:
+    """Store agent labels"""
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        store_agent_labels = cls()
+
+        store_agent_labels.additional_properties = d
+        return store_agent_labels
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties