anyscale 0.26.48__py3-none-any.whl → 0.26.50__py3-none-any.whl

anyscale/_private/anyscale_client/README.md

@@ -7,6 +7,17 @@ The purpose of centralizing this logic is to:
 - Keep all external dependencies in one place.
 - Enable writing comprehensive unit tests for upstream components (without using mocks!) using the `FakeAnyscaleClient`.
 
+## Installation
+
+To install the anyscale client package in development mode:
+
+```bash
+cd frontend/cli
+pip install -e .
+```
+
+This will install the `anyscale` package in editable mode, allowing you to make changes to the code and have them immediately available.
+
 ## Testing
 
 The `AnyscaleClient` is tested using a fake version of the internal and external OpenAPI clients.
@@ -14,3 +25,107 @@ The `AnyscaleClient` is tested using a fake version of the internal and external
 Upstream components should use the `FakeAnyscaleClient` to write their tests.
 This client should mirror the behavior of the real `AnyscaleClient` as closely as possible (avoid making methods and functionality
 complete "dummies").
+
+### Running Tests
+
+1. **Unit Tests**: Run the test suite to verify client functionality
+   ```bash
+   cd frontend/cli
+   python -m pytest tests/unit/test_anyscale_client.py
+   ```
+
+2. **Integration Tests**: Test the client against the actual Anyscale API
+   ```bash
+   # Set up your Anyscale credentials
+   export ANYSCALE_TOKEN="your_token_here"
+
+   # Run integration tests
+   python -m pytest tests/test_integrations.py
+   ```
+
+3. **All Client-Related Tests**: Run all tests that use the anyscale client
+   ```bash
+   cd frontend/cli
+   python -m pytest tests/ -k "anyscale_client" -v
+   ```
+
+### Testing Job Submission
+
+To test job submission functionality:
+
+1. **Create a test job configuration file** (`job.yaml`):
+   ```yaml
+   name: test-job
+   compute_config: default:1
+   working_dir: /path/to/working/directory
+   requirements:
+   - numpy==1.24.0
+   - pandas==2.0.0
+   entrypoint: python your_script.py
+   max_retries: 0
+   ```
+
+2. **Submit a test job**:
+   ```bash
+   anyscale job submit -f job.yaml
+   ```
+
+3. **Monitor job status**:
+   ```bash
+   anyscale job status <job_id>
+   ```
+
+4. **View job logs**:
+   ```bash
+   anyscale job logs <job_id>
+   ```
+
+### Example Test Job
+
+Here's an example job configuration for testing:
+
+```yaml
+name: generate-doggos-embeddings
+compute_config: doggos-azure:1
+working_dir: azure://cloud-dev-blob/org_7c1Kalm9WcX2bNIjW53GUT/cld_wgmfc248s6t7513awyubirlwu9/runtime_env_packages/pkg_b60e2d10615fb9845a9bad7d9307547a.zip
+requirements:
+- matplotlib==3.10.0
+- torch==2.7.1
+- transformers==4.52.3
+- scikit-learn==1.6.0
+- mlflow==2.19.0
+- ipywidgets==8.1.3
+entrypoint: python doggos/embed.py
+max_retries: 0
+```
+
+### Testing with Fake Client
+
+For unit testing components that depend on the Anyscale client:
+
+```python
+from anyscale._private.anyscale_client import FakeAnyscaleClient
+
+# Create a fake client for testing
+fake_client = FakeAnyscaleClient()
+
+# Use the fake client in your tests
+# The fake client should behave like the real client
+result = fake_client.submit_job(job_config)
+assert result.job_id is not None
+```
+
+### Debugging
+
+- Use `--verbose` flag for detailed output: `anyscale job submit -f job.yaml --verbose`
+- Check job status in the UI: The CLI will provide a URL to view the job in the Anyscale console
+- Use `--wait` flag to wait for job completion and stream logs: `anyscale job submit -f job.yaml --wait`
+
+### Common Issues
+
+1. **Authentication**: Ensure your Anyscale token is properly set
+2. **Network**: Check your internet connection and firewall settings
+3. **Dependencies**: Verify all required packages are installed
+4. **Job Configuration**: Ensure your YAML file is properly formatted
+
+For more detailed testing scenarios, refer to the test files in the `tests/` directory.

