anyscale 0.26.64__py3-none-any.whl → 0.26.65__py3-none-any.whl

anyscale/client/openapi_client/models/__init__.py

@@ -149,6 +149,7 @@ from openapi_client.models.compute_template_config import ComputeTemplateConfig
 from openapi_client.models.compute_template_query import ComputeTemplateQuery
 from openapi_client.models.computetemplate_response import ComputetemplateResponse
 from openapi_client.models.computetemplateconfig_response import ComputetemplateconfigResponse
+from openapi_client.models.connection_type import ConnectionType
 from openapi_client.models.create_aioa_cloud_waitlist import CreateAioaCloudWaitlist
 from openapi_client.models.create_analytics_event import CreateAnalyticsEvent
 from openapi_client.models.create_app_config import CreateAppConfig
@@ -204,6 +205,18 @@ from openapi_client.models.credit_type import CreditType
 from openapi_client.models.credits_v2 import CreditsV2
 from openapi_client.models.customer_alert_status import CustomerAlertStatus
 from openapi_client.models.dashboard_panel import DashboardPanel
+from openapi_client.models.data_catalog import DataCatalog
+from openapi_client.models.data_catalog_connection import DataCatalogConnection
+from openapi_client.models.data_catalog_connection_status import DataCatalogConnectionStatus
+from openapi_client.models.data_catalog_provider import DataCatalogProvider
+from openapi_client.models.databricks_connection_config import DatabricksConnectionConfig
+from openapi_client.models.databricks_connection_info import DatabricksConnectionInfo
+from openapi_client.models.databricks_connection_response import DatabricksConnectionResponse
+from openapi_client.models.databricks_register_request import DatabricksRegisterRequest
+from openapi_client.models.databricksconnectioninfo_response import DatabricksconnectioninfoResponse
+from openapi_client.models.databricksconnectionresponse_response import DatabricksconnectionresponseResponse
+from openapi_client.models.datacatalog_list_response import DatacatalogListResponse
+from openapi_client.models.datacatalogconnection_list_response import DatacatalogconnectionListResponse
 from openapi_client.models.dataplane_services import DataplaneServices
 from openapi_client.models.dataset import Dataset
 from openapi_client.models.dataset_dag import DatasetDag
@@ -283,6 +296,7 @@ from openapi_client.models.describemachinepoolresponse_response import Describem
 from openapi_client.models.describesystemworkloadresponse_response import DescribesystemworkloadresponseResponse
 from openapi_client.models.detach_machine_pool_from_cloud_request import DetachMachinePoolFromCloudRequest
 from openapi_client.models.detachmachinepoolfromcloudresponse_response import DetachmachinepoolfromcloudresponseResponse
+from openapi_client.models.domain_verification import DomainVerification
 from openapi_client.models.editable_cloud_resource import EditableCloudResource
 from openapi_client.models.editable_cloud_resource_gcp import EditableCloudResourceGCP
 from openapi_client.models.email_verification_request import EmailVerificationRequest
@@ -359,6 +373,7 @@ from openapi_client.models.kubernetes_manager_registration_response import Kuber
 from openapi_client.models.kubernetesmanagerregistrationresponse_response import KubernetesmanagerregistrationresponseResponse
 from openapi_client.models.lb_resource import LBResource
 from openapi_client.models.lbresource_response import LbresourceResponse
+from openapi_client.models.list_databricks_connections import ListDatabricksConnections
 from openapi_client.models.list_machine_pools_response import ListMachinePoolsResponse
 from openapi_client.models.list_machines_response import ListMachinesResponse
 from openapi_client.models.list_ray_sessions_response import ListRaySessionsResponse
@@ -429,6 +444,9 @@ from openapi_client.models.nodemetricsresponse_response import Nodemetricsrespon
 from openapi_client.models.notification_channel_email_config import NotificationChannelEmailConfig
 from openapi_client.models.notification_channel_slack_config import NotificationChannelSlackConfig
 from openapi_client.models.notification_channel_webhook_config import NotificationChannelWebhookConfig
