anyscale 0.24.86__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
@@ -162,6 +163,11 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**edit_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_patch**](docs/DefaultApi.md#edit_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_patch) | **PATCH** /api/v2/clouds_with_cloud_resource_router/{cloud_id} | Edit Cloud Resource
 *DefaultApi* | [**execute_interactive_command_api_v2_sessions_session_id_execute_interactive_command_post**](docs/DefaultApi.md#execute_interactive_command_api_v2_sessions_session_id_execute_interactive_command_post) | **POST** /api/v2/sessions/{session_id}/execute_interactive_command | Execute Interactive Command
 *DefaultApi* | [**execute_shell_command_api_v2_sessions_session_id_execute_shell_command_post**](docs/DefaultApi.md#execute_shell_command_api_v2_sessions_session_id_execute_shell_command_post) | **POST** /api/v2/sessions/{session_id}/execute_shell_command | Execute Shell Command
+*DefaultApi* | [**fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post**](docs/DefaultApi.md#fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post) | **POST** /api/v2/aggregated_instance_usage/cloud | Fetch Usage Group By Cloud
+*DefaultApi* | [**fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post**](docs/DefaultApi.md#fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post) | **POST** /api/v2/aggregated_instance_usage/cluster | Fetch Usage Group By Cluster
+*DefaultApi* | [**fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post**](docs/DefaultApi.md#fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post) | **POST** /api/v2/aggregated_instance_usage/instance_type | Fetch Usage Group By Instance Type
+*DefaultApi* | [**fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post**](docs/DefaultApi.md#fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post) | **POST** /api/v2/aggregated_instance_usage/project | Fetch Usage Group By Project
+*DefaultApi* | [**fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post**](docs/DefaultApi.md#fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post) | **POST** /api/v2/aggregated_instance_usage/user | Fetch Usage Group By User
 *DefaultApi* | [**find_cloud_by_name_api_v2_clouds_find_by_name_post**](docs/DefaultApi.md#find_cloud_by_name_api_v2_clouds_find_by_name_post) | **POST** /api/v2/clouds/find_by_name | Find Cloud By Name
 *DefaultApi* | [**find_dataset_api_v2_datasets_find_get**](docs/DefaultApi.md#find_dataset_api_v2_datasets_find_get) | **GET** /api/v2/datasets/find | Find Dataset
 *DefaultApi* | [**find_project_by_project_name_api_v2_projects_find_by_name_get**](docs/DefaultApi.md#find_project_by_project_name_api_v2_projects_find_by_name_get) | **GET** /api/v2/projects/find_by_name | Find Project By Project Name
@@ -277,6 +283,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_template_readme_api_v2_experimental_workspaces_template_readme_template_id_get**](docs/DefaultApi.md#get_template_readme_api_v2_experimental_workspaces_template_readme_template_id_get) | **GET** /api/v2/experimental_workspaces/template/readme/{template_id} | Get Template Readme
 *DefaultApi* | [**get_templatized_cluster_environments_api_v2_application_templates_templatized_get**](docs/DefaultApi.md#get_templatized_cluster_environments_api_v2_application_templates_templatized_get) | **GET** /api/v2/application_templates/templatized/ | Get Templatized Cluster Environments
 *DefaultApi* | [**get_templatized_compute_configs_api_v2_compute_templates_templatized_cloud_id_get**](docs/DefaultApi.md#get_templatized_compute_configs_api_v2_compute_templates_templatized_cloud_id_get) | **GET** /api/v2/compute_templates/templatized/{cloud_id} | Get Templatized Compute Configs
+*DefaultApi* | [**get_total_usage_api_v2_aggregated_instance_usage_get**](docs/DefaultApi.md#get_total_usage_api_v2_aggregated_instance_usage_get) | **GET** /api/v2/aggregated_instance_usage/ | Get Total Usage
 *DefaultApi* | [**get_user_info_api_v2_userinfo_get**](docs/DefaultApi.md#get_user_info_api_v2_userinfo_get) | **GET** /api/v2/userinfo/ | Get User Info
 *DefaultApi* | [**get_workspace_api_v2_experimental_workspaces_workspace_id_get**](docs/DefaultApi.md#get_workspace_api_v2_experimental_workspaces_workspace_id_get) | **GET** /api/v2/experimental_workspaces/{workspace_id} | Get Workspace
 *DefaultApi* | [**get_workspace_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_dataplane_artifacts_get**](docs/DefaultApi.md#get_workspace_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_dataplane_artifacts_get) | **GET** /api/v2/experimental_workspaces/{workspace_id}/dataplane_artifacts | Get Workspace Dataplane Artifacts
@@ -424,8 +431,14 @@ Class | Method | HTTP request | Description
  - [AWSCredentials](docs/AWSCredentials.md)
  - [AWSMemoryDBClusterConfig](docs/AWSMemoryDBClusterConfig.md)
  - [AccessConfig](docs/AccessConfig.md)
- - [AggregatedInstanceUsageWithCostModel](docs/AggregatedInstanceUsageWithCostModel.md)
- - [AggregatedinstanceusagewithcostmodelListResponse](docs/AggregatedinstanceusagewithcostmodelListResponse.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)
+ - [AggregatedinstanceusagecsvListResponse](docs/AggregatedinstanceusagecsvListResponse.md)
+ - [AggregatedusageResponse](docs/AggregatedusageResponse.md)
  - [AicaEndpoint](docs/AicaEndpoint.md)
  - [AicaEndpointEvent](docs/AicaEndpointEvent.md)
  - [AicaEndpointEventLevel](docs/AicaEndpointEventLevel.md)
@@ -1017,6 +1030,16 @@ Class | Method | HTTP request | Description
  - [UpdatemachinepoolresponseResponse](docs/UpdatemachinepoolresponseResponse.md)
  - [UploadSessionCommandLogsLocations](docs/UploadSessionCommandLogsLocations.md)
  - [UploadsessioncommandlogslocationsResponse](docs/UploadsessioncommandlogslocationsResponse.md)
+ - [UsageByCloud](docs/UsageByCloud.md)
+ - [UsageByCluster](docs/UsageByCluster.md)
+ - [UsageByInstanceType](docs/UsageByInstanceType.md)
+ - [UsageByProject](docs/UsageByProject.md)
+ - [UsageByUser](docs/UsageByUser.md)
+ - [UsagebycloudListResponse](docs/UsagebycloudListResponse.md)
+ - [UsagebyclusterListResponse](docs/UsagebyclusterListResponse.md)
+ - [UsagebyinstancetypeListResponse](docs/UsagebyinstancetypeListResponse.md)
+ - [UsagebyprojectListResponse](docs/UsagebyprojectListResponse.md)
+ - [UsagebyuserListResponse](docs/UsagebyuserListResponse.md)
  - [UserInfo](docs/UserInfo.md)
  - [UserResendEmailOptions](docs/UserResendEmailOptions.md)
  - [UserServiceAccessTypes](docs/UserServiceAccessTypes.md)

anyscale/client/openapi_client/__init__.py

@@ -31,8 +31,14 @@ 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.aggregated_instance_usage_with_cost_model import AggregatedInstanceUsageWithCostModel
-from openapi_client.models.aggregatedinstanceusagewithcostmodel_list_response import AggregatedinstanceusagewithcostmodelListResponse
+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
+from openapi_client.models.aggregatedinstanceusagecsv_list_response import AggregatedinstanceusagecsvListResponse
+from openapi_client.models.aggregatedusage_response import AggregatedusageResponse
 from openapi_client.models.aica_endpoint import AicaEndpoint
 from openapi_client.models.aica_endpoint_event import AicaEndpointEvent
 from openapi_client.models.aica_endpoint_event_level import AicaEndpointEventLevel
@@ -624,6 +630,16 @@ from openapi_client.models.update_resource_quota import UpdateResourceQuota
 from openapi_client.models.updatemachinepoolresponse_response import UpdatemachinepoolresponseResponse
 from openapi_client.models.upload_session_command_logs_locations import UploadSessionCommandLogsLocations
 from openapi_client.models.uploadsessioncommandlogslocations_response import UploadsessioncommandlogslocationsResponse
+from openapi_client.models.usage_by_cloud import UsageByCloud
+from openapi_client.models.usage_by_cluster import UsageByCluster
+from openapi_client.models.usage_by_instance_type import UsageByInstanceType
+from openapi_client.models.usage_by_project import UsageByProject
+from openapi_client.models.usage_by_user import UsageByUser
+from openapi_client.models.usagebycloud_list_response import UsagebycloudListResponse
+from openapi_client.models.usagebycluster_list_response import UsagebyclusterListResponse
+from openapi_client.models.usagebyinstancetype_list_response import UsagebyinstancetypeListResponse
+from openapi_client.models.usagebyproject_list_response import UsagebyprojectListResponse
+from openapi_client.models.usagebyuser_list_response import UsagebyuserListResponse
 from openapi_client.models.user_info import UserInfo
 from openapi_client.models.user_resend_email_options import UserResendEmailOptions
 from openapi_client.models.user_service_access_types import UserServiceAccessTypes