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

anyscale/_private/workload/workload_sdk.py

@@ -118,6 +118,7 @@ class WorkloadSDK(BaseSDK):
         autopopulate_in_workspace: bool = True,
         additional_py_modules: Optional[List[str]] = None,
         py_executable_override: Optional[str] = None,
+        has_multiple_cloud_deployments: bool = False,
     ) -> List[Dict[str, Any]]:
         """Returns modified runtime_envs with all local dirs converted to remote URIs.
 
@@ -137,6 +138,11 @@ class WorkloadSDK(BaseSDK):
             if target in local_path_to_uri:
                 return local_path_to_uri[target]
 
+            if has_multiple_cloud_deployments:
+                raise ValueError(
+                    "Local directory uploads are not supported for compute configurations with multiple cloud deployments."
+                )
+
             self.logger.info(f"Uploading local dir '{target}' to cloud storage.")
             assert cloud_id is not None
             uri = self._client.upload_local_dir_to_cloud_storage(

anyscale/client/README.md

@@ -147,6 +147,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**describe_machine_pool_api_v2_machine_pools_describe_post**](docs/DefaultApi.md#describe_machine_pool_api_v2_machine_pools_describe_post) | **POST** /api/v2/machine_pools/describe | Describe Machine Pool
 *DefaultApi* | [**describe_machine_pool_machines_api_v2_machine_pools_describe_machines_post**](docs/DefaultApi.md#describe_machine_pool_machines_api_v2_machine_pools_describe_machines_post) | **POST** /api/v2/machine_pools/describe_machines | Describe Machine Pool Machines
 *DefaultApi* | [**describe_machine_pool_requests_api_v2_machine_pools_describe_requests_post**](docs/DefaultApi.md#describe_machine_pool_requests_api_v2_machine_pools_describe_requests_post) | **POST** /api/v2/machine_pools/describe_requests | Describe Machine Pool Requests
+*DefaultApi* | [**describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post**](docs/DefaultApi.md#describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post) | **POST** /api/v2/machine_pools/describe_workloads | Describe Machine Pool Workloads
 *DefaultApi* | [**describe_system_workload_api_v2_system_workload_cloud_id_describe_post**](docs/DefaultApi.md#describe_system_workload_api_v2_system_workload_cloud_id_describe_post) | **POST** /api/v2/system_workload/{cloud_id}/describe | Describe System Workload
 *DefaultApi* | [**detach_machine_pool_from_cloud_api_v2_machine_pools_detach_post**](docs/DefaultApi.md#detach_machine_pool_from_cloud_api_v2_machine_pools_detach_post) | **POST** /api/v2/machine_pools/detach | Detach Machine Pool From Cloud
 *DefaultApi* | [**download_aggregated_instance_usage_csv_api_v2_aggregated_instance_usage_download_csv_get**](docs/DefaultApi.md#download_aggregated_instance_usage_csv_api_v2_aggregated_instance_usage_download_csv_get) | **GET** /api/v2/aggregated_instance_usage/download_csv | Download Aggregated Instance Usage Csv
@@ -336,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
@@ -626,8 +628,11 @@ 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)
+ - [DescribeMachinePoolWorkloadsRequest](docs/DescribeMachinePoolWorkloadsRequest.md)
  - [DescribeSystemWorkloadResponse](docs/DescribeSystemWorkloadResponse.md)
  - [DescribemachinepoolresponseResponse](docs/DescribemachinepoolresponseResponse.md)
  - [DescribesystemworkloadresponseResponse](docs/DescribesystemworkloadresponseResponse.md)
@@ -796,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)
@@ -963,7 +969,10 @@ Class | Method | HTTP request | Description
  - [WandBRunDetails](docs/WandBRunDetails.md)
  - [WorkerNodeType](docs/WorkerNodeType.md)
  - [WorkloadInfo](docs/WorkloadInfo.md)
+ - [WorkloadMachineInfo](docs/WorkloadMachineInfo.md)
+ - [WorkloadStateInfo](docs/WorkloadStateInfo.md)
  - [WorkloadType](docs/WorkloadType.md)
+ - [WorkloadstateinfoListResponse](docs/WorkloadstateinfoListResponse.md)
  - [WorkspaceDataplaneArtifact](docs/WorkspaceDataplaneArtifact.md)
  - [WorkspaceDataplaneArtifacts](docs/WorkspaceDataplaneArtifacts.md)
  - [WorkspaceDataplaneProxiedArtifacts](docs/WorkspaceDataplaneProxiedArtifacts.md)

anyscale/client/openapi_client/__init__.py

@@ -278,8 +278,11 @@ 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
+from openapi_client.models.describe_machine_pool_workloads_request import DescribeMachinePoolWorkloadsRequest
 from openapi_client.models.describe_system_workload_response import DescribeSystemWorkloadResponse
 from openapi_client.models.describemachinepoolresponse_response import DescribemachinepoolresponseResponse
 from openapi_client.models.describesystemworkloadresponse_response import DescribesystemworkloadresponseResponse
@@ -448,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
@@ -615,7 +619,10 @@ from openapi_client.models.waitliststatusresponse_response import Waitliststatus
 from openapi_client.models.wand_b_run_details import WandBRunDetails
 from openapi_client.models.worker_node_type import WorkerNodeType
 from openapi_client.models.workload_info import WorkloadInfo
+from openapi_client.models.workload_machine_info import WorkloadMachineInfo
+from openapi_client.models.workload_state_info import WorkloadStateInfo
 from openapi_client.models.workload_type import WorkloadType
+from openapi_client.models.workloadstateinfo_list_response import WorkloadstateinfoListResponse
 from openapi_client.models.workspace_dataplane_artifact import WorkspaceDataplaneArtifact
 from openapi_client.models.workspace_dataplane_artifacts import WorkspaceDataplaneArtifacts
 from openapi_client.models.workspace_dataplane_proxied_artifacts import WorkspaceDataplaneProxiedArtifacts

anyscale/client/openapi_client/api/default_api.py

@@ -8419,6 +8419,138 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post(self, describe_machine_pool_workloads_request, **kwargs):  # noqa: E501
+        """Describe Machine Pool Workloads  # noqa: E501
+
+        Describe the currently allocated workloads in a machine pool.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post(describe_machine_pool_workloads_request, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param DescribeMachinePoolWorkloadsRequest describe_machine_pool_workloads_request: (required)
+        :param str paging_token:
+        :param int count:
+        :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: WorkloadstateinfoListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post_with_http_info(describe_machine_pool_workloads_request, **kwargs)  # noqa: E501
+
+    def describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post_with_http_info(self, describe_machine_pool_workloads_request, **kwargs):  # noqa: E501
+        """Describe Machine Pool Workloads  # noqa: E501
+
+        Describe the currently allocated workloads in a machine pool.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post_with_http_info(describe_machine_pool_workloads_request, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param DescribeMachinePoolWorkloadsRequest describe_machine_pool_workloads_request: (required)
+        :param str paging_token:
+        :param int count:
+        :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: tuple(WorkloadstateinfoListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'describe_machine_pool_workloads_request',
+            'paging_token',
+            'count'
+        ]
+        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 describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'describe_machine_pool_workloads_request' is set
+        if self.api_client.client_side_validation and ('describe_machine_pool_workloads_request' not in local_var_params or  # noqa: E501
+                                                        local_var_params['describe_machine_pool_workloads_request'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `describe_machine_pool_workloads_request` when calling `describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post`")  # noqa: E501
+
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post`, must be a value less than or equal to `1000`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post`, must be a value greater than or equal to `0`")  # noqa: E501
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'paging_token' in local_var_params and local_var_params['paging_token'] is not None:  # noqa: E501
+            query_params.append(('paging_token', local_var_params['paging_token']))  # noqa: E501
+        if 'count' in local_var_params and local_var_params['count'] is not None:  # noqa: E501
+            query_params.append(('count', local_var_params['count']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'describe_machine_pool_workloads_request' in local_var_params:
+            body_params = local_var_params['describe_machine_pool_workloads_request']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/machine_pools/describe_workloads', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='WorkloadstateinfoListResponse',  # 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 describe_system_workload_api_v2_system_workload_cloud_id_describe_post(self, cloud_id, workload_name, **kwargs):  # noqa: E501
         """Describe System Workload  # noqa: E501
 
@@ -16939,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)
@@ -16953,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.
         """
@@ -16963,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)
@@ -16979,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.
         """
