anyscale 0.24.88__py3-none-any.whl → 0.25.5__py3-none-any.whl

anyscale/client/openapi_client/models/session_event.py

@@ -1,267 +0,0 @@
-# coding: utf-8
-
-"""
-    Managed Ray API
-
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
-
-    The version of the OpenAPI document: 0.1.0
-    Generated by: https://openapi-generator.tech
-"""
-
-
-import pprint
-import re  # noqa: F401
-
-import six
-
-from openapi_client.configuration import Configuration
-
-
-class SessionEvent(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
-
-    """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    openapi_types = {
-        'event_type': 'str',
-        'log_level': 'LogLevelTypes',
-        'timestamp': 'datetime',
-        'description': 'str',
-        'cause': 'SessionEventCause',
-        'id': 'str'
-    }
-
-    attribute_map = {
-        'event_type': 'event_type',
-        'log_level': 'log_level',
-        'timestamp': 'timestamp',
-        'description': 'description',
-        'cause': 'cause',
-        'id': 'id'
-    }
-
-    def __init__(self, event_type=None, log_level=None, timestamp=None, description=None, cause=None, id=None, local_vars_configuration=None):  # noqa: E501
-        """SessionEvent - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._event_type = None
-        self._log_level = None
-        self._timestamp = None
-        self._description = None
-        self._cause = None
-        self._id = None
-        self.discriminator = None
-
-        self.event_type = event_type
-        self.log_level = log_level
-        self.timestamp = timestamp
-        self.description = description
-        if cause is not None:
-            self.cause = cause
-        self.id = id
-
-    @property
-    def event_type(self):
-        """Gets the event_type of this SessionEvent.  # noqa: E501
-
-        The category of session event.  # noqa: E501
-
-        :return: The event_type of this SessionEvent.  # noqa: E501
-        :rtype: str
-        """
-        return self._event_type
-
-    @event_type.setter
-    def event_type(self, event_type):
-        """Sets the event_type of this SessionEvent.
-
-        The category of session event.  # noqa: E501
-
-        :param event_type: The event_type of this SessionEvent.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and event_type is None:  # noqa: E501
-            raise ValueError("Invalid value for `event_type`, must not be `None`")  # noqa: E501
-
-        self._event_type = event_type
-
-    @property
-    def log_level(self):
-        """Gets the log_level of this SessionEvent.  # noqa: E501
-
-        The severity of the session event.  # noqa: E501
-
-        :return: The log_level of this SessionEvent.  # noqa: E501
-        :rtype: LogLevelTypes
-        """
-        return self._log_level
-
-    @log_level.setter
-    def log_level(self, log_level):
-        """Sets the log_level of this SessionEvent.
-
-        The severity of the session event.  # noqa: E501
-
-        :param log_level: The log_level of this SessionEvent.  # noqa: E501
-        :type: LogLevelTypes
-        """
-        if self.local_vars_configuration.client_side_validation and log_level is None:  # noqa: E501
-            raise ValueError("Invalid value for `log_level`, must not be `None`")  # noqa: E501
-
-        self._log_level = log_level
-
-    @property
-    def timestamp(self):
-        """Gets the timestamp of this SessionEvent.  # noqa: E501
-
-        The time at which the session event occurred.  # noqa: E501
-
-        :return: The timestamp of this SessionEvent.  # noqa: E501
-        :rtype: datetime
-        """
-        return self._timestamp
-
-    @timestamp.setter
-    def timestamp(self, timestamp):
-        """Sets the timestamp of this SessionEvent.
-
-        The time at which the session event occurred.  # noqa: E501
-
-        :param timestamp: The timestamp of this SessionEvent.  # noqa: E501
-        :type: datetime
-        """
-        if self.local_vars_configuration.client_side_validation and timestamp is None:  # noqa: E501
-            raise ValueError("Invalid value for `timestamp`, must not be `None`")  # noqa: E501
-
-        self._timestamp = timestamp
-
-    @property
-    def description(self):
-        """Gets the description of this SessionEvent.  # noqa: E501
-
-        A human readable description of the session event.  # noqa: E501
-
-        :return: The description of this SessionEvent.  # noqa: E501
-        :rtype: str
-        """
-        return self._description
-
-    @description.setter
-    def description(self, description):
-        """Sets the description of this SessionEvent.
-
-        A human readable description of the session event.  # noqa: E501
-
-        :param description: The description of this SessionEvent.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and description is None:  # noqa: E501
-            raise ValueError("Invalid value for `description`, must not be `None`")  # noqa: E501
-
-        self._description = description
-
-    @property
-    def cause(self):
-        """Gets the cause of this SessionEvent.  # noqa: E501
-
-        The reason why the session event occurred. None indicates an unknown cause.  # noqa: E501
-
-        :return: The cause of this SessionEvent.  # noqa: E501
-        :rtype: SessionEventCause
-        """
-        return self._cause
-
-    @cause.setter
-    def cause(self, cause):
-        """Sets the cause of this SessionEvent.
-
-        The reason why the session event occurred. None indicates an unknown cause.  # noqa: E501
-
-        :param cause: The cause of this SessionEvent.  # noqa: E501
-        :type: SessionEventCause
-        """
-
-        self._cause = cause
-
-    @property
-    def id(self):
-        """Gets the id of this SessionEvent.  # noqa: E501
-
-        A unique identifier for the session event.  # noqa: E501
-
-        :return: The id of this SessionEvent.  # noqa: E501
-        :rtype: str
-        """
-        return self._id
-
-    @id.setter
-    def id(self, id):
-        """Sets the id of this SessionEvent.
-
-        A unique identifier for the session event.  # noqa: E501
-
-        :param id: The id of this SessionEvent.  # noqa: E501
-        :type: str
-        """
-        if self.local_vars_configuration.client_side_validation and id is None:  # noqa: E501
-            raise ValueError("Invalid value for `id`, must not be `None`")  # noqa: E501
-
-        self._id = id
-
-    def to_dict(self):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, SessionEvent):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, SessionEvent):
-            return True
-
-        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/session_event_cause.py

