anyscale 0.25.2__py3-none-any.whl → 0.25.3__py3-none-any.whl

anyscale/_private/anyscale_client/anyscale_client.py

@@ -1095,6 +1095,11 @@ class AnyscaleClient(AnyscaleClientInterface):
         result: ServiceModel = self._external_api_client.terminate_service(service_id)
         return result
 
+    @handle_api_exceptions
+    def archive_service(self, service_id: str) -> ServiceModel:
+        result: ServiceModel = self._external_api_client.archive_service(service_id)
+        return result
+
     @handle_api_exceptions
     def submit_job(self, model: CreateInternalProductionJob) -> InternalProductionJob:
         job: InternalProductionJob = self._internal_api_client.create_job_api_v2_decorated_ha_jobs_create_post(

anyscale/_private/anyscale_client/common.py

@@ -353,6 +353,11 @@ class AnyscaleClientInterface(ABC):
         """Mark the service to be terminated asynchronously."""
         raise NotImplementedError
 
+    @abstractmethod
+    def archive_service(self, service_id: str):
+        """Mark the service to be archived asynchronously."""
+        raise NotImplementedError
+
     @abstractmethod
     def submit_job(self, model: CreateInternalProductionJob) -> InternalProductionJob:
         """Submit the job to run."""

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -144,6 +144,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._workspace_cluster: Optional[Cluster] = None
         self._workspace_dependency_tracking_enabled: bool = False
         self._services: Dict[str, ServiceModel] = {}
+        self._archived_services: Dict[str, ServiceModel] = {}
         self._jobs: Dict[str, ProductionJob] = {}
         self._job_runs: Dict[str, List[APIJobRun]] = defaultdict(list)
         self._project_to_id: Dict[Optional[str] : Dict[Optional[str], str]] = {}
@@ -763,6 +764,16 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
                 service_id
             ].primary_version.current_state = ServiceVersionState.TERMINATED
 
+    @property
+    def archived_services(self) -> Dict[str, ServiceModel]:
+        return self._archived_services
+
+    def archive_service(self, service_id: str):
+        self._archived_services[service_id] = self._services.pop(service_id)
+
+    def is_archived_service(self, service_id: str) -> bool:
+        return service_id in self._archived_services
+
     @property
     def submitted_job(self) -> Optional[CreateInternalProductionJob]:
         return self._submitted_job

anyscale/_private/workload/workload_config.py

@@ -117,13 +117,27 @@ class WorkloadConfig(ModelBase):
     )
 
     def _validate_env_vars(self, env_vars: Optional[Dict[str, str]]):
