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

anyscale/resource_quota/models.py

@@ -0,0 +1,303 @@
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Dict, Optional
+
+from anyscale._private.models import ModelBase
+
+
+@dataclass(frozen=True)
+class CreateResourceQuota(ModelBase):
+    """Resource quota creation model.
+    """
+
+    __doc_py_example__ = """\
+import anyscale
+from anyscale.resource_quota.models import CreateResourceQuota
+
+create_resource_quota = CreateResourceQuota(
+    # Name of the resource quota to create
+    name="resource_quota_name",
+    # Name of the cloud that this resource quota applies to
+    cloud="cloud_name",
+    # Name of the project that this resource quota applies to (optional)
+    project="project_name",
+    # Email of the user that this resource quota applies to (optional)
+    user_email="test@anyscale.com",
+    # The quota limit for the number of CPUs (optional)
+    num_cpus=50,
+    # The quota limit for the number of instances (optional)
+    num_instances=100,
+    # The quota limit for the total number of GPUs (optional)
+    num_gpus=30,
+    # The quota limit for the number of accelerators (optional)
+    num_accelerators={"A100-80G": 10, "T4": 20},
+)
+"""
+
+    name: str = field(metadata={"docstring": "Name of the resource quota to create."})
+
+    def _validate_name(self, name: str):
+        if not isinstance(name, str):
+            raise TypeError("name must be a string.")
+
+    cloud: str = field(
+        metadata={"docstring": "Name of the cloud that this resource quota applies to."}
+    )
+
+    def _validate_cloud(self, cloud: str):
+        if not isinstance(cloud, str):
+            raise TypeError("cloud must be a string.")
+
+    project: Optional[str] = field(
+        default=None,
+        metadata={
+            "docstring": "Name of the project that this resource quota applies to (optional)."
+        },
+    )
+
+    def _validate_project(self, project: Optional[str]):
+        if project is not None and not isinstance(project, str):
+            raise TypeError("project must be a string.")
+
+    user_email: Optional[str] = field(
+        default=None,
+        metadata={
+            "docstring": "Email of the user that this resource quota applies to (optional)."
+        },
+    )
+
+    def _validate_user_email(self, user_email: Optional[str]):
+        if user_email is not None and not isinstance(user_email, str):
+            raise TypeError("user_email must be a string.")
+
+    num_cpus: Optional[int] = field(
+        default=None,
+        metadata={"docstring": "The quota limit for the number of CPUs (optional)."},
+    )
+
+    def _validate_num_cpus(self, num_cpus: Optional[int]):
+        if num_cpus is not None and not isinstance(num_cpus, int):
+            raise TypeError("num_cpus must be an integer.")
+
+    num_instances: Optional[int] = field(
+        default=None,
+        metadata={
+            "docstring": "The quota limit for the number of instances. (optional)."
+        },
+    )
+
+    def _validate_num_instances(self, num_instances: Optional[int]):
+        if num_instances is not None and not isinstance(num_instances, int):
+            raise TypeError("num_instances must be an integer.")
+
+    num_gpus: Optional[int] = field(
+        default=None,
+        metadata={
+            "docstring": "The quota limit for the total number of GPUs (optional)."
+        },
+    )
+
+    def _validate_num_gpus(self, num_gpus: Optional[int]):
+        if num_gpus is not None and not isinstance(num_gpus, int):
+            raise TypeError("num_gpus must be an integer.")
+
+    num_accelerators: Optional[Dict[str, int]] = field(
+        default=None,
+        metadata={
+            "docstring": "The quota limit for the number of accelerators (optional)."
+        },
+    )
+
+    def _validate_num_accelerators(self, num_accelerators: Optional[Dict[str, int]]):
+        if num_accelerators is not None and not isinstance(num_accelerators, dict):
+            raise TypeError("num_accelerators must be a dictionary.")
+        if num_accelerators is not None:
+            for key, value in num_accelerators.items():
+                if not isinstance(key, str):
+                    raise TypeError("num_accelerators keys must be strings.")
+                if not isinstance(value, int):
+                    raise TypeError("num_accelerators values must be integers.")
+
+
+@dataclass(frozen=True)
+class Quota(ModelBase):
+    """Resource quota limit
+    """
+
+    __doc_py_example__ = """\
+import anyscale
+from anyscale.resource_quota.models import CreateResourceQuota, ResourceQuota, Quota
+
+create_resource_quota = CreateResourceQuota(
+    # Name of the resource quota to create
+    name="resource_quota_name",
+    # Name of the cloud that this resource quota applies to
+    cloud="cloud_name",
+    # Name of the project that this resource quota applies to (optional)
+    project="project_name",
+    # Email of the user that this resource quota applies to (optional)
+    user_email="test@anyscale.com",
+    # The quota limit for the number of CPUs (optional)
+    num_cpus=50,
+    # The quota limit for the number of instances (optional)
+    num_instances=100,
+    # The quota limit for the total number of GPUs (optional)
+    num_gpus=30,
+    # The quota limit for the number of accelerators (optional)
+    num_accelerators={"A100-80G": 10, "T4": 20},
+)
+
+resource_quota: ResourceQuota = anyscale.resource_quota.create(create_resource_quota)
+
+quota: Quota = resource_quota.quota
+"""
+    num_cpus: Optional[int] = field(
+        default=None,
+        metadata={"docstring": "The quota limit for the number of CPUs (optional)."},
+    )
+
+    def _validate_num_cpus(self, num_cpus: Optional[int]):
+        if num_cpus is not None and not isinstance(num_cpus, int):
+            raise TypeError("num_cpus must be an integer.")
+
+    num_instances: Optional[int] = field(
+        default=None,
+        metadata={
+            "docstring": "The quota limit for the number of instances. (optional)."
+        },
+    )
+
+    def _validate_num_instances(self, num_instances: Optional[int]):
+        if num_instances is not None and not isinstance(num_instances, int):
+            raise TypeError("num_instances must be an integer.")
+
+    num_gpus: Optional[int] = field(
+        default=None,
+        metadata={
+            "docstring": "The quota limit for the total number of GPUs (optional)."
+        },
+    )
+
+    def _validate_num_gpus(self, num_gpus: Optional[int]):
+        if num_gpus is not None and not isinstance(num_gpus, int):
+            raise TypeError("num_gpus must be an integer.")
+
+    num_accelerators: Optional[Dict[str, int]] = field(
+        default=None,
+        metadata={
+            "docstring": "The quota limit for the number of accelerators (optional)."
+        },
+    )
+
+    def _validate_num_accelerators(self, num_accelerators: Optional[Dict[str, int]]):
+        if num_accelerators is not None and not isinstance(num_accelerators, dict):
+            raise TypeError("num_accelerators must be a dictionary.")
+        if num_accelerators is not None:
+            for key, value in num_accelerators.items():
+                if not isinstance(key, str):
+                    raise TypeError("num_accelerators keys must be strings.")
+                if not isinstance(value, int):
+                    raise TypeError("num_accelerators values must be integers.")
+
+
+@dataclass(frozen=True)
+class ResourceQuota(ModelBase):
+    """Resource quota
+    """
+
+    __doc_py_example__ = """\
+import anyscale
+from anyscale.resource_quota.models import CreateResourceQuota, ResourceQuota
+
+create_resource_quota = CreateResourceQuota(
+    # Name of the resource quota to create
+    name="resource_quota_name",
+    # Name of the cloud that this resource quota applies to
+    cloud="cloud_name",
+    # Name of the project that this resource quota applies to (optional)
+    project="project_name",
+    # Email of the user that this resource quota applies to (optional)
+    user_email="test@anyscale.com",
+    # The quota limit for the number of CPUs (optional)
+    num_cpus=50,
+    # The quota limit for the number of instances (optional)
+    num_instances=100,
+    # The quota limit for the total number of GPUs (optional)
+    num_gpus=30,
+    # The quota limit for the number of accelerators (optional)
+    num_accelerators={"A100-80G": 10, "T4": 20},
+)
+resource_quota: ResourceQuota = anyscale.resource_quota.create(create_resource_quota)
+"""
+    id: str = field(metadata={"docstring": "The ID of the resource quota."})
+
+    def _validate_id(self, id: str):  # noqa: A002
+        if not isinstance(id, str):
+            raise TypeError("id must be a string.")
+
+    name: str = field(metadata={"docstring": "Name of the resource quota."})
+
+    def _validate_name(self, name: str):
+        if not isinstance(name, str):
+            raise TypeError("name must be a string.")
+
+    quota: Quota = field(metadata={"docstring": "The quota limit."})
+
+    def _validate_quota(self, quota: Quota):
+        if not isinstance(quota, Quota):
+            raise TypeError("quota must be a Quota.")
+
+    created_at: datetime = field(
+        metadata={"docstring": "The timestamp when this resource quota was created."}
+    )
+
+    def _validate_created_at(self, created_at: datetime):
+        if not isinstance(created_at, datetime):
+            raise TypeError("created_at must be a datetime.")
+
+    cloud_id: str = field(
+        metadata={"docstring": "ID of the cloud that this resource quota applies to."}
+    )
+
+    def _validate_cloud_id(self, cloud_id: str):
+        if not isinstance(cloud_id, str):
+            raise TypeError("cloud_id must be a string.")
+
+    project_id: Optional[str] = field(
+        default=None,
+        metadata={
+            "docstring": "ID of the project that this resource quota applies to (optional)."
+        },
+    )
+
+    def _validate_project_id(self, project_id: Optional[str]):
+        if project_id is not None and not isinstance(project_id, str):
+            raise TypeError("project_id must be a string.")
+
+    user_id: Optional[str] = field(
+        default=None,
+        metadata={
+            "docstring": "ID of the user that this resource quota applies to (optional)."
+        },
+    )
+
+    def _validate_user_id(self, user_id: Optional[str]):
+        if user_id is not None and not isinstance(user_id, str):
+            raise TypeError("user_id must be a string.")
+
+    is_enabled: bool = field(
+        default=True, metadata={"docstring": "Whether the resource quota is enabled."}
+    )
+
+    def _validate_is_enabled(self, is_enabled: bool):
+        if not isinstance(is_enabled, bool):
+            raise TypeError("is_enabled must be a boolean.")
+
+    deleted_at: Optional[datetime] = field(
+        default=None,
+        metadata={"docstring": "The timestamp when this resource quota was deleted."},
+    )
+
+    def _validate_deleted_at(self, deleted_at: Optional[datetime]):
+        if deleted_at is not None and not isinstance(deleted_at, datetime):
+            raise TypeError("deleted_at must be a datetime.")

