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

anyscale/sdk/anyscale_client/__init__.py

@@ -137,7 +137,6 @@ from anyscale_client.models.list_service_model import ListServiceModel
 from anyscale_client.models.listservicemodel_list_response import ListservicemodelListResponse
 from anyscale_client.models.log_download_result import LogDownloadResult
 from anyscale_client.models.log_file_chunk import LogFileChunk
-from anyscale_client.models.log_level_types import LogLevelTypes
 from anyscale_client.models.log_stream import LogStream
 from anyscale_client.models.logdownloadresult_response import LogdownloadresultResponse
 from anyscale_client.models.logstream_response import LogstreamResponse
@@ -194,9 +193,6 @@ from anyscale_client.models.servicemodel_response import ServicemodelResponse
 from anyscale_client.models.session import Session
 from anyscale_client.models.session_command import SessionCommand
 from anyscale_client.models.session_command_types import SessionCommandTypes
-from anyscale_client.models.session_event import SessionEvent
-from anyscale_client.models.session_event_cause import SessionEventCause
-from anyscale_client.models.session_event_types import SessionEventTypes
 from anyscale_client.models.session_list_response import SessionListResponse
 from anyscale_client.models.session_operation import SessionOperation
 from anyscale_client.models.session_operation_type import SessionOperationType
@@ -207,7 +203,6 @@ from anyscale_client.models.session_state_data import SessionStateData
 from anyscale_client.models.session_stopping_data import SessionStoppingData
 from anyscale_client.models.sessioncommand_list_response import SessioncommandListResponse
 from anyscale_client.models.sessioncommand_response import SessioncommandResponse
-from anyscale_client.models.sessionevent_list_response import SessioneventListResponse
 from anyscale_client.models.sessionoperation_response import SessionoperationResponse
 from anyscale_client.models.sessions_query import SessionsQuery
 from anyscale_client.models.sort_by_clause_jobs_sort_field import SortByClauseJobsSortField

anyscale/sdk/anyscale_client/api/default_api.py