+from openapi_client.models.o_auth_connection_response import OAuthConnectionResponse
+from openapi_client.models.oauth_auth_url_response import OauthAuthUrlResponse
+from openapi_client.models.oauthconnectionresponse_response import OauthconnectionresponseResponse
 from openapi_client.models.object_storage import ObjectStorage
 from openapi_client.models.operation_error import OperationError
 from openapi_client.models.operation_progress import OperationProgress
@@ -525,6 +543,9 @@ from openapi_client.models.rollout_strategy import RolloutStrategy
 from openapi_client.models.run_attempt_status import RunAttemptStatus
 from openapi_client.models.run_status import RunStatus
 from openapi_client.models.s3_download_location import S3DownloadLocation
+from openapi_client.models.sso_config import SSOConfig
+from openapi_client.models.sso_connection import SSOConnection
+from openapi_client.models.sso_connection_state import SSOConnectionState
 from openapi_client.models.sso_login_info import SSOLoginInfo
 from openapi_client.models.sso_mode import SSOMode
 from openapi_client.models.supportedbaseimagesenum import SUPPORTEDBASEIMAGESENUM
@@ -568,6 +589,7 @@ from openapi_client.models.sessionsshkey_response import SessionsshkeyResponse
 from openapi_client.models.show_otp_source_return_api_model import ShowOTPSourceReturnApiModel
 from openapi_client.models.showotpsourcereturnapimodel_response import ShowotpsourcereturnapimodelResponse
 from openapi_client.models.sort_order import SortOrder
+from openapi_client.models.ssoconfig_response import SsoconfigResponse
 from openapi_client.models.ssologininfo_response import SsologininfoResponse
 from openapi_client.models.start_session_options import StartSessionOptions
 from openapi_client.models.stop_session_options import StopSessionOptions
@@ -625,17 +647,20 @@ from openapi_client.models.update_organization_collaborator import UpdateOrganiz
 from openapi_client.models.update_payment_info import UpdatePaymentInfo
 from openapi_client.models.update_project_collaborator import UpdateProjectCollaborator
 from openapi_client.models.update_resource_quota import UpdateResourceQuota
+from openapi_client.models.update_workspace_template import UpdateWorkspaceTemplate
 from openapi_client.models.updatemachinepoolresponse_response import UpdatemachinepoolresponseResponse
 from openapi_client.models.upload_session_command_logs_locations import UploadSessionCommandLogsLocations
 from openapi_client.models.uploadsessioncommandlogslocations_response import UploadsessioncommandlogslocationsResponse
 from openapi_client.models.upsert_resource_tags_request import UpsertResourceTagsRequest
 from openapi_client.models.usage_by_cloud import UsageByCloud
 from openapi_client.models.usage_by_cluster import UsageByCluster
+from openapi_client.models.usage_by_cluster_type import UsageByClusterType
 from openapi_client.models.usage_by_instance_type import UsageByInstanceType
 from openapi_client.models.usage_by_project import UsageByProject
 from openapi_client.models.usage_by_user import UsageByUser
 from openapi_client.models.usagebycloud_list_response import UsagebycloudListResponse
 from openapi_client.models.usagebycluster_list_response import UsagebyclusterListResponse
+from openapi_client.models.usagebyclustertype_list_response import UsagebyclustertypeListResponse
 from openapi_client.models.usagebyinstancetype_list_response import UsagebyinstancetypeListResponse
 from openapi_client.models.usagebyproject_list_response import UsagebyprojectListResponse
 from openapi_client.models.usagebyuser_list_response import UsagebyuserListResponse
@@ -648,6 +673,7 @@ from openapi_client.models.useworkosresponse_response import UseworkosresponseRe
 from openapi_client.models.utm_fields import UtmFields
 from openapi_client.models.validate_otp_params_api_model import ValidateOTPParamsApiModel
 from openapi_client.models.validation_error import ValidationError
+from openapi_client.models.validation_status import ValidationStatus
 from openapi_client.models.waitlist_status_response import WaitlistStatusResponse
 from openapi_client.models.waitlist_status_type import WaitlistStatusType
 from openapi_client.models.waitliststatusresponse_response import WaitliststatusresponseResponse

anyscale/client/openapi_client/models/connection_type.py

