anyscale 0.26.31__py3-none-any.whl → 0.26.33__py3-none-any.whl

anyscale/api.py

@@ -9,6 +9,7 @@ import click
 from urllib3.connection import HTTPConnection
 import wrapt
 
+from anyscale.cli_logger import BlockLogger
 from anyscale.client import openapi_client
 from anyscale.client.openapi_client.api.default_api import DefaultApi
 from anyscale.client.openapi_client.rest import ApiException as ApiExceptionInternal
@@ -16,9 +17,12 @@ from anyscale.sdk import anyscale_client
 from anyscale.sdk.anyscale_client.api.default_api import DefaultApi as AnyscaleApi
 from anyscale.sdk.anyscale_client.rest import ApiException as ApiExceptionExternal
 from anyscale.shared_anyscale_utils.headers import RequestHeaders
+from anyscale.telemetry import get_traceparent
 from anyscale.version import __version__ as version
 
 
+logger = BlockLogger()
+
 # NOTE: Some OSes don't implement all of these, so only include those that the OS supports.
 _SOCKET_OPTIONS = (
     HTTPConnection.default_socket_options
@@ -115,6 +119,15 @@ class ApiClientWrapperInternal(openapi_client.ApiClient):
         _request_timeout=None,
         _host=None,
     ):
+        # Add tracing correlation info
+        traceparent = get_traceparent()
+        if traceparent:
+            if header_params is None:
+                header_params = {}
+            header_params[RequestHeaders.TRACEPARENT] = traceparent
+
+        logger.debug(f"[API Internal] {method} {resource_path} (trace: {traceparent})")
+
         try:
             return openapi_client.ApiClient.call_api(
                 self,
@@ -173,6 +186,15 @@ class ApiClientWrapperExternal(anyscale_client.ApiClient):
         _request_timeout=None,
         _host=None,
     ):
