anyscale 0.26.66__py3-none-any.whl → 0.26.68__py3-none-any.whl

anyscale/client/openapi_client/models/lineageartifact_response.py

@@ -0,0 +1,121 @@
+# 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 LineageartifactResponse(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 = {
+        'result': 'LineageArtifact'
+    }
+
+    attribute_map = {
+        'result': 'result'
+    }
+
+    def __init__(self, result=None, local_vars_configuration=None):  # noqa: E501
+        """LineageartifactResponse - 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._result = None
+        self.discriminator = None
+
+        self.result = result
+
+    @property
+    def result(self):
+        """Gets the result of this LineageartifactResponse.  # noqa: E501
+
+
+        :return: The result of this LineageartifactResponse.  # noqa: E501
+        :rtype: LineageArtifact
+        """
+        return self._result
+
+    @result.setter
+    def result(self, result):
+        """Sets the result of this LineageartifactResponse.
+
+
+        :param result: The result of this LineageartifactResponse.  # noqa: E501
+        :type: LineageArtifact
+        """
+        if self.local_vars_configuration.client_side_validation and result is None:  # noqa: E501
+            raise ValueError("Invalid value for `result`, must not be `None`")  # noqa: E501
+
+        self._result = result
+
+    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, LineageartifactResponse):
+            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, LineageartifactResponse):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/lineagegraph_response.py

@@ -0,0 +1,121 @@
+# 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 LineagegraphResponse(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 = {
+        'result': 'LineageGraph'
+    }
+
+    attribute_map = {
+        'result': 'result'
+    }
+
+    def __init__(self, result=None, local_vars_configuration=None):  # noqa: E501
+        """LineagegraphResponse - 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._result = None
+        self.discriminator = None
+
+        self.result = result
+
+    @property
+    def result(self):
+        """Gets the result of this LineagegraphResponse.  # noqa: E501
+
+
+        :return: The result of this LineagegraphResponse.  # noqa: E501
+        :rtype: LineageGraph
+        """
+        return self._result
+
+    @result.setter
+    def result(self, result):
+        """Sets the result of this LineagegraphResponse.
+
+
+        :param result: The result of this LineagegraphResponse.  # noqa: E501
+        :type: LineageGraph
+        """
+        if self.local_vars_configuration.client_side_validation and result is None:  # noqa: E501
+            raise ValueError("Invalid value for `result`, must not be `None`")  # noqa: E501
+
+        self._result = result
+
+    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, LineagegraphResponse):
+            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, LineagegraphResponse):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/lineageworkload_list_response.py

@@ -0,0 +1,147 @@
+# 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 LineageworkloadListResponse(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 = {
+        'results': 'list[LineageWorkload]',
+        'metadata': 'ListResponseMetadata'
+    }
+
+    attribute_map = {
+        'results': 'results',
+        'metadata': 'metadata'
+    }
+
+    def __init__(self, results=None, metadata=None, local_vars_configuration=None):  # noqa: E501
+        """LineageworkloadListResponse - 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._results = None
+        self._metadata = None
+        self.discriminator = None
+
+        self.results = results
+        if metadata is not None:
+            self.metadata = metadata
+
+    @property
+    def results(self):
+        """Gets the results of this LineageworkloadListResponse.  # noqa: E501
+
+
+        :return: The results of this LineageworkloadListResponse.  # noqa: E501
+        :rtype: list[LineageWorkload]
+        """
+        return self._results
+
+    @results.setter
+    def results(self, results):
+        """Sets the results of this LineageworkloadListResponse.
+
+
+        :param results: The results of this LineageworkloadListResponse.  # noqa: E501
+        :type: list[LineageWorkload]
+        """
+        if self.local_vars_configuration.client_side_validation and results is None:  # noqa: E501
+            raise ValueError("Invalid value for `results`, must not be `None`")  # noqa: E501
+
+        self._results = results
+
+    @property
+    def metadata(self):
+        """Gets the metadata of this LineageworkloadListResponse.  # noqa: E501
+
+
+        :return: The metadata of this LineageworkloadListResponse.  # noqa: E501
+        :rtype: ListResponseMetadata
+        """
+        return self._metadata
+
+    @metadata.setter
+    def metadata(self, metadata):
+        """Sets the metadata of this LineageworkloadListResponse.
+
+
+        :param metadata: The metadata of this LineageworkloadListResponse.  # noqa: E501
+        :type: ListResponseMetadata
+        """
+
+        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, LineageworkloadListResponse):
+            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, LineageworkloadListResponse):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/lineageworkload_response.py

@@ -0,0 +1,121 @@
+# 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 LineageworkloadResponse(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 = {
+        'result': 'LineageWorkload'
+    }
+
+    attribute_map = {
+        'result': 'result'
+    }
+
+    def __init__(self, result=None, local_vars_configuration=None):  # noqa: E501
+        """LineageworkloadResponse - 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._result = None
+        self.discriminator = None
+
+        self.result = result
+
+    @property
+    def result(self):
+        """Gets the result of this LineageworkloadResponse.  # noqa: E501
+
+
+        :return: The result of this LineageworkloadResponse.  # noqa: E501
+        :rtype: LineageWorkload
+        """
+        return self._result
+
+    @result.setter
+    def result(self, result):
+        """Sets the result of this LineageworkloadResponse.
+
+
+        :param result: The result of this LineageworkloadResponse.  # noqa: E501
+        :type: LineageWorkload
+        """
+        if self.local_vars_configuration.client_side_validation and result is None:  # noqa: E501
+            raise ValueError("Invalid value for `result`, must not be `None`")  # noqa: E501
+
+        self._result = result
+
+    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, LineageworkloadResponse):
+            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, LineageworkloadResponse):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/commands/cloud_commands.py