anyscale/_private/anyscale_client/anyscale_client.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
 import contextlib
+from datetime import datetime
 from functools import wraps
 import io
 import json
@@ -1292,11 +1293,16 @@ class AnyscaleClient(AnyscaleClientInterface):
             else:
                 # Default to HTTP PUT.
                 internal_logger.debug(f"Uploading file '{file_name}' to cloud storage.")
-                headers = (
-                    {"x-ms-blob-type": "BlockBlob"}
-                    if info.file_uri.startswith("azure")
-                    else None
-                )
+                headers = None
+                if info.file_uri.startswith("azure"):
+                    headers = {
+                        "x-ms-blob-type": "BlockBlob",
+                        "x-ms-version": "2025-07-05",
+                        "x-ms-date": datetime.utcnow().strftime(
+                            "%a, %d %b %Y %H:%M:%S GMT"
+                        ),
+                        "x-ms-blob-content-type": "application/zip",
+                    }
                 requests.put(
                     info.url, data=zip_file_bytes, headers=headers
                 ).raise_for_status()

anyscale/_private/docgen/__main__.py

@@ -468,7 +468,13 @@ ALL_MODULES = [
             GCPConfig,
             KubernetesConfig,
         ],
-        cli_command_group_prefix={cloud_commands.cloud_config_update: "config"},
+        cli_command_group_prefix={
+            cloud_commands.cloud_deployment_create: "deployment",
+            cloud_commands.cloud_deployment_get: "deployment",
+            cloud_commands.cloud_deployment_update: "deployment",
+            cloud_commands.cloud_deployment_delete: "deployment",
+            cloud_commands.cloud_config_update: "config",
+        },
         legacy_sdk_commands={
             # limited support, no replacement yet
             "get_cloud": None,

anyscale/_private/workload/workload_sdk.py

@@ -143,8 +143,8 @@ class WorkloadSDK(BaseSDK):
                 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.warning(
+                    "For compute configurations with multiple cloud deployments, local directories will only be uploaded to the object storage of the primary cloud deployment."
                 )
 
             self.logger.info(f"Uploading local dir '{target}' to cloud storage.")

anyscale/client/README.md

@@ -308,6 +308,8 @@ 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* | [**query_api_v2_metrics_query_get**](docs/DefaultApi.md#query_api_v2_metrics_query_get) | **GET** /api/v2/metrics/query | Query
+*DefaultApi* | [**query_range_api_v2_metrics_query_range_get**](docs/DefaultApi.md#query_range_api_v2_metrics_query_range_get) | **GET** /api/v2/metrics/query_range | Query Range
 *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
@@ -484,7 +486,6 @@ Class | Method | HTTP request | Description
  - [ClouddatabucketpresigneduploadinfoResponse](docs/ClouddatabucketpresigneduploadinfoResponse.md)
  - [ClouddatabucketpresignedurlresponseResponse](docs/ClouddatabucketpresignedurlresponseResponse.md)
  - [ClouddeploymentListResponse](docs/ClouddeploymentListResponse.md)
- - [ClouddeploymentResponse](docs/ClouddeploymentResponse.md)
  - [ClouddeploymentconfigResponse](docs/ClouddeploymentconfigResponse.md)
  - [CloudoverviewdashboardResponse](docs/CloudoverviewdashboardResponse.md)
  - [CloudregionandzonesResponse](docs/CloudregionandzonesResponse.md)
@@ -583,6 +584,7 @@ Class | Method | HTTP request | Description
  - [DatasetuploadResponse](docs/DatasetuploadResponse.md)
  - [DecoratedApplicationTemplate](docs/DecoratedApplicationTemplate.md)
  - [DecoratedBuild](docs/DecoratedBuild.md)
+ - [DecoratedCloudDeployment](docs/DecoratedCloudDeployment.md)
  - [DecoratedComputeTemplate](docs/DecoratedComputeTemplate.md)
  - [DecoratedComputeTemplateConfig](docs/DecoratedComputeTemplateConfig.md)
  - [DecoratedInteractiveSession](docs/DecoratedInteractiveSession.md)
@@ -603,6 +605,7 @@ Class | Method | HTTP request | Description
  - [DecoratedapplicationtemplateResponse](docs/DecoratedapplicationtemplateResponse.md)
  - [DecoratedbuildListResponse](docs/DecoratedbuildListResponse.md)
  - [DecoratedbuildResponse](docs/DecoratedbuildResponse.md)
+ - [DecoratedclouddeploymentResponse](docs/DecoratedclouddeploymentResponse.md)
  - [DecoratedcomputetemplateListResponse](docs/DecoratedcomputetemplateListResponse.md)
  - [DecoratedcomputetemplateResponse](docs/DecoratedcomputetemplateResponse.md)
  - [DecoratedinteractivesessionListResponse](docs/DecoratedinteractivesessionListResponse.md)
@@ -748,6 +751,8 @@ Class | Method | HTTP request | Description
  - [MachinepoolsearchresultListResponse](docs/MachinepoolsearchresultListResponse.md)
  - [MachinestateinfoListResponse](docs/MachinestateinfoListResponse.md)
  - [Metric](docs/Metric.md)
+ - [MetricsQueryResponse](docs/MetricsQueryResponse.md)
+ - [MetricsqueryresponseResponse](docs/MetricsqueryresponseResponse.md)
  - [MetronomeCustomerInfoModel](docs/MetronomeCustomerInfoModel.md)
  - [MetronomeDashboardType](docs/MetronomeDashboardType.md)
  - [MetronomecustomerinfomodelListResponse](docs/MetronomecustomerinfomodelListResponse.md)

anyscale/client/openapi_client/__init__.py

@@ -130,7 +130,6 @@ from openapi_client.models.cloudcollaborator_list_response import Cloudcollabora
 from openapi_client.models.clouddatabucketpresigneduploadinfo_response import ClouddatabucketpresigneduploadinfoResponse
 from openapi_client.models.clouddatabucketpresignedurlresponse_response import ClouddatabucketpresignedurlresponseResponse
 from openapi_client.models.clouddeployment_list_response import ClouddeploymentListResponse
-from openapi_client.models.clouddeployment_response import ClouddeploymentResponse
 from openapi_client.models.clouddeploymentconfig_response import ClouddeploymentconfigResponse
 from openapi_client.models.cloudoverviewdashboard_response import CloudoverviewdashboardResponse
 from openapi_client.models.cloudregionandzones_response import CloudregionandzonesResponse
@@ -229,6 +228,7 @@ from openapi_client.models.dataset_upload import DatasetUpload
 from openapi_client.models.datasetupload_response import DatasetuploadResponse
 from openapi_client.models.decorated_application_template import DecoratedApplicationTemplate
 from openapi_client.models.decorated_build import DecoratedBuild
+from openapi_client.models.decorated_cloud_deployment import DecoratedCloudDeployment
 from openapi_client.models.decorated_compute_template import DecoratedComputeTemplate
 from openapi_client.models.decorated_compute_template_config import DecoratedComputeTemplateConfig
 from openapi_client.models.decorated_interactive_session import DecoratedInteractiveSession
@@ -249,6 +249,7 @@ from openapi_client.models.decoratedapplicationtemplate_list_response import Dec
 from openapi_client.models.decoratedapplicationtemplate_response import DecoratedapplicationtemplateResponse
 from openapi_client.models.decoratedbuild_list_response import DecoratedbuildListResponse
 from openapi_client.models.decoratedbuild_response import DecoratedbuildResponse
+from openapi_client.models.decoratedclouddeployment_response import DecoratedclouddeploymentResponse
 from openapi_client.models.decoratedcomputetemplate_list_response import DecoratedcomputetemplateListResponse
 from openapi_client.models.decoratedcomputetemplate_response import DecoratedcomputetemplateResponse
 from openapi_client.models.decoratedinteractivesession_list_response import DecoratedinteractivesessionListResponse
@@ -394,6 +395,8 @@ from openapi_client.models.machine_state_info import MachineStateInfo
 from openapi_client.models.machinepoolsearchresult_list_response import MachinepoolsearchresultListResponse
 from openapi_client.models.machinestateinfo_list_response import MachinestateinfoListResponse
 from openapi_client.models.metric import Metric
+from openapi_client.models.metrics_query_response import MetricsQueryResponse
+from openapi_client.models.metricsqueryresponse_response import MetricsqueryresponseResponse
 from openapi_client.models.metronome_customer_info_model import MetronomeCustomerInfoModel
 from openapi_client.models.metronome_dashboard_type import MetronomeDashboardType
 from openapi_client.models.metronomecustomerinfomodel_list_response import MetronomecustomerinfomodelListResponse

anyscale/client/openapi_client/api/default_api.py

@@ -11748,7 +11748,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: ClouddeploymentResponse
+        :return: DecoratedclouddeploymentResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -11775,7 +11775,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(ClouddeploymentResponse, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(DecoratedclouddeploymentResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -11843,7 +11843,7 @@ class DefaultApi(object):
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='ClouddeploymentResponse',  # noqa: E501
+            response_type='DecoratedclouddeploymentResponse',  # 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
@@ -21707,6 +21707,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cloud_id: (required)
+        :param bool via_edge_proxy:
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -21732,6 +21733,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cloud_id: (required)
+        :param bool via_edge_proxy:
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -21749,7 +21751,8 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'cloud_id'
+            'cloud_id',
+            'via_edge_proxy'
         ]
         all_params.extend(
             [
@@ -21780,6 +21783,8 @@ class DefaultApi(object):
             path_params['cloud_id'] = local_var_params['cloud_id']  # noqa: E501
 
         query_params = []
+        if 'via_edge_proxy' in local_var_params and local_var_params['via_edge_proxy'] is not None:  # noqa: E501
+            query_params.append(('via_edge_proxy', local_var_params['via_edge_proxy']))  # noqa: E501
 
         header_params = {}
 
@@ -28214,6 +28219,302 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def query_api_v2_metrics_query_get(self, promql_query, **kwargs):  # noqa: E501
+        """Query  # noqa: E501
+
+        Query metrics using PromQL query for various resource types  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.query_api_v2_metrics_query_get(promql_query, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str promql_query: PromQL query string to execute. This should be a valid Prometheus Query Language expression. (required)
+        :param str cluster_id: The cluster id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str workspace_id: The workspace id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str ha_job_id: The production job id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str service_id: The service id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param float time: Unix timestamp for the query time. If not provided, uses current time.
+        :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: MetricsqueryresponseResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.query_api_v2_metrics_query_get_with_http_info(promql_query, **kwargs)  # noqa: E501
+
+    def query_api_v2_metrics_query_get_with_http_info(self, promql_query, **kwargs):  # noqa: E501
+        """Query  # noqa: E501
+
+        Query metrics using PromQL query for various resource types  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.query_api_v2_metrics_query_get_with_http_info(promql_query, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str promql_query: PromQL query string to execute. This should be a valid Prometheus Query Language expression. (required)
+        :param str cluster_id: The cluster id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str workspace_id: The workspace id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str ha_job_id: The production job id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str service_id: The service id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param float time: Unix timestamp for the query time. If not provided, uses current time.
+        :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(MetricsqueryresponseResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'promql_query',
+            'cluster_id',
+            'workspace_id',
+            'ha_job_id',
+            'service_id',
+            'time'
+        ]
+        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 query_api_v2_metrics_query_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'promql_query' is set
+        if self.api_client.client_side_validation and ('promql_query' not in local_var_params or  # noqa: E501
+                                                        local_var_params['promql_query'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `promql_query` when calling `query_api_v2_metrics_query_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'promql_query' in local_var_params and local_var_params['promql_query'] is not None:  # noqa: E501
+            query_params.append(('promql_query', local_var_params['promql_query']))  # noqa: E501
+        if 'cluster_id' in local_var_params and local_var_params['cluster_id'] is not None:  # noqa: E501
+            query_params.append(('cluster_id', local_var_params['cluster_id']))  # noqa: E501
+        if 'workspace_id' in local_var_params and local_var_params['workspace_id'] is not None:  # noqa: E501
+            query_params.append(('workspace_id', local_var_params['workspace_id']))  # noqa: E501
+        if 'ha_job_id' in local_var_params and local_var_params['ha_job_id'] is not None:  # noqa: E501
+            query_params.append(('ha_job_id', local_var_params['ha_job_id']))  # noqa: E501
+        if 'service_id' in local_var_params and local_var_params['service_id'] is not None:  # noqa: E501
+            query_params.append(('service_id', local_var_params['service_id']))  # noqa: E501
+        if 'time' in local_var_params and local_var_params['time'] is not None:  # noqa: E501
+            query_params.append(('time', local_var_params['time']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/metrics/query', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='MetricsqueryresponseResponse',  # 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 query_range_api_v2_metrics_query_range_get(self, promql_query, start, end, **kwargs):  # noqa: E501
+        """Query Range  # noqa: E501
+
+        Query metrics using PromQL range query for various resource types  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.query_range_api_v2_metrics_query_range_get(promql_query, start, end, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str promql_query: PromQL query string to execute. This should be a valid Prometheus Query Language expression. (required)
+        :param float start: Start timestamp (Unix timestamp) for the range query. (required)
+        :param float end: End timestamp (Unix timestamp) for the range query. (required)
+        :param str cluster_id: The cluster id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str workspace_id: The workspace id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str ha_job_id: The production job id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str service_id: The service id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str step: Query resolution step width in seconds (e.g., '15s', '1m', '5m').
+        :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: MetricsqueryresponseResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.query_range_api_v2_metrics_query_range_get_with_http_info(promql_query, start, end, **kwargs)  # noqa: E501
+
+    def query_range_api_v2_metrics_query_range_get_with_http_info(self, promql_query, start, end, **kwargs):  # noqa: E501
+        """Query Range  # noqa: E501
+
+        Query metrics using PromQL range query for various resource types  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.query_range_api_v2_metrics_query_range_get_with_http_info(promql_query, start, end, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str promql_query: PromQL query string to execute. This should be a valid Prometheus Query Language expression. (required)
+        :param float start: Start timestamp (Unix timestamp) for the range query. (required)
+        :param float end: End timestamp (Unix timestamp) for the range query. (required)
+        :param str cluster_id: The cluster id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str workspace_id: The workspace id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str ha_job_id: The production job id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str service_id: The service id used to query metrics. Exactly one of cluster_id, workspace_id, ha_job_id, or service_id must be set.
+        :param str step: Query resolution step width in seconds (e.g., '15s', '1m', '5m').
+        :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(MetricsqueryresponseResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'promql_query',
+            'start',
+            'end',
+            'cluster_id',
+            'workspace_id',
+            'ha_job_id',
+            'service_id',
+            'step'
+        ]
+        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 query_range_api_v2_metrics_query_range_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'promql_query' is set
+        if self.api_client.client_side_validation and ('promql_query' not in local_var_params or  # noqa: E501
+                                                        local_var_params['promql_query'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `promql_query` when calling `query_range_api_v2_metrics_query_range_get`")  # noqa: E501
+        # verify the required parameter 'start' is set
+        if self.api_client.client_side_validation and ('start' not in local_var_params or  # noqa: E501
+                                                        local_var_params['start'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `start` when calling `query_range_api_v2_metrics_query_range_get`")  # noqa: E501
+        # verify the required parameter 'end' is set
+        if self.api_client.client_side_validation and ('end' not in local_var_params or  # noqa: E501
+                                                        local_var_params['end'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `end` when calling `query_range_api_v2_metrics_query_range_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'promql_query' in local_var_params and local_var_params['promql_query'] is not None:  # noqa: E501
+            query_params.append(('promql_query', local_var_params['promql_query']))  # noqa: E501
+        if 'start' in local_var_params and local_var_params['start'] is not None:  # noqa: E501
+            query_params.append(('start', local_var_params['start']))  # noqa: E501
+        if 'end' in local_var_params and local_var_params['end'] is not None:  # noqa: E501
+            query_params.append(('end', local_var_params['end']))  # noqa: E501
+        if 'cluster_id' in local_var_params and local_var_params['cluster_id'] is not None:  # noqa: E501
+            query_params.append(('cluster_id', local_var_params['cluster_id']))  # noqa: E501
+        if 'workspace_id' in local_var_params and local_var_params['workspace_id'] is not None:  # noqa: E501
+            query_params.append(('workspace_id', local_var_params['workspace_id']))  # noqa: E501
+        if 'ha_job_id' in local_var_params and local_var_params['ha_job_id'] is not None:  # noqa: E501
+            query_params.append(('ha_job_id', local_var_params['ha_job_id']))  # noqa: E501
+        if 'service_id' in local_var_params and local_var_params['service_id'] is not None:  # noqa: E501
+            query_params.append(('service_id', local_var_params['service_id']))  # noqa: E501
+        if 'step' in local_var_params and local_var_params['step'] is not None:  # noqa: E501
+            query_params.append(('step', local_var_params['step']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/metrics/query_range', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='MetricsqueryresponseResponse',  # 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 receive_cli_usage_api_v2_cli_usage_post(self, cli_usage_payload, **kwargs):  # noqa: E501
         """Receive Cli Usage  # noqa: E501
 

anyscale/client/openapi_client/models/__init__.py

@@ -116,7 +116,6 @@ from openapi_client.models.cloudcollaborator_list_response import Cloudcollabora
 from openapi_client.models.clouddatabucketpresigneduploadinfo_response import ClouddatabucketpresigneduploadinfoResponse
 from openapi_client.models.clouddatabucketpresignedurlresponse_response import ClouddatabucketpresignedurlresponseResponse
 from openapi_client.models.clouddeployment_list_response import ClouddeploymentListResponse
-from openapi_client.models.clouddeployment_response import ClouddeploymentResponse
 from openapi_client.models.clouddeploymentconfig_response import ClouddeploymentconfigResponse
 from openapi_client.models.cloudoverviewdashboard_response import CloudoverviewdashboardResponse
 from openapi_client.models.cloudregionandzones_response import CloudregionandzonesResponse
@@ -215,6 +214,7 @@ from openapi_client.models.dataset_upload import DatasetUpload
 from openapi_client.models.datasetupload_response import DatasetuploadResponse
 from openapi_client.models.decorated_application_template import DecoratedApplicationTemplate
 from openapi_client.models.decorated_build import DecoratedBuild
+from openapi_client.models.decorated_cloud_deployment import DecoratedCloudDeployment
 from openapi_client.models.decorated_compute_template import DecoratedComputeTemplate
 from openapi_client.models.decorated_compute_template_config import DecoratedComputeTemplateConfig
 from openapi_client.models.decorated_interactive_session import DecoratedInteractiveSession
@@ -235,6 +235,7 @@ from openapi_client.models.decoratedapplicationtemplate_list_response import Dec
 from openapi_client.models.decoratedapplicationtemplate_response import DecoratedapplicationtemplateResponse
 from openapi_client.models.decoratedbuild_list_response import DecoratedbuildListResponse
 from openapi_client.models.decoratedbuild_response import DecoratedbuildResponse
+from openapi_client.models.decoratedclouddeployment_response import DecoratedclouddeploymentResponse
 from openapi_client.models.decoratedcomputetemplate_list_response import DecoratedcomputetemplateListResponse
 from openapi_client.models.decoratedcomputetemplate_response import DecoratedcomputetemplateResponse
 from openapi_client.models.decoratedinteractivesession_list_response import DecoratedinteractivesessionListResponse
@@ -380,6 +381,8 @@ from openapi_client.models.machine_state_info import MachineStateInfo
 from openapi_client.models.machinepoolsearchresult_list_response import MachinepoolsearchresultListResponse
 from openapi_client.models.machinestateinfo_list_response import MachinestateinfoListResponse
 from openapi_client.models.metric import Metric
+from openapi_client.models.metrics_query_response import MetricsQueryResponse
+from openapi_client.models.metricsqueryresponse_response import MetricsqueryresponseResponse
 from openapi_client.models.metronome_customer_info_model import MetronomeCustomerInfoModel
 from openapi_client.models.metronome_dashboard_type import MetronomeDashboardType
 from openapi_client.models.metronomecustomerinfomodel_list_response import MetronomecustomerinfomodelListResponse