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

anyscale/__init__.py

@@ -27,6 +27,7 @@ from anyscale import (
     llm,
     schedule,
     service,
+    user,
 )
 from anyscale._private.anyscale_client import AnyscaleClient, AnyscaleClientInterface
 from anyscale._private.sdk.base_sdk import Timer
@@ -43,6 +44,7 @@ from anyscale.llm import LLMSDK
 from anyscale.schedule import ScheduleSDK
 from anyscale.sdk.anyscale_client.sdk import AnyscaleSDK
 from anyscale.service import ServiceSDK
+from anyscale.user import UserSDK
 from anyscale.workspace import WorkspaceSDK
 
 
@@ -108,6 +110,7 @@ class Anyscale:
         self._schedule_sdk = ScheduleSDK(client=self._anyscale_client)
         self._image_sdk = ImageSDK(client=self._anyscale_client)
         self._llm_sdk = LLMSDK(client=self._anyscale_client)
+        self._user_sdk = UserSDK(client=self._anyscale_client)
         self._workspace_sdk = WorkspaceSDK(client=self._anyscale_client)
 
     @classmethod
@@ -132,6 +135,9 @@ class Anyscale:
             client=client, logger=logger, timer=timer,
         )
         obj._image_sdk = ImageSDK(client=client, logger=logger)  # noqa: SLF001
+        obj._user_sdk = UserSDK(  # noqa: SLF001
+            client=client, logger=logger, timer=timer
+        )
         obj._workspace_sdk = WorkspaceSDK(  # noqa: SLF001
             client=client, logger=logger, timer=timer,
         )
@@ -165,6 +171,10 @@ class Anyscale:
     def llm(self) -> LLMSDK:  # noqa: F811
         return self._llm_sdk
 
+    @property
+    def user(self) -> UserSDK:  # noqa: F811
+        return self._user_sdk
+
     @property
     def workspace(self) -> WorkspaceSDK:  # noqa: F811
         return self._workspace_sdk

anyscale/_private/anyscale_client/anyscale_client.py

@@ -34,6 +34,8 @@ from anyscale.authenticate import AuthenticationBlock, get_auth_api_client
 from anyscale.cli_logger import BlockLogger
 from anyscale.client.openapi_client.api.default_api import DefaultApi as InternalApi
 from anyscale.client.openapi_client.models import (
+    AdminCreatedUser,
+    AdminCreateUser,
     ArchiveStatus,
     Cloud,
     CloudDataBucketAccessMode,
@@ -1845,3 +1847,11 @@ class AnyscaleClient(AnyscaleClientInterface):
             )
 
         return filepath
+
+    @handle_api_exceptions
+    def admin_batch_create_users(
+        self, admin_create_users: List[AdminCreateUser]
+    ) -> List[AdminCreatedUser]:
+        return self._internal_api_client.admin_batch_create_users_api_v2_users_admin_batch_create_post(
+            admin_create_users
+        ).results

anyscale/_private/anyscale_client/common.py

@@ -4,6 +4,8 @@ from typing import Dict, Generator, List, Optional, Tuple
 
 from anyscale._private.models.image_uri import ImageURI
 from anyscale.client.openapi_client.models import (
+    AdminCreatedUser,
+    AdminCreateUser,
     Cloud,
     ComputeTemplateConfig,
     CreateExperimentalWorkspace,
@@ -591,3 +593,10 @@ class AnyscaleClientInterface(ABC):
     ) -> str:
         """Download the aggregated instance usage csv."""
         raise NotImplementedError