@@ -23,6 +23,7 @@ from anyscale.client.openapi_client.models import (
 from anyscale.client.openapi_client.models.compute_stack import ComputeStack
 from anyscale.cloud.models import CreateCloudCollaborator, CreateCloudCollaborators
 from anyscale.commands import command_examples
+from anyscale.commands.setup_k8s import setup_kubernetes_cloud
 from anyscale.commands.util import AnyscaleCommand, OptionPromptNull
 from anyscale.controllers.cloud_controller import CloudController
 from anyscale.util import (
@@ -127,6 +128,27 @@ def default_region(provider: str) -> str:
     show_default=True,
 )
 @click.option("--name", "-n", help="Name of the cloud.", required=True, prompt="Name")
+@click.option(
+    "--stack",
+    help="The compute stack to use (vm or k8s).",
+    required=False,
+    type=click.Choice(["vm", "k8s"], case_sensitive=False),
+    default="vm",
+    show_default=True,
+)
+@click.option(
+    "--cluster-name",
+    help="Kubernetes cluster name (required for k8s stack).",
+    required=False,
+    type=str,
+)
+@click.option(
+    "--namespace",
+    help="Kubernetes namespace for Anyscale operator (for k8s stack).",
+    required=False,
+    type=str,
+    default="anyscale-operator",
+)
 @click.option(
     "--project-id",
     help="Globally Unique project ID for GCP clouds (e.g., my-project-abc123)",
@@ -173,10 +195,22 @@ def default_region(provider: str) -> str:
     show_default=True,
     help="The type of shared storage to use for the cloud. Use 'object-storage' for cloud bucket-based storage (e.g., S3, GCS), or 'nfs' for network file systems.",
 )
+@click.option(
+    "--values-file",
+    help="Path to save the generated Helm values file (for k8s stack, default: auto-generated with timestamp).",
+    required=False,
+    type=str,
+)
+@click.option(
+    "--debug", is_flag=True, default=False, help="Enable debug logging.",
+)
 def setup_cloud(  # noqa: PLR0913
     provider: str,
     region: str,
     name: str,
+    stack: str,
+    cluster_name: Optional[str],
+    namespace: str,
     project_id: str,
     functional_verify: Optional[str],
     anyscale_managed: bool,  # noqa: ARG001
@@ -184,9 +218,33 @@ def setup_cloud(  # noqa: PLR0913
     yes: bool,
     disable_auto_add_user: bool,
     shared_storage: str,
+    values_file: Optional[str],
+    debug: bool,
 ) -> None:
     # TODO (congding): remove `anyscale_managed` in the future, now keeping it for compatibility
 
+    # Handle Kubernetes stack
+    if stack == "k8s":
+        if not cluster_name:
+            raise click.ClickException(
+                "--cluster-name is required when using --stack=k8s"
+            )
+
+        setup_kubernetes_cloud(
+            provider=provider,
+            region=region,
+            name=name,
+            cluster_name=cluster_name,
+            namespace=namespace,
+            project_id=project_id,
+            functional_verify=bool(functional_verify),
+            yes=yes,
+            values_file=values_file,
+            debug=debug,
+        )
+        return
+
+    # Handle VM stack
     # Convert string to enum for type safety
     shared_storage_type = SharedStorageType(shared_storage)
     if provider == "aws":