anyscale 0.5.136__tar.gz → 0.5.137__tar.gz

{anyscale-0.5.136/anyscale.egg-info → anyscale-0.5.137}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.5.136
+Version: 0.5.137
 Summary: Command Line Interface for Anyscale
 Home-page: UNKNOWN
 Author: Anyscale Inc.

{anyscale-0.5.136 → anyscale-0.5.137}/anyscale/cli_logger.py

@@ -1,13 +1,21 @@
+import contextlib
 from contextlib import contextmanager
 import os
 import sys
 import time
-from typing import Any, Optional
+from typing import Any, List, Optional
 
+from botocore.exceptions import ClientError
 import click
-import colorama  # type: ignore
+import colorama
 from halo import Halo
 
+from anyscale.client.openapi_client.models import (
+    CloudAnalyticsEventCloudProviderError,
+    CloudAnalyticsEventCloudResource,
+)
+from anyscale.utils.imports.gcp import try_import_gcp_exceptions
+
 
 _process_start_time = time.time()
 
@@ -245,3 +253,97 @@ class LogsLogger(BlockLogger):
             text=msg, spinner="dots", stream=sys.stderr, enabled=enable_spinner
         ) as spinner:
             yield spinner
+
+
+class CloudSetupLogger(LogsLogger):
+    def __init__(
+        self,
+        log_output: bool = True,
+        t0: float = _process_start_time,
+        spinner_manager: Any = None,
+    ):
+        self.cloud_resource_errors: List[CloudAnalyticsEventCloudProviderError] = []
+        super().__init__(log_output, t0, spinner_manager)
+
+    def log_resource_exception(
+        self, resource: CloudAnalyticsEventCloudResource, exc: Exception
+    ):
+        """
+        Record the error in the logger if the exception is an unhandled exception
+        """
+        if resource in (
+            CloudAnalyticsEventCloudResource.GCP_DEPLOYMENT,
+            CloudAnalyticsEventCloudResource.GCP_FILESTORE,
+            CloudAnalyticsEventCloudResource.GCP_FIREWALL_POLICY,
+            CloudAnalyticsEventCloudResource.GCP_PROJECT,
+            CloudAnalyticsEventCloudResource.GCP_SERVICE_ACCOUNT,
+            CloudAnalyticsEventCloudResource.GCP_STORAGE_BUCKET,
+            CloudAnalyticsEventCloudResource.GCP_SUBNET,
+            CloudAnalyticsEventCloudResource.GCP_VPC,
+            CloudAnalyticsEventCloudResource.GCP_WORKLOAD_IDENTITY_PROVIDER,
+        ):
+            # exceptions from GCP
+            GoogleAPICallError, HttpError = try_import_gcp_exceptions()
+            if isinstance(exc, GoogleAPICallError):
+                self.log_resource_error(
+                    resource, exc.reason, exc.code, "GoogleAPICallError",
+                )
+                return
+            elif isinstance(exc, HttpError):
+                reason = None
+                with contextlib.suppress(Exception):
+                    for error_detail in exc.error_details:
+                        if error_detail.get("reason"):
+                            reason = str(error_detail["reason"])
+                            break
+                self.log_resource_error(resource, reason, exc.status_code, "HttpError")
+                return
+        elif isinstance(exc, ClientError):
+            # boto3 client error
+            self.log_resource_error(
+                resource,
+                exc.response.get("Error", {}).get("Code"),
+                exc.response.get("ResponseMetadata", {}).get("HTTPStatusCode"),
+                "ClientError",
+            )
+            return
+        else:
+            unhandled_exception = f"UnknownExceptionType_{exc.__class__.__name__}"
+            self.log_resource_error(
+                resource, None, None, unhandled_exception,
+            )
+            return
+
+    def log_resource_error(
+        self,
+        cloud_resource: CloudAnalyticsEventCloudResource,
+        error_reason: Optional[str] = None,
+        status_code: Optional[int] = None,
+        unhandled_exception: Optional[str] = None,
+    ):
+        """
+        Append the error in the logger. If the error is already logged, it will be deduplicated.
+        """
+        formatted_error_str = error_reason if error_reason else "unknown"
+        if status_code:
+            formatted_error_str = f"{formatted_error_str},{status_code}"
+        if unhandled_exception:
+            formatted_error_str = f"{formatted_error_str},{unhandled_exception}"
+        for error in self.cloud_resource_errors:
+            if (
+                error.cloud_resource == cloud_resource
+                and error.error_code == formatted_error_str
+            ):
+                # deduplicated errors
+                return
+        self.cloud_resource_errors.append(
+            CloudAnalyticsEventCloudProviderError(
+                cloud_resource=cloud_resource, error_code=formatted_error_str
+            )
+        )
+
+    def get_cloud_provider_errors(self) -> List[CloudAnalyticsEventCloudProviderError]:
+        return self.cloud_resource_errors
+
+    def clear_cloud_provider_errors(self):
+        self.cloud_resource_errors = []