@@ -1,150 +0,0 @@
-# coding: utf-8
-
-"""
-    Managed Ray API
-
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
-
-    The version of the OpenAPI document: 0.1.0
-    Generated by: https://openapi-generator.tech
-"""
-
-
-import pprint
-import re  # noqa: F401
-
-import six
-
-from openapi_client.configuration import Configuration
-
-
-class SessionEventCause(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
-
-    """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    openapi_types = {
-        'cause_user': 'str',
-        'cause_system': 'str'
-    }
-
-    attribute_map = {
-        'cause_user': 'cause_user',
-        'cause_system': 'cause_system'
-    }
-
-    def __init__(self, cause_user=None, cause_system=None, local_vars_configuration=None):  # noqa: E501
-        """SessionEventCause - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._cause_user = None
-        self._cause_system = None
-        self.discriminator = None
-
-        if cause_user is not None:
-            self.cause_user = cause_user
-        if cause_system is not None:
-            self.cause_system = cause_system
-
-    @property
-    def cause_user(self):
-        """Gets the cause_user of this SessionEventCause.  # noqa: E501
-
-        The username of the user who caused the session event.  # noqa: E501
-
-        :return: The cause_user of this SessionEventCause.  # noqa: E501
-        :rtype: str
-        """
-        return self._cause_user
-
-    @cause_user.setter
-    def cause_user(self, cause_user):
-        """Sets the cause_user of this SessionEventCause.
-
-        The username of the user who caused the session event.  # noqa: E501
-
-        :param cause_user: The cause_user of this SessionEventCause.  # noqa: E501
-        :type: str
-        """
-
-        self._cause_user = cause_user
-
-    @property
-    def cause_system(self):
-        """Gets the cause_system of this SessionEventCause.  # noqa: E501
-
-        The name of the internal anyscale system that caused the session event.  # noqa: E501
-
-        :return: The cause_system of this SessionEventCause.  # noqa: E501
-        :rtype: str
-        """
-        return self._cause_system
-
-    @cause_system.setter
-    def cause_system(self, cause_system):
-        """Sets the cause_system of this SessionEventCause.
-
-        The name of the internal anyscale system that caused the session event.  # noqa: E501
-
-        :param cause_system: The cause_system of this SessionEventCause.  # noqa: E501
-        :type: str
-        """
-
-        self._cause_system = cause_system
-
-    def to_dict(self):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, SessionEventCause):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, SessionEventCause):
-            return True
-
-        return self.to_dict() != other.to_dict()

anyscale/controllers/resource_quota_controller.py