@@ -6140,156 +6140,6 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def get_session_event_log(self, session_id, **kwargs):  # noqa: E501
-        """Get Session Event Log  # noqa: E501
-
-        Retrieves a session's event log.  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_session_event_log(session_id, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool: execute request asynchronously
-        :param str session_id: ID of the Session to retrieve event logs for. (required)
-        :param datetime before: Filters events occurring before this datetime.
-        :param datetime after: Filters events occurring after this datetime.
-        :param list[SessionEventTypes] event_types: Filters events to these types.
-        :param list[LogLevelTypes] log_level_types: Filters logs to these leves.
-        :param str paging_token:
-        :param int count:
-        :param _preload_content: if False, the urllib3.HTTPResponse object will
-                                 be returned without reading/decoding response
-                                 data. Default is True.
-        :param _request_timeout: timeout setting for this request. If one
-                                 number provided, it will be total request
-                                 timeout. It can also be a pair (tuple) of
-                                 (connection, read) timeouts.
-        :return: SessioneventListResponse
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-        kwargs['_return_http_data_only'] = True
-        return self.get_session_event_log_with_http_info(session_id, **kwargs)  # noqa: E501
-
-    def get_session_event_log_with_http_info(self, session_id, **kwargs):  # noqa: E501
-        """Get Session Event Log  # noqa: E501
-
-        Retrieves a session's event log.  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_session_event_log_with_http_info(session_id, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool: execute request asynchronously
-        :param str session_id: ID of the Session to retrieve event logs for. (required)
-        :param datetime before: Filters events occurring before this datetime.
-        :param datetime after: Filters events occurring after this datetime.
-        :param list[SessionEventTypes] event_types: Filters events to these types.
-        :param list[LogLevelTypes] log_level_types: Filters logs to these leves.
-        :param str paging_token:
-        :param int count:
-        :param _return_http_data_only: response data without head status code
-                                       and headers
-        :param _preload_content: if False, the urllib3.HTTPResponse object will
-                                 be returned without reading/decoding response
-                                 data. Default is True.
-        :param _request_timeout: timeout setting for this request. If one
-                                 number provided, it will be total request
-                                 timeout. It can also be a pair (tuple) of
-                                 (connection, read) timeouts.
-        :return: tuple(SessioneventListResponse, status_code(int), headers(HTTPHeaderDict))
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-
-        local_var_params = locals()
-
-        all_params = [
-            'session_id',
-            'before',
-            'after',
-            'event_types',
-            'log_level_types',
-            'paging_token',
-            'count'
-        ]
-        all_params.extend(
-            [
-                'async_req',
-                '_return_http_data_only',
-                '_preload_content',
-                '_request_timeout'
-            ]
-        )
-
-        for key, val in six.iteritems(local_var_params['kwargs']):
-            if key not in all_params:
-                raise ApiTypeError(
-                    "Got an unexpected keyword argument '%s'"
-                    " to method get_session_event_log" % key
-                )
-            local_var_params[key] = val
-        del local_var_params['kwargs']
-        # verify the required parameter 'session_id' is set
-        if self.api_client.client_side_validation and ('session_id' not in local_var_params or  # noqa: E501
-                                                        local_var_params['session_id'] is None):  # noqa: E501
-            raise ApiValueError("Missing the required parameter `session_id` when calling `get_session_event_log`")  # noqa: E501
-
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `get_session_event_log`, must be a value less than or equal to `1000`")  # noqa: E501
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `get_session_event_log`, must be a value greater than or equal to `0`")  # noqa: E501
-        collection_formats = {}
-
-        path_params = {}
-
-        query_params = []
-        if 'session_id' in local_var_params and local_var_params['session_id'] is not None:  # noqa: E501
-            query_params.append(('session_id', local_var_params['session_id']))  # noqa: E501
-        if 'before' in local_var_params and local_var_params['before'] is not None:  # noqa: E501
-            query_params.append(('before', local_var_params['before']))  # noqa: E501
-        if 'after' in local_var_params and local_var_params['after'] is not None:  # noqa: E501
-            query_params.append(('after', local_var_params['after']))  # noqa: E501
-        if 'event_types' in local_var_params and local_var_params['event_types'] is not None:  # noqa: E501
-            query_params.append(('event_types', local_var_params['event_types']))  # noqa: E501
-            collection_formats['event_types'] = 'multi'  # noqa: E501
-        if 'log_level_types' in local_var_params and local_var_params['log_level_types'] is not None:  # noqa: E501
-            query_params.append(('log_level_types', local_var_params['log_level_types']))  # noqa: E501
-            collection_formats['log_level_types'] = 'multi'  # noqa: E501
-        if 'paging_token' in local_var_params and local_var_params['paging_token'] is not None:  # noqa: E501
-            query_params.append(('paging_token', local_var_params['paging_token']))  # noqa: E501
-        if 'count' in local_var_params and local_var_params['count'] is not None:  # noqa: E501
-            query_params.append(('count', local_var_params['count']))  # noqa: E501
-
-        header_params = {}
-
-        form_params = []
-        local_var_files = {}
-
-        body_params = None
-        # HTTP header `Accept`
-        header_params['Accept'] = self.api_client.select_header_accept(
-            ['application/json'])  # noqa: E501
-
-        # Authentication setting
-        auth_settings = []  # noqa: E501
-
-        return self.api_client.call_api(
-            '/session_events/', 'GET',
-            path_params,
-            query_params,
-            header_params,
-            body=body_params,
-            post_params=form_params,
-            files=local_var_files,
-            response_type='SessioneventListResponse',  # noqa: E501
-            auth_settings=auth_settings,
-            async_req=local_var_params.get('async_req'),
-            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
-            _preload_content=local_var_params.get('_preload_content', True),
-            _request_timeout=local_var_params.get('_request_timeout'),
-            collection_formats=collection_formats)
-
     def get_session_for_job(self, production_job_id, **kwargs):  # noqa: E501
         """Get Session For Job  # noqa: E501
 

anyscale/sdk/anyscale_client/models/__init__.py

@@ -123,7 +123,6 @@ from anyscale_client.models.list_service_model import ListServiceModel
 from anyscale_client.models.listservicemodel_list_response import ListservicemodelListResponse
 from anyscale_client.models.log_download_result import LogDownloadResult
 from anyscale_client.models.log_file_chunk import LogFileChunk
-from anyscale_client.models.log_level_types import LogLevelTypes
 from anyscale_client.models.log_stream import LogStream
 from anyscale_client.models.logdownloadresult_response import LogdownloadresultResponse
 from anyscale_client.models.logstream_response import LogstreamResponse
@@ -180,9 +179,6 @@ from anyscale_client.models.servicemodel_response import ServicemodelResponse
 from anyscale_client.models.session import Session
 from anyscale_client.models.session_command import SessionCommand
 from anyscale_client.models.session_command_types import SessionCommandTypes
-from anyscale_client.models.session_event import SessionEvent
-from anyscale_client.models.session_event_cause import SessionEventCause
-from anyscale_client.models.session_event_types import SessionEventTypes
 from anyscale_client.models.session_list_response import SessionListResponse
 from anyscale_client.models.session_operation import SessionOperation
 from anyscale_client.models.session_operation_type import SessionOperationType
@@ -193,7 +189,6 @@ from anyscale_client.models.session_state_data import SessionStateData
 from anyscale_client.models.session_stopping_data import SessionStoppingData
 from anyscale_client.models.sessioncommand_list_response import SessioncommandListResponse
 from anyscale_client.models.sessioncommand_response import SessioncommandResponse
-from anyscale_client.models.sessionevent_list_response import SessioneventListResponse
 from anyscale_client.models.sessionoperation_response import SessionoperationResponse
 from anyscale_client.models.sessions_query import SessionsQuery
 from anyscale_client.models.sort_by_clause_jobs_sort_field import SortByClauseJobsSortField

anyscale/sdk/anyscale_client/models/cluster_status_details.py

@@ -30,8 +30,9 @@ class ClusterStatusDetails(object):
     """
     LAUNCHING_NODES = "LAUNCHING_NODES"
     CONFIGURING_HEAD_NODE = "CONFIGURING_HEAD_NODE"
