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

anyscale/_private/anyscale_client/anyscale_client.py

@@ -1668,7 +1668,7 @@ class AnyscaleClient(AnyscaleClientInterface):
     def update_workspace(
         self,
         *,
-        workspace_id: str,
+        workspace_id: Optional[str],
         name: Optional[str] = None,
         compute_config_id: Optional[str] = None,
         cluster_environment_build_id: Optional[str] = None,
@@ -1760,7 +1760,9 @@ class AnyscaleClient(AnyscaleClientInterface):
         )
 
     @handle_api_exceptions
-    def get_workspace_cluster(self, workspace_id: str) -> Optional[DecoratedSession]:
+    def get_workspace_cluster(
+        self, workspace_id: Optional[str]
+    ) -> Optional[DecoratedSession]:
         workspace = self.get_workspace(id=workspace_id)
         if not workspace:
             raise ValueError(f"Workspace '{workspace_id}' not found.")

anyscale/_private/anyscale_client/common.py

@@ -561,14 +561,14 @@ class AnyscaleClientInterface(ABC):
 
     @abstractmethod
     def update_workspace_dependencies_offline_only(
-        self, workspace_id: str, requirements: List[str]
+        self, workspace_id: Optional[str], requirements: List[str]
     ):
         """Updates the dynamic dependencies of a workspace while the workspace is offline"""
         raise NotImplementedError
 
     @abstractmethod
     def update_workspace_env_vars_offline_only(
-        self, workspace_id: str, env_vars: Dict[str, str]
+        self, workspace_id: Optional[str], env_vars: Dict[str, str]
     ):
         """Updates the dynamic dependencies of a workspace"""
         raise NotImplementedError
@@ -581,7 +581,9 @@ class AnyscaleClientInterface(ABC):
         raise NotImplementedError
 
     @abstractmethod
-    def get_workspace_cluster(self, workspace_id: str) -> Optional[DecoratedSession]:
+    def get_workspace_cluster(
+        self, workspace_id: Optional[str]
+    ) -> Optional[DecoratedSession]:
         """Get the cluster model for the provided workspace ID."""
         raise NotImplementedError
 
