anyscale 0.26.14__py3-none-any.whl → 0.26.15__py3-none-any.whl

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -1102,7 +1102,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         return self._clusters_headnode_ip.get(cluster_id, "")
 
     def get_cluster_ssh_key(self, cluster_id: str) -> SessionSshKey:
-        return self._clusters_ssh_key.get(cluster_id, None)
+        return self._clusters_ssh_key.get(cluster_id, None)  # type: ignore
 
     def get_workspace_default_dir_name(self, workspace_id) -> str:
         workspace = self._workspaces.get(workspace_id, None)

anyscale/_private/models/model_base.py

@@ -50,15 +50,18 @@ class ModelEnumType(EnumMeta):
         return cls
 
 
+M = TypeVar("M", bound="ModelEnum")
+
+
 class ModelEnum(str, Enum, metaclass=ModelEnumType):
     # Must be populated by subclasses to document every value.
-    __docstrings__: Dict["ModelEnum", str] = {}
+    __docstrings__: Dict[str, str] = {}
 
     def __str__(self):
         return self.name
 
     @classmethod
-    def validate(cls, value: Union[str, "ModelEnum"]) -> "ModelEnum":
+    def validate(cls: Type[M], value: Union[str, M]) -> M:
         allowed_values = [str(s) for s in cls.__members__ if not str(s).startswith("_")]
         if value.upper() in allowed_values:
             return cls(value.upper())

anyscale/_private/utils/progress_util.py