@@ -0,0 +1,99 @@
+# 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 ConnectionType(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    DATABRICKS_U2M = "databricks_U2M"
+
+    allowable_values = [DATABRICKS_U2M]  # noqa: E501
+
+    """
+    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 = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self, local_vars_configuration=None):  # noqa: E501
+        """ConnectionType - 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.discriminator = None
+
+    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, ConnectionType):
+            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, ConnectionType):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/create_experimental_workspace.py

@@ -44,6 +44,7 @@ class CreateExperimentalWorkspace(object):
         'cloned_job_id': 'str',
         'cloned_workspace_id': 'str',
         'template_id': 'str',
+        'template_version_id': 'str',
         'template_url': 'str',
         'tags': 'dict(str, str)',
         'skip_start': 'bool'
@@ -61,12 +62,13 @@ class CreateExperimentalWorkspace(object):
         'cloned_job_id': 'cloned_job_id',
         'cloned_workspace_id': 'cloned_workspace_id',
         'template_id': 'template_id',
+        'template_version_id': 'template_version_id',
         'template_url': 'template_url',
         'tags': 'tags',
         'skip_start': 'skip_start'
     }
 
-    def __init__(self, name=None, description=None, project_id=None, cloud_id=None, compute_config_id=None, base_snapshot=None, cluster_environment_build_id=None, idle_timeout_minutes=None, cloned_job_id=None, cloned_workspace_id=None, template_id=None, template_url=None, tags=None, skip_start=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, description=None, project_id=None, cloud_id=None, compute_config_id=None, base_snapshot=None, cluster_environment_build_id=None, idle_timeout_minutes=None, cloned_job_id=None, cloned_workspace_id=None, template_id=None, template_version_id=None, template_url=None, tags=None, skip_start=None, local_vars_configuration=None):  # noqa: E501
         """CreateExperimentalWorkspace - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -83,6 +85,7 @@ class CreateExperimentalWorkspace(object):
         self._cloned_job_id = None
         self._cloned_workspace_id = None
         self._template_id = None
+        self._template_version_id = None
         self._template_url = None
         self._tags = None
         self._skip_start = None
@@ -105,6 +108,8 @@ class CreateExperimentalWorkspace(object):
             self.cloned_workspace_id = cloned_workspace_id
         if template_id is not None:
             self.template_id = template_id
+        if template_version_id is not None:
+            self.template_version_id = template_version_id
         if template_url is not None:
             self.template_url = template_url
         if tags is not None:
@@ -375,6 +380,29 @@ class CreateExperimentalWorkspace(object):
 
         self._template_id = template_id
 
+    @property
+    def template_version_id(self):
+        """Gets the template_version_id of this CreateExperimentalWorkspace.  # noqa: E501
+
+        The Id of the template version to use.  # noqa: E501
+
+        :return: The template_version_id of this CreateExperimentalWorkspace.  # noqa: E501
+        :rtype: str
+        """
+        return self._template_version_id
+
+    @template_version_id.setter
+    def template_version_id(self, template_version_id):
+        """Sets the template_version_id of this CreateExperimentalWorkspace.
+
+        The Id of the template version to use.  # noqa: E501
+
+        :param template_version_id: The template_version_id of this CreateExperimentalWorkspace.  # noqa: E501
+        :type: str
+        """
+
+        self._template_version_id = template_version_id
+
     @property
     def template_url(self):
         """Gets the template_url of this CreateExperimentalWorkspace.  # noqa: E501

anyscale/client/openapi_client/models/data_catalog.py