+
+    @abstractmethod
+    def admin_batch_create_users(
+        self, admin_create_users: List[AdminCreateUser]
+    ) -> List[AdminCreatedUser]:
+        """Batch create users without email verification as an admin."""
+        raise NotImplementedError

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -14,6 +14,8 @@ from anyscale._private.models.image_uri import ImageURI
 from anyscale._private.models.model_base import ListResponse
 from anyscale.cli_logger import BlockLogger
 from anyscale.client.openapi_client.models import (
+    AdminCreatedUser,
+    AdminCreateUser,
     Cloud,
     ComputeTemplateConfig,
     CreateExperimentalWorkspace,
@@ -146,6 +148,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._controller_logs: Dict[str, str] = {}
         self._schedules: Dict[str, DecoratedSchedule] = {}
         self._schedule_trigger_counts: Dict[str, int] = defaultdict(int)
+        self._users: Dict[str, AdminCreatedUser] = {}
         self._workspaces: Dict[str, ExperimentalWorkspace] = {}
         self._workspaces_dependencies: Dict[str, List[str]] = {}
         self._workspaces_env_vars: Dict[str, Dict[str, str]] = {}
@@ -1078,3 +1081,28 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             filepath = filename
 
         return filepath
+
+    def admin_batch_create_users(
+        self, admin_create_users: List[AdminCreateUser]
+    ) -> List[AdminCreatedUser]:
+        created_users = []
+        for admin_create_user in admin_create_users:
+            if admin_create_user.email in self._users:
+                raise ValueError(
+                    f"User with email '{admin_create_user.email}' already exists."
+                )
+
+            user_id = f"user-id-{uuid.uuid4()!s}"
+            created_user = AdminCreatedUser(
+                user_id=user_id,
+                email=admin_create_user.email,
+                created_at=datetime.utcnow(),
+                name=admin_create_user.name,
+                lastname=admin_create_user.lastname,
+                title=admin_create_user.title,
+                is_sso_user=admin_create_user.is_sso_user,
+            )
+            created_users.append(created_user)
+            self._users[admin_create_user.email] = created_user
+
+        return created_users

anyscale/_private/docgen/__main__.py

@@ -27,6 +27,7 @@ from anyscale.commands import (
     schedule_commands,
     service_account_commands,
     service_commands,
+    user_commands,
     workspace_commands,
     workspace_commands_v2,
 )
@@ -67,6 +68,7 @@ from anyscale.service.models import (
     ServiceVersionStatus,
     TracingConfig,
 )
+from anyscale.user.models import AdminCreatedUser, AdminCreateUser
 from anyscale.workspace.models import WorkspaceConfig
 
 
@@ -81,6 +83,15 @@ ALL_MODULES = [
         sdk_commands=[anyscale.aggregated_instance_usage.download_csv,],
         models=[DownloadCSVFilters],
     ),
+    Module(
+        title="User",
+        filename="user.md",
+        cli_prefix="user",
+        cli_commands=[user_commands.admin_batch_create],
+        sdk_prefix="anyscale.user",
+        sdk_commands=[anyscale.user.admin_batch_create,],
+        models=[AdminCreateUser, AdminCreatedUser],
+    ),
     Module(
         title="Job",
         filename="job-api.md",
@@ -389,6 +400,7 @@ ALL_MODULES = [
             cloud_commands.cloud_verify,
             cloud_commands.list_cloud,
             cloud_commands.cloud_config_update,
+            cloud_commands.cloud_set_default,
         ],
         sdk_prefix="anyscale.cloud",
         sdk_commands=[],

anyscale/_private/docgen/generator.py

@@ -26,6 +26,7 @@ CLI_NO_EXAMPLES = {
     "anyscale",
     "anyscale cloud",
     "anyscale cloud config",
+    "anyscale cloud set-default",
     "anyscale logs",
     "anyscale service-account",
     "anyscale service",

anyscale/client/README.md

@@ -76,6 +76,7 @@ All URIs are relative to *http://localhost*
 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_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
@@ -430,6 +431,9 @@ Class | Method | HTTP request | Description
  - [AWSCredentials](docs/AWSCredentials.md)
  - [AWSMemoryDBClusterConfig](docs/AWSMemoryDBClusterConfig.md)
  - [AccessConfig](docs/AccessConfig.md)
+ - [AdminCreateUser](docs/AdminCreateUser.md)
+ - [AdminCreatedUser](docs/AdminCreatedUser.md)
+ - [AdmincreateduserListResponse](docs/AdmincreateduserListResponse.md)
  - [AggregatedInstanceUsageCSV](docs/AggregatedInstanceUsageCSV.md)
  - [AggregatedUsage](docs/AggregatedUsage.md)
  - [AggregatedUsageQuery](docs/AggregatedUsageQuery.md)
@@ -863,7 +867,6 @@ Class | Method | HTTP request | Description
  - [OrganizationpublicidentifierResponse](docs/OrganizationpublicidentifierResponse.md)
  - [OrganizationusagealertListResponse](docs/OrganizationusagealertListResponse.md)
  - [PageQuery](docs/PageQuery.md)
- - [PagingContext](docs/PagingContext.md)
  - [PauseSchedule](docs/PauseSchedule.md)
  - [PermissionLevel](docs/PermissionLevel.md)
  - [PlatformFineTuningJob](docs/PlatformFineTuningJob.md)

anyscale/client/openapi_client/__init__.py

@@ -31,6 +31,9 @@ from openapi_client.exceptions import ApiException
 from openapi_client.models.aws_credentials import AWSCredentials
 from openapi_client.models.aws_memory_db_cluster_config import AWSMemoryDBClusterConfig
 from openapi_client.models.access_config import AccessConfig
+from openapi_client.models.admin_create_user import AdminCreateUser
+from openapi_client.models.admin_created_user import AdminCreatedUser
+from openapi_client.models.admincreateduser_list_response import AdmincreateduserListResponse
 from openapi_client.models.aggregated_instance_usage_csv import AggregatedInstanceUsageCSV
 from openapi_client.models.aggregated_usage import AggregatedUsage
 from openapi_client.models.aggregated_usage_query import AggregatedUsageQuery
@@ -464,7 +467,6 @@ from openapi_client.models.organizationprojectcollaborator_list_response import
 from openapi_client.models.organizationpublicidentifier_response import OrganizationpublicidentifierResponse
 from openapi_client.models.organizationusagealert_list_response import OrganizationusagealertListResponse
 from openapi_client.models.page_query import PageQuery
-from openapi_client.models.paging_context import PagingContext
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
 from openapi_client.models.platform_fine_tuning_job import PlatformFineTuningJob

anyscale/client/openapi_client/api/default_api.py

@@ -152,6 +152,122 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def admin_batch_create_users_api_v2_users_admin_batch_create_post(self, admin_create_user, **kwargs):  # noqa: E501
+        """Admin Batch Create Users  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.admin_batch_create_users_api_v2_users_admin_batch_create_post(admin_create_user, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param list[AdminCreateUser] admin_create_user: (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: AdmincreateduserListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.admin_batch_create_users_api_v2_users_admin_batch_create_post_with_http_info(admin_create_user, **kwargs)  # noqa: E501
+
+    def admin_batch_create_users_api_v2_users_admin_batch_create_post_with_http_info(self, admin_create_user, **kwargs):  # noqa: E501
+        """Admin Batch Create Users  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.admin_batch_create_users_api_v2_users_admin_batch_create_post_with_http_info(admin_create_user, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param list[AdminCreateUser] admin_create_user: (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(AdmincreateduserListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'admin_create_user'
+        ]
+        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 admin_batch_create_users_api_v2_users_admin_batch_create_post" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'admin_create_user' is set
+        if self.api_client.client_side_validation and ('admin_create_user' not in local_var_params or  # noqa: E501
+                                                        local_var_params['admin_create_user'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `admin_create_user` when calling `admin_batch_create_users_api_v2_users_admin_batch_create_post`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'admin_create_user' in local_var_params:
+            body_params = local_var_params['admin_create_user']
+        # 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/users/admin_batch_create', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='AdmincreateduserListResponse',  # 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 admission_api_v2_kubernetes_manager_admission_cloud_resource_id_post(self, cloud_resource_id, **kwargs):  # noqa: E501
         """Admission  # noqa: E501
 
@@ -10258,6 +10374,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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.
@@ -10282,6 +10400,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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
@@ -10299,7 +10419,9 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'aggregated_usage_query'
+            'aggregated_usage_query',
+            'paging_token',
+            'count'
         ]
         all_params.extend(
             [
@@ -10323,11 +10445,19 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post`")  # noqa: E501
 
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post`, must be a value less than or equal to `50`")  # 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 `fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
         path_params = {}
 
         query_params = []
+        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 = {}
 
@@ -10374,6 +10504,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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.
@@ -10398,6 +10530,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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
@@ -10415,7 +10549,9 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'aggregated_usage_query'
+            'aggregated_usage_query',
+            'paging_token',
+            'count'
         ]
         all_params.extend(
             [
@@ -10439,11 +10575,19 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post`")  # noqa: E501
 
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post`, must be a value less than or equal to `50`")  # 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 `fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
         path_params = {}
 
         query_params = []
+        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 = {}
 
@@ -10490,6 +10634,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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.
@@ -10514,6 +10660,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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
@@ -10531,7 +10679,9 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'aggregated_usage_query'
+            'aggregated_usage_query',
+            'paging_token',
+            'count'
         ]
         all_params.extend(
             [
@@ -10555,11 +10705,19 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post`")  # noqa: E501
 
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post`, must be a value less than or equal to `50`")  # 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 `fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
         path_params = {}
 
         query_params = []
+        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 = {}
 
@@ -10606,6 +10764,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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.
@@ -10630,6 +10790,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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
@@ -10647,7 +10809,9 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'aggregated_usage_query'
+            'aggregated_usage_query',
+            'paging_token',
+            'count'
         ]
         all_params.extend(
             [
@@ -10671,11 +10835,19 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post`")  # noqa: E501
 
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post`, must be a value less than or equal to `50`")  # 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 `fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
         path_params = {}
 
         query_params = []
+        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 = {}
 
@@ -10722,6 +10894,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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.
@@ -10746,6 +10920,8 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param AggregatedUsageQuery aggregated_usage_query: (required)
+        :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
@@ -10763,7 +10939,9 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'aggregated_usage_query'
+            'aggregated_usage_query',
+            'paging_token',
+            'count'
         ]
         all_params.extend(
             [
@@ -10787,11 +10965,19 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post`")  # noqa: E501
 
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post`, must be a value less than or equal to `50`")  # 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 `fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
 
         path_params = {}
 
         query_params = []
+        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 = {}
 

anyscale/client/openapi_client/models/__init__.py

@@ -17,6 +17,9 @@ from __future__ import absolute_import
 from openapi_client.models.aws_credentials import AWSCredentials
 from openapi_client.models.aws_memory_db_cluster_config import AWSMemoryDBClusterConfig
 from openapi_client.models.access_config import AccessConfig
+from openapi_client.models.admin_create_user import AdminCreateUser
+from openapi_client.models.admin_created_user import AdminCreatedUser
+from openapi_client.models.admincreateduser_list_response import AdmincreateduserListResponse
 from openapi_client.models.aggregated_instance_usage_csv import AggregatedInstanceUsageCSV
 from openapi_client.models.aggregated_usage import AggregatedUsage
 from openapi_client.models.aggregated_usage_query import AggregatedUsageQuery
@@ -450,7 +453,6 @@ from openapi_client.models.organizationprojectcollaborator_list_response import
 from openapi_client.models.organizationpublicidentifier_response import OrganizationpublicidentifierResponse
 from openapi_client.models.organizationusagealert_list_response import OrganizationusagealertListResponse
 from openapi_client.models.page_query import PageQuery
-from openapi_client.models.paging_context import PagingContext
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
 from openapi_client.models.platform_fine_tuning_job import PlatformFineTuningJob