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

anyscale/service_account/commands.py

@@ -0,0 +1,147 @@
+from typing import List, Optional
+
+from anyscale._private.sdk import sdk_command
+from anyscale.service_account._private.service_account_sdk import (
+    PrivateServiceAccountSDK,
+)
+from anyscale.service_account.models import ServiceAccount
+
+
+_SERVICE_ACCOUNT_SDK_SINGLETON_KEY = "service_account_sdk"
+
+_CREATE_EXAMPLE = """
+import anyscale
+
+api_key = anyscale.service_account.create(
+    name="my-service-account",
+)
+"""
+
+_CREATE_DOCSTRINGS = {"name": "Name for the service account."}
+
+_CREATE_API_KEY_EXAMPLE = """
+import anyscale
+
+api_key = anyscale.service_account.create_api_key(
+    name="my-service-account",
+)
+"""
+
+_CREATE_API_KEY_DOCSTRINGS = {
+    "email": "Email of the service account to create the new key for.",
+    "name": "Name of the service account to create the new key for.",
+}
+
+_LIST_EXAMPLE = """
+import anyscale
+
+anyscale.service_account.list(
+    max_items=20,
+)
+"""
+
+_LIST_DOCSTRINGS = {
+    "max_items": "Maximum number of items to return.",
+}
+
+_DELETE_EXAMPLE = """
+import anyscale
+
+anyscale.service_account.delete(
+    name="my-service-account",
+)
+"""
+
+_DELETE_DOCSTRINGS = {
+    "email": "Email of the service account to delete.",
+    "name": "Name of the service account to delete.",
+}
+
+_ROTATE_API_KEYS_EXAMPLE = """
+import anyscale
+
+anyscale.service_account.rotate_api_keys(
+    name="my-service-account",
+)
+"""
+
+_ROTATE_API_KEYS_DOCSTRINGS = {
+    "email": "Rotate API keys for the service account with this email.",
+    "name": "Rotate API keys for the service account with this name.",
+}
+
+
+@sdk_command(
+    _SERVICE_ACCOUNT_SDK_SINGLETON_KEY,
+    PrivateServiceAccountSDK,
+    doc_py_example=_CREATE_EXAMPLE,
+    arg_docstrings=_CREATE_DOCSTRINGS,
+)
+def create(name: str, *, _sdk: PrivateServiceAccountSDK) -> str:
+    """Create a service account and return the API key.
+    """
+    return _sdk.create(name)
+
+
+@sdk_command(
+    _SERVICE_ACCOUNT_SDK_SINGLETON_KEY,
+    PrivateServiceAccountSDK,
+    doc_py_example=_CREATE_API_KEY_EXAMPLE,
+    arg_docstrings=_CREATE_API_KEY_DOCSTRINGS,
+)
+def create_api_key(
+    email: Optional[str] = None,
+    name: Optional[str] = None,
+    *,
+    _sdk: PrivateServiceAccountSDK
+) -> str:
+    """Create an API key for the service account and return the API key.
+    """
+    return _sdk.create_api_key(email, name)
+
+
+@sdk_command(
+    _SERVICE_ACCOUNT_SDK_SINGLETON_KEY,
+    PrivateServiceAccountSDK,
+    doc_py_example=_LIST_EXAMPLE,
+    arg_docstrings=_LIST_DOCSTRINGS,
+)
+def list(  # noqa: A001
+    max_items: int = 20, *, _sdk: PrivateServiceAccountSDK
+) -> List[ServiceAccount]:
+    """List service accounts. """
+    return _sdk.list(max_items)
+
+
+@sdk_command(
+    _SERVICE_ACCOUNT_SDK_SINGLETON_KEY,
+    PrivateServiceAccountSDK,
+    doc_py_example=_DELETE_EXAMPLE,
+    arg_docstrings=_DELETE_DOCSTRINGS,
+)
+def delete(
+    email: Optional[str] = None,
+    name: Optional[str] = None,
+    *,
+    _sdk: PrivateServiceAccountSDK
+):
+    """Delete a service account.
+    """
+    return _sdk.delete(email, name)
+
+
+@sdk_command(
+    _SERVICE_ACCOUNT_SDK_SINGLETON_KEY,
+    PrivateServiceAccountSDK,
+    doc_py_example=_ROTATE_API_KEYS_EXAMPLE,
+    arg_docstrings=_ROTATE_API_KEYS_DOCSTRINGS,
+)
+def rotate_api_keys(
+    email: Optional[str] = None,
+    name: Optional[str] = None,
+    *,
+    _sdk: PrivateServiceAccountSDK
+) -> str:
+    """Rotate all api keys of a service account and return the new API key.
+    """
+    return _sdk.rotate_api_keys(email, name)

