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

anyscale/__init__.py

@@ -20,11 +20,15 @@ path.append(os.path.join(anyscale_dir, "sdk"))
 import anyscale
 from anyscale import (
     aggregated_instance_usage,
+    cloud,
     compute_config,
     image,
     integrations,
     job,
     llm,
+    organization_invitation,
+    project,
+    resource_quota,
     schedule,
     service,
     user,
@@ -34,6 +38,7 @@ from anyscale._private.sdk.base_sdk import Timer
 from anyscale.aggregated_instance_usage import AggregatedInstanceUsageSDK
 from anyscale.authenticate import AuthenticationBlock
 from anyscale.cli_logger import BlockLogger
+from anyscale.cloud import CloudSDK
 from anyscale.cluster import get_job_submission_client_cluster_info
 from anyscale.cluster_compute import get_cluster_compute_from_name
 from anyscale.compute_config import ComputeConfigSDK
@@ -41,6 +46,9 @@ from anyscale.connect import ClientBuilder
 from anyscale.image import ImageSDK
 from anyscale.job import JobSDK
 from anyscale.llm import LLMSDK
+from anyscale.organization_invitation import OrganizationInvitationSDK
+from anyscale.project import ProjectSDK
+from anyscale.resource_quota import ResourceQuotaSDK
 from anyscale.schedule import ScheduleSDK
 from anyscale.sdk.anyscale_client.sdk import AnyscaleSDK
 from anyscale.service import ServiceSDK
@@ -58,6 +66,10 @@ for attr, _ in inspect.getmembers(ClientBuilder, inspect.isfunction):
     if attr.startswith("_"):
         continue
 
+    # This is exposed in anyscale/cloud/__init__.py since anyscale.cloud is used as the SDK module too.
+    if attr == "cloud":
+        continue
+
     def _new_builder(attr: str) -> Any:
         target = getattr(ClientBuilder, attr)
 
@@ -107,9 +119,15 @@ class Anyscale:
         self._job_sdk = JobSDK(client=self._anyscale_client)
         self._service_sdk = ServiceSDK(client=self._anyscale_client)
         self._compute_config_sdk = ComputeConfigSDK(client=self._anyscale_client)
+        self._cloud_sdk = CloudSDK(client=self._anyscale_client)
         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._organization_invitation_sdk = OrganizationInvitationSDK(
+            client=self._anyscale_client
+        )
+        self._project_sdk = ProjectSDK(client=self._anyscale_client)
+        self._resource_quota_sdk = ResourceQuotaSDK(client=self._anyscale_client)
         self._user_sdk = UserSDK(client=self._anyscale_client)
         self._workspace_sdk = WorkspaceSDK(client=self._anyscale_client)
 
@@ -131,13 +149,25 @@ class Anyscale:
         obj._compute_config_sdk = ComputeConfigSDK(  # noqa: SLF001
             client=client, logger=logger, timer=timer
         )
+        obj._cloud_sdk = CloudSDK(  # noqa: SLF001
+            client=client, logger=logger, timer=timer
+        )
         obj._schedule_sdk = ScheduleSDK(  # noqa: SLF001
             client=client, logger=logger, timer=timer,
         )
         obj._image_sdk = ImageSDK(client=client, logger=logger)  # noqa: SLF001
+        obj._organization_invitation_sdk = OrganizationInvitationSDK(  # noqa: SLF001
+            client=client, logger=logger
+        )
+        obj._project_sdk = ProjectSDK(  # noqa: SLF001
+            client=client, logger=logger, timer=timer
+        )
         obj._user_sdk = UserSDK(  # noqa: SLF001
             client=client, logger=logger, timer=timer
         )
+        obj._resource_quota_sdk = ResourceQuotaSDK(  # noqa: SLF001
+            client=client, logger=logger, timer=timer
+        )
         obj._workspace_sdk = WorkspaceSDK(  # noqa: SLF001
             client=client, logger=logger, timer=timer,
         )
@@ -159,6 +189,10 @@ class Anyscale:
     def compute_config(self) -> ComputeConfigSDK:  # noqa: F811
         return self._compute_config_sdk
 
+    @property
+    def cloud(self) -> CloudSDK:  # noqa: F811
+        return self._cloud_sdk
+
     @property
     def schedule(self) -> ScheduleSDK:  # noqa: F811
         return self._schedule_sdk
@@ -171,6 +205,18 @@ class Anyscale:
     def llm(self) -> LLMSDK:  # noqa: F811
         return self._llm_sdk
 
+    @property
+    def organization_invitation(self) -> OrganizationInvitationSDK:  # noqa: F811
+        return self._organization_invitation_sdk
+
+    @property
+    def project(self) -> ProjectSDK:  # noqa: F811
+        return self._project_sdk
+
+    @property
+    def resource_quota(self) -> ResourceQuotaSDK:  # noqa: F811
+        return self._resource_quota_sdk
+
     @property
     def user(self) -> UserSDK:  # noqa: F811
         return self._user_sdk

anyscale/_private/anyscale_client/anyscale_client.py

@@ -47,10 +47,14 @@ from anyscale.client.openapi_client.models import (
     ComputeTemplate,
     ComputeTemplateConfig,
     ComputeTemplateQuery,
+    CreateCloudCollaborator,
     CreateComputeTemplate,
     CreateDataset,
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
+    CreateOrganizationInvitation,
+    CreateResourceQuota,
+    CreateUserProjectCollaborator,
     Dataset as InternalDataset,
     DatasetUpload,
     DecoratedComputeTemplate,
@@ -60,6 +64,11 @@ from anyscale.client.openapi_client.models import (
     FineTunedModel,
     FinetunedmodelListResponse,
     InternalProductionJob,
+    ListResourceQuotasQuery,
+    OrganizationCollaborator,
+    OrganizationInvitation,
+    ResourceQuota,
+    ResourceQuotaStatus,
     SessionSshKey,
     StartSessionOptions,
     StopSessionOptions,
@@ -544,6 +553,14 @@ class AnyscaleClient(AnyscaleClientInterface):
 
             raise e from None
 
+    @handle_api_exceptions
+    def add_cloud_collaborators(
+        self, cloud_id: str, collaborators: List[CreateCloudCollaborator]
+    ) -> None:
+        self._internal_api_client.batch_create_cloud_collaborators_api_v2_clouds_cloud_id_collaborators_users_batch_create_post(
+            cloud_id, collaborators
+        )
+
     @handle_api_exceptions
     def create_compute_config(
         self, config: ComputeTemplateConfig, *, name: Optional[str] = None
@@ -986,6 +1003,14 @@ class AnyscaleClient(AnyscaleClientInterface):
             project_id
         ).result
 
+    @handle_api_exceptions
+    def add_project_collaborators(
+        self, project_id: str, collaborators: List[CreateUserProjectCollaborator]
+    ) -> None:
+        self._internal_api_client.batch_create_project_collaborators_api_v2_projects_project_id_collaborators_users_batch_create_post(
+            project_id, collaborators
+        )
+
     @handle_api_exceptions
     def get_job(
         self,
@@ -1122,39 +1147,26 @@ class AnyscaleClient(AnyscaleClientInterface):
                 # If the presigned URL scheme is SMART_OPEN, upload to cloud storage using the provided bucket name, path, & environment, and the smart_open library.
                 bucket_name = info.bucket_name
                 bucket_path = info.bucket_path
-                file_uri = info.file_uri
 
-                if file_uri and file_uri.startswith("azure"):
-
-                    from anyscale.utils.imports.azure import (
-                        try_import_azure_storage_blob_BlobServiceClient,
-                    )
-
-                    # Smartopen needs transport_params to be passed in for Azure.
-                    blob_service_client = (
-                        try_import_azure_storage_blob_BlobServiceClient()
-                    )
-
-                    transport_params = {
-                        "client": blob_service_client.from_connection_string(info.url),
-                    }
-                    with smart_open.open(
-                        f"{file_uri}", "wb", transport_params=transport_params,
-                    ) as fout:
-                        fout.write(zip_file_bytes)
-                else:
-                    env_vars: Dict[str, str] = {
-                        "AWS_ENDPOINT_URL": info.url,
-                    }
-                    with set_env(**env_vars), smart_open.open(
-                        f"{bucket_name}/{bucket_path}", "wb",
-                    ) as fout:
-                        fout.write(zip_file_bytes)
+                env_vars: Dict[str, str] = {
+                    "AWS_ENDPOINT_URL": info.url,
+                }
+                with set_env(**env_vars), smart_open.open(
+                    f"{bucket_name}/{bucket_path}", "wb",
+                ) as fout:
+                    fout.write(zip_file_bytes)
 
             else:
                 # Default to HTTP PUT.
                 internal_logger.debug(f"Uploading file '{file_name}' to cloud storage.")
-                requests.put(info.url, data=zip_file_bytes).raise_for_status()
+                headers = (
+                    {"x-ms-blob-type": "BlockBlob"}
+                    if info.file_uri.startswith("azure")
+                    else None
+                )
+                requests.put(
+                    info.url, data=zip_file_bytes, headers=headers
+                ).raise_for_status()
 
         return info.file_uri
 
@@ -1855,3 +1867,111 @@ class AnyscaleClient(AnyscaleClientInterface):
         return self._internal_api_client.admin_batch_create_users_api_v2_users_admin_batch_create_post(
             admin_create_users
         ).results
+
+    @handle_api_exceptions
+    def create_organization_invitations(
+        self, emails: List[str]
+    ) -> Tuple[List[str], List[str]]:
+        results = self._internal_api_client.batch_create_invitations_api_v2_organization_invitations_batch_create_post(
+            [CreateOrganizationInvitation(email=email) for email in emails]
+        ).results
+
+        success_emails = []
+        error_messages = []
+
+        for idx, result in enumerate(results):
+            if result.data:
+                success_emails.append(emails[idx])
+            else:
+                error_messages.append(result.error.detail)
+
+        return success_emails, error_messages
+
+    @handle_api_exceptions
+    def list_organization_invitations(self) -> List[OrganizationInvitation]:
+        results = (
+            self._internal_api_client.list_invitations_api_v2_organization_invitations_get().results
+        )
+
+        return results
+
+    @handle_api_exceptions
+    def delete_organization_invitation(self, email: str) -> OrganizationInvitation:
+        invitation = self._internal_api_client.list_invitations_api_v2_organization_invitations_get(
+            email=email
+        ).results
+
+        if len(invitation) == 0:
+            raise ValueError(f"Invitation for email '{email}' not found.")
+        elif len(invitation) > 1:
+            raise ValueError(
+                f"Multiple invitations found for email '{email}'. Please contact Anyscale support."
+            )
+
+        invitation_id = invitation[0].id
+
+        return self._internal_api_client.invalidate_invitation_api_v2_organization_invitations_invitation_id_invalidate_post(
+            invitation_id
+        ).result
+
+    @handle_api_exceptions
+    def get_organization_collaborator(self, email: str) -> OrganizationCollaborator:
+        results = self._internal_api_client.list_organization_collaborators_api_v2_organization_collaborators_get(
+            email=email
+        ).results
+
+        if len(results) == 0:
+            raise ValueError(f"User with email '{email}' not found.")
+
+        if len(results) > 1:
+            raise ValueError(
+                f"Multiple users found for email '{email}'. Please contact Anyscale support."
+            )
+
+        return results[0]
+
+    @handle_api_exceptions
+    def create_resource_quota(
+        self, create_resource_quota: CreateResourceQuota
+    ) -> ResourceQuota:
+        return self._internal_api_client.create_resource_quota_api_v2_resource_quotas_post(
+            create_resource_quota
+        ).result
+
+    @handle_api_exceptions
+    def list_resource_quotas(
+        self,
+        name: Optional[str] = None,
+        cloud_id: Optional[str] = None,
+        creator_id: Optional[str] = None,
+        is_enabled: Optional[bool] = None,
+        max_items: int = 20,
+    ) -> List[ResourceQuota]:
+
+        query = ListResourceQuotasQuery(
+            name=TextQuery(equals=name) if name else None,
+            cloud_id=cloud_id,
+            creator_id=creator_id,
+            is_enabled=is_enabled,
+            paging=PageQuery(count=max_items),
+        )
+
+        resource_quotas = self._internal_api_client.search_resource_quotas_api_v2_resource_quotas_search_post(
+            query
+        ).results
+
+        return resource_quotas
+
+    @handle_api_exceptions
+    def delete_resource_quota(self, resource_quota_id: str) -> None:
+        self._internal_api_client.delete_resource_quota_api_v2_resource_quotas_resource_quota_id_delete(
+            resource_quota_id
+        )
+
+    @handle_api_exceptions
+    def set_resource_quota_status(
+        self, resource_quota_id: str, is_enabled: bool
+    ) -> None:
+        self._internal_api_client.set_resource_quota_status_api_v2_resource_quotas_resource_quota_id_status_patch(
+            resource_quota_id, ResourceQuotaStatus(is_enabled=is_enabled)
+        ).result

anyscale/_private/anyscale_client/common.py

@@ -8,13 +8,19 @@ from anyscale.client.openapi_client.models import (
     AdminCreateUser,
     Cloud,
     ComputeTemplateConfig,
+    CreateCloudCollaborator,
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
+    CreateResourceQuota,
+    CreateUserProjectCollaborator,
     DecoratedComputeTemplate,
     DeletedPlatformFineTunedModel,
     FineTunedModel,
     InternalProductionJob,
+    OrganizationCollaborator,
+    OrganizationInvitation,
     Project,
+    ResourceQuota,
     WorkspaceDataplaneProxiedArtifacts,
 )
 from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
@@ -142,6 +148,13 @@ class AnyscaleClientInterface(ABC):
         """
         raise NotImplementedError
 
+    @abstractmethod
+    def add_cloud_collaborators(
+        self, cloud_id: str, collaborators: List[CreateCloudCollaborator]
+    ) -> None:
+        """Batch add collaborators to a cloud."""
+        raise NotImplementedError
+
     @abstractmethod
     def create_compute_config(
         self, config: ComputeTemplateConfig, *, name: Optional[str] = None
@@ -285,6 +298,13 @@ class AnyscaleClientInterface(ABC):
         """
         raise NotImplementedError
 
+    @abstractmethod
+    def add_project_collaborators(
+        self, project_id: str, collaborators: List[CreateUserProjectCollaborator]
+    ) -> None:
+        """Batch add collaborators to a project."""
+        raise NotImplementedError
+
     @abstractmethod
     def get_job(
         self,
@@ -598,5 +618,58 @@ class AnyscaleClientInterface(ABC):
     def admin_batch_create_users(
         self, admin_create_users: List[AdminCreateUser]
     ) -> List[AdminCreatedUser]:
-        """Batch create users without email verification as an admin."""
+        """Batch create, as an admin, users without email verification."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def create_organization_invitations(
+        self, emails: List[str]
+    ) -> Tuple[List[str], List[str]]:
+        """Create organization invitations."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def list_organization_invitations(self) -> List[OrganizationInvitation]:
+        """List organization invitations."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_organization_invitation(self, email: str) -> OrganizationInvitation:
+        """Delete organization invitation."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_organization_collaborator(self, email: str) -> OrganizationCollaborator:
+        """Get organization collaborator."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def create_resource_quota(
+        self, create_resource_quota: CreateResourceQuota
+    ) -> ResourceQuota:
+        """Create a resource quota."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def list_resource_quotas(
+        self,
+        name: Optional[str] = None,
+        cloud_id: Optional[str] = None,
+        creator_id: Optional[str] = None,
+        is_enabled: Optional[bool] = None,
+        max_items: int = 20,
+    ) -> List[ResourceQuota]:
+        """List resource quotas."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_resource_quota(self, resource_quota_id: str) -> None:
+        """Delete a resource quota."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def set_resource_quota_status(
+        self, resource_quota_id: str, is_enabled: bool
+    ) -> None:
+        """Set the status of a resource quota."""
         raise NotImplementedError

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -17,9 +17,13 @@ from anyscale.client.openapi_client.models import (
     AdminCreatedUser,
     AdminCreateUser,
     Cloud,
+    CloudProviders,
     ComputeTemplateConfig,
+    CreateCloudCollaborator,
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
+    CreateResourceQuota,
+    CreateUserProjectCollaborator,
     DecoratedComputeTemplate,
     DeletedPlatformFineTunedModel,
     ExperimentalWorkspace,
@@ -28,9 +32,15 @@ from anyscale.client.openapi_client.models import (
     HaJobGoalStates,
     HaJobStates,
     InternalProductionJob,
+    MiniCloud,
+    MiniUser,
+    OrganizationCollaborator,
+    OrganizationInvitation,
+    OrganizationPermissionLevel,
     ProductionJob,
     ProductionJobStateTransition,
     Project,
+    ResourceQuota,
     WorkspaceDataplaneProxiedArtifacts,
 )
 from anyscale.client.openapi_client.models.create_schedule import CreateSchedule
@@ -135,6 +145,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._jobs: Dict[str, ProductionJob] = {}
         self._job_runs: Dict[str, List[APIJobRun]] = defaultdict(list)
         self._project_to_id: Dict[Optional[str] : Dict[Optional[str], str]] = {}
+        self._project_collaborators: Dict[str, List[CreateUserProjectCollaborator]] = {}
         self._rolled_out_model: Optional[ApplyServiceModel] = None
         self._sent_workspace_notifications: List[WorkspaceNotification] = []
         self._rolled_back_service: Optional[Tuple[str, Optional[int]]] = None
@@ -154,6 +165,8 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._workspaces_env_vars: Dict[str, Dict[str, str]] = {}
         self._clusters_headnode_ip: Dict[str, str] = {}
         self._clusters_ssh_key: Dict[str, SessionSshKey] = {}
+        self._organization_invitations: Dict[str, OrganizationInvitation] = {}
+        self._resource_quotas: Dict[str, ResourceQuota] = {}
 
         # Cloud ID -> Cloud.
         self._clouds: Dict[str, Cloud] = {
@@ -164,6 +177,9 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             ),
         }
 
+        # Cloud ID -> collaborators
+        self._cloud_collaborators: Dict[str, List[CreateCloudCollaborator]] = {}
+
         # Cloud ID -> default ClusterCompute.
         compute_config = ClusterCompute(
             id=self.DEFAULT_CLUSTER_COMPUTE_ID,
@@ -347,6 +363,25 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     def get_cloud(self, *, cloud_id: str) -> Optional[Cloud]:
         return self._clouds.get(cloud_id, None)
 
+    def add_cloud_collaborators(
+        self, cloud_id: str, collaborators: List[CreateCloudCollaborator]
+    ) -> None:
+        existing_collaborators = [
+            collaborator.email
+            for collaborator in self._cloud_collaborators.get(cloud_id, [])
+        ]
+
+        for collaborator in collaborators:
+            if collaborator.email in existing_collaborators:
+                raise ValueError(
+                    f"Collaborator with email '{collaborator.email}' already exists in cloud '{cloud_id}'."
+                )
+
+        if cloud_id not in self._cloud_collaborators:
+            self._cloud_collaborators[cloud_id] = collaborators
+        else:
+            self._cloud_collaborators[cloud_id].extend(collaborators)
+
     def add_compute_config(self, compute_config: DecoratedComputeTemplate) -> int:
         compute_config.version = (
             len(self._compute_config_name_to_ids[compute_config.name]) + 1
@@ -579,6 +614,28 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
                     )
         return None
 
+    def add_project_collaborators(
+        self, project_id: str, collaborators: List[CreateUserProjectCollaborator]
+    ):
+        existing_collaborators = [
+            collaborator.value.email if collaborator.value else None
+            for collaborator in self._project_collaborators.get(project_id, [])
+        ]
+
+        for collaborator in collaborators:
+            if (
+                collaborator.value
+                and collaborator.value.email in existing_collaborators
+            ):
+                raise ValueError(
+                    f"Collaborator with email '{collaborator.value.email}' already exists in project '{project_id}'."
+                )
+
+        if project_id not in self._project_collaborators:
+            self._project_collaborators[project_id] = collaborators
+        else:
+            self._project_collaborators[project_id].extend(collaborators)
+
     def get_job(
         self,
         *,
@@ -908,7 +965,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             compute_config_id=model.compute_config_id,
             environment_id=model.cluster_environment_build_id,
             cloud_id=model.cloud_id or self.get_cloud_id(),
-            created_at=datetime.now(),
+            created_at=datetime.utcnow(),
             creator_id=self.DEFAULT_USER_ID,
             creator_email=self.DEFAULT_USER_EMAIL,
             organization_id=self.DEFAULT_ORGANIZATION_ID,
@@ -1106,3 +1163,110 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             self._users[admin_create_user.email] = created_user
 
         return created_users
+
+    def create_organization_invitations(
+        self, emails: List[str]
+    ) -> Tuple[List[str], List[str]]:
+        success_emails, error_messages = [], []
+        for email in emails:
+            if email in self._users:
+                error_messages.append(
+                    f"User with email {email} is already a member of your organization."
+                )
+            else:
+                orginv_id = f"orginv_{uuid.uuid4()!s}"
+                self._organization_invitations[email] = OrganizationInvitation(
+                    email=email,
+                    id=orginv_id,
+                    organization_id="org_1",
+                    created_at=datetime.utcnow(),
+                    expires_at=datetime.utcnow(),
+                )
+                success_emails.append(email)
+
+        return success_emails, error_messages
+
+    def list_organization_invitations(self) -> List[OrganizationInvitation]:
+        return list(self._organization_invitations.values())
+
+    def delete_organization_invitation(self, email: str) -> OrganizationInvitation:
+        if email not in self._organization_invitations:
+            raise ValueError(f"Invitation for email '{email}' not found.")
+
+        return self._organization_invitations.pop(email)
+
+    def get_organization_collaborator(self, email):
+        return OrganizationCollaborator(
+            id="identity_1",
+            permission_level=OrganizationPermissionLevel.COLLABORATOR,
+            name="Test User",
+            created_at=datetime.utcnow(),
+            email=email,
+            user_id="usr_1",
+        )
+
+    def create_resource_quota(
+        self, create_resource_quota: CreateResourceQuota
+    ) -> ResourceQuota:
+        resource_quota_id = f"rq_{uuid.uuid4()!s}"
+        resource_quota = ResourceQuota(
+            id=resource_quota_id,
+            name=create_resource_quota.name,
+            cloud_id=create_resource_quota.cloud_id,
+            project_id=create_resource_quota.project_id,
+            quota=create_resource_quota.quota,
+            is_enabled=True,
+            created_at=datetime.utcnow(),
+            creator=MiniUser(
+                id="user_1",
+                email="test@anyscale.com",
+                name="Test User",
+                username="testuser",
+            ),
+            cloud=MiniCloud(
+                id=create_resource_quota.cloud_id,
+                name="Test Cloud",
+                provider=CloudProviders.AWS,
+            ),
+        )
+
+        self._resource_quotas[resource_quota_id] = resource_quota
+
+        return self._resource_quotas[resource_quota_id]
+
+    def list_resource_quotas(
+        self,
+        name: Optional[str] = None,
+        cloud_id: Optional[str] = None,
+        creator_id: Optional[str] = None,
+        is_enabled: Optional[bool] = None,
+        max_items: int = 20,
+    ) -> List[ResourceQuota]:
+        results = []
+        for resource_quota in self._resource_quotas.values():
+            if name and resource_quota.name != name:
+                continue
+            if cloud_id and resource_quota.cloud_id != cloud_id:
+                continue
+            if creator_id and resource_quota.creator.id != creator_id:
+                continue
+            if is_enabled is not None and resource_quota.is_enabled != is_enabled:
+                continue
+            results.append(resource_quota)
+
+        return results[:max_items]
+
+    def delete_resource_quota(self, resource_quota_id: str) -> None:
+        if resource_quota_id not in self._resource_quotas:
+            raise ValueError(f"Resource Quota with id '{resource_quota_id}' not found.")
+
+        self._resource_quotas.pop(resource_quota_id)
+
+    def set_resource_quota_status(
+        self, resource_quota_id: str, is_enabled: bool
+    ) -> None:
+        resource_quota = self._resource_quotas.get(resource_quota_id)
+        if resource_quota is None:
+            raise ValueError(f"Resource Quota with id '{resource_quota_id}' not found.")
+
+        resource_quota.is_enabled = is_enabled

anyscale/_private/docgen/README.md

@@ -1,4 +1,4 @@
-Tool to autogenerate documentation for the Anyscale SDK/CLI that can be embedded in a [Docusarus](https://docusaurus.io/) site.
+Tool to autogenerate documentation for the Anyscale SDK/CLI that can be embedded in a [Docusarus](https://docusaurus.io/) site. If you're adding a new API reference in anyscale/product, follow this guide to see what changes you need to make: https://www.notion.so/anyscale-hq/How-to-document-new-anyscale-API-references-173027c809cb8080b8e3e53fc244d70b.
 
 This is used by: https://github.com/anyscale/docs/tree/master/docs/reference.