anyscale 0.26.44__py3-none-any.whl → 0.26.45__py3-none-any.whl

anyscale/client/README.md

@@ -337,6 +337,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**sso_login_test_api_v2_users_sso_login_test_get**](docs/DefaultApi.md#sso_login_test_api_v2_users_sso_login_test_get) | **GET** /api/v2/users/sso_login_test | Sso Login Test
 *DefaultApi* | [**start_session_api_v2_sessions_session_id_start_post**](docs/DefaultApi.md#start_session_api_v2_sessions_session_id_start_post) | **POST** /api/v2/sessions/{session_id}/start | Start Session
 *DefaultApi* | [**stop_session_api_v2_sessions_session_id_stop_post**](docs/DefaultApi.md#stop_session_api_v2_sessions_session_id_stop_post) | **POST** /api/v2/sessions/{session_id}/stop | Stop Session
+*DefaultApi* | [**stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post**](docs/DefaultApi.md#stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post) | **POST** /api/v2/organization_billing/stripe_success_webhook | Stripe Success Webhook
 *DefaultApi* | [**summarize_machine_pool_api_v2_machine_pools_summary_post**](docs/DefaultApi.md#summarize_machine_pool_api_v2_machine_pools_summary_post) | **POST** /api/v2/machine_pools/summary | Summarize Machine Pool
 *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
@@ -627,6 +628,7 @@ Class | Method | HTTP request | Description
  - [DescribeMachinePoolMachinesFilters](docs/DescribeMachinePoolMachinesFilters.md)
  - [DescribeMachinePoolMachinesRequest](docs/DescribeMachinePoolMachinesRequest.md)
  - [DescribeMachinePoolRequest](docs/DescribeMachinePoolRequest.md)
+ - [DescribeMachinePoolRequestsFilters](docs/DescribeMachinePoolRequestsFilters.md)
  - [DescribeMachinePoolRequestsRequest](docs/DescribeMachinePoolRequestsRequest.md)
  - [DescribeMachinePoolResponse](docs/DescribeMachinePoolResponse.md)
  - [DescribeMachinePoolWorkloadsFilters](docs/DescribeMachinePoolWorkloadsFilters.md)
@@ -799,6 +801,7 @@ Class | Method | HTTP request | Description
  - [PartitionInfo](docs/PartitionInfo.md)
  - [PauseSchedule](docs/PauseSchedule.md)
  - [PermissionLevel](docs/PermissionLevel.md)
+ - [PlanStatus](docs/PlanStatus.md)
  - [ProductionJob](docs/ProductionJob.md)
  - [ProductionJobConfig](docs/ProductionJobConfig.md)
  - [ProductionJobEvent](docs/ProductionJobEvent.md)

anyscale/client/openapi_client/__init__.py

@@ -278,6 +278,7 @@ from openapi_client.models.deletemachinepoolresponse_response import Deletemachi
 from openapi_client.models.describe_machine_pool_machines_filters import DescribeMachinePoolMachinesFilters
 from openapi_client.models.describe_machine_pool_machines_request import DescribeMachinePoolMachinesRequest
 from openapi_client.models.describe_machine_pool_request import DescribeMachinePoolRequest
+from openapi_client.models.describe_machine_pool_requests_filters import DescribeMachinePoolRequestsFilters
 from openapi_client.models.describe_machine_pool_requests_request import DescribeMachinePoolRequestsRequest
 from openapi_client.models.describe_machine_pool_response import DescribeMachinePoolResponse
 from openapi_client.models.describe_machine_pool_workloads_filters import DescribeMachinePoolWorkloadsFilters
@@ -450,6 +451,7 @@ from openapi_client.models.page_query import PageQuery
 from openapi_client.models.partition_info import PartitionInfo
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
+from openapi_client.models.plan_status import PlanStatus
 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

anyscale/client/openapi_client/api/default_api.py

@@ -17071,7 +17071,7 @@ class DefaultApi(object):
     def get_plan_status_api_v2_organization_billing_plan_status_get(self, **kwargs):  # noqa: E501
         """Get Plan Status  # noqa: E501
 
-        Returns the plan status for this user: True if the organization has ever been on a plan, False otherwise.  # noqa: E501
+        Returns the plan status for this user.  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
         >>> thread = api.get_plan_status_api_v2_organization_billing_plan_status_get(async_req=True)
@@ -17085,7 +17085,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: bool
+        :return: PlanStatus
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -17095,7 +17095,7 @@ class DefaultApi(object):
     def get_plan_status_api_v2_organization_billing_plan_status_get_with_http_info(self, **kwargs):  # noqa: E501
         """Get Plan Status  # noqa: E501
 
-        Returns the plan status for this user: True if the organization has ever been on a plan, False otherwise.  # noqa: E501
+        Returns the plan status for this user.  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
         >>> thread = api.get_plan_status_api_v2_organization_billing_plan_status_get_with_http_info(async_req=True)
@@ -17111,7 +17111,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: tuple(bool, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(PlanStatus, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -17165,7 +17165,7 @@ class DefaultApi(object):
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='bool',  # noqa: E501
+            response_type='PlanStatus',  # noqa: E501
             auth_settings=auth_settings,
             async_req=local_var_params.get('async_req'),
             _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
@@ -31704,6 +31704,107 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post(self, **kwargs):  # noqa: E501
+        """Stripe Success Webhook  # noqa: E501
+
+        Adds payment info for the first time or updates existing payment info.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: None
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post_with_http_info(**kwargs)  # noqa: E501
+
+    def stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post_with_http_info(self, **kwargs):  # noqa: E501
+        """Stripe Success Webhook  # noqa: E501
+
+        Adds payment info for the first time or updates existing payment info.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: None
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method stripe_success_webhook_api_v2_organization_billing_stripe_success_webhook_post" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/organization_billing/stripe_success_webhook', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type=None,  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def summarize_machine_pool_api_v2_machine_pools_summary_post(self, summarize_machine_pool_request, **kwargs):  # noqa: E501
         """Summarize Machine Pool  # noqa: E501
 

anyscale/client/openapi_client/models/__init__.py

@@ -264,6 +264,7 @@ from openapi_client.models.deletemachinepoolresponse_response import Deletemachi
 from openapi_client.models.describe_machine_pool_machines_filters import DescribeMachinePoolMachinesFilters
 from openapi_client.models.describe_machine_pool_machines_request import DescribeMachinePoolMachinesRequest
 from openapi_client.models.describe_machine_pool_request import DescribeMachinePoolRequest
+from openapi_client.models.describe_machine_pool_requests_filters import DescribeMachinePoolRequestsFilters
 from openapi_client.models.describe_machine_pool_requests_request import DescribeMachinePoolRequestsRequest
 from openapi_client.models.describe_machine_pool_response import DescribeMachinePoolResponse
 from openapi_client.models.describe_machine_pool_workloads_filters import DescribeMachinePoolWorkloadsFilters
@@ -436,6 +437,7 @@ from openapi_client.models.page_query import PageQuery
 from openapi_client.models.partition_info import PartitionInfo
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
+from openapi_client.models.plan_status import PlanStatus
 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

anyscale/client/openapi_client/models/describe_machine_pool_requests_filters.py

@@ -0,0 +1,150 @@
+# coding: utf-8
+
+"""
+    Managed Ray API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by: https://openapi-generator.tech
+"""
+
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+from openapi_client.configuration import Configuration
+
+
+class DescribeMachinePoolRequestsFilters(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    Attributes:
+      openapi_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    openapi_types = {
+        'partitions': 'list[str]',
+        'workload_name': 'TextQuery'
+    }
+
+    attribute_map = {
+        'partitions': 'partitions',
+        'workload_name': 'workload_name'
+    }
+
+    def __init__(self, partitions=None, workload_name=None, local_vars_configuration=None):  # noqa: E501
+        """DescribeMachinePoolRequestsFilters - a model defined in OpenAPI"""  # noqa: E501
+        if local_vars_configuration is None:
+            local_vars_configuration = Configuration()
+        self.local_vars_configuration = local_vars_configuration
+
+        self._partitions = None
+        self._workload_name = None
+        self.discriminator = None
+
+        if partitions is not None:
+            self.partitions = partitions
+        if workload_name is not None:
+            self.workload_name = workload_name
+
+    @property
+    def partitions(self):
+        """Gets the partitions of this DescribeMachinePoolRequestsFilters.  # noqa: E501
+
+        The partitions to filter by.  # noqa: E501
+
+        :return: The partitions of this DescribeMachinePoolRequestsFilters.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._partitions
+
+    @partitions.setter
+    def partitions(self, partitions):
+        """Sets the partitions of this DescribeMachinePoolRequestsFilters.
+
+        The partitions to filter by.  # noqa: E501
+
+        :param partitions: The partitions of this DescribeMachinePoolRequestsFilters.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._partitions = partitions
+
+    @property
+    def workload_name(self):
+        """Gets the workload_name of this DescribeMachinePoolRequestsFilters.  # noqa: E501
+
+        Filters requests by workload name. If this field is absent, no filtering is done.  # noqa: E501
+
+        :return: The workload_name of this DescribeMachinePoolRequestsFilters.  # noqa: E501
+        :rtype: TextQuery
+        """
+        return self._workload_name
+
+    @workload_name.setter
+    def workload_name(self, workload_name):
+        """Sets the workload_name of this DescribeMachinePoolRequestsFilters.
+
+        Filters requests by workload name. If this field is absent, no filtering is done.  # noqa: E501
+
+        :param workload_name: The workload_name of this DescribeMachinePoolRequestsFilters.  # noqa: E501
+        :type: TextQuery
+        """
+
+        self._workload_name = workload_name
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.openapi_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+
+        return result
+
+    def to_str(self):
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self):
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other):
+        """Returns true if both objects are equal"""
+        if not isinstance(other, DescribeMachinePoolRequestsFilters):
+            return False
+
+        return self.to_dict() == other.to_dict()
+
+    def __ne__(self, other):
+        """Returns true if both objects are not equal"""
+        if not isinstance(other, DescribeMachinePoolRequestsFilters):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/describe_machine_pool_requests_request.py

@@ -34,27 +34,27 @@ class DescribeMachinePoolRequestsRequest(object):
     """
     openapi_types = {
         'machine_pool_name': 'str',
-        'workload_name': 'TextQuery'
+        'filters': 'DescribeMachinePoolRequestsFilters'
     }
 
     attribute_map = {
         'machine_pool_name': 'machine_pool_name',
-        'workload_name': 'workload_name'
+        'filters': 'filters'
     }
 
-    def __init__(self, machine_pool_name=None, workload_name=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, machine_pool_name=None, filters=None, local_vars_configuration=None):  # noqa: E501
         """DescribeMachinePoolRequestsRequest - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
         self.local_vars_configuration = local_vars_configuration
 
         self._machine_pool_name = None
-        self._workload_name = None
+        self._filters = None
         self.discriminator = None
 
         self.machine_pool_name = machine_pool_name
-        if workload_name is not None:
-            self.workload_name = workload_name
+        if filters is not None:
+            self.filters = filters
 
     @property
     def machine_pool_name(self):
@@ -82,27 +82,27 @@ class DescribeMachinePoolRequestsRequest(object):
         self._machine_pool_name = machine_pool_name
 
     @property
-    def workload_name(self):
-        """Gets the workload_name of this DescribeMachinePoolRequestsRequest.  # noqa: E501
+    def filters(self):
+        """Gets the filters of this DescribeMachinePoolRequestsRequest.  # noqa: E501
 
-        Filters requests by workload name. If this field is absent, no filtering is done.  # noqa: E501
+        Filters to apply to the requests.  # noqa: E501
 
-        :return: The workload_name of this DescribeMachinePoolRequestsRequest.  # noqa: E501
-        :rtype: TextQuery
+        :return: The filters of this DescribeMachinePoolRequestsRequest.  # noqa: E501
+        :rtype: DescribeMachinePoolRequestsFilters
         """
-        return self._workload_name
+        return self._filters
 
-    @workload_name.setter
-    def workload_name(self, workload_name):
-        """Sets the workload_name of this DescribeMachinePoolRequestsRequest.
+    @filters.setter
+    def filters(self, filters):
+        """Sets the filters of this DescribeMachinePoolRequestsRequest.
 
-        Filters requests by workload name. If this field is absent, no filtering is done.  # noqa: E501
+        Filters to apply to the requests.  # noqa: E501
 
-        :param workload_name: The workload_name of this DescribeMachinePoolRequestsRequest.  # noqa: E501
-        :type: TextQuery
+        :param filters: The filters of this DescribeMachinePoolRequestsRequest.  # noqa: E501
+        :type: DescribeMachinePoolRequestsFilters
         """
 
-        self._workload_name = workload_name
+        self._filters = filters
 
     def to_dict(self):
         """Returns the model properties as a dict"""

anyscale/client/openapi_client/models/plan_status.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    Managed Ray API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by: https://openapi-generator.tech
+"""
+
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+from openapi_client.configuration import Configuration
+
+
+class PlanStatus(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    Attributes:
+      openapi_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    openapi_types = {
+        'has_ever_had_a_plan': 'bool'
+    }
+
+    attribute_map = {
+        'has_ever_had_a_plan': 'has_ever_had_a_plan'
+    }
+
+    def __init__(self, has_ever_had_a_plan=None, local_vars_configuration=None):  # noqa: E501
+        """PlanStatus - a model defined in OpenAPI"""  # noqa: E501
+        if local_vars_configuration is None:
+            local_vars_configuration = Configuration()
+        self.local_vars_configuration = local_vars_configuration
+
+        self._has_ever_had_a_plan = None
+        self.discriminator = None
+
+        self.has_ever_had_a_plan = has_ever_had_a_plan
+
+    @property
+    def has_ever_had_a_plan(self):
+        """Gets the has_ever_had_a_plan of this PlanStatus.  # noqa: E501
+
+        Whether the organization has ever had a plan.  # noqa: E501
+
+        :return: The has_ever_had_a_plan of this PlanStatus.  # noqa: E501
+        :rtype: bool
+        """
+        return self._has_ever_had_a_plan
+
+    @has_ever_had_a_plan.setter
+    def has_ever_had_a_plan(self, has_ever_had_a_plan):
+        """Sets the has_ever_had_a_plan of this PlanStatus.
+
+        Whether the organization has ever had a plan.  # noqa: E501
+
+        :param has_ever_had_a_plan: The has_ever_had_a_plan of this PlanStatus.  # noqa: E501
+        :type: bool
+        """
+        if self.local_vars_configuration.client_side_validation and has_ever_had_a_plan is None:  # noqa: E501
+            raise ValueError("Invalid value for `has_ever_had_a_plan`, must not be `None`")  # noqa: E501
+
+        self._has_ever_had_a_plan = has_ever_had_a_plan
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.openapi_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+
+        return result
+
+    def to_str(self):
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self):
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other):
+        """Returns true if both objects are equal"""
+        if not isinstance(other, PlanStatus):
+            return False
+
+        return self.to_dict() == other.to_dict()
+
+    def __ne__(self, other):
+        """Returns true if both objects are not equal"""
+        if not isinstance(other, PlanStatus):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/telemetry.py

@@ -366,55 +366,47 @@ def _patch_click() -> None:
 
         @functools.wraps(original_invoke)
         def instrumented_invoke(self, ctx, *args, **kwargs):
-            try:
-                # Sampling
-                if SAMPLE_RATE <= 0 or random.random() > SAMPLE_RATE:
-                    return original_invoke(self, ctx, *args, **kwargs)
-                # Only instrument leaf commands
-                if isinstance(self, click.Group):
-                    return original_invoke(self, ctx, *args, **kwargs)
+            if (
+                isinstance(self, click.Group)
+                or SAMPLE_RATE <= 0
+                or random.random() > SAMPLE_RATE
+            ):
+                return original_invoke(self, ctx, *args, **kwargs)
 
+            try:
                 trace_id = _setup_trace_context()
-                start = time.perf_counter()
-
-                # Store start time in context for interactive commands
-                ctx.telemetry_start_time = start
-
-                code, exc = 0, None
-                should_emit_telemetry = True
-
-                try:
-                    result = original_invoke(self, ctx, *args, **kwargs)
-                    return result
-                except Exception as e:
-                    code, exc = 1, e.__class__.__name__
-                    raise
-                finally:
-                    # Only emit telemetry once per command invocation
-                    if _skip_click_patch_var.get():
-                        should_emit_telemetry = False
-
-                    if should_emit_telemetry:
-                        try:
-                            # Use actual end time for non-interactive commands
-                            dur = (time.perf_counter() - start) * 1_000
-                            body = _create_payload(
-                                trace_id=trace_id,
-                                ctx=ctx,
-                                duration_ms=dur,
-                                exit_code=code,
-                                exception_type=exc,
-                                event_type="command",
-                            )
-                            _emit_telemetry(body)
-                            _skip_click_patch_var.set(True)
-                        except Exception:  # noqa: BLE001
-                            # Telemetry should never crash the CLI
-                            pass
             except Exception:  # noqa: BLE001
-                # If telemetry setup fails, just run the original command
                 return original_invoke(self, ctx, *args, **kwargs)
 
+            start = time.perf_counter()
+            ctx.telemetry_start_time = start
+            exit_code, exc_name = 0, None
+
+            try:
+                result = original_invoke(self, ctx, *args, **kwargs)
+                return result
+            except Exception as e:  # noqa: BLE001
+                exit_code, exc_name = 1, e.__class__.__name__
+                raise
+            finally:
+                # Only emit telemetry once per command invocation
+                if not _skip_click_patch_var.get():
+                    try:
+                        duration_ms = (time.perf_counter() - start) * 1000
+                        body = _create_payload(
+                            trace_id=trace_id,
+                            ctx=ctx,
+                            duration_ms=duration_ms,
+                            exit_code=exit_code,
+                            exception_type=exc_name,
+                            event_type="command",
+                        )
+                        _emit_telemetry(body)
+                        # Prevent Click patch from emitting again
+                        _skip_click_patch_var.set(True)
+                    except Exception:  # noqa: BLE001
+                        pass
+
         click.Command.invoke = instrumented_invoke
         click._anyscale_telemetry_patched = (  # noqa: SLF001  # type: ignore[attr-defined]
             True

anyscale/version.py

@@ -1 +1 @@
-__version__ = "0.26.44"
+__version__ = "0.26.45"

{anyscale-0.26.44.dist-info → anyscale-0.26.45.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.26.44
+Version: 0.26.45
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.26.44.dist-info → anyscale-0.26.45.dist-info}/RECORD

@@ -26,9 +26,9 @@ anyscale/project_utils.py,sha256=SBwkD5B10ku2kkmp6y-Cr5RL7xf52b9zELP35kfg2PE,176
 anyscale/scripts.py,sha256=dG_JgWuq8TrM4XLRCCXhccka41qn95qbEgt0GG_1x5M,5678
 anyscale/snapshot.py,sha256=UGJT5C1s_4xmQxjWODK5DFpGxHRBX5jOCdSCqXESH8E,1685
 anyscale/tables.py,sha256=TV4F2uLnwehvbkAfaP7iuLlT2wLIo6ORH2LVdRGXW5g,2840
-anyscale/telemetry.py,sha256=wklWABN8wtq8Cq3ptPFEeQQC9xjKP5ytxybCuWAvNI0,15229
+anyscale/telemetry.py,sha256=U90C2Vgx48z9PMTI6EbzHFbP3jWnDUutbIfMPBb8-SI,14711
 anyscale/util.py,sha256=0iSmKKsDFV0i2eRmVQ1W60Sgh2g-vPeBoBw3ra3m6eI,42782
-anyscale/version.py,sha256=kFdTi66Rz_z8qEhFVAixi2LQhtItihocbCWXIM2RhBw,24
+anyscale/version.py,sha256=54xC63YC2aqC0E8WKTGvo_4_EUotBQoaAqxATidvy4k,24
 anyscale/workspace_utils.py,sha256=OViE88CnIF5ruVxd3kazQ0Mf2BxqtMq6wx-XQ5A2cp8,1204
 anyscale/_private/anyscale_client/README.md,sha256=gk8obk7kqg6VWoUHcqDMwJULh35tYKEZFC0UF_dixGA,718
 anyscale/_private/anyscale_client/__init__.py,sha256=807Blx3RHQeS8BmKZcsOQQ4dYoKlCnpm6Bdsif2CrHg,337
@@ -106,7 +106,7 @@ anyscale/background/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 anyscale/background/job_runner.py,sha256=LTuv9JOahyv6C9i7DLQAONgQF6--FfYZEmJrKy-sUG8,2687
 anyscale/client/.gitignore,sha256=JZyvYEtT2DCSK9V5Joi6lQofhMik4PXiJRCWsg7SvqI,807
 anyscale/client/.openapi-generator-ignore,sha256=pu2PTide7pJtJ-DFLzDy0cTYQJRlrB-8RRH3zGLeUds,1040
-anyscale/client/README.md,sha256=SKPbFw_-rfKiUbNU9saO9ALTxcsDROcMHxu7n5rzJ4A,113443
+anyscale/client/README.md,sha256=AyYt2MgcDjh43zpzdCsNF4XP6r9ZLxidYBiw0DLbB48,113850
 anyscale/client/git_push.sh,sha256=EDCZOTTiLxbtPHmiU63qC99rGH67B7dhdPZdNUKivF0,1827
 anyscale/client/requirements.txt,sha256=dkVKYUStC5h_g_87SH7pRdhXCj7ySozAJMGAFEzGgFc,126
 anyscale/client/setup.cfg,sha256=l7bdKSIedeBhhoDtupsBwx1xPrlBf2yYeTH7a8kMga4,28
@@ -114,14 +114,14 @@ anyscale/client/setup.py,sha256=tSxqw1kAL1B9adnrnOarjnQfSbwGmnTr_kg8ZXhlm5A,1109
 anyscale/client/test-requirements.txt,sha256=sTjmDTj5W9fh1ZAeo8UT2EBdeGDBNttj_PHiPBXg1D4,111
 anyscale/client/tox.ini,sha256=M6L3UmvAdvU65LsoAF-Oi7oRjwZlCJZn8I7ofdXn5Ok,156
 anyscale/client/.openapi-generator/VERSION,sha256=J0RzX-4u4jfin1kviKtmncjUePyjHm2kyvmkobOrt_E,5
-anyscale/client/openapi_client/__init__.py,sha256=0jf4aDO7V9rqC52onEAovmFnQqyrLoblVABTXB0uG9I,51402
+anyscale/client/openapi_client/__init__.py,sha256=FSu4sK3Ahq7O-DCpSnYufPH7m2pxvmNlJyqxAAY58lc,51567
 anyscale/client/openapi_client/api_client.py,sha256=d8Un6j2Ny2vlS2qBXPVFj6_ql0k36DFahpWt_28TfCk,25563
 anyscale/client/openapi_client/configuration.py,sha256=Dd5XrlHwv-wxnf0C35PG_-HBQoY3Yaz6hKrmkZz-m0E,12363
 anyscale/client/openapi_client/exceptions.py,sha256=3egwsXQG2j_vARbqgBxUO1xSltAhpfiHTYVP7VXTvU0,3792
 anyscale/client/openapi_client/rest.py,sha256=Ehj37v7GHW6SXV067Hze5HE42ayKaGi6a6ZlkR7u3Lg,12501
 anyscale/client/openapi_client/api/__init__.py,sha256=i8u7BI2xX1GrXTL3hN0pKpYIlnT-D_uDxH2ElOfYG1I,141
-anyscale/client/openapi_client/api/default_api.py,sha256=cIVbFBzkYTExvbetShjCJ_B3yTm1Vsl6-wFTTe2QuaI,1816202
-anyscale/client/openapi_client/models/__init__.py,sha256=FM3XoQvOenXjSpH4qmavVIyjn9OHckwKl3abpzargK8,50912
+anyscale/client/openapi_client/api/default_api.py,sha256=Q2LJZhR5tZUkQXXTW1EcRGWd7YeN5JQ7eEK3aI5lpDw,1820642
+anyscale/client/openapi_client/models/__init__.py,sha256=lR19uidnrUeHLq9PwBMcAOCJBKcmFaa7dIruhe29RpQ,51077
 anyscale/client/openapi_client/models/access_config.py,sha256=b2mA0qtuTA5PFbp6C61Jc_T2zUMaojM1v32IhZo0MfY,3648
 anyscale/client/openapi_client/models/actor_status.py,sha256=6xyX_aIqURj2raBdY9DmBxsdDACFrqqYvElGiM6YG2E,2813
 anyscale/client/openapi_client/models/admin_create_user.py,sha256=9DPr8D0lKgoEZ3Z2kGsAd8L7ocFCiP6woOGLVs8SRb8,7251
@@ -372,7 +372,8 @@ anyscale/client/openapi_client/models/deletemachinepoolresponse_response.py,sha2
 anyscale/client/openapi_client/models/describe_machine_pool_machines_filters.py,sha256=o1wQrxdp8JBvjNsagCPlInhQ3KmsMMm5oW8h2FrvYXk,4795
 anyscale/client/openapi_client/models/describe_machine_pool_machines_request.py,sha256=2hDulfPYPu92gM_3-2z53sHwnKj3tDChtkA1UJdLbSk,4924
 anyscale/client/openapi_client/models/describe_machine_pool_request.py,sha256=7ECyUCAlwLFLdcIQrvyNiAOILBzNGUTN5_nK7EN84yg,3739
-anyscale/client/openapi_client/models/describe_machine_pool_requests_request.py,sha256=HfYseu-BjeZpbAb49AlFeH-be8mTSvYmOXJpHZm_XHc,5065
+anyscale/client/openapi_client/models/describe_machine_pool_requests_filters.py,sha256=W8B_a34oWMaZcC2K13XDD8qdnlQICqLkPNu6_RQtu0c,4750
+anyscale/client/openapi_client/models/describe_machine_pool_requests_request.py,sha256=z39iFIWq9tMz7A0sB9rw0oEwl3VTLD2K3p7R9YUFi9E,4924
 anyscale/client/openapi_client/models/describe_machine_pool_response.py,sha256=dtcGBS48Vz07CbIVPfeU4XBS9aQMpt9ADEhG_KCNl48,3887
 anyscale/client/openapi_client/models/describe_machine_pool_workloads_filters.py,sha256=5AFLJDmIdhWn6khTjWhiQSa4JLUCNEsF4DD10_UFAhQ,4876
 anyscale/client/openapi_client/models/describe_machine_pool_workloads_request.py,sha256=J4T9L1sEGPBkufc74W6z3QvZyPQ3_5QDXKlrOjkc6YA,4941
@@ -544,6 +545,7 @@ anyscale/client/openapi_client/models/page_query.py,sha256=dw5HgH5gVYi2OcLW3Kg0Z
 anyscale/client/openapi_client/models/partition_info.py,sha256=XaAsTffyjTxdCZN3wQtLxUch5oFs3XwqOtaeHjyEt14,4916
 anyscale/client/openapi_client/models/pause_schedule.py,sha256=ru7_OzH7oDw5y2h1o5QNfKW8TXbkF1h-NDwccbUDRd0,3661
 anyscale/client/openapi_client/models/permission_level.py,sha256=fGnBgKrcZkGLYl7RpQQm0_gaGzkTehnJjOnIX8G4jAw,2868
+anyscale/client/openapi_client/models/plan_status.py,sha256=4Gl0dZDiPeP4ar87viHvggfjZwQeqWuMZ78i8bRWV9k,3855
 anyscale/client/openapi_client/models/production_job.py,sha256=fVDrDRonEiH6OVfKl3Xwc1RPaPs0S4EKLqBEzvRIraY,13672
 anyscale/client/openapi_client/models/production_job_config.py,sha256=cW2TxSTUw-dDm3kGWa_tNet7dmJM3VVPJ75pWumosiA,12527
 anyscale/client/openapi_client/models/production_job_event.py,sha256=VljH0Qm09GbEHsb9je3LDcENu9MwD3pRISRJipwrV_s,11432
@@ -1098,10 +1100,10 @@ anyscale/workspace/__init__.py,sha256=Innbm5ZhCyADEVBiYSo_vbpKwUNcMzVSAfxIGKOYe6
 anyscale/workspace/commands.py,sha256=b1sqNseoPj-1VXznqQOLe0V_a663bOTvJX-TaOMJa1Y,14590
 anyscale/workspace/models.py,sha256=HBvM9ybOdJjqQeViQ30C36gdKT_AwH_JHPoL-DTkESo,9841
 anyscale/workspace/_private/workspace_sdk.py,sha256=2CMeYfJt0UtIFCocDn1ukw1iI5esKHdopLe6duEs-qE,27599
-anyscale-0.26.44.dist-info/LICENSE,sha256=UOPu974Wzsna6frFv1mu4VrZgNdZT7lbcNPzo5ue3qs,3494
-anyscale-0.26.44.dist-info/METADATA,sha256=ThYc65jtlZTAKd8MRW7YeqORkhHCxW8kT8F9zu51yyg,3032
-anyscale-0.26.44.dist-info/NOTICE,sha256=gHqDhSnUYlRXX-mDOL5FtE7774oiKyV_HO80qM3r9Xo,196
-anyscale-0.26.44.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-anyscale-0.26.44.dist-info/entry_points.txt,sha256=NqO18sCZn6zG6J0S38itjcN00s7aE3C3v3k5lMAfCLk,51
-anyscale-0.26.44.dist-info/top_level.txt,sha256=g3NVNS8Oh0NZwbFFgeX696C5MZZkS5dqV2NqcsbDRJE,9
-anyscale-0.26.44.dist-info/RECORD,,
+anyscale-0.26.45.dist-info/LICENSE,sha256=UOPu974Wzsna6frFv1mu4VrZgNdZT7lbcNPzo5ue3qs,3494
+anyscale-0.26.45.dist-info/METADATA,sha256=7cVr8LcD7lkncSk24pvdzWEYUmXYoPRPPDfp-V27xtg,3032
+anyscale-0.26.45.dist-info/NOTICE,sha256=gHqDhSnUYlRXX-mDOL5FtE7774oiKyV_HO80qM3r9Xo,196
+anyscale-0.26.45.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+anyscale-0.26.45.dist-info/entry_points.txt,sha256=NqO18sCZn6zG6J0S38itjcN00s7aE3C3v3k5lMAfCLk,51
+anyscale-0.26.45.dist-info/top_level.txt,sha256=g3NVNS8Oh0NZwbFFgeX696C5MZZkS5dqV2NqcsbDRJE,9
+anyscale-0.26.45.dist-info/RECORD,,