@@ -1,183 +0,0 @@
-from typing import Dict, List, Optional
-
-import tabulate
-
-from anyscale.cli_logger import LogsLogger
-from anyscale.client.openapi_client.models.create_resource_quota import (
-    CreateResourceQuota,
-)
-from anyscale.client.openapi_client.models.list_resource_quotas_query import (
-    ListResourceQuotasQuery,
-)
-from anyscale.client.openapi_client.models.quota import Quota
-from anyscale.client.openapi_client.models.resource_quota import ResourceQuota
-from anyscale.client.openapi_client.models.resource_quota_status import (
-    ResourceQuotaStatus,
-)
-from anyscale.cloud import get_cloud_id_and_name
-from anyscale.controllers.base_controller import BaseController
-from anyscale.project import get_project_id_for_cloud_from_name
-from anyscale.sdk.anyscale_client.models.page_query import PageQuery
-from anyscale.sdk.anyscale_client.models.text_query import TextQuery
-from anyscale.utils.user_utils import get_user_id_by_email
-
-
-class ResourceQuotaController(BaseController):
-    def __init__(self):
-        super().__init__()
-        self.log = LogsLogger()
-
-    def create(  # noqa: PLR0913
-        self,
-        name: str,
-        cloud: str,
-        project: Optional[str] = None,
-        user_email: Optional[str] = None,
-        num_cpus: Optional[int] = None,
-        num_instances: Optional[int] = None,
-        num_gpus: Optional[int] = None,
-        num_accelerators: Optional[Dict[str, int]] = None,
-    ) -> str:
-
-        cloud_id, _ = get_cloud_id_and_name(self.api_client, cloud_name=cloud)
-        project_id = (
-            get_project_id_for_cloud_from_name(
-                project, cloud_id, self.api_client, self.anyscale_api_client
-            )
-            if project
-            else None
-        )
-        user_id = (
-            get_user_id_by_email(self.api_client, user_email) if user_email else None
-        )
-
-        create_resource_quota = CreateResourceQuota(
-            name=name,
-            cloud_id=cloud_id,
-            project_id=project_id,
-            user_id=user_id,
-            quota=Quota(
-                num_cpus=num_cpus,
-                num_instances=num_instances,
-                num_gpus=num_gpus,
-                num_accelerators=num_accelerators,
-            ),
-        )
-
-        with self.log.spinner("Creating resource quota..."):
-            resource_quota_id = self.api_client.create_resource_quota_api_v2_resource_quotas_post(
-                create_resource_quota
-            ).result.id
-
-        create_resource_quota_message = [f"Name: {name}\nCloud name: {cloud}"]
-        if project:
-            create_resource_quota_message.append(f"Project name: {project}")
-        if user_email:
-            create_resource_quota_message.append(f"User email: {user_email}")
-        if num_cpus:
-            create_resource_quota_message.append(f"Number of CPUs: {num_cpus}")
-        if num_instances:
-            create_resource_quota_message.append(
-                f"Number of instances: {num_instances}"
-            )
-        if num_gpus:
-            create_resource_quota_message.append(f"Number of GPUs: {num_gpus}")
-        if num_accelerators:
-            create_resource_quota_message.append(
-                f"Number of accelerators: {dict(num_accelerators)}"
-            )
-
-        self.log.info("\n".join(create_resource_quota_message))
-        self.log.info(f"Resource quota created successfully ID: {resource_quota_id}")
-
-        return resource_quota_id
-
-    def _format_resource_quotas(self, resource_quotas: List[ResourceQuota]) -> str:
-        table_rows = []
-        for resource_quota in resource_quotas:
-            table_rows.append(
-                [
-                    resource_quota.id,
-                    resource_quota.name,
-                    resource_quota.cloud_id,
-                    resource_quota.project_id,
-                    resource_quota.user_id,
-                    resource_quota.is_enabled,
-                    resource_quota.created_at.strftime("%m/%d/%Y"),
-                    resource_quota.deleted_at.strftime("%m/%d/%Y")
-                    if resource_quota.deleted_at
-                    else None,
-                    resource_quota.quota,
-                ]
-            )
-        table = tabulate.tabulate(
-            table_rows,
-            headers=[
-                "ID",
-                "NAME",
-                "CLOUD ID",
-                "PROJECT ID",
-                "USER ID",
-                "IS ENABLED",
-                "CREATED AT",
-                "DELETED AT",
-                "QUOTA",
-            ],
-            tablefmt="plain",
-        )
-
-        return f"Resource quotas:\n{table}"
-
-    def list_resource_quotas(
-        self,
-        *,
-        name: Optional[str] = None,
-        cloud: Optional[str] = None,
-        creator_id: Optional[str] = None,
-        is_enabled: Optional[bool] = None,
-        max_items: int = 20,
-    ) -> List[ResourceQuota]:
-        cloud_id, _ = (
-            get_cloud_id_and_name(self.api_client, cloud_name=cloud)
-            if cloud
-            else (None, None)
-        )
-
-        resource_quotas = self.api_client.search_resource_quotas_api_v2_resource_quotas_search_post(
-            ListResourceQuotasQuery(
-                name=TextQuery(equals=name) if name else None,
-                cloud_id=cloud_id,
-                creator_id=creator_id,
-                is_enabled=is_enabled,
-                paging=PageQuery(count=max_items),
-            )
-        ).results
-
-        print(self._format_resource_quotas(resource_quotas))
-        return resource_quotas
-
-    def delete(self, resource_quota_id: str,) -> None:
-        """
-        Delete a resource quota.
-        """
-        with self.log.spinner("Deleting resource quota..."):
-            self.api_client.delete_resource_quota_api_v2_resource_quotas_resource_quota_id_delete(
-                resource_quota_id
-            )
-
-        self.log.info(
-            f"Resource quota with ID {resource_quota_id} deleted successfully."
-        )
-
-    def set_status(self, resource_quota_id: str, *, is_enabled: bool,) -> None:
-        """
-        Set the status of a resource quota.
-        """
-        with self.log.spinner("Setting resource quota status..."):
-            self.api_client.set_resource_quota_status_api_v2_resource_quotas_resource_quota_id_status_patch(
-                resource_quota_id, ResourceQuotaStatus(is_enabled=is_enabled)
-            )
-
-        self.log.info(
-            f"{'Enabled' if is_enabled else 'Disabled'} resource quota with ID {resource_quota_id} successfully."
-        )