anyscale 0.26.14__py3-none-any.whl → 0.26.16__py3-none-any.whl

anyscale/image/commands.py

@@ -36,13 +36,13 @@ def build(
     *,
     name: str,
     ray_version: Optional[str] = None,
-    _sdk: PrivateImageSDK
+    _private_sdk: Optional[PrivateImageSDK] = None,
 ) -> str:
     """Build an image from a Containerfile.
 
     Returns the URI of the image.
     """
-    return _sdk.build_image_from_containerfile_with_image_uri(
+    return _private_sdk.build_image_from_containerfile_with_image_uri(  # type: ignore
         name, containerfile, ray_version=ray_version
     )
 
@@ -68,12 +68,12 @@ _GET_ARG_DOCSTRINGS = {
     doc_py_example=_GET_EXAMPLE,
     arg_docstrings=_GET_ARG_DOCSTRINGS,
 )
-def get(*, name: str, _sdk: PrivateImageSDK) -> ImageBuild:
+def get(*, name: str, _private_sdk: Optional[PrivateImageSDK] = None) -> ImageBuild:
     """The name can contain an optional version tag, i.e., 'name:version'.
 
     If no version is provided, the latest one will be returned.
     """
-    return _sdk.get(name)
+    return _private_sdk.get(name)  # type: ignore
 
 
 _REGISTER_EXAMPLE = """
@@ -102,14 +102,14 @@ def register(
     name: str,
     ray_version: Optional[str] = None,
     registry_login_secret: Optional[str] = None,
-    _sdk: PrivateImageSDK
+    _private_sdk: Optional[PrivateImageSDK] = None,
 ) -> str:
     """
     Registers a BYOD image with a container image name.
 
     Returns the URI of the image.
     """
-    return _sdk.register_byod_image_with_name(
+    return _private_sdk.register_byod_image_with_name(  # type: ignore
         image_uri,
         registry_login_secret=registry_login_secret,
         ray_version=ray_version,

anyscale/job/commands.py

@@ -54,12 +54,12 @@ _SUBMIT_ARG_DOCSTRINGS = {"config": "The config options defining the job."}
     doc_py_example=_SUBMIT_EXAMPLE,
     arg_docstrings=_SUBMIT_ARG_DOCSTRINGS,
 )
-def submit(config: JobConfig, *, _sdk: PrivateJobSDK) -> str:
+def submit(config: JobConfig, *, _private_sdk: Optional[PrivateJobSDK] = None) -> str:
     """Submit a job.
 
     Returns the id of the submitted job.
     """
-    return _sdk.submit(config)
+    return _private_sdk.submit(config)  # type: ignore
 
 
 _STATUS_EXAMPLE = """
@@ -89,12 +89,12 @@ def status(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> JobStatus:
     """Get the status of a job."""
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.status(name=name, job_id=id, cloud=cloud, project=project)
+    return _private_sdk.status(name=name, job_id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _TERMINATE_EXAMPLE = """
@@ -123,7 +123,7 @@ def terminate(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> str:
     """Terminate a job.
@@ -133,7 +133,7 @@ def terminate(
     Returns the id of the terminated job.
     """
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.terminate(name=name, job_id=id, cloud=cloud, project=project)
+    return _private_sdk.terminate(name=name, job_id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _ARCHIVE_EXAMPLE = """
@@ -162,7 +162,7 @@ def archive(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> str:
     """Archive a job.
@@ -172,7 +172,7 @@ def archive(
     Returns the id of the archived job.
     """
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.archive(name=name, job_id=id, cloud=cloud, project=project)
+    return _private_sdk.archive(name=name, job_id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _WAIT_EXAMPLE = """\
@@ -204,12 +204,12 @@ def wait(
     project: Optional[str] = None,
     state: Union[JobState, str] = JobState.SUCCEEDED,
     timeout_s: float = 1800,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ):
     """"Wait for a job to enter a specific state."""
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    _sdk.wait(
+    _private_sdk.wait(  # type: ignore
         name=name,
         job_id=id,
         cloud=cloud,
@@ -251,12 +251,12 @@ def get_logs(
     run: Optional[str] = None,
     mode: Union[str, JobLogMode] = JobLogMode.TAIL,
     max_lines: Optional[int] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> str:
     """Query the jobs for a job run."""
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.get_logs(
+    return _private_sdk.get_logs(  # type: ignore
         job_id=id,
         name=name,
         cloud=cloud,

anyscale/organization_invitation/commands.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Tuple
+from typing import Dict, List, Optional, Tuple
 
 from anyscale._private.sdk import sdk_command
 from anyscale.organization_invitation._private.organization_invitation_sdk import (
@@ -45,13 +45,15 @@ _DELETE_ARG_DOCSTRINGS = {
     arg_docstrings=_CREATE_ARG_DOCSTRINGS,
 )
 def create(
-    emails: List[str], *, _sdk: PrivateOrganizationInvitationSDK
+    emails: List[str],
+    *,
+    _private_sdk: Optional[PrivateOrganizationInvitationSDK] = None
 ) -> Tuple[List[str], List[str]]:
     """Creates organization invitations for the provided emails.
 
     Returns a tuple of successful emails and error messages.
     """
-    return _sdk.create(emails=emails)
+    return _private_sdk.create(emails=emails)  # type: ignore
 
 
 @sdk_command(
@@ -61,13 +63,13 @@ def create(
     arg_docstrings=_LIST_ARG_DOCSTRINGS,
 )
 def list(  # noqa: A001
-    *, _sdk: PrivateOrganizationInvitationSDK
+    *, _private_sdk: Optional[PrivateOrganizationInvitationSDK] = None
 ) -> List[OrganizationInvitation]:
     """Lists organization invitations.
 
     Returns a list of organization invitations.
     """
-    return _sdk.list()
+    return _private_sdk.list()  # type: ignore
 
 
 @sdk_command(
@@ -76,9 +78,11 @@ def list(  # noqa: A001
     doc_py_example=_DELETE_EXAMPLE,
     arg_docstrings=_DELETE_ARG_DOCSTRINGS,
 )
-def delete(email: str, *, _sdk: PrivateOrganizationInvitationSDK) -> str:
+def delete(
+    email: str, *, _private_sdk: Optional[PrivateOrganizationInvitationSDK] = None
+) -> str:
     """Deletes an organization invitation.
 
     Returns the email of the deleted organization invitation.
     """
-    return _sdk.delete(email)
+    return _private_sdk.delete(email)  # type: ignore

anyscale/project/__init__.py

@@ -29,7 +29,7 @@ class ProjectSDK:
     )
     def add_collaborators(  # noqa: F811
         self, cloud: str, project: str, collaborators: List[CreateProjectCollaborator],
-    ) -> str:
+    ) -> None:
         """Batch add collaborators to a project.
         """
-        return self._private_sdk.add_collaborators(cloud, project, collaborators)
+        self._private_sdk.add_collaborators(cloud, project, collaborators)

anyscale/project/_private/project_sdk.py

@@ -11,11 +11,11 @@ from anyscale.project.models import CreateProjectCollaborator
 class PrivateProjectSDK(BaseSDK):
     def add_collaborators(
         self, cloud: str, project: str, collaborators: List[CreateProjectCollaborator]
-    ) -> str:
+    ) -> None:
         cloud_id = self.client.get_cloud_id(cloud_name=cloud, compute_config_id=None)
         project_id = self.client.get_project_id(parent_cloud_id=cloud_id, name=project)
 
-        return self.client.add_project_collaborators(
+        self.client.add_project_collaborators(
             project_id=project_id,
             collaborators=[
                 CreateUserProjectCollaborator(

anyscale/project/commands.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from anyscale._private.sdk import sdk_command
 from anyscale.project._private.project_sdk import PrivateProjectSDK
@@ -49,8 +49,8 @@ def add_collaborators(
     project: str,
     collaborators: List[CreateProjectCollaborator],
     *,
-    _sdk: PrivateProjectSDK
+    _private_sdk: Optional[PrivateProjectSDK] = None
 ) -> str:
     """Batch add collaborators to a project.
     """
-    return _sdk.add_collaborators(cloud, project, collaborators)
+    return _private_sdk.add_collaborators(cloud, project, collaborators)  # type: ignore

anyscale/project_utils.py

@@ -14,6 +14,7 @@ from anyscale.cluster_compute import (
     get_cluster_compute_from_name,
     get_selected_cloud_id_or_default,
 )
+from anyscale.sdk.anyscale_client.api.default_api import DefaultApi as SDKDefaultApi
 from anyscale.sdk.anyscale_client.models.cluster_compute_config import (
     ClusterComputeConfig,
 )
@@ -395,7 +396,7 @@ def get_and_validate_project_id(
     project_name: Optional[str],
     parent_cloud_id: Optional[str],
     api_client: DefaultApi,
-    anyscale_api_client: DefaultApi,
+    anyscale_api_client: Union[DefaultApi, SDKDefaultApi],
 ) -> str:
     project_name = project_name or os.environ.get(PROJECT_NAME_ENV_VAR)
     if project_id:
@@ -421,7 +422,7 @@ def get_and_validate_project_id(
 
 def get_default_project(
     api_client: Optional[DefaultApi] = None,
-    anyscale_api_client: Optional[DefaultApi] = None,
+    anyscale_api_client: Optional[Union[DefaultApi, SDKDefaultApi]] = None,
     parent_cloud_id: Optional[str] = None,
 ):
     """
@@ -441,14 +442,14 @@ def get_default_project(
             "Please specify a cloud for this command either through the cloud or "
             "compute config arguments."
         )
-    default_project = anyscale_api_client.get_default_project(
+    default_project = anyscale_api_client.get_default_project(  # type: ignore
         parent_cloud_id=parent_cloud_id
     ).result
     return default_project
 
 
 def infer_project_id(  # noqa PLR0913
-    anyscale_api_client: DefaultApi,
+    anyscale_api_client: Union[DefaultApi, SDKDefaultApi],
     api_client: DefaultApi,
     log: BlockLogger,
     project_id: Optional[str],

anyscale/resource_quota/commands.py

@@ -87,11 +87,13 @@ _DISABLE_DOCSTRINGS = {
     arg_docstrings=_CREATE_DOCSTRINGS,
 )
 def create(
-    create_resource_quota: CreateResourceQuota, *, _sdk: PrivateResourceQuotaSDK
+    create_resource_quota: CreateResourceQuota,
+    *,
+    _private_sdk: Optional[PrivateResourceQuotaSDK] = None
 ) -> ResourceQuota:
     """Create a resource quota.
     """
-    return _sdk.create(create_resource_quota)
+    return _private_sdk.create(create_resource_quota)  # type: ignore
 
 
 @sdk_command(
@@ -107,10 +109,10 @@ def list(  # noqa: A001
     is_enabled: Optional[bool] = None,
     max_items: int = 20,
     *,
-    _sdk: PrivateResourceQuotaSDK
+    _private_sdk: Optional[PrivateResourceQuotaSDK] = None
 ) -> List[ResourceQuota]:
     """List resource quotas. """
-    return _sdk.list(name, cloud, creator_id, is_enabled, max_items,)
+    return _private_sdk.list(name, cloud, creator_id, is_enabled, max_items,)  # type: ignore
 
 
 @sdk_command(
@@ -119,10 +121,12 @@ def list(  # noqa: A001
     doc_py_example=_DELETE_EXAMPLE,
     arg_docstrings=_DELETE_DOCSTRINGS,
 )
-def delete(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
+def delete(
+    resource_quota_id: str, *, _private_sdk: Optional[PrivateResourceQuotaSDK] = None
+):
     """Delete a resource quota.
     """
-    return _sdk.delete(resource_quota_id)
+    return _private_sdk.delete(resource_quota_id)  # type: ignore
 
 
 @sdk_command(
@@ -131,10 +135,12 @@ def delete(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
     doc_py_example=_ENABLE_EXAMPLE,
     arg_docstrings=_ENABLE_DOCSTRINGS,
 )
-def enable(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
+def enable(
+    resource_quota_id: str, *, _private_sdk: Optional[PrivateResourceQuotaSDK] = None
+):
     """Enable a resource quota.
     """
-    return _sdk.set_status(resource_quota_id, True)
+    return _private_sdk.set_status(resource_quota_id, True)  # type: ignore
 
 
 @sdk_command(
@@ -143,7 +149,9 @@ def enable(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
     doc_py_example=_DISABLE_EXAMPLE,
     arg_docstrings=_DISABLE_DOCSTRINGS,
 )
-def disable(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
+def disable(
+    resource_quota_id: str, *, _private_sdk: Optional[PrivateResourceQuotaSDK] = None
+):
     """Disable a resource quota.
     """
-    return _sdk.set_status(resource_quota_id, False)
+    return _private_sdk.set_status(resource_quota_id, False)  # type: ignore

anyscale/schedule/commands.py

@@ -36,12 +36,14 @@ _APPLY_ARG_DOCSTRINGS = {"config": "The config options defining the schedule."}
     doc_py_example=_APPLY_EXAMPLE,
     arg_docstrings=_APPLY_ARG_DOCSTRINGS,
 )
-def apply(config: ScheduleConfig, *, _sdk: PrivateScheduleSDK) -> str:
+def apply(
+    config: ScheduleConfig, *, _private_sdk: Optional[PrivateScheduleSDK] = None
+) -> str:
     """Apply or update a schedule.
 
     Returns the id of the schedule.
     """
-    return _sdk.apply(config)
+    return _private_sdk.apply(config)  # type: ignore
 
 
 _SET_STATE_EXAMPLE = """
@@ -76,13 +78,15 @@ def set_state(
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     state: ScheduleState,
-    _sdk: PrivateScheduleSDK
+    _private_sdk: Optional[PrivateScheduleSDK] = None
 ) -> str:
     """Set the state of a schedule.
 
     Returns the id of the schedule.
     """
-    return _sdk.set_state(id=id, name=name, cloud=cloud, project=project, state=state,)
+    return _private_sdk.set_state(  # type: ignore
+        id=id, name=name, cloud=cloud, project=project, state=state,
+    )
 
 
 _STATUS_EXAMPLE = """
@@ -110,11 +114,11 @@ def status(
     name: Optional[str] = None,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateScheduleSDK
+    _private_sdk: Optional[PrivateScheduleSDK] = None
 ) -> ScheduleStatus:
     """Return the status of the schedule.
     """
-    return _sdk.status(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.status(id=id, name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _TRIGGER_EXAMPLE = """
@@ -142,8 +146,8 @@ def trigger(
     name: Optional[str] = None,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateScheduleSDK
+    _private_sdk: Optional[PrivateScheduleSDK] = None
 ) -> str:
     """Trigger the execution of the schedule.
     """
-    return _sdk.trigger(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.trigger(id=id, name=name, cloud=cloud, project=project)  # type: ignore

anyscale/sdk/anyscale_client/sdk.py

@@ -68,7 +68,7 @@ def _upload_and_rewrite_working_dir_in_create_production_job(create_production_j
 
 def _upload_and_rewrite_working_dir_ray_serve_config(
     ray_serve_config: Optional[Dict[str, Any]]
-) -> Dict[str, Any]:
+) -> Optional[Dict[str, Any]]:
     """
     If a local working_dir and upload_path are specified, this method
     will upload the working_dir and rewrites the working_dir to the remote uri.
@@ -159,7 +159,7 @@ class AnyscaleSDK(DefaultApi):  # type: ignore
         poll_rate_seconds: int = 15,
         timeout_seconds: Optional[int] = None,
         log_output: bool = False,
-    ) -> ClusterEnvironmentBuild:
+    ) -> Optional[ClusterEnvironmentBuild]:
         """
         Creates a new Cluster Environment and waits for build to complete.
 
@@ -199,7 +199,7 @@ class AnyscaleSDK(DefaultApi):  # type: ignore
                 correct_base_image_name = get_correct_name_for_base_image_name(base_image_name)
                 if base_image_name != correct_base_image_name: # We need to make sure base_image_name after ray 2.7.0, before 2.8.0 have a optimized suffix
                     self.log.error(f"Starting from Ray version 2.7.0 and before version 2.8.0, you might be required to include the suffix 'optimized' in the base_image name for that specific Ray version. Try using {self.log.highlight(correct_base_image_name)}")
-                    return
+                    return  # type: ignore
                 cluster_environment = self.create_cluster_environment(
                     create_cluster_environment
                 ).result
@@ -224,7 +224,7 @@ class AnyscaleSDK(DefaultApi):  # type: ignore
                 correct_base_image_name = get_correct_name_for_base_image_name(base_image_name)
                 if base_image_name != correct_base_image_name: # We need to make sure base_image_name after ray 2.7.0, before 2.8.0 have a optimized suffix
                     self.log.error(f"Starting from Ray version 2.7.0 and before version 2.8.0, you might be required to include the suffix 'optimized' in the base_image name for that specific Ray version. Try using {self.log.highlight(correct_base_image_name)}")
-                    return
+                    return  # type: ignore
                 build = self.create_cluster_environment_build(
                     {
                         "cluster_environment_id": cluster_environment.id,
@@ -477,6 +477,7 @@ class AnyscaleSDK(DefaultApi):  # type: ignore
             create_cluster_environment, poll_rate_seconds, timeout_seconds
         )
 
+        assert cluster_environment_build is not None
         return self.launch_cluster(
             project_id,
             cluster_name,

anyscale/service/commands.py

@@ -48,7 +48,7 @@ def deploy(
     in_place: bool = False,
     canary_percent: Optional[int] = None,
     max_surge_percent: Optional[int] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Deploy a service.
 
@@ -58,7 +58,7 @@ def deploy(
 
     Returns the id of the deployed service.
     """
-    return _sdk.deploy(
+    return _private_sdk.deploy(  # type: ignore
         config,
         in_place=in_place,
         canary_percent=canary_percent,
@@ -92,7 +92,7 @@ def rollback(
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     max_surge_percent: Optional[int] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Rollback to the primary version of the service.
 
@@ -103,7 +103,7 @@ def rollback(
 
     Returns the id of the rolled back service.
     """
-    return _sdk.rollback(
+    return _private_sdk.rollback(  # type: ignore
         name=name, cloud=cloud, project=project, max_surge_percent=max_surge_percent
     )
 
@@ -132,7 +132,7 @@ def terminate(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Terminate a service.
 
@@ -140,7 +140,7 @@ def terminate(
 
     Returns the id of the terminated service.
     """
-    return _sdk.terminate(name=name, cloud=cloud, project=project)
+    return _private_sdk.terminate(name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _ARCHIVE_EXAMPLE = """
@@ -169,7 +169,7 @@ def archive(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Archive a service.
 
@@ -177,7 +177,7 @@ def archive(
 
     Returns the ID of the archived service.
     """
-    return _sdk.archive(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.archive(id=id, name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _DELETE_EXAMPLE = """
@@ -206,7 +206,7 @@ def delete(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Delete a service.
 
@@ -214,7 +214,7 @@ def delete(
 
     Returns the ID of the deleted service.
     """
-    return _sdk.delete(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.delete(id=id, name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _STATUS_EXAMPLE = """
@@ -242,10 +242,10 @@ def status(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> ServiceStatus:
     """Get the status of a service."""
-    return _sdk.status(name=name, cloud=cloud, project=project)
+    return _private_sdk.status(name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _WAIT_EXAMPLE = """
@@ -277,18 +277,18 @@ def wait(
     project: Optional[str] = None,
     state: Union[str, ServiceState] = ServiceState.RUNNING,
     timeout_s: float = 600,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
     _interval_s: float = 5,
 ):
     """Wait for a service to reach a target state."""
-    _sdk.wait(
+    _private_sdk.wait(  # type: ignore
         name=name,
         cloud=cloud,
         project=project,
         state=ServiceState(state),
         timeout_s=timeout_s,
         interval_s=_interval_s,
-    )
+    )  # type: ignore
 
 
 _CONTROLLER_LOGS_EXAMPLE = """
@@ -322,10 +322,10 @@ def _controller_logs(
     canary: bool = False,
     mode: Union[str, ServiceLogMode] = ServiceLogMode.TAIL,
     max_lines: int = 1000,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ):
     """Wait for a service to reach a target state."""
-    return _sdk.controller_logs(
+    return _private_sdk.controller_logs(  # type: ignore
         name,
         cloud=cloud,
         project=project,

anyscale/service_account/commands.py

@@ -77,10 +77,12 @@ _ROTATE_API_KEYS_DOCSTRINGS = {
     doc_py_example=_CREATE_EXAMPLE,
     arg_docstrings=_CREATE_DOCSTRINGS,
 )
-def create(name: str, *, _sdk: PrivateServiceAccountSDK) -> str:
+def create(
+    name: str, *, _private_sdk: Optional[PrivateServiceAccountSDK] = None
+) -> str:
     """Create a service account and return the API key.
     """
-    return _sdk.create(name)
+    return _private_sdk.create(name)  # type: ignore
 
 
 @sdk_command(
@@ -93,11 +95,11 @@ def create_api_key(
     email: Optional[str] = None,
     name: Optional[str] = None,
     *,
-    _sdk: PrivateServiceAccountSDK
+    _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ) -> str:
     """Create an API key for the service account and return the API key.
     """
-    return _sdk.create_api_key(email, name)
+    return _private_sdk.create_api_key(email, name)  # type: ignore
 
 
 @sdk_command(
@@ -107,10 +109,10 @@ def create_api_key(
     arg_docstrings=_LIST_DOCSTRINGS,
 )
 def list(  # noqa: A001
-    max_items: int = 20, *, _sdk: PrivateServiceAccountSDK
+    max_items: int = 20, *, _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ) -> List[ServiceAccount]:
     """List service accounts. """
-    return _sdk.list(max_items)
+    return _private_sdk.list(max_items)  # type: ignore
 
 
 @sdk_command(
@@ -123,11 +125,11 @@ def delete(
     email: Optional[str] = None,
     name: Optional[str] = None,
     *,
-    _sdk: PrivateServiceAccountSDK
+    _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ):
     """Delete a service account.
     """
-    return _sdk.delete(email, name)
+    return _private_sdk.delete(email, name)  # type: ignore
 
 
 @sdk_command(
@@ -140,8 +142,8 @@ def rotate_api_keys(
     email: Optional[str] = None,
     name: Optional[str] = None,
     *,
-    _sdk: PrivateServiceAccountSDK
+    _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ) -> str:
     """Rotate all api keys of a service account and return the new API key.
     """
-    return _sdk.rotate_api_keys(email, name)
+    return _private_sdk.rotate_api_keys(email, name)  # type: ignore

anyscale/user/commands.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from anyscale._private.sdk import sdk_command
 from anyscale.user._private.user_sdk import PrivateUserSDK
@@ -35,8 +35,10 @@ _ADMIN_BATCH_CREATE_ARG_DOCSTRINGS = {
     arg_docstrings=_ADMIN_BATCH_CREATE_ARG_DOCSTRINGS,
 )
 def admin_batch_create(
-    admin_create_users: List[AdminCreateUser], *, _sdk: PrivateUserSDK
+    admin_create_users: List[AdminCreateUser],
+    *,
+    _private_sdk: Optional[PrivateUserSDK] = None
 ) -> List[AdminCreatedUser]:
     """Batch create, as an admin, users without email verification.
     """
-    return _sdk.admin_batch_create(admin_create_users)
+    return _private_sdk.admin_batch_create(admin_create_users)  # type: ignore