+        # Add tracing correlation info
+        traceparent = get_traceparent()
+        if traceparent:
+            if header_params is None:
+                header_params = {}
+            header_params[RequestHeaders.TRACEPARENT] = traceparent
+
+        logger.debug(f"[API External] {method} {resource_path} (trace: {traceparent})")
+
         try:
             return anyscale_client.ApiClient.call_api(
                 self,

anyscale/aws_iam_policies.py

@@ -303,9 +303,6 @@ ANYSCALE_IAM_PERMISSIONS_SERVICE_STEADY_STATE = {
                 "elasticloadbalancing:SetSubnets",
             ],
             "Resource": "*",
-            "Condition": {
-                "StringEquals": {"aws:CalledViaFirst": "cloudformation.amazonaws.com"}
-            },
         },
         {
             "Sid": "CreateELBServiceLinkedRole",

anyscale/client/README.md

@@ -88,7 +88,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**archive_service_api_v2_services_v2_service_id_archive_post**](docs/DefaultApi.md#archive_service_api_v2_services_v2_service_id_archive_post) | **POST** /api/v2/services-v2/{service_id}/archive | Archive Service
 *DefaultApi* | [**attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post**](docs/DefaultApi.md#attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post) | **POST** /api/v2/machine_pools/attach | Attach Machine Pool To Cloud
 *DefaultApi* | [**aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post**](docs/DefaultApi.md#aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post) | **POST** /api/v2/organization_billing/aws_marketplace_registration | Aws Marketplace Registration
-*DefaultApi* | [**ban_organization_api_v2_organizations_organization_id_ban_put**](docs/DefaultApi.md#ban_organization_api_v2_organizations_organization_id_ban_put) | **PUT** /api/v2/organizations/{organization_id}/ban | Ban Organization
 *DefaultApi* | [**batch_create_cloud_collaborators_api_v2_clouds_cloud_id_collaborators_users_batch_create_post**](docs/DefaultApi.md#batch_create_cloud_collaborators_api_v2_clouds_cloud_id_collaborators_users_batch_create_post) | **POST** /api/v2/clouds/{cloud_id}/collaborators/users/batch_create | Batch Create Cloud Collaborators
 *DefaultApi* | [**batch_create_invitations_api_v2_organization_invitations_batch_create_post**](docs/DefaultApi.md#batch_create_invitations_api_v2_organization_invitations_batch_create_post) | **POST** /api/v2/organization_invitations/batch_create | Batch Create Invitations
 *DefaultApi* | [**batch_create_project_collaborators_api_v2_projects_project_id_collaborators_users_batch_create_post**](docs/DefaultApi.md#batch_create_project_collaborators_api_v2_projects_project_id_collaborators_users_batch_create_post) | **POST** /api/v2/projects/{project_id}/collaborators/users/batch_create | Batch Create Project Collaborators
@@ -179,6 +178,8 @@ Class | Method | HTTP request | Description
 *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_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_credits_v2_api_v2_organization_billing_credits_v2_get**](docs/DefaultApi.md#get_credits_v2_api_v2_organization_billing_credits_v2_get) | **GET** /api/v2/organization_billing/credits_v2 | Get Credits V2
+*DefaultApi* | [**get_credits_v2_by_organization_api_v2_organization_billing_credits_v2_organization_id_get**](docs/DefaultApi.md#get_credits_v2_by_organization_api_v2_organization_billing_credits_v2_organization_id_get) | **GET** /api/v2/organization_billing/credits_v2/{organization_id} | Get Credits V2 By Organization
 *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
 *DefaultApi* | [**get_dashboard_api_v2_dataset_runs_get**](docs/DefaultApi.md#get_dashboard_api_v2_dataset_runs_get) | **GET** /api/v2/dataset_runs/ | Get Dashboard
@@ -213,6 +214,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_operator_metrics_api_v2_dataset_runs_operator_get**](docs/DefaultApi.md#get_operator_metrics_api_v2_dataset_runs_operator_get) | **GET** /api/v2/dataset_runs/operator | Get Operator Metrics
 *DefaultApi* | [**get_or_create_build_from_image_uri_api_v2_builds_get_or_create_build_from_image_uri_post**](docs/DefaultApi.md#get_or_create_build_from_image_uri_api_v2_builds_get_or_create_build_from_image_uri_post) | **POST** /api/v2/builds/get_or_create_build_from_image_uri | Get Or Create Build From Image Uri
 *DefaultApi* | [**get_organization_metronome_usage_alerts_api_v2_organization_billing_alerts_get**](docs/DefaultApi.md#get_organization_metronome_usage_alerts_api_v2_organization_billing_alerts_get) | **GET** /api/v2/organization_billing/alerts | Get Organization Metronome Usage Alerts
+*DefaultApi* | [**get_plan_status_api_v2_organization_billing_plan_status_get**](docs/DefaultApi.md#get_plan_status_api_v2_organization_billing_plan_status_get) | **GET** /api/v2/organization_billing/plan_status | Get Plan Status
 *DefaultApi* | [**get_project_api_v2_projects_project_id_get**](docs/DefaultApi.md#get_project_api_v2_projects_project_id_get) | **GET** /api/v2/projects/{project_id} | Get Project
 *DefaultApi* | [**get_project_default_session_name_api_v2_projects_project_id_default_session_name_get**](docs/DefaultApi.md#get_project_default_session_name_api_v2_projects_project_id_default_session_name_get) | **GET** /api/v2/projects/{project_id}/default_session_name | Get Project Default Session Name
 *DefaultApi* | [**get_recent_cluster_compute_configs_api_v2_recent_activity_cluster_compute_configs_get**](docs/DefaultApi.md#get_recent_cluster_compute_configs_api_v2_recent_activity_cluster_compute_configs_get) | **GET** /api/v2/recent_activity/cluster_compute_configs | Get Recent Cluster Compute Configs
@@ -300,6 +302,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**put_workspace_proxied_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_proxied_dataplane_artifacts_put**](docs/DefaultApi.md#put_workspace_proxied_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_proxied_dataplane_artifacts_put) | **PUT** /api/v2/experimental_workspaces/{workspace_id}/proxied_dataplane_artifacts | Put Workspace Proxied Dataplane Artifacts
 *DefaultApi* | [**put_workspace_wandb_run_details_api_v2_integrations_workspace_wandb_run_details_workspace_id_put**](docs/DefaultApi.md#put_workspace_wandb_run_details_api_v2_integrations_workspace_wandb_run_details_workspace_id_put) | **PUT** /api/v2/integrations/workspace_wandb_run_details/{workspace_id} | Put Workspace Wandb Run Details
 *DefaultApi* | [**query_aggregated_logs_api_v2_logs_query_aggregated_logs_get**](docs/DefaultApi.md#query_aggregated_logs_api_v2_logs_query_aggregated_logs_get) | **GET** /api/v2/logs/query_aggregated_logs | Query Aggregated Logs
+*DefaultApi* | [**receive_cli_usage_api_v2_cli_usage_post**](docs/DefaultApi.md#receive_cli_usage_api_v2_cli_usage_post) | **POST** /api/v2/cli_usage/ | Receive Cli Usage
 *DefaultApi* | [**redirect_to_service_api_v2_sessions_cluster_id_services_get**](docs/DefaultApi.md#redirect_to_service_api_v2_sessions_cluster_id_services_get) | **GET** /api/v2/sessions/{cluster_id}/services | Redirect To Service
 *DefaultApi* | [**redirect_to_tools_api_v2_services_v2_service_id_tools_get**](docs/DefaultApi.md#redirect_to_tools_api_v2_services_v2_service_id_tools_get) | **GET** /api/v2/services-v2/{service_id}/tools | Redirect To Tools
 *DefaultApi* | [**register_api_v2_kubernetes_manager_register_post**](docs/DefaultApi.md#register_api_v2_kubernetes_manager_register_post) | **POST** /api/v2/kubernetes_manager/register | Register
@@ -331,6 +334,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**sync_organization_with_metronome_api_v2_organization_billing_organization_id_sync_with_metronome_post**](docs/DefaultApi.md#sync_organization_with_metronome_api_v2_organization_billing_organization_id_sync_with_metronome_post) | **POST** /api/v2/organization_billing/{organization_id}/sync_with_metronome | Sync Organization With Metronome
 *DefaultApi* | [**terminate_job_api_v2_decorated_ha_jobs_production_job_id_terminate_post**](docs/DefaultApi.md#terminate_job_api_v2_decorated_ha_jobs_production_job_id_terminate_post) | **POST** /api/v2/decorated_ha_jobs/{production_job_id}/terminate | Terminate Job
 *DefaultApi* | [**terminate_service_api_v2_services_v2_service_id_terminate_post**](docs/DefaultApi.md#terminate_service_api_v2_services_v2_service_id_terminate_post) | **POST** /api/v2/services-v2/{service_id}/terminate | Terminate Service
+*DefaultApi* | [**terminate_system_cluster_api_v2_system_workload_cloud_id_terminate_post**](docs/DefaultApi.md#terminate_system_cluster_api_v2_system_workload_cloud_id_terminate_post) | **POST** /api/v2/system_workload/{cloud_id}/terminate | Terminate System Cluster
 *DefaultApi* | [**test_saml_acs_api_v2_organizations_organization_id_test_saml_acs_post**](docs/DefaultApi.md#test_saml_acs_api_v2_organizations_organization_id_test_saml_acs_post) | **POST** /api/v2/organizations/{organization_id}/test_saml_acs | Test Saml Acs
 *DefaultApi* | [**toggle_instance_usage_budget_is_enabled_api_v2_instance_usage_budgets_instance_usage_budget_id_toggle_is_enabled_post**](docs/DefaultApi.md#toggle_instance_usage_budget_is_enabled_api_v2_instance_usage_budgets_instance_usage_budget_id_toggle_is_enabled_post) | **POST** /api/v2/instance_usage_budgets/{instance_usage_budget_id}/toggle_is_enabled | Toggle Instance Usage Budget Is Enabled
 *DefaultApi* | [**trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post**](docs/DefaultApi.md#trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post) | **POST** /api/v2/experimental_cron_jobs/{cron_job_id}/trigger | Trigger Cron Job
@@ -413,6 +417,7 @@ Class | Method | HTTP request | Description
  - [BuildResponse](docs/BuildResponse.md)
  - [BuildStatus](docs/BuildStatus.md)
  - [BuildlogresponseResponse](docs/BuildlogresponseResponse.md)
+ - [CLIUsagePayload](docs/CLIUsagePayload.md)
  - [ChangePasswordParams](docs/ChangePasswordParams.md)
  - [CleanupLeakedGrafanaDashboardResponse](docs/CleanupLeakedGrafanaDashboardResponse.md)
  - [CleanupleakedgrafanadashboardresponseResponse](docs/CleanupleakedgrafanadashboardresponseResponse.md)
@@ -479,12 +484,18 @@ Class | Method | HTTP request | Description
  - [ClusterEventSource](docs/ClusterEventSource.md)
  - [ClusterEventsOutput](docs/ClusterEventsOutput.md)
  - [ClusterManagementStackVersions](docs/ClusterManagementStackVersions.md)
+ - [ClusterOperation](docs/ClusterOperation.md)
+ - [ClusterOperationType](docs/ClusterOperationType.md)
  - [ClusterState](docs/ClusterState.md)
  - [ClusterStatus](docs/ClusterStatus.md)
  - [ClusterStatusDetails](docs/ClusterStatusDetails.md)
  - [ClusterauthresponseResponse](docs/ClusterauthresponseResponse.md)
  - [ClustereventListResponse](docs/ClustereventListResponse.md)
  - [ClustereventsoutputResponse](docs/ClustereventsoutputResponse.md)
+ - [ClusteroperationResponse](docs/ClusteroperationResponse.md)
+ - [CommitLedgerItemType](docs/CommitLedgerItemType.md)
+ - [CommitLedgerRecordV2](docs/CommitLedgerRecordV2.md)
+ - [ComplexityLevel](docs/ComplexityLevel.md)
  - [ComputeNodeType](docs/ComputeNodeType.md)
  - [ComputeStack](docs/ComputeStack.md)
  - [ComputeTemplate](docs/ComputeTemplate.md)
@@ -536,6 +547,13 @@ Class | Method | HTTP request | Description
  - [CreatemachinepoolresponseResponse](docs/CreatemachinepoolresponseResponse.md)
  - [CreatemachineresponseResponse](docs/CreatemachineresponseResponse.md)
  - [CreateotpreturnapimodelResponse](docs/CreateotpreturnapimodelResponse.md)
+ - [CreditGrantRecordV2](docs/CreditGrantRecordV2.md)
+ - [CreditLedgerItemType](docs/CreditLedgerItemType.md)
+ - [CreditLedgerRecordV2](docs/CreditLedgerRecordV2.md)
+ - [CreditRecordCommitV2](docs/CreditRecordCommitV2.md)
+ - [CreditRecordCreditV2](docs/CreditRecordCreditV2.md)
+ - [CreditType](docs/CreditType.md)
+ - [CreditsV2](docs/CreditsV2.md)
  - [CustomerAlertStatus](docs/CustomerAlertStatus.md)
  - [DataplaneServices](docs/DataplaneServices.md)
  - [Dataset](docs/Dataset.md)
@@ -735,6 +753,9 @@ Class | Method | HTTP request | Description
  - [NotificationChannelSlackConfig](docs/NotificationChannelSlackConfig.md)
  - [NotificationChannelWebhookConfig](docs/NotificationChannelWebhookConfig.md)
  - [ObjectStorage](docs/ObjectStorage.md)
+ - [OperationError](docs/OperationError.md)
+ - [OperationProgress](docs/OperationProgress.md)
+ - [OperationResult](docs/OperationResult.md)
  - [OperatorBranch](docs/OperatorBranch.md)
  - [OperatorId](docs/OperatorId.md)
  - [OperatorMetrics](docs/OperatorMetrics.md)

anyscale/client/openapi_client/__init__.py

@@ -77,6 +77,7 @@ from openapi_client.models.build_registration import BuildRegistration
 from openapi_client.models.build_response import BuildResponse
 from openapi_client.models.build_status import BuildStatus
 from openapi_client.models.buildlogresponse_response import BuildlogresponseResponse
+from openapi_client.models.cli_usage_payload import CLIUsagePayload
 from openapi_client.models.change_password_params import ChangePasswordParams
 from openapi_client.models.cleanup_leaked_grafana_dashboard_response import CleanupLeakedGrafanaDashboardResponse
 from openapi_client.models.cleanupleakedgrafanadashboardresponse_response import CleanupleakedgrafanadashboardresponseResponse
@@ -143,12 +144,18 @@ from openapi_client.models.cluster_event import ClusterEvent
 from openapi_client.models.cluster_event_source import ClusterEventSource
 from openapi_client.models.cluster_events_output import ClusterEventsOutput
 from openapi_client.models.cluster_management_stack_versions import ClusterManagementStackVersions
+from openapi_client.models.cluster_operation import ClusterOperation
+from openapi_client.models.cluster_operation_type import ClusterOperationType
 from openapi_client.models.cluster_state import ClusterState
 from openapi_client.models.cluster_status import ClusterStatus
 from openapi_client.models.cluster_status_details import ClusterStatusDetails
 from openapi_client.models.clusterauthresponse_response import ClusterauthresponseResponse
 from openapi_client.models.clusterevent_list_response import ClustereventListResponse
 from openapi_client.models.clustereventsoutput_response import ClustereventsoutputResponse
+from openapi_client.models.clusteroperation_response import ClusteroperationResponse
+from openapi_client.models.commit_ledger_item_type import CommitLedgerItemType
+from openapi_client.models.commit_ledger_record_v2 import CommitLedgerRecordV2
+from openapi_client.models.complexity_level import ComplexityLevel
 from openapi_client.models.compute_node_type import ComputeNodeType
 from openapi_client.models.compute_stack import ComputeStack
 from openapi_client.models.compute_template import ComputeTemplate
@@ -200,6 +207,13 @@ from openapi_client.models.createcomputetemplateconfig_response import Createcom
 from openapi_client.models.createmachinepoolresponse_response import CreatemachinepoolresponseResponse
 from openapi_client.models.createmachineresponse_response import CreatemachineresponseResponse
 from openapi_client.models.createotpreturnapimodel_response import CreateotpreturnapimodelResponse
+from openapi_client.models.credit_grant_record_v2 import CreditGrantRecordV2
+from openapi_client.models.credit_ledger_item_type import CreditLedgerItemType
+from openapi_client.models.credit_ledger_record_v2 import CreditLedgerRecordV2
+from openapi_client.models.credit_record_commit_v2 import CreditRecordCommitV2
+from openapi_client.models.credit_record_credit_v2 import CreditRecordCreditV2
+from openapi_client.models.credit_type import CreditType
+from openapi_client.models.credits_v2 import CreditsV2
 from openapi_client.models.customer_alert_status import CustomerAlertStatus
 from openapi_client.models.dataplane_services import DataplaneServices
 from openapi_client.models.dataset import Dataset
@@ -399,6 +413,9 @@ from openapi_client.models.notification_channel_email_config import Notification
 from openapi_client.models.notification_channel_slack_config import NotificationChannelSlackConfig
 from openapi_client.models.notification_channel_webhook_config import NotificationChannelWebhookConfig
 from openapi_client.models.object_storage import ObjectStorage
+from openapi_client.models.operation_error import OperationError
+from openapi_client.models.operation_progress import OperationProgress
+from openapi_client.models.operation_result import OperationResult
 from openapi_client.models.operator_branch import OperatorBranch
 from openapi_client.models.operator_id import OperatorId
 from openapi_client.models.operator_metrics import OperatorMetrics