{anyscale-0.5.136 → anyscale-0.5.137}/anyscale/client/README.md

@@ -103,6 +103,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**create_connect_session_api_v2_sessions_create_connect_session_post**](docs/DefaultApi.md#create_connect_session_api_v2_sessions_create_connect_session_post) | **POST** /api/v2/sessions/create_connect_session | Create Connect Session
 *DefaultApi* | [**create_invitation_api_v2_organization_invitations_post**](docs/DefaultApi.md#create_invitation_api_v2_organization_invitations_post) | **POST** /api/v2/organization_invitations/ | Create Invitation
 *DefaultApi* | [**create_job_api_v2_decorated_ha_jobs_create_post**](docs/DefaultApi.md#create_job_api_v2_decorated_ha_jobs_create_post) | **POST** /api/v2/decorated_ha_jobs/create | Create Job
+*DefaultApi* | [**create_machine_api_v2_machines_create_post**](docs/DefaultApi.md#create_machine_api_v2_machines_create_post) | **POST** /api/v2/machines/create | Create Machine
 *DefaultApi* | [**create_new_session_api_v2_sessions_create_new_session_post**](docs/DefaultApi.md#create_new_session_api_v2_sessions_create_new_session_post) | **POST** /api/v2/sessions/create_new_session | Create New Session
 *DefaultApi* | [**create_one_time_password_api_v2_users_create_otp_token_get**](docs/DefaultApi.md#create_one_time_password_api_v2_users_create_otp_token_get) | **GET** /api/v2/users/create_otp_token | Create One Time Password
 *DefaultApi* | [**create_or_update_job_api_v2_experimental_cron_jobs_put**](docs/DefaultApi.md#create_or_update_job_api_v2_experimental_cron_jobs_put) | **PUT** /api/v2/experimental_cron_jobs/ | Create Or Update Job
@@ -120,6 +121,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**delete_cloud_api_v2_clouds_cloud_id_delete**](docs/DefaultApi.md#delete_cloud_api_v2_clouds_cloud_id_delete) | **DELETE** /api/v2/clouds/{cloud_id} | Delete Cloud
 *DefaultApi* | [**delete_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_identity_id_delete**](docs/DefaultApi.md#delete_cloud_collaborator_api_v2_clouds_cloud_id_collaborators_identity_id_delete) | **DELETE** /api/v2/clouds/{cloud_id}/collaborators/{identity_id} | Delete Cloud Collaborator
 *DefaultApi* | [**delete_compute_template_api_v2_compute_templates_template_id_delete**](docs/DefaultApi.md#delete_compute_template_api_v2_compute_templates_template_id_delete) | **DELETE** /api/v2/compute_templates/{template_id} | Delete Compute Template
+*DefaultApi* | [**delete_machine_api_v2_machines_delete_post**](docs/DefaultApi.md#delete_machine_api_v2_machines_delete_post) | **POST** /api/v2/machines/delete | Delete Machine
 *DefaultApi* | [**delete_project_api_v2_projects_project_id_delete**](docs/DefaultApi.md#delete_project_api_v2_projects_project_id_delete) | **DELETE** /api/v2/projects/{project_id} | Delete Project
 *DefaultApi* | [**delete_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_delete**](docs/DefaultApi.md#delete_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_delete) | **DELETE** /api/v2/projects/{project_id}/collaborators/{role_or_identity_id} | Delete Project Collaborator
 *DefaultApi* | [**delete_session_api_v2_sessions_session_id_delete**](docs/DefaultApi.md#delete_session_api_v2_sessions_session_id_delete) | **DELETE** /api/v2/sessions/{session_id} | Delete Session
@@ -129,6 +131,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**describe_session_api_v2_sessions_session_id_describe_get**](docs/DefaultApi.md#describe_session_api_v2_sessions_session_id_describe_get) | **GET** /api/v2/sessions/{session_id}/describe | Describe Session
 *DefaultApi* | [**dismiss_user_card_api_v2_onboarding_cards_card_id_dismiss_post**](docs/DefaultApi.md#dismiss_user_card_api_v2_onboarding_cards_card_id_dismiss_post) | **POST** /api/v2/onboarding_cards/{card_id}/dismiss | Dismiss User Card
 *DefaultApi* | [**echo_ip_api_v2_health_echo_ip_get**](docs/DefaultApi.md#echo_ip_api_v2_health_echo_ip_get) | **GET** /api/v2/health/echo_ip | Echo Ip
+*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* | [**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
@@ -151,6 +154,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_cloud_project_collaborator_api_v2_projects_project_id_collaborators_clouds_get**](docs/DefaultApi.md#get_cloud_project_collaborator_api_v2_projects_project_id_collaborators_clouds_get) | **GET** /api/v2/projects/{project_id}/collaborators/clouds | Get Cloud Project Collaborator
 *DefaultApi* | [**get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_get**](docs/DefaultApi.md#get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_get) | **GET** /api/v2/clouds_with_cloud_resource_gcp_router/{cloud_id} | Get Cloud With Cloud Resource
 *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_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_decorated_cluster_api_v2_decorated_sessions_cluster_id_get**](docs/DefaultApi.md#get_decorated_cluster_api_v2_decorated_sessions_cluster_id_get) | **GET** /api/v2/decorated_sessions/{cluster_id} | Get Decorated Cluster
@@ -402,6 +406,7 @@ Class | Method | HTTP request | Description
  - [CloudWithCloudResourceGCP](docs/CloudWithCloudResourceGCP.md)
  - [CloudcollaboratorListResponse](docs/CloudcollaboratorListResponse.md)
  - [CloudregionandzonesResponse](docs/CloudregionandzonesResponse.md)
+ - [CloudresourceResponse](docs/CloudresourceResponse.md)
  - [CloudwithcloudresourceResponse](docs/CloudwithcloudresourceResponse.md)
  - [CloudwithcloudresourcegcpResponse](docs/CloudwithcloudresourcegcpResponse.md)
  - [ClusterAuthResponse](docs/ClusterAuthResponse.md)
@@ -442,6 +447,8 @@ Class | Method | HTTP request | Description
  - [CreateExperimentalWorkspace](docs/CreateExperimentalWorkspace.md)
  - [CreateExperimentalWorkspaceFromJob](docs/CreateExperimentalWorkspaceFromJob.md)
  - [CreateInternalProductionJob](docs/CreateInternalProductionJob.md)
+ - [CreateMachineRequest](docs/CreateMachineRequest.md)
+ - [CreateMachineResponse](docs/CreateMachineResponse.md)
  - [CreateOTPReturnApiModel](docs/CreateOTPReturnApiModel.md)
  - [CreateOrganizationInvitation](docs/CreateOrganizationInvitation.md)
  - [CreateProductionJobConfig](docs/CreateProductionJobConfig.md)
@@ -455,6 +462,7 @@ Class | Method | HTTP request | Description
  - [CreateUserProjectCollaborator](docs/CreateUserProjectCollaborator.md)
  - [CreateUserProjectCollaboratorValue](docs/CreateUserProjectCollaboratorValue.md)
  - [CreateWorkspaceFromTemplate](docs/CreateWorkspaceFromTemplate.md)
+ - [CreatemachineresponseResponse](docs/CreatemachineresponseResponse.md)
  - [CreateotpreturnapimodelResponse](docs/CreateotpreturnapimodelResponse.md)
  - [CreatesessionresponseResponse](docs/CreatesessionresponseResponse.md)
  - [CreditCardInformation](docs/CreditCardInformation.md)
@@ -509,9 +517,11 @@ Class | Method | HTTP request | Description
  - [DecoratedsupportrequestResponse](docs/DecoratedsupportrequestResponse.md)
  - [DecoratedunifiedjobListResponse](docs/DecoratedunifiedjobListResponse.md)
  - [DecoratedunifiedjobResponse](docs/DecoratedunifiedjobResponse.md)
+ - [DeleteMachineRequest](docs/DeleteMachineRequest.md)
  - [DeploymentStatus](docs/DeploymentStatus.md)
  - [DismissalType](docs/DismissalType.md)
  - [EbsBlockDevice](docs/EbsBlockDevice.md)
+ - [EditableCloudResource](docs/EditableCloudResource.md)
  - [Error](docs/Error.md)
  - [ExecuteCommandResponse](docs/ExecuteCommandResponse.md)
  - [ExecuteInteractiveCommandOptions](docs/ExecuteInteractiveCommandOptions.md)
@@ -623,7 +633,9 @@ 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)
  - [ProductionJobStateTransition](docs/ProductionJobStateTransition.md)
@@ -667,6 +679,7 @@ Class | Method | HTTP request | Description
  - [ServerSessionToken](docs/ServerSessionToken.md)
  - [ServersessiontokenResponse](docs/ServersessiontokenResponse.md)
  - [ServiceAccount](docs/ServiceAccount.md)
+ - [ServiceConfig](docs/ServiceConfig.md)
  - [ServiceEventCurrentState](docs/ServiceEventCurrentState.md)
  - [ServiceEventLevel](docs/ServiceEventLevel.md)
  - [ServiceEventOrigin](docs/ServiceEventOrigin.md)

{anyscale-0.5.136 → anyscale-0.5.137}/anyscale/client/openapi_client/__init__.py

@@ -109,6 +109,7 @@ from openapi_client.models.cloud_with_cloud_resource import CloudWithCloudResour
 from openapi_client.models.cloud_with_cloud_resource_gcp import CloudWithCloudResourceGCP
 from openapi_client.models.cloudcollaborator_list_response import CloudcollaboratorListResponse
 from openapi_client.models.cloudregionandzones_response import CloudregionandzonesResponse
+from openapi_client.models.cloudresource_response import CloudresourceResponse
 from openapi_client.models.cloudwithcloudresource_response import CloudwithcloudresourceResponse
 from openapi_client.models.cloudwithcloudresourcegcp_response import CloudwithcloudresourcegcpResponse
 from openapi_client.models.cluster_auth_response import ClusterAuthResponse
@@ -149,6 +150,8 @@ from openapi_client.models.create_compute_template_config import CreateComputeTe
 from openapi_client.models.create_experimental_workspace import CreateExperimentalWorkspace
 from openapi_client.models.create_experimental_workspace_from_job import CreateExperimentalWorkspaceFromJob
 from openapi_client.models.create_internal_production_job import CreateInternalProductionJob
+from openapi_client.models.create_machine_request import CreateMachineRequest
+from openapi_client.models.create_machine_response import CreateMachineResponse
 from openapi_client.models.create_otp_return_api_model import CreateOTPReturnApiModel
 from openapi_client.models.create_organization_invitation import CreateOrganizationInvitation
 from openapi_client.models.create_production_job_config import CreateProductionJobConfig
@@ -162,6 +165,7 @@ from openapi_client.models.create_user import CreateUser
 from openapi_client.models.create_user_project_collaborator import CreateUserProjectCollaborator
 from openapi_client.models.create_user_project_collaborator_value import CreateUserProjectCollaboratorValue
 from openapi_client.models.create_workspace_from_template import CreateWorkspaceFromTemplate
+from openapi_client.models.createmachineresponse_response import CreatemachineresponseResponse
 from openapi_client.models.createotpreturnapimodel_response import CreateotpreturnapimodelResponse
 from openapi_client.models.createsessionresponse_response import CreatesessionresponseResponse
 from openapi_client.models.credit_card_information import CreditCardInformation
@@ -216,9 +220,11 @@ from openapi_client.models.decoratedsupportrequest_list_response import Decorate
 from openapi_client.models.decoratedsupportrequest_response import DecoratedsupportrequestResponse
 from openapi_client.models.decoratedunifiedjob_list_response import DecoratedunifiedjobListResponse
 from openapi_client.models.decoratedunifiedjob_response import DecoratedunifiedjobResponse
+from openapi_client.models.delete_machine_request import DeleteMachineRequest
 from openapi_client.models.deployment_status import DeploymentStatus
 from openapi_client.models.dismissal_type import DismissalType
 from openapi_client.models.ebs_block_device import EbsBlockDevice
+from openapi_client.models.editable_cloud_resource import EditableCloudResource
 from openapi_client.models.error import Error
 from openapi_client.models.execute_command_response import ExecuteCommandResponse
 from openapi_client.models.execute_interactive_command_options import ExecuteInteractiveCommandOptions
@@ -330,7 +336,9 @@ from openapi_client.models.organizationprojectcollaborator_list_response import
 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_state_transition import ProductionJobStateTransition
@@ -374,6 +382,7 @@ from openapi_client.models.servedeploymentlogs_response import Servedeploymentlo
 from openapi_client.models.server_session_token import ServerSessionToken
 from openapi_client.models.serversessiontoken_response import ServersessiontokenResponse
 from openapi_client.models.service_account import ServiceAccount
+from openapi_client.models.service_config import ServiceConfig
 from openapi_client.models.service_event_current_state import ServiceEventCurrentState
 from openapi_client.models.service_event_level import ServiceEventLevel
 from openapi_client.models.service_event_origin import ServiceEventOrigin