-        if env_vars is not None and (
-            not isinstance(env_vars, dict)
-            or not all(
-                isinstance(k, str) and isinstance(v, str) for k, v in env_vars.items()
+        if env_vars is None:
+            return
+
+        if not isinstance(env_vars, dict):
+            raise TypeError(
+                "'env_vars' must be a Dict[str, str], "
+                f"but got type {type(env_vars)}."
             )
-        ):
-            raise TypeError("'env_vars' must be a Dict[str, str].")
+
+        for k, v in env_vars.items():
+            if not isinstance(k, str):
+                raise TypeError(
+                    "'env_vars' must be a Dict[str, str], "
+                    f"but got key of type {type(k)}: {k}."
+                )
+
+            if not isinstance(v, str):
+                raise TypeError(
+                    "'env_vars' must be a Dict[str, str], "
+                    f"but got value of type {type(v)} for key '{k}': {v}."
+                )
 
     py_modules: Optional[List[str]] = field(
         default=None,

anyscale/client/README.md

@@ -78,6 +78,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**add_to_waitlist_api_v2_aioa_cloud_waitlist_post**](docs/DefaultApi.md#add_to_waitlist_api_v2_aioa_cloud_waitlist_post) | **POST** /api/v2/aioa_cloud_waitlist/ | Add To Waitlist
 *DefaultApi* | [**admin_batch_create_users_api_v2_users_admin_batch_create_post**](docs/DefaultApi.md#admin_batch_create_users_api_v2_users_admin_batch_create_post) | **POST** /api/v2/users/admin_batch_create | Admin Batch Create Users
 *DefaultApi* | [**admission_api_v2_kubernetes_manager_admission_cloud_resource_id_post**](docs/DefaultApi.md#admission_api_v2_kubernetes_manager_admission_cloud_resource_id_post) | **POST** /api/v2/kubernetes_manager/admission/{cloud_resource_id} | Admission
+*DefaultApi* | [**alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put**](docs/DefaultApi.md#alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put) | **PUT** /api/v2/clouds/{cloud_id}/collaborators/users/{identity_id} | Alter Cloud Collaborator
 *DefaultApi* | [**alter_organization_collaborator_api_v2_organization_collaborators_identity_id_put**](docs/DefaultApi.md#alter_organization_collaborator_api_v2_organization_collaborators_identity_id_put) | **PUT** /api/v2/organization_collaborators/{identity_id} | Alter Organization Collaborator
 *DefaultApi* | [**alter_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_put**](docs/DefaultApi.md#alter_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_put) | **PUT** /api/v2/projects/{project_id}/collaborators/{role_or_identity_id} | Alter Project Collaborator
 *DefaultApi* | [**apply_service_api_v2_services_v2_apply_put**](docs/DefaultApi.md#apply_service_api_v2_services_v2_apply_put) | **PUT** /api/v2/services-v2/apply | Apply Service
@@ -195,6 +196,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_cloud_project_collaborator_api_v2_projects_project_id_collaborators_clouds_get**](docs/DefaultApi.md#get_cloud_project_collaborator_api_v2_projects_project_id_collaborators_clouds_get) | **GET** /api/v2/projects/{project_id}/collaborators/clouds | Get Cloud Project Collaborator
 *DefaultApi* | [**get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_get**](docs/DefaultApi.md#get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_get) | **GET** /api/v2/clouds_with_cloud_resource_gcp_router/{cloud_id} | Get Cloud With Cloud Resource
 *DefaultApi* | [**get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_get**](docs/DefaultApi.md#get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_get) | **GET** /api/v2/clouds_with_cloud_resource_router/{cloud_id} | Get Cloud With Cloud Resource
+*DefaultApi* | [**get_cluster_events_api_v2_sessions_session_id_cluster_events_get**](docs/DefaultApi.md#get_cluster_events_api_v2_sessions_session_id_cluster_events_get) | **GET** /api/v2/sessions/{session_id}/cluster_events | Get Cluster Events
 *DefaultApi* | [**get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get**](docs/DefaultApi.md#get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get) | **GET** /api/v2/logs/cluster_product_autoscaler_flag/{session_id} | Get Cluster Product Autoscaler Flag
 *DefaultApi* | [**get_compute_template_api_v2_compute_templates_template_id_get**](docs/DefaultApi.md#get_compute_template_api_v2_compute_templates_template_id_get) | **GET** /api/v2/compute_templates/{template_id} | Get Compute Template
 *DefaultApi* | [**get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get**](docs/DefaultApi.md#get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get) | **GET** /api/v2/experimental_cron_jobs/{cron_job_id} | Get Cron Job
@@ -559,6 +561,7 @@ Class | Method | HTTP request | Description
  - [ClusterConfigWithSessionIdleTimeout](docs/ClusterConfigWithSessionIdleTimeout.md)
  - [ClusterEnvironmentsQuery](docs/ClusterEnvironmentsQuery.md)
  - [ClusterEvent](docs/ClusterEvent.md)
+ - [ClusterEventSource](docs/ClusterEventSource.md)
  - [ClusterEventsOutput](docs/ClusterEventsOutput.md)
  - [ClusterFeatures](docs/ClusterFeatures.md)
  - [ClusterManagementStackVersions](docs/ClusterManagementStackVersions.md)
@@ -569,6 +572,7 @@ Class | Method | HTTP request | Description
  - [ClusterauthresponseResponse](docs/ClusterauthresponseResponse.md)
  - [ClusterconfigResponse](docs/ClusterconfigResponse.md)
  - [ClusterconfigwithsessionidletimeoutResponse](docs/ClusterconfigwithsessionidletimeoutResponse.md)
+ - [ClustereventListResponse](docs/ClustereventListResponse.md)
  - [ClustereventsoutputResponse](docs/ClustereventsoutputResponse.md)
  - [ClusterfeaturesResponse](docs/ClusterfeaturesResponse.md)
  - [ComputeNodeType](docs/ComputeNodeType.md)
@@ -1012,6 +1016,7 @@ Class | Method | HTTP request | Description
  - [UnifiedJobSortField](docs/UnifiedJobSortField.md)
  - [UnifiedJobStatus](docs/UnifiedJobStatus.md)
  - [UnifiedJobType](docs/UnifiedJobType.md)
+ - [UpdateCloudCollaborator](docs/UpdateCloudCollaborator.md)
  - [UpdateCloudWithCloudResource](docs/UpdateCloudWithCloudResource.md)
  - [UpdateCloudWithCloudResourceGCP](docs/UpdateCloudWithCloudResourceGCP.md)
  - [UpdateClusterDns](docs/UpdateClusterDns.md)

anyscale/client/openapi_client/__init__.py

@@ -160,6 +160,7 @@ from openapi_client.models.cluster_config import ClusterConfig
 from openapi_client.models.cluster_config_with_session_idle_timeout import ClusterConfigWithSessionIdleTimeout
 from openapi_client.models.cluster_environments_query import ClusterEnvironmentsQuery
 from openapi_client.models.cluster_event import ClusterEvent
+from openapi_client.models.cluster_event_source import ClusterEventSource
 from openapi_client.models.cluster_events_output import ClusterEventsOutput
 from openapi_client.models.cluster_features import ClusterFeatures
 from openapi_client.models.cluster_management_stack_versions import ClusterManagementStackVersions
@@ -170,6 +171,7 @@ from openapi_client.models.cluster_status_details import ClusterStatusDetails
 from openapi_client.models.clusterauthresponse_response import ClusterauthresponseResponse
 from openapi_client.models.clusterconfig_response import ClusterconfigResponse
 from openapi_client.models.clusterconfigwithsessionidletimeout_response import ClusterconfigwithsessionidletimeoutResponse
+from openapi_client.models.clusterevent_list_response import ClustereventListResponse
 from openapi_client.models.clustereventsoutput_response import ClustereventsoutputResponse
 from openapi_client.models.clusterfeatures_response import ClusterfeaturesResponse
 from openapi_client.models.compute_node_type import ComputeNodeType
@@ -613,6 +615,7 @@ from openapi_client.models.ux_instance import UXInstance
 from openapi_client.models.unified_job_sort_field import UnifiedJobSortField
 from openapi_client.models.unified_job_status import UnifiedJobStatus
 from openapi_client.models.unified_job_type import UnifiedJobType
+from openapi_client.models.update_cloud_collaborator import UpdateCloudCollaborator
 from openapi_client.models.update_cloud_with_cloud_resource import UpdateCloudWithCloudResource
 from openapi_client.models.update_cloud_with_cloud_resource_gcp import UpdateCloudWithCloudResourceGCP
 from openapi_client.models.update_cluster_dns import UpdateClusterDns

anyscale/client/openapi_client/api/default_api.py

@@ -382,6 +382,140 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put(self, cloud_id, identity_id, update_cloud_collaborator, **kwargs):  # noqa: E501
+        """Alter Cloud Collaborator  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put(cloud_id, identity_id, update_cloud_collaborator, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param str identity_id: (required)
+        :param UpdateCloudCollaborator update_cloud_collaborator: (required)
+        :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: None
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put_with_http_info(cloud_id, identity_id, update_cloud_collaborator, **kwargs)  # noqa: E501
+
+    def alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put_with_http_info(self, cloud_id, identity_id, update_cloud_collaborator, **kwargs):  # noqa: E501
+        """Alter Cloud Collaborator  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put_with_http_info(cloud_id, identity_id, update_cloud_collaborator, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param str identity_id: (required)
+        :param UpdateCloudCollaborator update_cloud_collaborator: (required)
+        :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: None
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'cloud_id',
+            'identity_id',
+            'update_cloud_collaborator'
+        ]
+        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 alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'cloud_id' is set
+        if self.api_client.client_side_validation and ('cloud_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['cloud_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `cloud_id` when calling `alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put`")  # noqa: E501
+        # verify the required parameter 'identity_id' is set
+        if self.api_client.client_side_validation and ('identity_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['identity_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `identity_id` when calling `alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put`")  # noqa: E501
+        # verify the required parameter 'update_cloud_collaborator' is set
+        if self.api_client.client_side_validation and ('update_cloud_collaborator' not in local_var_params or  # noqa: E501
+                                                        local_var_params['update_cloud_collaborator'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `update_cloud_collaborator` when calling `alter_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_users_identity_id_put`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'cloud_id' in local_var_params:
+            path_params['cloud_id'] = local_var_params['cloud_id']  # noqa: E501
+        if 'identity_id' in local_var_params:
+            path_params['identity_id'] = local_var_params['identity_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'update_cloud_collaborator' in local_var_params:
+            body_params = local_var_params['update_cloud_collaborator']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/clouds/{cloud_id}/collaborators/users/{identity_id}', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type=None,  # 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 alter_organization_collaborator_api_v2_organization_collaborators_identity_id_put(self, identity_id, update_organization_collaborator, **kwargs):  # noqa: E501
         """Alter Organization Collaborator  # noqa: E501
 
@@ -14147,6 +14281,142 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def get_cluster_events_api_v2_sessions_session_id_cluster_events_get(self, session_id, **kwargs):  # noqa: E501
+        """Get Cluster Events  # noqa: E501
+
+        Return events for this cluster.  There are multiple endpoints to return events. This endpoint is for the UI to fetch events. This should power general observability event views shared by different workload types. Like the task view or the Train dashboard view.  The /events endpoint is used by the dataplane to fetch events and has cluster auth requirements.  Finally, there are events endpoints for each workload type (workspaces, jobs, services). Those endpoints have some additional logic to stitch together cluster events with workload-specific events.  # 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_cluster_events_api_v2_sessions_session_id_cluster_events_get(session_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str session_id: (required)
+        :param datetime start_time: Filter for events that have occurred since this value
+        :param datetime end_time: Filter for events that have occurred before this value
+        :param list[ClusterEventSource] sources: Filter for events from these sources
+        :param list[str] autoscaler_resource_bundles: Filter for autoscaler events relevant to specific resource bundles.Each string should be a JSON-encoded resource bundle. We accept a string here because query params do not support dictionaries. Ex: ['{\"CPU\": 1, \"GPU\": 1}', '{\"GPU\": 1}']
+        :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: ClustereventListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.get_cluster_events_api_v2_sessions_session_id_cluster_events_get_with_http_info(session_id, **kwargs)  # noqa: E501
+
+    def get_cluster_events_api_v2_sessions_session_id_cluster_events_get_with_http_info(self, session_id, **kwargs):  # noqa: E501
+        """Get Cluster Events  # noqa: E501
+
+        Return events for this cluster.  There are multiple endpoints to return events. This endpoint is for the UI to fetch events. This should power general observability event views shared by different workload types. Like the task view or the Train dashboard view.  The /events endpoint is used by the dataplane to fetch events and has cluster auth requirements.  Finally, there are events endpoints for each workload type (workspaces, jobs, services). Those endpoints have some additional logic to stitch together cluster events with workload-specific events.  # 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_cluster_events_api_v2_sessions_session_id_cluster_events_get_with_http_info(session_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str session_id: (required)
+        :param datetime start_time: Filter for events that have occurred since this value
+        :param datetime end_time: Filter for events that have occurred before this value
+        :param list[ClusterEventSource] sources: Filter for events from these sources
+        :param list[str] autoscaler_resource_bundles: Filter for autoscaler events relevant to specific resource bundles.Each string should be a JSON-encoded resource bundle. We accept a string here because query params do not support dictionaries. Ex: ['{\"CPU\": 1, \"GPU\": 1}', '{\"GPU\": 1}']
+        :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(ClustereventListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'session_id',
+            'start_time',
+            'end_time',
+            'sources',
+            'autoscaler_resource_bundles'
+        ]
+        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_cluster_events_api_v2_sessions_session_id_cluster_events_get" % 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_cluster_events_api_v2_sessions_session_id_cluster_events_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'session_id' in local_var_params:
+            path_params['session_id'] = local_var_params['session_id']  # noqa: E501
+
+        query_params = []
+        if 'start_time' in local_var_params and local_var_params['start_time'] is not None:  # noqa: E501
+            query_params.append(('start_time', local_var_params['start_time']))  # noqa: E501
+        if 'end_time' in local_var_params and local_var_params['end_time'] is not None:  # noqa: E501
+            query_params.append(('end_time', local_var_params['end_time']))  # noqa: E501
+        if 'sources' in local_var_params and local_var_params['sources'] is not None:  # noqa: E501
+            query_params.append(('sources', local_var_params['sources']))  # noqa: E501
+            collection_formats['sources'] = 'multi'  # noqa: E501
+        if 'autoscaler_resource_bundles' in local_var_params and local_var_params['autoscaler_resource_bundles'] is not None:  # noqa: E501
+            query_params.append(('autoscaler_resource_bundles', local_var_params['autoscaler_resource_bundles']))  # noqa: E501
+            collection_formats['autoscaler_resource_bundles'] = 'multi'  # 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(
+            '/api/v2/sessions/{session_id}/cluster_events', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='ClustereventListResponse',  # 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_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get(self, session_id, **kwargs):  # noqa: E501
         """Get Cluster Product Autoscaler Flag  # noqa: E501
 
@@ -16783,7 +17053,7 @@ class DefaultApi(object):
     def get_events_api_v2_sessions_session_id_events_get(self, session_id, start_index, end_index, **kwargs):  # noqa: E501
         """Get Events  # noqa: E501
 
-        Return events for this cluster.  # noqa: E501
+        Return events for this cluster.  There are multiple endpoints to return events.  This endpoint is for the dataplane to fetch events to power the oss ray dashboard.  The /cluster_events endpoint is used by the UI to fetch events to power general debugging UI features like train dashboard.  Finally, there are events endpoints for each workload type (workspaces, jobs, services). Those endpoints have some additional logic to stitch together cluster events with workload-specific events.  # 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_events_api_v2_sessions_session_id_events_get(session_id, start_index, end_index, async_req=True)
@@ -16811,7 +17081,7 @@ class DefaultApi(object):
     def get_events_api_v2_sessions_session_id_events_get_with_http_info(self, session_id, start_index, end_index, **kwargs):  # noqa: E501
         """Get Events  # noqa: E501
 
-        Return events for this cluster.  # noqa: E501
+        Return events for this cluster.  There are multiple endpoints to return events.  This endpoint is for the dataplane to fetch events to power the oss ray dashboard.  The /cluster_events endpoint is used by the UI to fetch events to power general debugging UI features like train dashboard.  Finally, there are events endpoints for each workload type (workspaces, jobs, services). Those endpoints have some additional logic to stitch together cluster events with workload-specific events.  # 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_events_api_v2_sessions_session_id_events_get_with_http_info(session_id, start_index, end_index, async_req=True)

anyscale/client/openapi_client/models/__init__.py

@@ -146,6 +146,7 @@ from openapi_client.models.cluster_config import ClusterConfig
 from openapi_client.models.cluster_config_with_session_idle_timeout import ClusterConfigWithSessionIdleTimeout
 from openapi_client.models.cluster_environments_query import ClusterEnvironmentsQuery
 from openapi_client.models.cluster_event import ClusterEvent
+from openapi_client.models.cluster_event_source import ClusterEventSource
 from openapi_client.models.cluster_events_output import ClusterEventsOutput
 from openapi_client.models.cluster_features import ClusterFeatures
 from openapi_client.models.cluster_management_stack_versions import ClusterManagementStackVersions
@@ -156,6 +157,7 @@ from openapi_client.models.cluster_status_details import ClusterStatusDetails
 from openapi_client.models.clusterauthresponse_response import ClusterauthresponseResponse
 from openapi_client.models.clusterconfig_response import ClusterconfigResponse
 from openapi_client.models.clusterconfigwithsessionidletimeout_response import ClusterconfigwithsessionidletimeoutResponse
+from openapi_client.models.clusterevent_list_response import ClustereventListResponse
 from openapi_client.models.clustereventsoutput_response import ClustereventsoutputResponse
 from openapi_client.models.clusterfeatures_response import ClusterfeaturesResponse
 from openapi_client.models.compute_node_type import ComputeNodeType
@@ -599,6 +601,7 @@ from openapi_client.models.ux_instance import UXInstance
 from openapi_client.models.unified_job_sort_field import UnifiedJobSortField
 from openapi_client.models.unified_job_status import UnifiedJobStatus
 from openapi_client.models.unified_job_type import UnifiedJobType
+from openapi_client.models.update_cloud_collaborator import UpdateCloudCollaborator
 from openapi_client.models.update_cloud_with_cloud_resource import UpdateCloudWithCloudResource
 from openapi_client.models.update_cloud_with_cloud_resource_gcp import UpdateCloudWithCloudResourceGCP
 from openapi_client.models.update_cluster_dns import UpdateClusterDns

anyscale/client/openapi_client/models/cluster_event_source.py

@@ -0,0 +1,105 @@
+# 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 ClusterEventSource(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    AUTOSCALER = "AUTOSCALER"
+    CLUSTER_LIFECYCLE = "CLUSTER_LIFECYCLE"
+    IDLE_TERMINATION = "IDLE_TERMINATION"
+    JOBS = "JOBS"
+    WORKSPACE_SNAPSHOT = "WORKSPACE_SNAPSHOT"
+    OS = "OS"
+    UNKNOWN = "UNKNOWN"
+
+    allowable_values = [AUTOSCALER, CLUSTER_LIFECYCLE, IDLE_TERMINATION, JOBS, WORKSPACE_SNAPSHOT, OS, UNKNOWN]  # 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
+        """ClusterEventSource - 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, ClusterEventSource):
+            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, ClusterEventSource):
+            return True
+
+        return self.to_dict() != other.to_dict()