anyscale/service_account/models.py

@@ -0,0 +1,66 @@
+from dataclasses import dataclass, field
+from datetime import datetime
+
+from anyscale._private.models import ModelBase, ModelEnum
+
+
+# TODO(cynthiakwu): Move this when we have organization collaborator sdk
+class OrganizationPermissionLevel(ModelEnum):
+    OWNER = "OWNER"
+    COLLABORATOR = "COLLABORATOR"
+
+    __docstrings__ = {
+        OWNER: "Owner permission level for the organization",
+        COLLABORATOR: "Collaborator permission level for the organization",
+    }
+
+
+@dataclass(frozen=True)
+class ServiceAccount(ModelBase):
+    """Service account
+    """
+
+    __doc_py_example__ = """\
+import anyscale
+from anyscale.service_account.models import ServiceAccount
+
+service_accounts: List[ServiceAccount] = anyscale.service_account.list()
+"""
+
+    name: str = field(metadata={"docstring": "Name of the service account."})
+
+    def _validate_name(self, name: str):
+        if not isinstance(name, str):
+            raise TypeError("name must be a string.")
+
+    created_at: datetime = field(
+        metadata={"docstring": "The timestamp when this service account was created."}
+    )
+
+    def _validate_created_at(self, created_at: datetime):
+        if not isinstance(created_at, datetime):
+            raise TypeError("created_at must be a datetime.")
+
+    permission_level: OrganizationPermissionLevel = field(
+        metadata={
+            "docstring": "The organization permission level of the service account."
+        }
+    )
+
+    def _validate_permission_level(
+        self, permission_level: OrganizationPermissionLevel
+    ) -> OrganizationPermissionLevel:
+        if isinstance(permission_level, str):
+            return OrganizationPermissionLevel.validate(permission_level)
+        elif isinstance(permission_level, OrganizationPermissionLevel):
+            return permission_level
+        else:
+            raise TypeError(
+                f"'permission_level' must be a 'OrganizationPermissionLevel' (it is {type(permission_level)})."
+            )
+
+    email: str = field(metadata={"docstring": "Email of the service account."})
+
+    def _validate_email(self, email: str):
+        if not isinstance(email, str):
+            raise TypeError("email must be a string.")

anyscale/shared_anyscale_utils/latest_ray_version.py

@@ -1,2 +1,2 @@
 # AUTOGENERATED - modify shared_anyscale_util in root directory to make changes
-LATEST_RAY_VERSION = "2.40.0"
+LATEST_RAY_VERSION = "2.41.0"

anyscale/shared_anyscale_utils/utils/id_gen.py

@@ -111,7 +111,9 @@ class IDTypes(Enum):
     instance_usage_budgets = "iub"
     datasets = "dataset"
     resource_quotas = "rsq"
+    resource_alerts = "rsa"
     aggregated_instance_usage = "aiu"
+    readonly_collaborators = "rc"
 
 
 _default_id_length: int = 26

anyscale/user/__init__.py