@@ -17033,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
@@ -31572,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,8 +264,11 @@ 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
+from openapi_client.models.describe_machine_pool_workloads_request import DescribeMachinePoolWorkloadsRequest
 from openapi_client.models.describe_system_workload_response import DescribeSystemWorkloadResponse
 from openapi_client.models.describemachinepoolresponse_response import DescribemachinepoolresponseResponse
 from openapi_client.models.describesystemworkloadresponse_response import DescribesystemworkloadresponseResponse
@@ -434,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
@@ -601,7 +605,10 @@ from openapi_client.models.waitliststatusresponse_response import Waitliststatus
 from openapi_client.models.wand_b_run_details import WandBRunDetails
 from openapi_client.models.worker_node_type import WorkerNodeType
 from openapi_client.models.workload_info import WorkloadInfo
+from openapi_client.models.workload_machine_info import WorkloadMachineInfo
+from openapi_client.models.workload_state_info import WorkloadStateInfo
 from openapi_client.models.workload_type import WorkloadType
+from openapi_client.models.workloadstateinfo_list_response import WorkloadstateinfoListResponse
 from openapi_client.models.workspace_dataplane_artifact import WorkspaceDataplaneArtifact
 from openapi_client.models.workspace_dataplane_artifacts import WorkspaceDataplaneArtifacts
 from openapi_client.models.workspace_dataplane_proxied_artifacts import WorkspaceDataplaneProxiedArtifacts