anyscale 0.26.21__py3-none-any.whl → 0.26.23__py3-none-any.whl

anyscale/_private/models/model_base.py

@@ -1,9 +1,13 @@
+import asyncio
+from collections import deque
 from dataclasses import asdict, fields
 from enum import Enum, EnumMeta
 import inspect
 from typing import (
     Any,
+    Awaitable,
     Callable,
+    Deque,
     Dict,
     Generic,
     Iterable,
@@ -252,3 +256,94 @@ class ListResponse(List[TModelBase]):
                 return
             if index >= len(self) and self.has_more:
                 self._fetch_next_page()
+
+
+RT = TypeVar("RT")
+
+
+class ResultIterator(Generic[RT]):
+    """
+    Lazily fetch and parse pages from a paged-list API that returns
+    Pydantic models with `.results` and `.metadata.next_paging_token`.
+    """
+
+    def __init__(
+        self,
+        *,
+        page_token: Optional[str],
+        max_items: Optional[int],
+        fetch_page: Callable[[Optional[str]], Any],
+        parse_fn: Optional[Callable[[Any], RT]] = None,
+        async_parse_fn: Optional[Callable[[Any], Awaitable[RT]]] = None,
+    ):
+        if parse_fn and async_parse_fn:
+            raise ValueError("Only one of parse_fn or async_parse_fn may be provided")
+
+        self._token = page_token
+        self._max = max_items
+        self._fetch = fetch_page
+        self._parse = parse_fn
+        self._aparse = async_parse_fn
+        self._buffer: Deque[RT] = deque()
+        self._count = 0
+        self._finished = False
+
+    def __iter__(self) -> Iterator[RT]:
+        while True:
+            # 1) Drain the buffer
+            while self._buffer:
+                if self._max is not None and self._count >= self._max:
+                    return
+                self._count += 1
+                yield self._buffer.popleft()
+
+            # 2) Done?
+            if self._finished or (self._max is not None and self._count >= self._max):
+                return
+
+            # 3) Fetch the next page (Pydantic model)
+            page = self._fetch(self._token)
+            raw_results = page.results
+            self._token = page.metadata.next_paging_token
+
+            # 4) No more data?
+            if not raw_results:
+                self._finished = True
+                return
+
+            # 5) Parse—sync or async
+            if self._aparse:
+                processed = asyncio.run(
+                    ResultIterator._process_items_async(raw_results, self._aparse)
+                )
+                self._buffer.extend(processed)
+
+            elif self._parse:
+                try:
+                    for raw in raw_results:
+                        self._buffer.append(self._parse(raw))
+                except Exception as e:  # noqa: BLE001
+                    raise RuntimeError(f"sync parse error: {e}") from e
+
+            else:
+                # No parser: assume items are already RT
+                self._buffer.extend(raw_results)  # type: ignore
+
+            # 6) If no next token, finish on next loop
+            if self._token is None:
+                self._finished = True
+
+    @staticmethod
+    async def _process_items_async(
+        items: List[Any], parser: Callable[[Any], Awaitable[RT]],
+    ) -> List[RT]:
+        if not items:
+            return []
+        tasks = [parser(item) for item in items]
+        results = await asyncio.gather(*tasks, return_exceptions=True)
+        processed: List[RT] = []
+        for idx, res in enumerate(results):
+            if isinstance(res, Exception):
+                raise RuntimeError(f"async parse failed on item {idx}: {res}") from res
+            processed.append(res)
+        return processed

anyscale/_private/workload/workload_sdk.py

@@ -328,4 +328,6 @@ class WorkloadSDK(BaseSDK):
         if not compute_config.anonymous:
             return compute_config_name
 
-        return self._compute_config_sdk.get_compute_config(id=compute_config_id).config
+        return self._compute_config_sdk._convert_api_model_to_compute_config_version(  # noqa: SLF001
+            compute_config
+        ).config

anyscale/aggregated_instance_usage/models.py

@@ -15,9 +15,9 @@ import anyscale
 from anyscale.aggregated_instance_usage.models import DownloadCSVFilters
 
 download_csv_filters = DownloadCSVFilters(
-    # Start date (inclusive) for the usage CSV.
+    # Start date (UTC inclusive) for the usage CSV.
     start_date="2024-10-01",
-    # End date (inclusive) for the usage CSV.
+    # End date (UTC inclusive) for the usage CSV.
     end_date="2024-10-31",
     # Optional cloud name to filter by.
     cloud="cloud_name",
@@ -39,14 +39,14 @@ download_csv_filters = DownloadCSVFilters(
             raise ValueError("Incorrect date format, should be YYYY-MM-DD")
 
     start_date: str = field(
-        metadata={"docstring": "Start date (inclusive) for the usage CSV."}
+        metadata={"docstring": "Start date (UTC inclusive) for the usage CSV."}
     )
 
     def _validate_start_date(self, start_date: str):
         self._validate_date(start_date)
 
     end_date: str = field(
-        metadata={"docstring": "End date (inclusive) for the usage CSV."}
+        metadata={"docstring": "End date (UTC inclusive) for the usage CSV."}
     )
 
     def _validate_end_date(self, end_date: str):

anyscale/client/README.md

@@ -294,6 +294,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_project_collaborators_api_v2_projects_project_id_collaborators_users_get**](docs/DefaultApi.md#list_project_collaborators_api_v2_projects_project_id_collaborators_users_get) | **GET** /api/v2/projects/{project_id}/collaborators/users | List Project Collaborators
 *DefaultApi* | [**list_projects_api_v2_projects_get**](docs/DefaultApi.md#list_projects_api_v2_projects_get) | **GET** /api/v2/projects/ | List Projects
 *DefaultApi* | [**list_ray_sessions_api_v2_tasks_dashboard_list_ray_sessions_get**](docs/DefaultApi.md#list_ray_sessions_api_v2_tasks_dashboard_list_ray_sessions_get) | **GET** /api/v2/tasks_dashboard/list_ray_sessions | List Ray Sessions
+*DefaultApi* | [**list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get**](docs/DefaultApi.md#list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get) | **GET** /api/v2/experimental_workspaces/templates/recommended | List Recommended Workspace Templates
 *DefaultApi* | [**list_services_api_v2_services_v2_get**](docs/DefaultApi.md#list_services_api_v2_services_v2_get) | **GET** /api/v2/services-v2/ | List Services
 *DefaultApi* | [**list_sessions_api_v2_sessions_get**](docs/DefaultApi.md#list_sessions_api_v2_sessions_get) | **GET** /api/v2/sessions/ | List Sessions
 *DefaultApi* | [**list_workspace_templates_api_v2_experimental_workspaces_templates_get**](docs/DefaultApi.md#list_workspace_templates_api_v2_experimental_workspaces_templates_get) | **GET** /api/v2/experimental_workspaces/templates | List Workspace Templates

anyscale/client/openapi_client/api/default_api.py

@@ -20980,6 +20980,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cluster_id: (required)
+        :param bool skip_job_details: Skip decorating job details, which can be an expensive operation.
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -21004,6 +21005,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cluster_id: (required)
+        :param bool skip_job_details: Skip decorating job details, which can be an expensive operation.
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -21021,7 +21023,8 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'cluster_id'
+            'cluster_id',
+            'skip_job_details'
         ]
         all_params.extend(
             [
@@ -21052,6 +21055,8 @@ class DefaultApi(object):
         query_params = []
         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 'skip_job_details' in local_var_params and local_var_params['skip_job_details'] is not None:  # noqa: E501
+            query_params.append(('skip_job_details', local_var_params['skip_job_details']))  # noqa: E501
 
         header_params = {}
 
@@ -26320,6 +26325,122 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get(self, **kwargs):  # noqa: E501
+        """List Recommended Workspace Templates  # noqa: E501
+
+        Lists all workspace templates ranked by user's organization marketing questions  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param int count: Maximum number of templates to return
+        :param list[str] oa_group_names: Search for templates that belong to the provided group name
+        :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: WorkspacetemplateListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get_with_http_info(**kwargs)  # noqa: E501
+
+    def list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get_with_http_info(self, **kwargs):  # noqa: E501
+        """List Recommended Workspace Templates  # noqa: E501
+
+        Lists all workspace templates ranked by user's organization marketing questions  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get_with_http_info(async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param int count: Maximum number of templates to return
+        :param list[str] oa_group_names: Search for templates that belong to the provided group name
+        :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(WorkspacetemplateListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'count',
+            'oa_group_names'
+        ]
+        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 list_recommended_workspace_templates_api_v2_experimental_workspaces_templates_recommended_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        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
+        if 'oa_group_names' in local_var_params and local_var_params['oa_group_names'] is not None:  # noqa: E501
+            query_params.append(('oa_group_names', local_var_params['oa_group_names']))  # noqa: E501
+            collection_formats['oa_group_names'] = 'multi'  # 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/experimental_workspaces/templates/recommended', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='WorkspacetemplateListResponse',  # 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 list_services_api_v2_services_v2_get(self, **kwargs):  # noqa: E501
         """List Services  # noqa: E501