@@ -0,0 +1,281 @@
+# 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 DataCatalog(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 = {
+        'id': 'str',
+        'name': 'str',
+        'connection_id': 'str',
+        'description': 'str',
+        'created_at': 'str',
+        'updated_at': 'str',
+        'metadata': 'object'
+    }
+
+    attribute_map = {
+        'id': 'id',
+        'name': 'name',
+        'connection_id': 'connection_id',
+        'description': 'description',
+        'created_at': 'created_at',
+        'updated_at': 'updated_at',
+        'metadata': 'metadata'
+    }
+
+    def __init__(self, id=None, name=None, connection_id=None, description=None, created_at=None, updated_at=None, metadata=None, local_vars_configuration=None):  # noqa: E501
+        """DataCatalog - 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._id = None
+        self._name = None
+        self._connection_id = None
+        self._description = None
+        self._created_at = None
+        self._updated_at = None
+        self._metadata = None
+        self.discriminator = None
+
+        self.id = id
+        self.name = name
+        self.connection_id = connection_id
+        if description is not None:
+            self.description = description
+        self.created_at = created_at
+        self.updated_at = updated_at
+        if metadata is not None:
+            self.metadata = metadata
+
+    @property
+    def id(self):
+        """Gets the id of this DataCatalog.  # noqa: E501
+
+
+        :return: The id of this DataCatalog.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id):
+        """Sets the id of this DataCatalog.
+
+
+        :param id: The id of this DataCatalog.  # 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
+
+    @property
+    def name(self):
+        """Gets the name of this DataCatalog.  # noqa: E501
+
+
+        :return: The name of this DataCatalog.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name):
+        """Sets the name of this DataCatalog.
+
+
+        :param name: The name of this DataCatalog.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and name is None:  # noqa: E501
+            raise ValueError("Invalid value for `name`, must not be `None`")  # noqa: E501
+
+        self._name = name
+
+    @property
+    def connection_id(self):
+        """Gets the connection_id of this DataCatalog.  # noqa: E501
+
+
+        :return: The connection_id of this DataCatalog.  # noqa: E501
+        :rtype: str
+        """
+        return self._connection_id
+
+    @connection_id.setter
+    def connection_id(self, connection_id):
+        """Sets the connection_id of this DataCatalog.
+
+
+        :param connection_id: The connection_id of this DataCatalog.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and connection_id is None:  # noqa: E501
+            raise ValueError("Invalid value for `connection_id`, must not be `None`")  # noqa: E501
+
+        self._connection_id = connection_id
+
+    @property
+    def description(self):
+        """Gets the description of this DataCatalog.  # noqa: E501
+
+
+        :return: The description of this DataCatalog.  # noqa: E501
+        :rtype: str
+        """
+        return self._description
+
+    @description.setter
+    def description(self, description):
+        """Sets the description of this DataCatalog.
+
+
+        :param description: The description of this DataCatalog.  # noqa: E501
+        :type: str
+        """
+
+        self._description = description
+
+    @property
+    def created_at(self):
+        """Gets the created_at of this DataCatalog.  # noqa: E501
+
+
+        :return: The created_at of this DataCatalog.  # noqa: E501
+        :rtype: str
+        """
+        return self._created_at
+
+    @created_at.setter
+    def created_at(self, created_at):
+        """Sets the created_at of this DataCatalog.
+
+
+        :param created_at: The created_at of this DataCatalog.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and created_at is None:  # noqa: E501
+            raise ValueError("Invalid value for `created_at`, must not be `None`")  # noqa: E501
+
+        self._created_at = created_at
+
+    @property
+    def updated_at(self):
+        """Gets the updated_at of this DataCatalog.  # noqa: E501
+
+
+        :return: The updated_at of this DataCatalog.  # noqa: E501
+        :rtype: str
+        """
+        return self._updated_at
+
+    @updated_at.setter
+    def updated_at(self, updated_at):
+        """Sets the updated_at of this DataCatalog.
+
+
+        :param updated_at: The updated_at of this DataCatalog.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and updated_at is None:  # noqa: E501
+            raise ValueError("Invalid value for `updated_at`, must not be `None`")  # noqa: E501
+
+        self._updated_at = updated_at
+
+    @property
+    def metadata(self):
+        """Gets the metadata of this DataCatalog.  # noqa: E501
+
+
+        :return: The metadata of this DataCatalog.  # noqa: E501
+        :rtype: object
+        """
+        return self._metadata
+
+    @metadata.setter
+    def metadata(self, metadata):
+        """Sets the metadata of this DataCatalog.
+
+
+        :param metadata: The metadata of this DataCatalog.  # noqa: E501
+        :type: object
+        """
+
+        self._metadata = metadata
+
+    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, DataCatalog):
+            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, DataCatalog):
+            return True
+
+        return self.to_dict() != other.to_dict()