+    UPDATING_CONTAINERS = "UPDATING_CONTAINERS"
 
-    allowable_values = [LAUNCHING_NODES, CONFIGURING_HEAD_NODE]  # noqa: E501
+    allowable_values = [LAUNCHING_NODES, CONFIGURING_HEAD_NODE, UPDATING_CONTAINERS]  # noqa: E501
 
     """
     Attributes:

anyscale/sdk/anyscale_client/sdk.py

@@ -39,7 +39,7 @@ from anyscale.util import get_endpoint
 from anyscale.authenticate import AuthenticationBlock
 from anyscale_client.models import ClusterEnvironmentBuildStatus
 from anyscale.util import get_ray_and_py_version_for_default_cluster_env
-from anyscale.project import get_default_project
+from anyscale.project_utils import get_default_project
 from anyscale.utils.ray_version_utils import get_correct_name_for_base_image_name
 
 logger = logging.getLogger(__file__)

anyscale/user/__init__.py

@@ -30,6 +30,6 @@ class UserSDK:
     def admin_batch_create(  # noqa: F811
         self, admin_create_users: List[AdminCreateUser],
     ) -> List[AdminCreatedUser]:
-        """Batch create users without email verification as an admin.
+        """Batch create, as an admin, users without email verification.
         """
         return self._private_sdk.admin_batch_create(admin_create_users)

anyscale/user/commands.py

