anyscale 0.24.60__tar.gz → 0.24.61__tar.gz

{anyscale-0.24.60 → anyscale-0.24.61}/PKG-INFO

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

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/_private/anyscale_client/anyscale_client.py

@@ -19,6 +19,7 @@ from anyscale._private.anyscale_client.common import (
     RUNTIME_ENV_PACKAGE_FORMAT,
 )
 from anyscale._private.models.image_uri import ImageURI
+from anyscale.api_utils.common_utils import source_cloud_id_and_project_id
 from anyscale.api_utils.logs_util import _download_log_from_s3_url_sync
 from anyscale.authenticate import AuthenticationBlock, get_auth_api_client
 from anyscale.cli_logger import BlockLogger
@@ -37,8 +38,10 @@ from anyscale.client.openapi_client.models import (
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
+    DeletedPlatformFineTunedModel,
     ExperimentalWorkspace,
     FineTunedModel,
+    FinetunedmodelListResponse,
     InternalProductionJob,
     SessionSshKey,
     SessionState,
@@ -1115,8 +1118,11 @@ class AnyscaleClient(AnyscaleClientInterface):
         if not os.path.isfile(dataset_file):
             raise ValueError(f"Path '{dataset_file}' is not a valid file.")
 
-        cloud_id, project_id = self._source_cloud_id_and_project_id(
-            cloud_id, project_id
+        cloud_id, project_id = source_cloud_id_and_project_id(
+            internal_api=self._internal_api_client,
+            external_api=self._external_api_client,
+            cloud_id=cloud_id,
+            project_id=project_id,
         )
         print(existing_dataset_id_or_name, description)
 
@@ -1129,72 +1135,19 @@ class AnyscaleClient(AnyscaleClientInterface):
         dataset_bytes = response.content
         return dataset_bytes
 
-    def _source_cloud_id_and_project_id(
-        self, cloud_id: Optional[str], project_id: Optional[str]
-    ) -> Tuple[str, str]:
-        """
-        Returns `Tuple[cloud_id, project_id]` based on optionally provided `cloud_id` and `project_id`.
-
-        - If neither `cloud_id` nor `project_id` are provided, the default cloud and its default project are used
-        (unless inside a workspace)
-        - If both `cloud_id` and `project_id` are provided, we verify that the project is under the cloud.
-
-        Note that the relationship hierarchy is:
-        ```
-        Cloud -----------<- Project
-        1                     Many
-        ```
-        - Each cloud will have at least one project (the default project).
-        - Each project will belong to exactly one cloud.
-        """
-        if not cloud_id:
-            if project_id:
-                # 1. No cloud, project
-                # (Use the cloud of the project)
-                project: Project = self._external_api_client.get_project(
-                    project_id
-                ).result
-                cloud_id = project.cloud_id
-            else:
-                # 2. No cloud, no project
-                # (Use the default cloud and its default project)
-                # (unless we are in an Anyscale workspace, then use the workspace's cloud and project)
-                workspace_id = self.get_current_workspace_id()
-                is_anyscale_workspace = workspace_id is not None
-                if is_anyscale_workspace:
-                    workspace: ExperimentalWorkspace = self._internal_api_client.get_workspace_api_v2_experimental_workspaces_workspace_id_get(
-                        workspace_id
-                    ).result
-                    cloud_id = workspace.cloud_id
-                    project_id = workspace.project_id
-                else:
-                    cloud: Cloud = self._external_api_client.get_default_cloud().result
-                    cloud_id = cloud.id
-                    project = self._external_api_client.get_default_project(
-                        parent_cloud_id=cloud_id
-                    ).result
-                    project_id = project.id
+    def get_finetuned_model(
+        self, model_id: Optional[str], job_id: Optional[str]
+    ) -> FineTunedModel:
+        if model_id:
+            return self._internal_api_client.get_model_api_v2_llm_models_model_id_get(
+                model_id
+            ).result
+        elif job_id:
+            return self._internal_api_client.get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get(
+                job_id
+            ).result
         else:
-            if not project_id:
-                # 3. Cloud, no project
-                # (Use the default project of the cloud)
-                project = self._external_api_client.get_default_project(
-                    parent_cloud_id=cloud_id
-                ).result
-                project_id = project.id
-            else:
-                # 4. Cloud, project
-                # (Verify that the project is in the cloud)
-                project = self._external_api_client.get_project(project_id).result
-                assert (
-                    project.cloud_id == cloud_id
-                ), f"Project {project_id} is not in cloud {cloud_id}."
-        return (cloud_id, project_id)  # type: ignore
-
-    def retrieve_finetuned_model(self, model_id: str) -> FineTunedModel:
-        return self._internal_api_client.get_model_api_v2_llm_models_model_id_get(
-            model_id
-        )
+            raise ValueError("Atleast one of `model_id` or `job_id` must be provided")
 
     @handle_api_exceptions
     def create_workspace(self, model: CreateExperimentalWorkspace) -> str:
@@ -1330,3 +1283,43 @@ class AnyscaleClient(AnyscaleClientInterface):
             return project.directory_name
         else:
             return project.name
+
+    def delete_finetuned_model(self, model_id: str) -> DeletedPlatformFineTunedModel:
+        deleted_model = self._internal_api_client.delete_model_api_v2_llm_models_model_id_delete(
+            model_id
+        ).result
+        return deleted_model
+
+    def list_finetuned_models(
+        self, cloud_id: Optional[str], project_id: Optional[str], max_items: int,
+    ) -> List[FineTunedModel]:
+        if self.inside_workspace():
+            # Resolve `cloud_id` and `project_id`. If not provided and if this is being run in a workspace,
+            # we use the `cloud_id` and `project_id` of the workspace
+            cloud_id, project_id = source_cloud_id_and_project_id(
+                internal_api=self._internal_api_client,
+                external_api=self._external_api_client,
+                cloud_id=cloud_id,
+                project_id=project_id,
+            )
+
+        paging_token = None
+        results = []
+        while True:
+            count = min(self.LIST_ENDPOINT_COUNT, max_items)
+            resp: FinetunedmodelListResponse = self._internal_api_client.list_models_api_v2_llm_models_get(
+                cloud_id=cloud_id,
+                project_id=project_id,
+                paging_token=paging_token,
+                count=count,
+            )
+            models = resp.results
+            results.extend(models)
+            if not len(models) or not resp.metadata.next_paging_token:
+                break
+
+            if max_items and len(results) >= max_items:
+                break
+            paging_token = resp.metadata.next_paging_token
+
+        return results[:max_items] if max_items else results

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/_private/anyscale_client/common.py

@@ -8,6 +8,7 @@ from anyscale.client.openapi_client.models import (
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
+    DeletedPlatformFineTunedModel,
     FineTunedModel,
     InternalProductionJob,
     Project,
@@ -472,7 +473,9 @@ class AnyscaleClientInterface(ABC):
         raise NotImplementedError
 
     @abstractmethod
-    def retrieve_finetuned_model(self, id: str) -> FineTunedModel:  # noqa: A002
+    def get_finetuned_model(
+        self, model_id: Optional[str], job_id: Optional[str]
+    ) -> FineTunedModel:  # noqa: A002
         """Returns LLM model information for the given model ID"""
         raise NotImplementedError
 
@@ -495,3 +498,14 @@ class AnyscaleClientInterface(ABC):
     def get_workspace_default_dir_name(self, workspace_id) -> str:
         """Get the default directory name for a workspace."""
         raise NotImplementedError
+
+    @abstractmethod
+    def delete_finetuned_model(self, model_id: str) -> DeletedPlatformFineTunedModel:
+        """Deletes a finetuned model from the model registry given the model ID"""
+        raise NotImplementedError
+
+    @abstractmethod
+    def list_finetuned_models(
+        self, cloud_id: Optional[str], project_id: Optional[str], max_items: int,
+    ) -> List[FineTunedModel]:
+        raise NotImplementedError

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -16,7 +16,7 @@ from anyscale.client.openapi_client.models import (
     CreateExperimentalWorkspace,
     CreateInternalProductionJob,
     DecoratedComputeTemplate,
-    EndpointsFineTunedModelStatus,
+    DeletedPlatformFineTunedModel,
     ExperimentalWorkspace,
     FineTunedModel,
     FineTuneType,
@@ -833,12 +833,13 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     ) -> bytes:
         return b""
 
-    def retrieve_finetuned_model(self, model_id: str) -> FineTunedModel:
+    def get_finetuned_model(
+        self, model_id: Optional[str], job_id: Optional[str]  # noqa: ARG002
+    ) -> FineTunedModel:
         return FineTunedModel(
-            id=model_id,
-            model_id=model_id,
+            id=model_id if model_id else "test-model-id",
+            model_id=model_id if model_id else "test-model-id",
             base_model_id="my_base_model_id",
-            status=EndpointsFineTunedModelStatus.READY,
             ft_type=FineTuneType.LORA,
             creator_id="",
             creator_email="",
@@ -944,3 +945,26 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         if workspace is None:
             raise ValueError(f"Workspace '{workspace_id}' not found.")
         return "/home/ray/default"
+
+    def delete_finetuned_model(self, model_id: str) -> DeletedPlatformFineTunedModel:
+        return DeletedPlatformFineTunedModel(id=model_id, deleted_at=datetime.utcnow())
+
+    def list_finetuned_models(
+        self,
+        cloud_id: Optional[str],  # noqa: ARG002
+        project_id: Optional[str],  # noqa: ARG002
+        max_items: int,
+    ) -> List[FineTunedModel]:
+        return [
+            FineTunedModel(
+                id="test-model-id",
+                model_id="test-model-id",
+                base_model_id="my_base_model_id",
+                ft_type=FineTuneType.LORA,
+                creator_id="",
+                creator_email="",
+                created_at=datetime.utcnow(),
+                storage_uri="s3://fake_bucket/fake_folder/",
+            )
+            for _ in range(max_items)
+        ]

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/_private/docgen/generator.py

@@ -8,6 +8,7 @@ import click
 import yaml
 
 from anyscale._private.models.model_base import ModelBaseType, ModelEnumType
+from anyscale.commands.util import AnyscaleCommand
 
 
 ModelType = Union[ModelBaseType, ModelEnumType]
@@ -32,11 +33,11 @@ class Module:
     title: str
     filename: str
     cli_prefix: str
-    cli_commands: List[click.Command]
+    cli_commands: List[AnyscaleCommand]
     sdk_prefix: str
     sdk_commands: List[Callable]
     models: List[ModelType]
-    cli_command_group_prefix: Optional[Dict[click.Command, str]] = None
+    cli_command_group_prefix: Optional[Dict[AnyscaleCommand, str]] = None
 
 
 class MarkdownGenerator:
@@ -172,13 +173,15 @@ class MarkdownGenerator:
             f"Unhandled type: {t}. Either this type should not be in our public APIs, or you must add handling for it to the doc generator."
         )
 
-    def _gen_example_tabs(self, t: Union[Callable, ModelBaseType]) -> str:
+    def _gen_example_tabs(
+        self, t: Union[Callable, ModelBaseType, AnyscaleCommand]
+    ) -> str:
         """Generate a tab section that contains yaml, python, and/or CLI examples for the type.
 
         The examples are pulled from magic attributes:
             - __doc_yaml_example__ (required for models ending with "Config")
-            - __doc_py_example__ (required in all cases)
-            - __doc_cli_example__ (required for models)
+            - __doc_py_example__ (required in sdks)
+            - __doc_cli_example__ (required for models and cli commands)
         """
         yaml_example: Optional[str] = getattr(t, "__doc_yaml_example__", None)
         py_example: Optional[str] = getattr(t, "__doc_py_example__", None)
@@ -193,7 +196,15 @@ class MarkdownGenerator:
                 raise ValueError(
                     f"Config model '{t.__name__}' is missing a '__doc_yaml_example__'."
                 )
-        elif not py_example:
+        if isinstance(t, AnyscaleCommand) and not cli_example:
+            raise ValueError(
+                f"CLI command '{t.name}' is missing a '__doc_cli_example__'."
+            )
+        if (
+            not isinstance(t, ModelBaseType)
+            and not isinstance(t, AnyscaleCommand)
+            and not py_example
+        ):
             raise ValueError(
                 f"SDK command '{t.__name__}' is missing a '__doc_py_example__'."
             )
@@ -297,7 +308,7 @@ class MarkdownGenerator:
         return md
 
     def _gen_markdown_for_cli_command(
-        self, c: click.Command, *, cli_prefix: str
+        self, c: AnyscaleCommand, *, cli_prefix: str
     ) -> str:
         """Generate a markdown section for a CLI command.
 
@@ -335,6 +346,9 @@ class MarkdownGenerator:
                 md += f"- **`{name}`**: {help_str}\n"
             md += "\n"
 
+        if hasattr(c, "__doc_cli_example__"):
+            md += self._gen_example_tabs(c)
+
         return md
 
     def _gen_markdown_for_sdk_command(self, c: Callable, *, sdk_prefix: str) -> str:

anyscale-0.24.61/anyscale/api_utils/common_utils.py

@@ -0,0 +1,81 @@
+import os
+from typing import Optional, Tuple
+
+from anyscale.client.openapi_client.api.default_api import DefaultApi as InternalApi
+from anyscale.client.openapi_client.models import (
+    Cloud,
+    ExperimentalWorkspace,
+)
+from anyscale.sdk.anyscale_client.api.default_api import DefaultApi as ExternalApi
+from anyscale.sdk.anyscale_client.models import Project
+
+
+WORKSPACE_ID_ENV_VAR = "ANYSCALE_EXPERIMENTAL_WORKSPACE_ID"
+
+# TODO (sumanthrh): Currently this same functionality of getting workspace id is repeated in many places, we should try to use this util everywhere
+def get_current_workspace_id() -> Optional[str]:
+    """Returns the ID of the workspace the client is running in (or `None`)."""
+    return os.environ.get(WORKSPACE_ID_ENV_VAR, None)
+
+
+def source_cloud_id_and_project_id(
+    *,
+    internal_api: InternalApi,
+    external_api: ExternalApi,
+    cloud_id: Optional[str],
+    project_id: Optional[str],
+) -> Tuple[str, str]:
+    """
+        Returns `Tuple[cloud_id, project_id]` based on optionally provided `cloud_id` and `project_id`.
+
+        - If neither `cloud_id` nor `project_id` are provided, the default cloud and its default project are used
+        (unless inside a workspace)
+        - If both `cloud_id` and `project_id` are provided, we verify that the project is under the cloud.
+
+        Note that the relationship hierarchy is:
+        ```
+        Cloud -----------<- Project
+        1                     Many
+        ```
+        - Each cloud will have at least one project (the default project).
+        - Each project will belong to exactly one cloud.
+        """
+    if not cloud_id:
+        if project_id:
+            # 1. No cloud, project
+            # (Use the cloud of the project)
+            project: Project = external_api.get_project(project_id).result
+            cloud_id = project.cloud_id
+        else:
+            # 2. No cloud, no project
+            # (Use the default cloud and its default project)
+            # (unless we are in an Anyscale workspace, then use the workspace's cloud and project)
+            workspace_id = get_current_workspace_id()
+            is_anyscale_workspace = workspace_id is not None
+            if is_anyscale_workspace:
+                workspace: ExperimentalWorkspace = internal_api.get_workspace_api_v2_experimental_workspaces_workspace_id_get(
+                    workspace_id
+                ).result
+                cloud_id = workspace.cloud_id
+                project_id = workspace.project_id
+            else:
+                cloud: Cloud = external_api.get_default_cloud().result
+                cloud_id = cloud.id
+                project = external_api.get_default_project(
+                    parent_cloud_id=cloud_id
+                ).result
+                project_id = project.id
+    else:
+        if not project_id:
+            # 3. Cloud, no project
+            # (Use the default project of the cloud)
+            project = external_api.get_default_project(parent_cloud_id=cloud_id).result
+            project_id = project.id
+        else:
+            # 4. Cloud, project
+            # (Verify that the project is in the cloud)
+            project = external_api.get_project(project_id).result
+            assert (
+                project.cloud_id == cloud_id
+            ), f"Project {project_id} is not in cloud {cloud_id}."
+    return (cloud_id, project_id)  # type: ignore

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/client/README.md

@@ -224,6 +224,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_metronome_embedded_usage_dashboard_api_v2_organization_billing_metronome_embedded_dashboard_url_dashboard_type_get**](docs/DefaultApi.md#get_metronome_embedded_usage_dashboard_api_v2_organization_billing_metronome_embedded_dashboard_url_dashboard_type_get) | **GET** /api/v2/organization_billing/metronome_embedded_dashboard_url/{dashboard_type} | Get Metronome Embedded Usage Dashboard
 *DefaultApi* | [**get_metronome_embedded_usage_dashboard_by_organization_api_v2_organization_billing_organization_id_metronome_embedded_dashboard_url_dashboard_type_get**](docs/DefaultApi.md#get_metronome_embedded_usage_dashboard_by_organization_api_v2_organization_billing_organization_id_metronome_embedded_dashboard_url_dashboard_type_get) | **GET** /api/v2/organization_billing/{organization_id}/metronome_embedded_dashboard_url/{dashboard_type} | Get Metronome Embedded Usage Dashboard By Organization
 *DefaultApi* | [**get_model_api_v2_llm_models_model_id_get**](docs/DefaultApi.md#get_model_api_v2_llm_models_model_id_get) | **GET** /api/v2/llm/models/{model_id} | Get Model
+*DefaultApi* | [**get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get**](docs/DefaultApi.md#get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get) | **GET** /api/v2/llm/models/get_by_job_id/{job_id} | Get Model By Job Id
 *DefaultApi* | [**get_model_configuration_api_v2_aica_endpoints_model_config_get**](docs/DefaultApi.md#get_model_configuration_api_v2_aica_endpoints_model_config_get) | **GET** /api/v2/aica_endpoints/model_config | Get Model Configuration
 *DefaultApi* | [**get_model_schema_api_v2_aica_endpoints_model_schema_get**](docs/DefaultApi.md#get_model_schema_api_v2_aica_endpoints_model_schema_get) | **GET** /api/v2/aica_endpoints/model_schema | Get Model Schema
 *DefaultApi* | [**get_monitor_logs_api_v2_sessions_session_id_monitor_logs_get**](docs/DefaultApi.md#get_monitor_logs_api_v2_sessions_session_id_monitor_logs_get) | **GET** /api/v2/sessions/{session_id}/monitor_logs | Get Monitor Logs
@@ -663,7 +664,6 @@ Class | Method | HTTP request | Description
  - [EbsBlockDevice](docs/EbsBlockDevice.md)
  - [EditableCloudResource](docs/EditableCloudResource.md)
  - [EditableCloudResourceGCP](docs/EditableCloudResourceGCP.md)
- - [EndpointsFineTunedModelStatus](docs/EndpointsFineTunedModelStatus.md)
  - [Error](docs/Error.md)
  - [EventLevel](docs/EventLevel.md)
  - [ExecuteCommandResponse](docs/ExecuteCommandResponse.md)
@@ -686,6 +686,7 @@ Class | Method | HTTP request | Description
  - [FineTunedModel](docs/FineTunedModel.md)
  - [FineTuningJobStatus](docs/FineTuningJobStatus.md)
  - [FinetunedmodelListResponse](docs/FinetunedmodelListResponse.md)
+ - [FinetunedmodelResponse](docs/FinetunedmodelResponse.md)
  - [FinishFTJobRequest](docs/FinishFTJobRequest.md)
  - [FinishFTJobRequestV2](docs/FinishFTJobRequestV2.md)
  - [GCPFileStoreConfig](docs/GCPFileStoreConfig.md)

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/client/openapi_client/__init__.py

@@ -285,7 +285,6 @@ from openapi_client.models.dismissal_type import DismissalType
 from openapi_client.models.ebs_block_device import EbsBlockDevice
 from openapi_client.models.editable_cloud_resource import EditableCloudResource
 from openapi_client.models.editable_cloud_resource_gcp import EditableCloudResourceGCP
-from openapi_client.models.endpoints_fine_tuned_model_status import EndpointsFineTunedModelStatus
 from openapi_client.models.error import Error
 from openapi_client.models.event_level import EventLevel
 from openapi_client.models.execute_command_response import ExecuteCommandResponse
@@ -308,6 +307,7 @@ from openapi_client.models.fine_tune_type import FineTuneType
 from openapi_client.models.fine_tuned_model import FineTunedModel
 from openapi_client.models.fine_tuning_job_status import FineTuningJobStatus
 from openapi_client.models.finetunedmodel_list_response import FinetunedmodelListResponse
+from openapi_client.models.finetunedmodel_response import FinetunedmodelResponse
 from openapi_client.models.finish_ft_job_request import FinishFTJobRequest
 from openapi_client.models.finish_ft_job_request_v2 import FinishFTJobRequestV2
 from openapi_client.models.gcp_file_store_config import GCPFileStoreConfig

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/client/openapi_client/api/default_api.py

@@ -17300,7 +17300,7 @@ class DefaultApi(object):
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
                                  (connection, read) timeouts.
-        :return: FineTunedModel
+        :return: FinetunedmodelResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -17327,7 +17327,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(FineTunedModel, status_code(int), headers(HTTPHeaderDict))
+        :return: tuple(FinetunedmodelResponse, status_code(int), headers(HTTPHeaderDict))
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -17388,7 +17388,121 @@ class DefaultApi(object):
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='FineTunedModel',  # noqa: E501
+            response_type='FinetunedmodelResponse',  # 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 get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get(self, job_id, **kwargs):  # noqa: E501
+        """Get Model By Job Id  # noqa: E501
+
+        Retrieves a model from the model registry by job id  # 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_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get(job_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str job_id: (required)
+        :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: FinetunedmodelResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get_with_http_info(job_id, **kwargs)  # noqa: E501
+
+    def get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get_with_http_info(self, job_id, **kwargs):  # noqa: E501
+        """Get Model By Job Id  # noqa: E501
+
+        Retrieves a model from the model registry by job id  # 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_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get_with_http_info(job_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str job_id: (required)
+        :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(FinetunedmodelResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'job_id'
+        ]
+        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 get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'job_id' is set
+        if self.api_client.client_side_validation and ('job_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['job_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `job_id` when calling `get_model_by_job_id_api_v2_llm_models_get_by_job_id_job_id_get`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'job_id' in local_var_params:
+            path_params['job_id'] = local_var_params['job_id']  # noqa: E501
+
+        query_params = []
+
+        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/llm/models/get_by_job_id/{job_id}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='FinetunedmodelResponse',  # 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
@@ -27784,6 +27898,7 @@ class DefaultApi(object):
         :param async_req bool: execute request asynchronously
         :param str cloud_id:
         :param str project_id:
+        :param str workspace_id:
         :param str paging_token:
         :param int count:
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -27812,6 +27927,7 @@ class DefaultApi(object):
         :param async_req bool: execute request asynchronously
         :param str cloud_id:
         :param str project_id:
+        :param str workspace_id:
         :param str paging_token:
         :param int count:
         :param _return_http_data_only: response data without head status code
@@ -27833,6 +27949,7 @@ class DefaultApi(object):
         all_params = [
             'cloud_id',
             'project_id',
+            'workspace_id',
             'paging_token',
             'count'
         ]
@@ -27867,6 +27984,8 @@ class DefaultApi(object):
             query_params.append(('cloud_id', local_var_params['cloud_id']))  # noqa: E501
         if 'project_id' in local_var_params and local_var_params['project_id'] is not None:  # noqa: E501
             query_params.append(('project_id', local_var_params['project_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 '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

{anyscale-0.24.60 → anyscale-0.24.61}/anyscale/client/openapi_client/models/__init__.py

@@ -271,7 +271,6 @@ from openapi_client.models.dismissal_type import DismissalType
 from openapi_client.models.ebs_block_device import EbsBlockDevice
 from openapi_client.models.editable_cloud_resource import EditableCloudResource
 from openapi_client.models.editable_cloud_resource_gcp import EditableCloudResourceGCP
-from openapi_client.models.endpoints_fine_tuned_model_status import EndpointsFineTunedModelStatus
 from openapi_client.models.error import Error
 from openapi_client.models.event_level import EventLevel
 from openapi_client.models.execute_command_response import ExecuteCommandResponse
@@ -294,6 +293,7 @@ from openapi_client.models.fine_tune_type import FineTuneType
 from openapi_client.models.fine_tuned_model import FineTunedModel
 from openapi_client.models.fine_tuning_job_status import FineTuningJobStatus
 from openapi_client.models.finetunedmodel_list_response import FinetunedmodelListResponse
+from openapi_client.models.finetunedmodel_response import FinetunedmodelResponse
 from openapi_client.models.finish_ft_job_request import FinishFTJobRequest
 from openapi_client.models.finish_ft_job_request_v2 import FinishFTJobRequestV2
 from openapi_client.models.gcp_file_store_config import GCPFileStoreConfig