@@ -622,7 +624,7 @@ class AnyscaleClientInterface(ABC):
     def update_workspace(
         self,
         *,
-        workspace_id: str,
+        workspace_id: Optional[str] = None,  # noqa: A002
         name: Optional[str] = None,
         compute_config_id: Optional[str] = None,
         cluster_environment_build_id: Optional[str] = None,

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -701,7 +701,10 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._job_runs[prod_job_id].append(model)
 
     def register_project_by_name(
-        self, name: str, cloud: str = DEFAULT_CLOUD_NAME, project_id: str = None
+        self,
+        name: str,
+        cloud: str = DEFAULT_CLOUD_NAME,
+        project_id: Optional[str] = None,
     ) -> str:
         """Helper method to create project name to project id mapping."""
         cloud_id = self.get_cloud_id(cloud_name=cloud)
@@ -1063,17 +1066,23 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         workspace.state = SessionState.TERMINATED
 
     def update_workspace_dependencies_offline_only(
-        self, workspace_id: str, requirements: List[str]
+        self, workspace_id: Optional[str], requirements: List[str]
     ):
+        assert workspace_id is not None
         self._workspaces_dependencies[workspace_id] = requirements
 
     def update_workspace_env_vars_offline_only(
-        self, workspace_id: str, env_vars: Dict[str, str]
+        self, workspace_id: Optional[str], env_vars: Dict[str, str]
     ):
+        assert workspace_id is not None
         self._workspaces_env_vars[workspace_id] = env_vars
 
-    def get_workspace_cluster(self, workspace_id: str) -> Optional[DecoratedSession]:
-        workspace_model = self._workspaces.get(workspace_id, None)
+    def get_workspace_cluster(
+        self, workspace_id: Optional[str]
+    ) -> Optional[DecoratedSession]:
+        workspace_model = (
+            self._workspaces.get(workspace_id, None) if workspace_id else None
+        )
         compute_config = self.get_compute_config(workspace_model.compute_config_id)
         assert compute_config is not None
         return Mock(
@@ -1136,7 +1145,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     def update_workspace(
         self,
         *,
-        workspace_id: str,
+        workspace_id: Optional[str] = None,  # noqa: A002
         name: Optional[str] = None,
         compute_config_id: Optional[str] = None,
         cluster_environment_build_id: Optional[str] = None,

anyscale/_private/docgen/__main__.py

@@ -233,7 +233,7 @@ ALL_MODULES = [
             "list_schedules": None,
         },
         legacy_cli_prefix="anyscale schedule",
-        legacy_cli_commands={schedule_commands.create, schedule_commands.update,},
+        legacy_cli_commands=[schedule_commands.create, schedule_commands.update],
     ),
     Module(
         title="Service",
@@ -298,7 +298,7 @@ ALL_MODULES = [
             "ServicemodelResponse",
         ],
         legacy_cli_prefix="anyscale service",
-        legacy_cli_commands={service_commands.rollout},
+        legacy_cli_commands=[service_commands.rollout],
     ),
     Module(
         title="Compute Config",
@@ -391,12 +391,12 @@ ALL_MODULES = [
         models=[ImageBuild, ImageBuildStatus],
         legacy_title="Cluster environment",
         legacy_cli_prefix="anyscale image",
-        legacy_cli_commands={
+        legacy_cli_commands=[
             cluster_env_commands.archive,
             cluster_env_commands.build,
             cluster_env_commands.get,
             cluster_env_commands.list,
-        },
+        ],
         legacy_sdk_commands={
             "get_cluster_environment_build_operation": None,
             "create_byod_cluster_environment_build": anyscale.image.build,

anyscale/_private/docgen/generator.py

@@ -60,7 +60,7 @@ class Module:
     legacy_title: Optional[str] = None
     legacy_cli_prefix: Optional[str] = None
     legacy_cli_commands: Optional[List[LegacyAnyscaleCommand]] = None
-    legacy_sdk_commands: Optional[Dict[str, Callable]] = None
+    legacy_sdk_commands: Optional[Dict[str, Optional[Callable]]] = None
     legacy_sdk_models: Optional[List[str]] = None
 
 
@@ -259,7 +259,7 @@ class MarkdownGenerator:
         if typing.get_origin(t) is not None:
             return self._type_container_to_string(t)
         if isinstance(t, (ModelBaseType, ModelEnumType)):
-            return f"[{t.__name__}]({self._get_anchor(t)})"
+            return f"[{t.__name__}]({self._get_anchor(t)})"  # type: ignore
 
         # Avoid poor rendering of unhandled types.
         raise NotImplementedError(

anyscale/_private/sdk/__init__.py

@@ -35,11 +35,11 @@ def sdk_command(
         def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
             # We disable the mypy linter here because it treats kwargs as a
             # P.kwargs object. mypy wrongly thinks kwargs can't be indexed.
-            if "_sdk" not in kwargs:  # type: ignore
+            if "_private_sdk" not in kwargs:  # type: ignore
                 if key not in _LAZY_SDK_SINGLETONS:
                     _LAZY_SDK_SINGLETONS[key] = sdk_cls()
 
-                kwargs["_sdk"] = _LAZY_SDK_SINGLETONS[key]  # type: ignore
+                kwargs["_private_sdk"] = _LAZY_SDK_SINGLETONS[key]  # type: ignore
 
             return f(*args, **kwargs)
 

anyscale/_private/workload/workload_sdk.py

@@ -211,7 +211,7 @@ class WorkloadSDK(BaseSDK):
     def resolve_compute_config_and_cloud_id(
         self,
         *,
-        compute_config: Union[None, str, ComputeConfig],
+        compute_config: Union[ComputeConfig, Dict, str, None],
         cloud: Union[None, str],
     ) -> Tuple[str, str]:
         """Resolve the passed compute config to its ID and corresponding cloud ID.
@@ -237,7 +237,7 @@ class WorkloadSDK(BaseSDK):
             return (compute_config_id, cloud_id)  # type: ignore
         elif cloud is None:
             compute_config_id = self._resolve_compute_config_id(
-                compute_config=compute_config
+                compute_config=compute_config  # type: ignore
             )
             return (
                 compute_config_id,
@@ -252,7 +252,7 @@ class WorkloadSDK(BaseSDK):
                 compute_config = replace(compute_config, cloud=cloud)
 
             compute_config_id = self._resolve_compute_config_id(
-                compute_config=compute_config
+                compute_config=compute_config  # type: ignore
             )
             cloud_id_from_cc = self.client.get_cloud_id(
                 compute_config_id=compute_config_id
@@ -270,7 +270,9 @@ class WorkloadSDK(BaseSDK):
         if not self._client.inside_workspace():
             return None
 
-        name = self._client.get_current_workspace_cluster().name
+        workspace = self._client.get_current_workspace_cluster()
+        assert workspace is not None
+        name = workspace.name
         # Defensively default to the workspace cluster name as-is if it doesn't
         # start with the expected prefix.
         if name.startswith(WORKSPACE_CLUSTER_NAME_PREFIX):

anyscale/aggregated_instance_usage/commands.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from anyscale._private.sdk import sdk_command
 from anyscale.aggregated_instance_usage._private.aggregated_instance_usage_sdk import (
     PrivateAggregatedInstanceUsageSDK,
@@ -33,8 +35,10 @@ _DOWNLOAD_ARG_DOCSTRINGS = {"filters": "The filter of the instance usage to down
     arg_docstrings=_DOWNLOAD_ARG_DOCSTRINGS,
 )
 def download_csv(
-    filters: DownloadCSVFilters, *, _sdk: PrivateAggregatedInstanceUsageSDK
+    filters: DownloadCSVFilters,
+    *,
+    _private_sdk: Optional[PrivateAggregatedInstanceUsageSDK] = None
 ) -> str:
     """Download an aggregated instance usage report as a zipped CSV to the provided directory.
     """
-    return _sdk.download_csv(filters)
+    return _private_sdk.download_csv(filters)  # type: ignore

anyscale/client/README.md

@@ -172,7 +172,6 @@ Class | Method | HTTP request | Description
 *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
 *DefaultApi* | [**find_with_invitation_api_v2_organizations_find_with_invitation_get**](docs/DefaultApi.md#find_with_invitation_api_v2_organizations_find_with_invitation_get) | **GET** /api/v2/organizations/find_with_invitation | Find With Invitation
 *DefaultApi* | [**find_with_user_api_v2_organizations_find_with_user_get**](docs/DefaultApi.md#find_with_user_api_v2_organizations_find_with_user_get) | **GET** /api/v2/organizations/find_with_user | Find With User
-*DefaultApi* | [**fine_tuning_job_callback_v2_api_v2_fine_tuning_v2_jobs_internal_finish_post**](docs/DefaultApi.md#fine_tuning_job_callback_v2_api_v2_fine_tuning_v2_jobs_internal_finish_post) | **POST** /api/v2/fine_tuning_v2/jobs/internal/finish | Fine Tuning Job Callback V2
 *DefaultApi* | [**finish_session_command_api_v2_session_commands_session_command_id_finish_post**](docs/DefaultApi.md#finish_session_command_api_v2_session_commands_session_command_id_finish_post) | **POST** /api/v2/session_commands/{session_command_id}/finish | Finish Session Command
 *DefaultApi* | [**generate_cloud_data_bucket_presigned_upload_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_upload_url_post**](docs/DefaultApi.md#generate_cloud_data_bucket_presigned_upload_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_upload_url_post) | **POST** /api/v2/clouds/{cloud_id}/generate_cloud_data_bucket_presigned_upload_url | Generate Cloud Data Bucket Presigned Upload Url
 *DefaultApi* | [**generate_cloud_data_bucket_presigned_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_url_post**](docs/DefaultApi.md#generate_cloud_data_bucket_presigned_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_url_post) | **POST** /api/v2/clouds/{cloud_id}/generate_cloud_data_bucket_presigned_url | Generate Cloud Data Bucket Presigned Url
@@ -191,7 +190,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_get**](docs/DefaultApi.md#get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_get) | **GET** /api/v2/clouds_with_cloud_resource_router/{cloud_id} | Get Cloud With Cloud Resource
 *DefaultApi* | [**get_cluster_access_token_api_v2_authentication_cluster_id_cluster_access_token_get**](docs/DefaultApi.md#get_cluster_access_token_api_v2_authentication_cluster_id_cluster_access_token_get) | **GET** /api/v2/authentication/{cluster_id}/cluster_access_token | Get Cluster Access Token
 *DefaultApi* | [**get_cluster_events_api_v2_sessions_session_id_cluster_events_get**](docs/DefaultApi.md#get_cluster_events_api_v2_sessions_session_id_cluster_events_get) | **GET** /api/v2/sessions/{session_id}/cluster_events | Get Cluster Events
-*DefaultApi* | [**get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get**](docs/DefaultApi.md#get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get) | **GET** /api/v2/logs/cluster_product_autoscaler_flag/{session_id} | Get Cluster Product Autoscaler Flag
 *DefaultApi* | [**get_compute_template_api_v2_compute_templates_template_id_get**](docs/DefaultApi.md#get_compute_template_api_v2_compute_templates_template_id_get) | **GET** /api/v2/compute_templates/{template_id} | Get Compute Template
 *DefaultApi* | [**get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get**](docs/DefaultApi.md#get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get) | **GET** /api/v2/experimental_cron_jobs/{cron_job_id} | Get Cron Job
 *DefaultApi* | [**get_dags_api_v2_dataset_runs_dags_get**](docs/DefaultApi.md#get_dags_api_v2_dataset_runs_dags_get) | **GET** /api/v2/dataset_runs/dags | Get Dags
@@ -312,6 +310,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_organization_project_collaborators_api_v2_projects_project_id_collaborators_organizations_get**](docs/DefaultApi.md#list_organization_project_collaborators_api_v2_projects_project_id_collaborators_organizations_get) | **GET** /api/v2/projects/{project_id}/collaborators/organizations | List Organization Project Collaborators
 *DefaultApi* | [**list_project_collaborators_api_v2_projects_project_id_collaborators_users_get**](docs/DefaultApi.md#list_project_collaborators_api_v2_projects_project_id_collaborators_users_get) | **GET** /api/v2/projects/{project_id}/collaborators/users | List Project Collaborators
 *DefaultApi* | [**list_projects_api_v2_projects_get**](docs/DefaultApi.md#list_projects_api_v2_projects_get) | **GET** /api/v2/projects/ | List Projects
+*DefaultApi* | [**list_ray_sessions_api_v2_tasks_dashboard_list_ray_sessions_get**](docs/DefaultApi.md#list_ray_sessions_api_v2_tasks_dashboard_list_ray_sessions_get) | **GET** /api/v2/tasks_dashboard/list_ray_sessions | List Ray Sessions
 *DefaultApi* | [**list_services_api_v2_services_v2_get**](docs/DefaultApi.md#list_services_api_v2_services_v2_get) | **GET** /api/v2/services-v2/ | List Services
 *DefaultApi* | [**list_sessions_api_v2_sessions_get**](docs/DefaultApi.md#list_sessions_api_v2_sessions_get) | **GET** /api/v2/sessions/ | List Sessions
 *DefaultApi* | [**list_web_terminals_api_v2_sessions_session_id_web_terminals_get**](docs/DefaultApi.md#list_web_terminals_api_v2_sessions_session_id_web_terminals_get) | **GET** /api/v2/sessions/{session_id}/web_terminals | List Web Terminals
@@ -430,7 +429,6 @@ Class | Method | HTTP request | Description
  - [ArchivedlogsinfoResponse](docs/ArchivedlogsinfoResponse.md)
  - [AttachMachinePoolToCloudRequest](docs/AttachMachinePoolToCloudRequest.md)
  - [AttachmachinepooltocloudresponseResponse](docs/AttachmachinepooltocloudresponseResponse.md)
- - [AviaryModelConfigV2](docs/AviaryModelConfigV2.md)
  - [AwsRegionAndZones](docs/AwsRegionAndZones.md)
  - [AwsRegionInfo](docs/AwsRegionInfo.md)
  - [AwsregionandzonesResponse](docs/AwsregionandzonesResponse.md)
@@ -665,7 +663,6 @@ Class | Method | HTTP request | Description
  - [FineTunedModel](docs/FineTunedModel.md)
  - [FinetunedmodelListResponse](docs/FinetunedmodelListResponse.md)
  - [FinetunedmodelResponse](docs/FinetunedmodelResponse.md)
- - [FinishFTJobRequestV2](docs/FinishFTJobRequestV2.md)
  - [GCPFileStoreConfig](docs/GCPFileStoreConfig.md)
  - [GCPMemorystoreInstanceConfig](docs/GCPMemorystoreInstanceConfig.md)
  - [GrafanaDashboard](docs/GrafanaDashboard.md)
@@ -713,11 +710,13 @@ Class | Method | HTTP request | Description
  - [LbresourceResponse](docs/LbresourceResponse.md)
  - [ListMachinePoolsResponse](docs/ListMachinePoolsResponse.md)
  - [ListMachinesResponse](docs/ListMachinesResponse.md)
+ - [ListRaySessionsResponse](docs/ListRaySessionsResponse.md)
  - [ListResourceNotificationsQuery](docs/ListResourceNotificationsQuery.md)
  - [ListResourceQuotasQuery](docs/ListResourceQuotasQuery.md)
  - [ListResponseMetadata](docs/ListResponseMetadata.md)
  - [ListmachinepoolsresponseResponse](docs/ListmachinepoolsresponseResponse.md)
  - [ListmachinesresponseResponse](docs/ListmachinesresponseResponse.md)
+ - [ListraysessionsresponseResponse](docs/ListraysessionsresponseResponse.md)
  - [LogDownloadConfig](docs/LogDownloadConfig.md)
  - [LogDownloadRequest](docs/LogDownloadRequest.md)
  - [LogDownloadResult](docs/LogDownloadResult.md)
@@ -802,9 +801,7 @@ Class | Method | HTTP request | Description
  - [PageQuery](docs/PageQuery.md)
  - [PauseSchedule](docs/PauseSchedule.md)
  - [PermissionLevel](docs/PermissionLevel.md)
- - [ProductAutoscalerFlag](docs/ProductAutoscalerFlag.md)
  - [ProductType](docs/ProductType.md)
- - [ProductautoscalerflagResponse](docs/ProductautoscalerflagResponse.md)
  - [ProductionJob](docs/ProductionJob.md)
  - [ProductionJobConfig](docs/ProductionJobConfig.md)
  - [ProductionJobEvent](docs/ProductionJobEvent.md)
@@ -828,6 +825,7 @@ Class | Method | HTTP request | Description
  - [Quota](docs/Quota.md)
  - [RayGCSExternalStorageConfig](docs/RayGCSExternalStorageConfig.md)
  - [RayRuntimeEnvConfig](docs/RayRuntimeEnvConfig.md)
+ - [RaySession](docs/RaySession.md)
  - [ReadBillingVersion](docs/ReadBillingVersion.md)
  - [ReadbillingversionListResponse](docs/ReadbillingversionListResponse.md)
  - [ReplicaDetails](docs/ReplicaDetails.md)

anyscale/client/openapi_client/__init__.py

@@ -61,7 +61,6 @@ from openapi_client.models.archived_logs_info import ArchivedLogsInfo
 from openapi_client.models.archivedlogsinfo_response import ArchivedlogsinfoResponse
 from openapi_client.models.attach_machine_pool_to_cloud_request import AttachMachinePoolToCloudRequest
 from openapi_client.models.attachmachinepooltocloudresponse_response import AttachmachinepooltocloudresponseResponse
-from openapi_client.models.aviary_model_config_v2 import AviaryModelConfigV2
 from openapi_client.models.aws_region_and_zones import AwsRegionAndZones
 from openapi_client.models.aws_region_info import AwsRegionInfo
 from openapi_client.models.awsregionandzones_response import AwsregionandzonesResponse
@@ -296,7 +295,6 @@ from openapi_client.models.fine_tune_type import FineTuneType
 from openapi_client.models.fine_tuned_model import FineTunedModel
 from openapi_client.models.finetunedmodel_list_response import FinetunedmodelListResponse
 from openapi_client.models.finetunedmodel_response import FinetunedmodelResponse
-from openapi_client.models.finish_ft_job_request_v2 import FinishFTJobRequestV2
 from openapi_client.models.gcp_file_store_config import GCPFileStoreConfig
 from openapi_client.models.gcp_memorystore_instance_config import GCPMemorystoreInstanceConfig
 from openapi_client.models.grafana_dashboard import GrafanaDashboard
@@ -344,11 +342,13 @@ from openapi_client.models.lb_resource import LBResource
 from openapi_client.models.lbresource_response import LbresourceResponse
 from openapi_client.models.list_machine_pools_response import ListMachinePoolsResponse
 from openapi_client.models.list_machines_response import ListMachinesResponse
+from openapi_client.models.list_ray_sessions_response import ListRaySessionsResponse
 from openapi_client.models.list_resource_notifications_query import ListResourceNotificationsQuery
 from openapi_client.models.list_resource_quotas_query import ListResourceQuotasQuery
 from openapi_client.models.list_response_metadata import ListResponseMetadata
 from openapi_client.models.listmachinepoolsresponse_response import ListmachinepoolsresponseResponse
 from openapi_client.models.listmachinesresponse_response import ListmachinesresponseResponse
+from openapi_client.models.listraysessionsresponse_response import ListraysessionsresponseResponse
 from openapi_client.models.log_download_config import LogDownloadConfig
 from openapi_client.models.log_download_request import LogDownloadRequest
 from openapi_client.models.log_download_result import LogDownloadResult
@@ -433,9 +433,7 @@ from openapi_client.models.organizationusagealert_list_response import Organizat
 from openapi_client.models.page_query import PageQuery
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
-from openapi_client.models.product_autoscaler_flag import ProductAutoscalerFlag
 from openapi_client.models.product_type import ProductType
-from openapi_client.models.productautoscalerflag_response import ProductautoscalerflagResponse
 from openapi_client.models.production_job import ProductionJob
 from openapi_client.models.production_job_config import ProductionJobConfig
 from openapi_client.models.production_job_event import ProductionJobEvent
@@ -459,6 +457,7 @@ from openapi_client.models.python_modules import PythonModules
 from openapi_client.models.quota import Quota
 from openapi_client.models.ray_gcs_external_storage_config import RayGCSExternalStorageConfig
 from openapi_client.models.ray_runtime_env_config import RayRuntimeEnvConfig
+from openapi_client.models.ray_session import RaySession
 from openapi_client.models.read_billing_version import ReadBillingVersion
 from openapi_client.models.readbillingversion_list_response import ReadbillingversionListResponse
 from openapi_client.models.replica_details import ReplicaDetails