@@ -30,6 +30,6 @@ class UserSDK:
     def admin_batch_create(  # noqa: F811
         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.
         """
         return self._private_sdk.admin_batch_create(admin_create_users)

anyscale/user/commands.py

@@ -37,6 +37,6 @@ _ADMIN_BATCH_CREATE_ARG_DOCSTRINGS = {
 def admin_batch_create(
     admin_create_users: List[AdminCreateUser], *, _sdk: PrivateUserSDK
 ) -> List[AdminCreatedUser]:
-    """Batch create users without email verification as an admin.
+    """Batch create, as an admin, users without email verification.
     """
     return _sdk.admin_batch_create(admin_create_users)

anyscale/user/models.py

@@ -15,13 +15,13 @@ import anyscale
 from anyscale.user.models import AdminCreateUser
 
 admin_create_user = AdminCreateUser(
-    # Name of the user to be created.
+    # First name of the user to be created.
     name="name",
     # Email of the user to be created.
     email="test@anyscale.com",
     # Password for the user account being created.
     password="",
-    # Whether the user is an SSO user.
+    # Whether the user is an SSO user. SSO users can log in using SSO.
     is_sso_user=False,
     # Optional last name of the user to be created.
     lastname="lastname",
@@ -29,7 +29,7 @@ admin_create_user = AdminCreateUser(
     title="title",
 )
 """
-    name: str = field(metadata={"docstring": "Name of the user to be created."})
+    name: str = field(metadata={"docstring": "First name of the user to be created."})
 
     def _validate_name(self, name: str):
         if not isinstance(name, str):
@@ -50,7 +50,9 @@ admin_create_user = AdminCreateUser(
             raise TypeError("password must be a string.")
 
     is_sso_user: bool = field(
-        metadata={"docstring": "Whether the user is an SSO user."},
+        metadata={
+            "docstring": "Whether the user is an SSO user. SSO users can log in using SSO."
+        },
     )
 
     def _validate_is_sso_user(self, is_sso_user: bool):
@@ -86,13 +88,13 @@ import anyscale
 from anyscale.user.models import AdminCreateUser
 
 admin_create_user = AdminCreateUser(
-    # Name of the user to be created.
+    # First name of the user to be created.
     name="name",
     # Email of the user to be created.
     email="test@anyscale.com",
     # Password for the user account being created.
     password="",
-    # Whether the user is an SSO user.
+    # Whether the user is an SSO user. SSO users can log in using SSO.
     is_sso_user=False,
     # Optional last name of the user to be created.
     lastname="lastname",
@@ -115,7 +117,7 @@ admin_create_users = AdminCreateUsers(
 
 @dataclass(frozen=True)
 class AdminCreatedUser(ModelBase):
-    """User account created by an admin.
+    """User account created by an admin that has organization collaborator permissions.
     """
 
     __doc_py_example__ = """\
@@ -123,13 +125,13 @@ import anyscale
 from anyscale.user.models import AdminCreatedUser
 
 admin_create_user = AdminCreateUser(
-    # Name of the user to be created.
+    # First name of the user to be created.
     name="name",
     # Email of the user to be created.
     email="test@anyscale.com",
     # Password for the user account being created.
     password="",
-    # Whether the user is an SSO user.
+    # Whether the user is an SSO user. SSO users can log in using SSO.
     is_sso_user=False,
     # Optional last name of the user to be created.
     lastname="lastname",
@@ -138,19 +140,25 @@ admin_create_user = AdminCreateUser(
 )
 admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([admin_create_user])
 """
-    user_id: str = field(metadata={"docstring": "Name of the user to be created."})
+    user_id: str = field(
+        metadata={"docstring": "ID of the user that has been created."}
+    )
 
     def _validate_user_id(self, user_id: str):
         if not isinstance(user_id, str):
             raise TypeError("user_id must be a string.")
 
-    name: str = field(metadata={"docstring": "Name of the user to be created."})
+    name: str = field(
+        metadata={"docstring": "First name of the user that has been created."}
+    )
 
     def _validate_name(self, name: str):
         if not isinstance(name, str):
             raise TypeError("name must be a string.")
 
-    email: str = field(metadata={"docstring": "Email of the user to be created."})
+    email: str = field(
+        metadata={"docstring": "Email of the user that has been created."}
+    )
 
     def _validate_email(self, email: str):
         if not isinstance(email, str):
@@ -165,7 +173,9 @@ admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([
             raise TypeError("created_at must be a datetime.")
 
     is_sso_user: bool = field(
-        metadata={"docstring": "Whether the user is an SSO user."},
+        metadata={
+            "docstring": "Whether the user is an SSO user. SSO users can log in using SSO."
+        },
     )
 
     def _validate_is_sso_user(self, is_sso_user: bool):
@@ -174,7 +184,7 @@ admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([
 
     lastname: Optional[str] = field(
         default=None,
-        metadata={"docstring": "Optional last name of the user to be created."},
+        metadata={"docstring": "Optional last name of the user that has been created."},
     )
 
     def _validate_lastname(self, lastname: Optional[str]):
@@ -183,7 +193,7 @@ admin_created_users: List[AdminCreatedUser] = anyscale.user.admin_batch_create([
 
     title: Optional[str] = field(
         default=None,
-        metadata={"docstring": "Optional title of the user to be created."},
+        metadata={"docstring": "Optional title of the user that has been created."},
     )
 
     def _validate_title(self, title: Optional[str]):

anyscale/util.py

@@ -44,6 +44,7 @@ from anyscale.client.openapi_client.models.cloud_analytics_event_cloud_resource
 from anyscale.client.openapi_client.models.decorated_compute_template import (
     DecoratedComputeTemplate,
 )
+from anyscale.client.openapi_client.models.ha_job_states import HaJobStates
 from anyscale.client.openapi_client.models.user_info import UserInfo
 from anyscale.cluster_compute import get_cluster_compute_from_name
 import anyscale.conf
@@ -986,6 +987,20 @@ def validate_job_config_dict(
         )
 
 
+def validate_list_jobs_state_filter(ctx, param, value) -> List[str]:  # noqa: ARG001
+    """
+    Validate the job state filter for list jobs CLI method
+    """
+    if not value:
+        return []
+    for each_value in value:
+        if each_value.upper() not in HaJobStates.allowable_values:
+            raise click.ClickException(
+                f"{each_value} is not a valid value for {param.opts[0]}. Allowed values: {', '.join(HaJobStates.allowable_values)}"
+            )
+    return [each_value.upper() for each_value in value]
+
+
 def get_cluster_model_for_current_workspace(
     anyscale_api_client: BaseApi,
 ) -> Optional[Cluster]:
@@ -1153,3 +1168,11 @@ T = TypeVar("T")
 
 def _coerce_to_list(maybe_list: Union[T, List[T]]) -> List[T]:
     return maybe_list if isinstance(maybe_list, list) else [maybe_list]
+
+
+def allow_optional_file_storage(api_client: Optional[ProductApi] = None) -> bool:
+    if api_client is None:
+        api_client = get_auth_api_client().api_client
+    return api_client.check_is_feature_flag_on_api_v2_userinfo_check_is_feature_flag_on_get(
+        "allow-optional-file-storage"
+    ).result.is_on

anyscale/utils/cloud_utils.py

@@ -20,7 +20,7 @@ from anyscale.client.openapi_client.models import (
     CloudProviders,
     CreateAnalyticsEvent,
 )
-from anyscale.cloud import get_cloud_id_and_name
+from anyscale.cloud_utils import get_cloud_id_and_name
 from anyscale.shared_anyscale_utils.utils.collections import flatten
 
 

anyscale/version.py

@@ -1 +1 @@
-__version__ = "0.24.88"
+__version__ = "0.25.5"

anyscale/workspace_utils.py

@@ -3,7 +3,7 @@ import os
 from click import ClickException
 import yaml
 
-from anyscale.project import ANYSCALE_PROJECT_FILE, find_project_root
+from anyscale.project_utils import ANYSCALE_PROJECT_FILE, find_project_root
 
 
 def write_workspace_id_to_disk(workspace_id: str, directory: str) -> None:

{anyscale-0.24.88.dist-info → anyscale-0.25.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.88
+Version: 0.25.5
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License
@@ -47,7 +47,6 @@ Requires-Dist: google-cloud-filestore; extra == "backend"
 Requires-Dist: google-cloud-storage; extra == "backend"
 Requires-Dist: google-cloud-redis; extra == "backend"
 Requires-Dist: google-cloud-certificate-manager; extra == "backend"
-Requires-Dist: azure-storage-blob; extra == "backend"
 Provides-Extra: gcp
 Requires-Dist: protobuf<4; python_version < "3.11" and extra == "gcp"
 Requires-Dist: google-api-python-client; extra == "gcp"
@@ -58,8 +57,6 @@ Requires-Dist: google-cloud-filestore; extra == "gcp"
 Requires-Dist: google-cloud-storage; extra == "gcp"
 Requires-Dist: google-cloud-redis; extra == "gcp"
 Requires-Dist: google-cloud-certificate-manager; extra == "gcp"
-Provides-Extra: azure
-Requires-Dist: azure-storage-blob; extra == "azure"
 Provides-Extra: all
 Requires-Dist: ray>=1.4.0; extra == "all"
 Requires-Dist: protobuf<4; python_version < "3.11" and extra == "all"
@@ -71,7 +68,6 @@ Requires-Dist: google-cloud-filestore; extra == "all"
 Requires-Dist: google-cloud-storage; extra == "all"
 Requires-Dist: google-cloud-redis; extra == "all"
 Requires-Dist: google-cloud-certificate-manager; extra == "all"
-Requires-Dist: azure-storage-blob; extra == "all"
 
 # Anyscale