@@ -37,6 +37,6 @@ _ADMIN_BATCH_CREATE_ARG_DOCSTRINGS = {
 def admin_batch_create(
     admin_create_users: List[AdminCreateUser], *, _sdk: PrivateUserSDK
 ) -> List[AdminCreatedUser]:
-    """Batch create users without email verification as an admin.
+    """Batch create, as an admin, users without email verification.
     """
     return _sdk.admin_batch_create(admin_create_users)

anyscale/user/models.py

@@ -15,13 +15,13 @@ import anyscale
 from anyscale.user.models import AdminCreateUser
 
 admin_create_user = AdminCreateUser(
-    # Name of the user to be created.
+    # First name of the user to be created.
     name="name",
     # Email of the user to be created.
     email="test@anyscale.com",
     # Password for the user account being created.
     password="",
-    # Whether the user is an SSO user.
+    # Whether the user is an SSO user. SSO users can log in using SSO.
     is_sso_user=False,
     # Optional last name of the user to be created.
     lastname="lastname",
@@ -29,7 +29,7 @@ admin_create_user = AdminCreateUser(
     title="title",
 )
 """
-    name: str = field(metadata={"docstring": "Name of the user to be created."})
+    name: str = field(metadata={"docstring": "First name of the user to be created."})
 
     def _validate_name(self, name: str):
         if not isinstance(name, str):
@@ -50,7 +50,9 @@ admin_create_user = AdminCreateUser(
             raise TypeError("password must be a string.")
 
     is_sso_user: bool = field(
-        metadata={"docstring": "Whether the user is an SSO user."},
+        metadata={
+            "docstring": "Whether the user is an SSO user. SSO users can log in using SSO."
+        },
     )
 
     def _validate_is_sso_user(self, is_sso_user: bool):
@@ -86,13 +88,13 @@ import anyscale
 from anyscale.user.models import AdminCreateUser
 
 admin_create_user = AdminCreateUser(
-    # Name of the user to be created.
+    # First name of the user to be created.
     name="name",
     # Email of the user to be created.
     email="test@anyscale.com",
     # Password for the user account being created.
     password="",
-    # Whether the user is an SSO user.
+    # Whether the user is an SSO user. SSO users can log in using SSO.
     is_sso_user=False,
     # Optional last name of the user to be created.
     lastname="lastname",
@@ -115,7 +117,7 @@ admin_create_users = AdminCreateUsers(
 
 @dataclass(frozen=True)
 class AdminCreatedUser(ModelBase):
-    """User account created by an admin.
+    """User account created by an admin that has organization collaborator permissions.
     """
 
     __doc_py_example__ = """\
@@ -123,13 +125,13 @@ import anyscale
 from anyscale.user.models import AdminCreatedUser
 
 admin_create_user = AdminCreateUser(
-    # Name of the user to be created.
+    # First name of the user to be created.
     name="name",
     # Email of the user to be created.
     email="test@anyscale.com",
     # Password for the user account being created.
     password="",
-    # Whether the user is an SSO user.
+    # Whether the user is an SSO user. SSO users can log in using SSO.
     is_sso_user=False,
     # Optional last name of the user to be created.
     lastname="lastname",
@@ -138,19 +140,25 @@ admin_create_user = AdminCreateUser(
 )
 admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([admin_create_user])
 """
-    user_id: str = field(metadata={"docstring": "Name of the user to be created."})
+    user_id: str = field(
+        metadata={"docstring": "ID of the user that has been created."}
+    )
 
     def _validate_user_id(self, user_id: str):
         if not isinstance(user_id, str):
             raise TypeError("user_id must be a string.")
 
-    name: str = field(metadata={"docstring": "Name of the user to be created."})
+    name: str = field(
+        metadata={"docstring": "First name of the user that has been created."}
+    )
 
     def _validate_name(self, name: str):
         if not isinstance(name, str):
             raise TypeError("name must be a string.")
 
-    email: str = field(metadata={"docstring": "Email of the user to be created."})
+    email: str = field(
+        metadata={"docstring": "Email of the user that has been created."}
+    )
 
     def _validate_email(self, email: str):
         if not isinstance(email, str):
@@ -165,7 +173,9 @@ admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([
             raise TypeError("created_at must be a datetime.")
 
     is_sso_user: bool = field(
-        metadata={"docstring": "Whether the user is an SSO user."},
+        metadata={
+            "docstring": "Whether the user is an SSO user. SSO users can log in using SSO."
+        },
     )
 
     def _validate_is_sso_user(self, is_sso_user: bool):
@@ -174,7 +184,7 @@ admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([
 
     lastname: Optional[str] = field(
         default=None,
-        metadata={"docstring": "Optional last name of the user to be created."},
+        metadata={"docstring": "Optional last name of the user that has been created."},
     )
 
     def _validate_lastname(self, lastname: Optional[str]):
@@ -183,7 +193,7 @@ admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([
 
     title: Optional[str] = field(
         default=None,
-        metadata={"docstring": "Optional title of the user to be created."},
+        metadata={"docstring": "Optional title of the user that has been created."},
     )
 
     def _validate_title(self, title: Optional[str]):

anyscale/util.py

@@ -44,6 +44,7 @@ from anyscale.client.openapi_client.models.cloud_analytics_event_cloud_resource
 from anyscale.client.openapi_client.models.decorated_compute_template import (
     DecoratedComputeTemplate,
 )
+from anyscale.client.openapi_client.models.ha_job_states import HaJobStates
 from anyscale.client.openapi_client.models.user_info import UserInfo
 from anyscale.cluster_compute import get_cluster_compute_from_name
 import anyscale.conf
@@ -986,6 +987,20 @@ def validate_job_config_dict(
         )
 
 
+def validate_list_jobs_state_filter(ctx, param, value) -> List[str]:  # noqa: ARG001
+    """
+    Validate the job state filter for list jobs CLI method
+    """
+    if not value:
+        return []
+    for each_value in value:
+        if each_value.upper() not in HaJobStates.allowable_values:
+            raise click.ClickException(
+                f"{each_value} is not a valid value for {param.opts[0]}. Allowed values: {', '.join(HaJobStates.allowable_values)}"
+            )
+    return [each_value.upper() for each_value in value]
+
+
 def get_cluster_model_for_current_workspace(
     anyscale_api_client: BaseApi,
 ) -> Optional[Cluster]:

anyscale/utils/cloud_utils.py

@@ -20,7 +20,7 @@ from anyscale.client.openapi_client.models import (
     CloudProviders,
     CreateAnalyticsEvent,
 )
-from anyscale.cloud import get_cloud_id_and_name
+from anyscale.cloud_utils import get_cloud_id_and_name
 from anyscale.shared_anyscale_utils.utils.collections import flatten
 
 

anyscale/version.py

@@ -1 +1 @@
-__version__ = "0.24.88"
+__version__ = "0.25.0"

anyscale/workspace_utils.py

@@ -3,7 +3,7 @@ import os
 from click import ClickException
 import yaml
 
-from anyscale.project import ANYSCALE_PROJECT_FILE, find_project_root
+from anyscale.project_utils import ANYSCALE_PROJECT_FILE, find_project_root
 
 
 def write_workspace_id_to_disk(workspace_id: str, directory: str) -> None:

{anyscale-0.24.88.dist-info → anyscale-0.25.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.88
+Version: 0.25.0
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License
@@ -47,7 +47,6 @@ Requires-Dist: google-cloud-filestore; extra == "backend"
 Requires-Dist: google-cloud-storage; extra == "backend"
 Requires-Dist: google-cloud-redis; extra == "backend"
 Requires-Dist: google-cloud-certificate-manager; extra == "backend"
-Requires-Dist: azure-storage-blob; extra == "backend"
 Provides-Extra: gcp
 Requires-Dist: protobuf<4; python_version < "3.11" and extra == "gcp"
 Requires-Dist: google-api-python-client; extra == "gcp"
@@ -58,8 +57,6 @@ Requires-Dist: google-cloud-filestore; extra == "gcp"
 Requires-Dist: google-cloud-storage; extra == "gcp"
 Requires-Dist: google-cloud-redis; extra == "gcp"
 Requires-Dist: google-cloud-certificate-manager; extra == "gcp"
-Provides-Extra: azure
-Requires-Dist: azure-storage-blob; extra == "azure"
 Provides-Extra: all
 Requires-Dist: ray>=1.4.0; extra == "all"
 Requires-Dist: protobuf<4; python_version < "3.11" and extra == "all"
@@ -71,7 +68,6 @@ Requires-Dist: google-cloud-filestore; extra == "all"
 Requires-Dist: google-cloud-storage; extra == "all"
 Requires-Dist: google-cloud-redis; extra == "all"
 Requires-Dist: google-cloud-certificate-manager; extra == "all"
-Requires-Dist: azure-storage-blob; extra == "all"
 
 # Anyscale