@@ -1,4 +1,4 @@
-from io import FileIO
+from io import BufferedReader
 import os
 
 from rich.progress import (
@@ -57,7 +57,7 @@ class ProgressFileReader:
     """
 
     def __init__(
-        self, file: FileIO, progress: Progress, task_id: TaskID,
+        self, file: BufferedReader, progress: Progress, task_id: TaskID,
     ):
         self.file = file
         self.progress = progress

anyscale/_private/workload/workload_sdk.py

@@ -138,6 +138,7 @@ class WorkloadSDK(BaseSDK):
                 return local_path_to_uri[target]
 
             self.logger.info(f"Uploading local dir '{target}' to cloud storage.")
+            assert cloud_id is not None
             uri = self._client.upload_local_dir_to_cloud_storage(
                 target, cloud_id=cloud_id, excludes=excludes,
             )
@@ -229,11 +230,11 @@ class WorkloadSDK(BaseSDK):
                 self.client.get_cloud_id(compute_config_id=compute_config_id),
             )
         elif compute_config is None:
-            cloud_id = self.client.get_cloud_id(cloud_name=cloud)
+            cloud_id = self.client.get_cloud_id(cloud_name=cloud)  # type: ignore
             compute_config_id = self._client.get_default_compute_config(
                 cloud_id=cloud_id
             ).id
-            return (compute_config_id, cloud_id)
+            return (compute_config_id, cloud_id)  # type: ignore
         elif cloud is None:
             compute_config_id = self._resolve_compute_config_id(
                 compute_config=compute_config
@@ -256,7 +257,7 @@ class WorkloadSDK(BaseSDK):
             cloud_id_from_cc = self.client.get_cloud_id(
                 compute_config_id=compute_config_id
             )
-            cloud_id_from_cloud = self.client.get_cloud_id(cloud_name=cloud)
+            cloud_id_from_cloud = self.client.get_cloud_id(cloud_name=cloud)  # type: ignore
             if cloud_id_from_cc != cloud_id_from_cloud:
                 raise ValueError(
                     "Cloud does not match from provided `cloud` and `compute_config`. Either pass one of these or ensure they match."

anyscale/anyscale-cloud-setup.yaml

@@ -193,6 +193,10 @@ $SUBNETS_ROUTE_TABLE_ASSOCIATION
               - "s3:DeleteObject"
               - "s3:GetObject"
               - "s3:ListBucket"
+              - "s3:ListBucketMultipartUploads"
+              - "s3:ListMultipartUploadParts"
+              - "s3:AbortMultipartUpload"
+              - "s3:GetBucketLocation"
             Effect: Allow
             Resource:
               - !Sub arn:aws:s3:::${S3Bucket}

anyscale/client/README.md

@@ -199,6 +199,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_dataset_api_v2_datasets_dataset_id_get**](docs/DefaultApi.md#get_dataset_api_v2_datasets_dataset_id_get) | **GET** /api/v2/datasets/{dataset_id} | Get Dataset
 *DefaultApi* | [**get_dataset_download_info_api_v2_datasets_download_info_get**](docs/DefaultApi.md#get_dataset_download_info_api_v2_datasets_download_info_get) | **GET** /api/v2/datasets/download_info | Get Dataset Download Info
 *DefaultApi* | [**get_dataset_download_url_api_v2_datasets_download_get**](docs/DefaultApi.md#get_dataset_download_url_api_v2_datasets_download_get) | **GET** /api/v2/datasets/download | Get Dataset Download Url
+*DefaultApi* | [**get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get**](docs/DefaultApi.md#get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get) | **GET** /api/v2/dataset_runs/dataset_logs | Get Dataset Logs Download
 *DefaultApi* | [**get_decorated_cluster_api_v2_decorated_sessions_cluster_id_get**](docs/DefaultApi.md#get_decorated_cluster_api_v2_decorated_sessions_cluster_id_get) | **GET** /api/v2/decorated_sessions/{cluster_id} | Get Decorated Cluster
 *DefaultApi* | [**get_decorated_interactive_session_api_v2_decorated_interactive_sessions_interactive_session_id_get**](docs/DefaultApi.md#get_decorated_interactive_session_api_v2_decorated_interactive_sessions_interactive_session_id_get) | **GET** /api/v2/decorated_interactive_sessions/{interactive_session_id} | Get Decorated Interactive Session
 *DefaultApi* | [**get_decorated_job_api_v2_decorated_jobs_job_id_get**](docs/DefaultApi.md#get_decorated_job_api_v2_decorated_jobs_job_id_get) | **GET** /api/v2/decorated_jobs/{job_id} | Get Decorated Job
@@ -334,7 +335,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**redirect_to_service_api_v2_sessions_cluster_id_services_get**](docs/DefaultApi.md#redirect_to_service_api_v2_sessions_cluster_id_services_get) | **GET** /api/v2/sessions/{cluster_id}/services | Redirect To Service
 *DefaultApi* | [**redirect_to_tools_api_v2_services_v2_service_id_tools_get**](docs/DefaultApi.md#redirect_to_tools_api_v2_services_v2_service_id_tools_get) | **GET** /api/v2/services-v2/{service_id}/tools | Redirect To Tools
 *DefaultApi* | [**register_api_v2_kubernetes_manager_register_post**](docs/DefaultApi.md#register_api_v2_kubernetes_manager_register_post) | **POST** /api/v2/kubernetes_manager/register | Register
-*DefaultApi* | [**register_buffer_api_v2_anyscaled_register_buffer_post**](docs/DefaultApi.md#register_buffer_api_v2_anyscaled_register_buffer_post) | **POST** /api/v2/anyscaled/register_buffer | Register Buffer
 *DefaultApi* | [**register_build_api_v2_anyscaled_register_build_post**](docs/DefaultApi.md#register_build_api_v2_anyscaled_register_build_post) | **POST** /api/v2/anyscaled/register_build | Register Build
 *DefaultApi* | [**register_machine_api_v2_anyscaled_register_machine_post**](docs/DefaultApi.md#register_machine_api_v2_anyscaled_register_machine_post) | **POST** /api/v2/anyscaled/register_machine | Register Machine
 *DefaultApi* | [**register_user_api_v2_users_post**](docs/DefaultApi.md#register_user_api_v2_users_post) | **POST** /api/v2/users/ | Register User
@@ -444,7 +444,6 @@ Class | Method | HTTP request | Description
  - [BillingInformation](docs/BillingInformation.md)
  - [BillingVersionCode](docs/BillingVersionCode.md)
  - [BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost](docs/BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost.md)
- - [BufferRegistration](docs/BufferRegistration.md)
  - [Build](docs/Build.md)
  - [BuildLogResponse](docs/BuildLogResponse.md)
  - [BuildRegistration](docs/BuildRegistration.md)
@@ -913,18 +912,18 @@ Class | Method | HTTP request | Description
  - [SupportRequestsQuery](docs/SupportRequestsQuery.md)
  - [SystemWorkloadName](docs/SystemWorkloadName.md)
  - [TaskExceptionGroupAggregate](docs/TaskExceptionGroupAggregate.md)
+ - [TaskExceptionGroupAggregateResponse](docs/TaskExceptionGroupAggregateResponse.md)
  - [TaskFunctionNameGroupAggregate](docs/TaskFunctionNameGroupAggregate.md)
+ - [TaskFunctionNameGroupAggregateResponse](docs/TaskFunctionNameGroupAggregateResponse.md)
  - [TaskGroupedAggregateMetrics](docs/TaskGroupedAggregateMetrics.md)
  - [TaskJobGroupAggregate](docs/TaskJobGroupAggregate.md)
+ - [TaskJobGroupAggregateResponse](docs/TaskJobGroupAggregateResponse.md)
  - [TaskState](docs/TaskState.md)
  - [TaskSummary](docs/TaskSummary.md)
  - [TaskTableConfig](docs/TaskTableConfig.md)
  - [TaskTableResponse](docs/TaskTableResponse.md)
  - [TaskTableRow](docs/TaskTableRow.md)
  - [TaskType](docs/TaskType.md)
- - [TaskexceptiongroupaggregateListResponse](docs/TaskexceptiongroupaggregateListResponse.md)
- - [TaskfunctionnamegroupaggregateListResponse](docs/TaskfunctionnamegroupaggregateListResponse.md)
- - [TaskjobgroupaggregateListResponse](docs/TaskjobgroupaggregateListResponse.md)
  - [TasksummaryResponse](docs/TasksummaryResponse.md)
  - [TemplatizedComputeConfigs](docs/TemplatizedComputeConfigs.md)
  - [TemplatizedDecoratedApplicationTemplates](docs/TemplatizedDecoratedApplicationTemplates.md)

anyscale/client/openapi_client/__init__.py

@@ -75,7 +75,6 @@ from openapi_client.models.batched_result_organization_invitation_base import Ba
 from openapi_client.models.billing_information import BillingInformation
 from openapi_client.models.billing_version_code import BillingVersionCode
 from openapi_client.models.body_aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post import BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost
-from openapi_client.models.buffer_registration import BufferRegistration
 from openapi_client.models.build import Build
 from openapi_client.models.build_log_response import BuildLogResponse
 from openapi_client.models.build_registration import BuildRegistration
@@ -544,18 +543,18 @@ from openapi_client.models.subnet_id_with_availability_zone_aws import SubnetIdW
 from openapi_client.models.support_requests_query import SupportRequestsQuery
 from openapi_client.models.system_workload_name import SystemWorkloadName
 from openapi_client.models.task_exception_group_aggregate import TaskExceptionGroupAggregate
+from openapi_client.models.task_exception_group_aggregate_response import TaskExceptionGroupAggregateResponse
 from openapi_client.models.task_function_name_group_aggregate import TaskFunctionNameGroupAggregate
+from openapi_client.models.task_function_name_group_aggregate_response import TaskFunctionNameGroupAggregateResponse
 from openapi_client.models.task_grouped_aggregate_metrics import TaskGroupedAggregateMetrics
 from openapi_client.models.task_job_group_aggregate import TaskJobGroupAggregate
+from openapi_client.models.task_job_group_aggregate_response import TaskJobGroupAggregateResponse
 from openapi_client.models.task_state import TaskState
 from openapi_client.models.task_summary import TaskSummary
 from openapi_client.models.task_table_config import TaskTableConfig
 from openapi_client.models.task_table_response import TaskTableResponse
 from openapi_client.models.task_table_row import TaskTableRow
 from openapi_client.models.task_type import TaskType
-from openapi_client.models.taskexceptiongroupaggregate_list_response import TaskexceptiongroupaggregateListResponse
-from openapi_client.models.taskfunctionnamegroupaggregate_list_response import TaskfunctionnamegroupaggregateListResponse
-from openapi_client.models.taskjobgroupaggregate_list_response import TaskjobgroupaggregateListResponse
 from openapi_client.models.tasksummary_response import TasksummaryResponse
 from openapi_client.models.templatized_compute_configs import TemplatizedComputeConfigs
 from openapi_client.models.templatized_decorated_application_templates import TemplatizedDecoratedApplicationTemplates

anyscale/client/openapi_client/api/default_api.py

@@ -14694,6 +14694,155 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get(self, cluster_id, session_name, dataset_id, **kwargs):  # noqa: E501
+        """Get Dataset Logs Download  # noqa: E501
+
+        Fetches logs of the dataset. The logs will be fetched from the streaming logs s3 bucket.  If a pagination token is provided, we use the streaming logs or downloaded logs bucket depending on the format of the pagination token.  # 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_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get(cluster_id, session_name, dataset_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cluster_id: (required)
+        :param str session_name: (required)
+        :param str dataset_id: (required)
+        :param int page_size: Number of chunks to fetch. Defaults to 400. Max of 1000.
+        :param str next_page_token: Pagination token for getting newer logs.
+        :param str previous_page_token: Pagination token for getting older logs.
+        :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: LogdownloadresultResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get_with_http_info(cluster_id, session_name, dataset_id, **kwargs)  # noqa: E501
+
+    def get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get_with_http_info(self, cluster_id, session_name, dataset_id, **kwargs):  # noqa: E501
+        """Get Dataset Logs Download  # noqa: E501
+
+        Fetches logs of the dataset. The logs will be fetched from the streaming logs s3 bucket.  If a pagination token is provided, we use the streaming logs or downloaded logs bucket depending on the format of the pagination token.  # 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_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get_with_http_info(cluster_id, session_name, dataset_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cluster_id: (required)
+        :param str session_name: (required)
+        :param str dataset_id: (required)
+        :param int page_size: Number of chunks to fetch. Defaults to 400. Max of 1000.
+        :param str next_page_token: Pagination token for getting newer logs.
+        :param str previous_page_token: Pagination token for getting older logs.
+        :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(LogdownloadresultResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'cluster_id',
+            'session_name',
+            'dataset_id',
+            'page_size',
+            'next_page_token',
+            'previous_page_token'
+        ]
+        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_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'cluster_id' is set
+        if self.api_client.client_side_validation and ('cluster_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['cluster_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `cluster_id` when calling `get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get`")  # noqa: E501
+        # verify the required parameter 'session_name' is set
+        if self.api_client.client_side_validation and ('session_name' not in local_var_params or  # noqa: E501
+                                                        local_var_params['session_name'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `session_name` when calling `get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get`")  # noqa: E501
+        # verify the required parameter 'dataset_id' is set
+        if self.api_client.client_side_validation and ('dataset_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['dataset_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `dataset_id` when calling `get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get`")  # noqa: E501
+
+        if self.api_client.client_side_validation and 'page_size' in local_var_params and local_var_params['page_size'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `page_size` when calling `get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get`, must be a value less than or equal to `1000`")  # noqa: E501
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'cluster_id' in local_var_params and local_var_params['cluster_id'] is not None:  # noqa: E501
+            query_params.append(('cluster_id', local_var_params['cluster_id']))  # noqa: E501
+        if 'session_name' in local_var_params and local_var_params['session_name'] is not None:  # noqa: E501
+            query_params.append(('session_name', local_var_params['session_name']))  # noqa: E501
+        if 'dataset_id' in local_var_params and local_var_params['dataset_id'] is not None:  # noqa: E501
+            query_params.append(('dataset_id', local_var_params['dataset_id']))  # noqa: E501
+        if 'page_size' in local_var_params and local_var_params['page_size'] is not None:  # noqa: E501
+            query_params.append(('page_size', local_var_params['page_size']))  # noqa: E501
+        if 'next_page_token' in local_var_params and local_var_params['next_page_token'] is not None:  # noqa: E501
+            query_params.append(('next_page_token', local_var_params['next_page_token']))  # noqa: E501
+        if 'previous_page_token' in local_var_params and local_var_params['previous_page_token'] is not None:  # noqa: E501
+            query_params.append(('previous_page_token', local_var_params['previous_page_token']))  # 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/dataset_runs/dataset_logs', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='LogdownloadresultResponse',  # 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_decorated_cluster_api_v2_decorated_sessions_cluster_id_get(self, cluster_id, **kwargs):  # noqa: E501
         """Get Decorated Cluster  # noqa: E501
 
@@ -21881,7 +22030,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: TaskexceptiongroupaggregateListResponse
+        :return: TaskExceptionGroupAggregateResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -21907,7 +22056,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(TaskexceptiongroupaggregateListResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(TaskExceptionGroupAggregateResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -21968,7 +22117,7 @@ class DefaultApi(object):
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='TaskexceptiongroupaggregateListResponse',  # noqa: E501
+            response_type='TaskExceptionGroupAggregateResponse',  # 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
@@ -21993,7 +22142,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: TaskfunctionnamegroupaggregateListResponse
+        :return: TaskFunctionNameGroupAggregateResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -22019,7 +22168,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(TaskfunctionnamegroupaggregateListResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(TaskFunctionNameGroupAggregateResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -22080,7 +22229,7 @@ class DefaultApi(object):
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='TaskfunctionnamegroupaggregateListResponse',  # noqa: E501
+            response_type='TaskFunctionNameGroupAggregateResponse',  # 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
@@ -22105,7 +22254,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: TaskjobgroupaggregateListResponse
+        :return: TaskJobGroupAggregateResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -22131,7 +22280,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(TaskjobgroupaggregateListResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(TaskJobGroupAggregateResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -22192,7 +22341,7 @@ class DefaultApi(object):
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='TaskjobgroupaggregateListResponse',  # noqa: E501
+            response_type='TaskJobGroupAggregateResponse',  # 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
@@ -31209,124 +31358,6 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def register_buffer_api_v2_anyscaled_register_buffer_post(self, buffer_registration, **kwargs):  # noqa: E501
-        """Register Buffer  # noqa: E501
-
-        Register a node to receive a per-buffer token.  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.register_buffer_api_v2_anyscaled_register_buffer_post(buffer_registration, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool: execute request asynchronously
-        :param BufferRegistration buffer_registration: (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: AnyscaledcredentialresponseResponse
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-        kwargs['_return_http_data_only'] = True
-        return self.register_buffer_api_v2_anyscaled_register_buffer_post_with_http_info(buffer_registration, **kwargs)  # noqa: E501
-
-    def register_buffer_api_v2_anyscaled_register_buffer_post_with_http_info(self, buffer_registration, **kwargs):  # noqa: E501
-        """Register Buffer  # noqa: E501
-
-        Register a node to receive a per-buffer token.  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.register_buffer_api_v2_anyscaled_register_buffer_post_with_http_info(buffer_registration, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool: execute request asynchronously
-        :param BufferRegistration buffer_registration: (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: tuple(AnyscaledcredentialresponseResponse, status_code(int), headers(HTTPHeaderDict))
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-
-        local_var_params = locals()
-
-        all_params = [
-            'buffer_registration'
-        ]
-        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 register_buffer_api_v2_anyscaled_register_buffer_post" % key
-                )
-            local_var_params[key] = val
-        del local_var_params['kwargs']
-        # verify the required parameter 'buffer_registration' is set
-        if self.api_client.client_side_validation and ('buffer_registration' not in local_var_params or  # noqa: E501
-                                                        local_var_params['buffer_registration'] is None):  # noqa: E501
-            raise ApiValueError("Missing the required parameter `buffer_registration` when calling `register_buffer_api_v2_anyscaled_register_buffer_post`")  # noqa: E501
-
-        collection_formats = {}
-
-        path_params = {}
-
-        query_params = []
-
-        header_params = {}
-
-        form_params = []
-        local_var_files = {}
-
-        body_params = None
-        if 'buffer_registration' in local_var_params:
-            body_params = local_var_params['buffer_registration']
-        # 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/anyscaled/register_buffer', 'POST',
-            path_params,
-            query_params,
-            header_params,
-            body=body_params,
-            post_params=form_params,
-            files=local_var_files,
-            response_type='AnyscaledcredentialresponseResponse',  # 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 register_build_api_v2_anyscaled_register_build_post(self, build_registration, **kwargs):  # noqa: E501
         """Register Build  # noqa: E501
 

anyscale/client/openapi_client/models/__init__.py

@@ -61,7 +61,6 @@ from openapi_client.models.batched_result_organization_invitation_base import Ba
 from openapi_client.models.billing_information import BillingInformation
 from openapi_client.models.billing_version_code import BillingVersionCode
 from openapi_client.models.body_aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post import BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost
-from openapi_client.models.buffer_registration import BufferRegistration
 from openapi_client.models.build import Build
 from openapi_client.models.build_log_response import BuildLogResponse
 from openapi_client.models.build_registration import BuildRegistration
@@ -530,18 +529,18 @@ from openapi_client.models.subnet_id_with_availability_zone_aws import SubnetIdW
 from openapi_client.models.support_requests_query import SupportRequestsQuery
 from openapi_client.models.system_workload_name import SystemWorkloadName
 from openapi_client.models.task_exception_group_aggregate import TaskExceptionGroupAggregate
+from openapi_client.models.task_exception_group_aggregate_response import TaskExceptionGroupAggregateResponse
 from openapi_client.models.task_function_name_group_aggregate import TaskFunctionNameGroupAggregate
+from openapi_client.models.task_function_name_group_aggregate_response import TaskFunctionNameGroupAggregateResponse
 from openapi_client.models.task_grouped_aggregate_metrics import TaskGroupedAggregateMetrics
 from openapi_client.models.task_job_group_aggregate import TaskJobGroupAggregate
+from openapi_client.models.task_job_group_aggregate_response import TaskJobGroupAggregateResponse
 from openapi_client.models.task_state import TaskState
 from openapi_client.models.task_summary import TaskSummary
 from openapi_client.models.task_table_config import TaskTableConfig
 from openapi_client.models.task_table_response import TaskTableResponse
 from openapi_client.models.task_table_row import TaskTableRow
 from openapi_client.models.task_type import TaskType
-from openapi_client.models.taskexceptiongroupaggregate_list_response import TaskexceptiongroupaggregateListResponse
-from openapi_client.models.taskfunctionnamegroupaggregate_list_response import TaskfunctionnamegroupaggregateListResponse
-from openapi_client.models.taskjobgroupaggregate_list_response import TaskjobgroupaggregateListResponse
 from openapi_client.models.tasksummary_response import TasksummaryResponse
 from openapi_client.models.templatized_compute_configs import TemplatizedComputeConfigs
 from openapi_client.models.templatized_decorated_application_templates import TemplatizedDecoratedApplicationTemplates

anyscale/client/openapi_client/models/task_exception_group_aggregate.py

@@ -34,25 +34,29 @@ class TaskExceptionGroupAggregate(object):
     """
     openapi_types = {
         'exception_type': 'str',
+        'ray_session_name': 'str',
         'metrics': 'TaskGroupedAggregateMetrics'
     }
 
     attribute_map = {
         'exception_type': 'exception_type',
+        'ray_session_name': 'ray_session_name',
         'metrics': 'metrics'
     }
 
-    def __init__(self, exception_type=None, metrics=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, exception_type=None, ray_session_name=None, metrics=None, local_vars_configuration=None):  # noqa: E501
         """TaskExceptionGroupAggregate - 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._exception_type = None
+        self._ray_session_name = None
         self._metrics = None
         self.discriminator = None
 
         self.exception_type = exception_type
+        self.ray_session_name = ray_session_name
         self.metrics = metrics
 
     @property
@@ -78,6 +82,29 @@ class TaskExceptionGroupAggregate(object):
 
         self._exception_type = exception_type
 
+    @property
+    def ray_session_name(self):
+        """Gets the ray_session_name of this TaskExceptionGroupAggregate.  # noqa: E501
+
+
+        :return: The ray_session_name of this TaskExceptionGroupAggregate.  # noqa: E501
+        :rtype: str
+        """
+        return self._ray_session_name
+
+    @ray_session_name.setter
+    def ray_session_name(self, ray_session_name):
+        """Sets the ray_session_name of this TaskExceptionGroupAggregate.
+
+
+        :param ray_session_name: The ray_session_name of this TaskExceptionGroupAggregate.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and ray_session_name is None:  # noqa: E501
+            raise ValueError("Invalid value for `ray_session_name`, must not be `None`")  # noqa: E501
+
+        self._ray_session_name = ray_session_name
+
     @property
     def metrics(self):
         """Gets the metrics of this TaskExceptionGroupAggregate.  # noqa: E501