anyscale/scripts.py

@@ -29,12 +29,16 @@ from anyscale.commands.logs_commands import log_cli
 from anyscale.commands.machine_commands import machine_cli
 from anyscale.commands.machine_pool_commands import machine_pool_cli
 from anyscale.commands.migrate_commands import migrate_cli
+from anyscale.commands.organization_invitation_commands import (
+    organization_invitation_cli,
+)
 from anyscale.commands.project_commands import anyscale_init, project_cli
 from anyscale.commands.resource_quota_commands import resource_quota_cli
 from anyscale.commands.schedule_commands import schedule_cli
 from anyscale.commands.service_account_commands import service_account_cli
 from anyscale.commands.service_commands import service_cli
 from anyscale.commands.session_commands_hidden import session_cli
+from anyscale.commands.user_commands import user_cli
 from anyscale.commands.workspace_commands import workspace_cli
 from anyscale.commands.workspace_commands_v2 import workspace_cli as workspace_cli_v2
 import anyscale.conf
@@ -142,6 +146,8 @@ cli.add_command(machine_pool_cli)
 cli.add_command(service_account_cli)
 cli.add_command(resource_quota_cli)
 cli.add_command(aggregated_instance_usage_cli)
+cli.add_command(user_cli)
+cli.add_command(organization_invitation_cli)
 
 ALIASES = {
     "h": anyscale_help,

anyscale/sdk/anyscale_client/__init__.py

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

anyscale/sdk/anyscale_client/api/default_api.py

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

anyscale/sdk/anyscale_client/models/__init__.py

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

anyscale/sdk/anyscale_client/models/cluster_status_details.py

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

anyscale/sdk/anyscale_client/sdk.py

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

anyscale/user/__init__.py

@@ -0,0 +1,35 @@
+from typing import List, Optional
+
+from anyscale._private.anyscale_client import AnyscaleClientInterface
+from anyscale._private.sdk import sdk_docs
+from anyscale._private.sdk.base_sdk import Timer
+from anyscale.cli_logger import BlockLogger
+from anyscale.user._private.user_sdk import PrivateUserSDK
+from anyscale.user.commands import (
+    _ADMIN_BATCH_CREATE_ARG_DOCSTRINGS,
+    _ADMIN_BATCH_CREATE_EXAMPLE,
+    admin_batch_create,
+)
+from anyscale.user.models import AdminCreatedUser, AdminCreateUser
+
+
+class UserSDK:
+    def __init__(
+        self,
+        *,
+        client: Optional[AnyscaleClientInterface] = None,
+        logger: Optional[BlockLogger] = None,
+        timer: Optional[Timer] = None,
+    ):
+        self._private_sdk = PrivateUserSDK(client=client, logger=logger, timer=timer)
+
+    @sdk_docs(
+        doc_py_example=_ADMIN_BATCH_CREATE_EXAMPLE,
+        arg_docstrings=_ADMIN_BATCH_CREATE_ARG_DOCSTRINGS,
+    )
+    def admin_batch_create(  # noqa: F811
+        self, admin_create_users: List[AdminCreateUser],
+    ) -> List[AdminCreatedUser]:
+        """Batch create, as an admin, users without email verification.
+        """
+        return self._private_sdk.admin_batch_create(admin_create_users)

anyscale/user/_private/user_sdk.py

@@ -0,0 +1,32 @@
+from typing import List
+
+from anyscale._private.sdk.base_sdk import BaseSDK
+from anyscale.client.openapi_client.models import (
+    AdminCreateUser as AdminCreateUserModel,
+)
+from anyscale.user.models import AdminCreatedUser, AdminCreateUser
+
+
+class PrivateUserSDK(BaseSDK):
+    def admin_batch_create(
+        self, admin_create_users: List[AdminCreateUser]
+    ) -> List[AdminCreatedUser]:
+        created_users = self.client.admin_batch_create_users(
+            [
+                AdminCreateUserModel(**admin_create_user.to_dict())
+                for admin_create_user in admin_create_users
+            ]
+        )
+
+        return [
+            AdminCreatedUser(
+                user_id=created_user.user_id,
+                name=created_user.name,
+                email=created_user.email,
+                created_at=created_user.created_at,
+                is_sso_user=created_user.is_sso_user,
+                lastname=created_user.lastname,
+                title=created_user.title,
+            )
+            for created_user in created_users
+        ]

anyscale/user/commands.py

@@ -0,0 +1,42 @@
+from typing import List
+
+from anyscale._private.sdk import sdk_command
+from anyscale.user._private.user_sdk import PrivateUserSDK
+from anyscale.user.models import AdminCreatedUser, AdminCreateUser
+
+
+_USER_SDK_SINGLETON_KEY = "user_sdk"
+
+_ADMIN_BATCH_CREATE_EXAMPLE = """
+import anyscale
+from anyscale.user.models import AdminCreateUser
+
+anyscale.user.admin_batch_create(
+    [AdminCreateUser(
+        name="name",
+        email="test@anyscale.com",
+        password="",
+        is_sso_user=False,
+        lastname="lastname",
+        title="title",
+    ),],
+)
+"""
+
+_ADMIN_BATCH_CREATE_ARG_DOCSTRINGS = {
+    "admin_create_users": "Users to be created by an admin.",
+}
+
+
+@sdk_command(
+    _USER_SDK_SINGLETON_KEY,
+    PrivateUserSDK,
+    doc_py_example=_ADMIN_BATCH_CREATE_EXAMPLE,
+    arg_docstrings=_ADMIN_BATCH_CREATE_ARG_DOCSTRINGS,
+)
+def admin_batch_create(
+    admin_create_users: List[AdminCreateUser], *, _sdk: PrivateUserSDK
+) -> List[AdminCreatedUser]:
+    """Batch create, as an admin, users without email verification.
+    """
+    return _sdk.admin_batch_create(admin_create_users)