lightning-sdk 0.1.29__py3-none-any.whl → 0.1.31__py3-none-any.whl

lightning_sdk/__init__.py

@@ -25,5 +25,5 @@ __all__ = [
     "Agent",
 ]
 
-__version__ = "0.1.29"
+__version__ = "0.1.31"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/api/teamspace_api.py

@@ -6,6 +6,7 @@ from typing import Dict, List, Optional
 from lightning_sdk.api.utils import _download_model_files, _DummyBody, _ModelFileUploader
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
+    ModelIdVersionsBody,
     ModelsStoreApi,
     ProjectIdAgentsBody,
     ProjectIdModelsBody,
@@ -28,6 +29,7 @@ class TeamspaceApi:
 
     def __init__(self) -> None:
         self._client = LightningClient(max_tries=7)
+        self._models: Optional[ModelsStoreApi] = None
 
     def get_teamspace(self, name: str, owner_id: str) -> V1Project:
         """Get the current teamspace from the owner."""
@@ -165,9 +167,36 @@ class TeamspaceApi:
         teamspace_id: str,
         cluster_id: str,
     ) -> V1ModelVersionArchive:
-        api = ModelsStoreApi(self._client.api_client)
-        body = ProjectIdModelsBody(cluster_id=cluster_id, metadata=metadata, name=name, private=private)
-        return api.models_store_create_model(body, project_id=teamspace_id)
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        # ask if such model already exists by listing models with specific name
+        models = self._models.models_store_list_models(project_id=teamspace_id, name=name).models
+        if len(models) == 0:
+            return self._models.models_store_create_model(
+                body=ProjectIdModelsBody(cluster_id=cluster_id, metadata=metadata, name=name, private=private),
+                project_id=teamspace_id,
+            )
+        assert len(models) == 1, "Multiple models with the same name found"
+        return self._models.models_store_create_model_version(
+            body=ModelIdVersionsBody(cluster_id=cluster_id),
+            project_id=teamspace_id,
+            model_id=models[0].id,
+        )
+
+    def delete_model(self, name: str, version: Optional[str], teamspace_id: str) -> None:
+        """Delete a model or a version from the model store."""
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        models = self._models.models_store_list_models(project_id=teamspace_id, name=name).models
+        assert len(models) == 1, "Multiple models with the same name found"
+        model_id = models[0].id
+        # decide if delete only version of whole model
+        if version:
+            if version == "latest":
+                version = models[0].latest_version
+            self._models.models_store_delete_model_version(project_id=teamspace_id, model_id=model_id, version=version)
+        else:
+            self._models.models_store_delete_model(project_id=teamspace_id, model_id=model_id)
 
     def upload_model_file(
         self,
@@ -213,8 +242,9 @@ class TeamspaceApi:
             )
 
     def complete_model_upload(self, model_id: str, version: str, teamspace_id: str) -> None:
-        api = ModelsStoreApi(self._client.api_client)
-        api.models_store_complete_model_upload(
+        if not self._models:
+            self._models = ModelsStoreApi(self._client.api_client)
+        self._models.models_store_complete_model_upload(
             body=_DummyBody(),
             project_id=teamspace_id,
             model_id=model_id,

lightning_sdk/cli/download.py

@@ -4,6 +4,7 @@ from pathlib import Path
 from typing import Optional
 
 from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.cli.models import _get_teamspace, _parse_model_name
 from lightning_sdk.cli.studios_menu import _StudiosMenu
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
@@ -12,6 +13,22 @@ from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 class _Downloads(_StudiosMenu):
     """Download files and folders from Lightning AI."""
 
+    def model(self, name: str, path: Optional[str] = None) -> None:
+        """Download a Model.
+
+        Args:
+          name: The name of the Model you want to download.
+            This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
+          path: The path to the directory where the Model should be downloaded.
+        """
+        org_name, teamspace_name, model_name = _parse_model_name(name)
+        teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
+        teamspace.download_model(
+            name=model_name,
+            download_dir=path or ".",
+            progress_bar=True,
+        )
+
     def _resolve_studio(self, studio: Optional[str]) -> Studio:
         user = _get_authed_user()
         # if no studio specify suggest/filter only user's studios
@@ -53,7 +70,7 @@ class _Downloads(_StudiosMenu):
             return Studio(**selected_studio)
 
     def folder(self, path: str = "", studio: Optional[str] = None, local_path: str = ".") -> None:
-        """Download a folder from a studio.
+        """Download a folder from a Studio.
 
         Args:
           path: The relative path within the Studio you want to download.
@@ -86,7 +103,7 @@ class _Downloads(_StudiosMenu):
             ) from e
 
     def file(self, path: str = "", studio: Optional[str] = None, local_path: str = ".") -> None:
-        """Download a file from a studio.
+        """Download a file from a Studio.
 
         Args:
           path: The relative path within the Studio you want to download.

lightning_sdk/cli/models.py

@@ -0,0 +1,38 @@
+from typing import Tuple
+
+from lightning_sdk.api import OrgApi, UserApi
+from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.teamspace import Teamspace
+from lightning_sdk.utils.resolve import _get_authed_user
+
+
+def _parse_model_name(name: str) -> Tuple[str, str, str]:
+    """Parse the name argument into its components."""
+    try:
+        org_name, teamspace_name, model_name = name.split("/")
+    except ValueError as err:
+        raise StudioCliError(
+            f"Model name must be in the format 'organization/teamspace/model' but you provided '{name}'."
+        ) from err
+    return org_name, teamspace_name, model_name
+
+
+def _get_teamspace(name: str, organization: str) -> Teamspace:
+    """Get a Teamspace object from the SDK."""
+    org_api = OrgApi()
+    user = _get_authed_user()
+    teamspaces = {}
+    for ts in UserApi()._get_all_teamspace_memberships(""):
+        if ts.owner_type == "organization":
+            org = org_api._get_org_by_id(ts.owner_id)
+            teamspaces[f"{org.name}/{ts.name}"] = {"name": ts.name, "org": org.name}
+        elif ts.owner_type == "user":  # todo: check also the name
+            teamspaces[f"{user.name}/{ts.name}"] = {"name": ts.name, "user": user}
+        else:
+            raise StudioCliError(f"Unknown organization type {ts.owner_type}")
+
+    requested_teamspace = f"{organization}/{name}".lower()
+    if requested_teamspace not in teamspaces:
+        options = "\n\t".join(teamspaces.keys())
+        raise StudioCliError(f"Teamspace `{requested_teamspace}` not found. Available teamspaces: \n\t{options}")
+    return Teamspace(**teamspaces[requested_teamspace])

lightning_sdk/cli/upload.py

@@ -9,6 +9,7 @@ from tqdm import tqdm
 
 from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.cli.models import _get_teamspace, _parse_model_name
 from lightning_sdk.cli.studios_menu import _StudiosMenu
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
@@ -19,6 +20,24 @@ class _Uploads(_StudiosMenu):
 
     _studio_upload_status_path = "~/.lightning/studios/uploads"
 
+    def model(self, name: str, path: Optional[str] = None, cloud_account: Optional[str] = None) -> None:
+        """Upload a Model.
+
+        Args:
+          name: The name of the Model you want to upload.
+            This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
+          path: The path to the file or directory you want to upload. Defaults to the current directory.
+          cloud_account: The name of the cloud account to store the Model in.
+        """
+        org_name, teamspace_name, model_name = _parse_model_name(name)
+        teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
+        teamspace.upload_model(
+            path=path or ".",
+            name=model_name,
+            progress_bar=True,
+            cluster_id=cloud_account,
+        )
+
     def _resolve_studio(self, studio: Optional[str]) -> Studio:
         user = _get_authed_user()
         possible_studios = self._get_possible_studios(user)
@@ -43,7 +62,7 @@ class _Uploads(_StudiosMenu):
             return Studio(**selected_studio)
 
     def folder(self, path: str, studio: Optional[str] = None, remote_path: Optional[str] = None) -> None:
-        """Upload a file or folder to a studio.
+        """Upload a file or folder to a Studio.
 
         Args:
           path: The path to the file or directory you want to upload
@@ -97,7 +116,7 @@ class _Uploads(_StudiosMenu):
         print(f"See your files at {studio_url}")
 
     def file(self, path: str, studio: Optional[str] = None, remote_path: Optional[str] = None) -> None:
-        """Upload a file to a studio.
+        """Upload a file to a Studio.
 
         Args:
           path: The path to the file you want to upload

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -104,6 +104,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCo
 from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
 from lightning_sdk.lightning_cloud.openapi.models.id_complete_body import IdCompleteBody
 from lightning_sdk.lightning_cloud.openapi.models.id_engage_body import IdEngageBody
+from lightning_sdk.lightning_cloud.openapi.models.id_engage_body1 import IdEngageBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body import IdExecuteBody
 from lightning_sdk.lightning_cloud.openapi.models.id_execute_body1 import IdExecuteBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_fork_body import IdForkBody
@@ -133,6 +134,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metrics_stream_id_loggerartifa
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body import MetricsstreamCreateBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body import MetricsstreamDeleteBody
 from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
+from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
 from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
@@ -359,6 +361,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_state import V1D
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_status import V1DeploymentStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_strategy import V1DeploymentStrategy
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template import V1DeploymentTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_engagement_response import V1DeploymentTemplateEngagementResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import V1DeploymentTemplateGalleryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter import V1DeploymentTemplateParameter
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter_placement import V1DeploymentTemplateParameterPlacement
@@ -591,6 +594,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_org_role import V1OrgRole
 from lightning_sdk.lightning_cloud.openapi.models.v1_organization import V1Organization
 from lightning_sdk.lightning_cloud.openapi.models.v1_owner_type import V1OwnerType
 from lightning_sdk.lightning_cloud.openapi.models.v1_package_manager import V1PackageManager
+from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec import V1ParameterizationSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin

lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py

@@ -1955,7 +1955,7 @@ class CloudSpaceServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def cloud_space_service_engage_cloud_space(self, body: 'IdEngageBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
+    def cloud_space_service_engage_cloud_space(self, body: 'IdEngageBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
         """cloud_space_service_engage_cloud_space  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1964,7 +1964,7 @@ class CloudSpaceServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdEngageBody body: (required)
+        :param IdEngageBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1CloudSpaceEngagementResponse
@@ -1978,7 +1978,7 @@ class CloudSpaceServiceApi(object):
             (data) = self.cloud_space_service_engage_cloud_space_with_http_info(body, project_id, id, **kwargs)  # noqa: E501
             return data
 
-    def cloud_space_service_engage_cloud_space_with_http_info(self, body: 'IdEngageBody', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
+    def cloud_space_service_engage_cloud_space_with_http_info(self, body: 'IdEngageBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1CloudSpaceEngagementResponse':  # noqa: E501
         """cloud_space_service_engage_cloud_space  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -1987,7 +1987,7 @@ class CloudSpaceServiceApi(object):
         >>> result = thread.get()
 
         :param async_req bool
-        :param IdEngageBody body: (required)
+        :param IdEngageBody1 body: (required)
         :param str project_id: (required)
         :param str id: (required)
         :return: V1CloudSpaceEngagementResponse

lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py

@@ -1106,10 +1106,10 @@ class ClusterServiceApi(object):
         :param str cluster_id: (required)
         :param str instance_type:
         :param int instance_count:
-        :param int capacity_block_duration_hours:
         :param datetime start_date:
         :param str timezone:
         :param str org_id:
+        :param int capacity_block_duration_days:
         :return: V1FindCapacityBlockOfferingResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1134,16 +1134,16 @@ class ClusterServiceApi(object):
         :param str cluster_id: (required)
         :param str instance_type:
         :param int instance_count:
-        :param int capacity_block_duration_hours:
         :param datetime start_date:
         :param str timezone:
         :param str org_id:
+        :param int capacity_block_duration_days:
         :return: V1FindCapacityBlockOfferingResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'capacity_block_duration_hours', 'start_date', 'timezone', 'org_id']  # noqa: E501
+        all_params = ['project_id', 'cluster_id', 'instance_type', 'instance_count', 'start_date', 'timezone', 'org_id', 'capacity_block_duration_days']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1180,14 +1180,14 @@ class ClusterServiceApi(object):
             query_params.append(('instanceType', params['instance_type']))  # noqa: E501
         if 'instance_count' in params:
             query_params.append(('instanceCount', params['instance_count']))  # noqa: E501
-        if 'capacity_block_duration_hours' in params:
-            query_params.append(('capacityBlockDurationHours', params['capacity_block_duration_hours']))  # noqa: E501
         if 'start_date' in params:
             query_params.append(('startDate', params['start_date']))  # noqa: E501
         if 'timezone' in params:
             query_params.append(('timezone', params['timezone']))  # noqa: E501
         if 'org_id' in params:
             query_params.append(('orgId', params['org_id']))  # noqa: E501
+        if 'capacity_block_duration_days' in params:
+            query_params.append(('capacityBlockDurationDays', params['capacity_block_duration_days']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py

@@ -140,6 +140,111 @@ class DeploymentTemplatesServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def deployment_templates_service_engage_deployment_template(self, body: 'IdEngageBody', id: 'str', **kwargs) -> 'V1DeploymentTemplateEngagementResponse':  # noqa: E501
+        """deployment_templates_service_engage_deployment_template  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.deployment_templates_service_engage_deployment_template(body, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdEngageBody body: (required)
+        :param str id: (required)
+        :return: V1DeploymentTemplateEngagementResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.deployment_templates_service_engage_deployment_template_with_http_info(body, id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.deployment_templates_service_engage_deployment_template_with_http_info(body, id, **kwargs)  # noqa: E501
+            return data
+
+    def deployment_templates_service_engage_deployment_template_with_http_info(self, body: 'IdEngageBody', id: 'str', **kwargs) -> 'V1DeploymentTemplateEngagementResponse':  # noqa: E501
+        """deployment_templates_service_engage_deployment_template  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.deployment_templates_service_engage_deployment_template_with_http_info(body, id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param IdEngageBody body: (required)
+        :param str id: (required)
+        :return: V1DeploymentTemplateEngagementResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method deployment_templates_service_engage_deployment_template" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `deployment_templates_service_engage_deployment_template`")  # noqa: E501
+        # verify the required parameter 'id' is set
+        if ('id' not in params or
+                params['id'] is None):
+            raise ValueError("Missing the required parameter `id` when calling `deployment_templates_service_engage_deployment_template`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'id' in params:
+            path_params['id'] = params['id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # 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(
+            '/v1/deployment-templates/{id}/engage', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1DeploymentTemplateEngagementResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def deployment_templates_service_get_deployment_template(self, id: 'str', **kwargs) -> 'V1DeploymentTemplate':  # noqa: E501
         """deployment_templates_service_get_deployment_template  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py

@@ -365,6 +365,7 @@ class JobsServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str id: (required)
+        :param bool delete_templates:
         :return: V1DeleteDeploymentResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -387,12 +388,13 @@ class JobsServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str id: (required)
+        :param bool delete_templates:
         :return: V1DeleteDeploymentResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'id']  # noqa: E501
+        all_params = ['project_id', 'id', 'delete_templates']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -425,6 +427,8 @@ class JobsServiceApi(object):
             path_params['id'] = params['id']  # noqa: E501
 
         query_params = []
+        if 'delete_templates' in params:
+            query_params.append(('deleteTemplates', params['delete_templates']))  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/models_store_api.py

@@ -28,7 +28,6 @@ import six
 from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
 
 if TYPE_CHECKING:
-    from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
 class ModelsStoreApi(object):
@@ -398,6 +397,119 @@ class ModelsStoreApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def models_store_create_model_version(self, body: 'ModelIdVersionsBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """models_store_create_model_version  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.models_store_create_model_version(body, project_id, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelIdVersionsBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :return: V1ModelVersionArchive
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.models_store_create_model_version_with_http_info(body, project_id, model_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_create_model_version_with_http_info(body, project_id, model_id, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_create_model_version_with_http_info(self, body: 'ModelIdVersionsBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """models_store_create_model_version  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.models_store_create_model_version_with_http_info(body, project_id, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelIdVersionsBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :return: V1ModelVersionArchive
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'model_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method models_store_create_model_version" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `models_store_create_model_version`")  # noqa: E501
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `models_store_create_model_version`")  # noqa: E501
+        # verify the required parameter 'model_id' is set
+        if ('model_id' not in params or
+                params['model_id'] is None):
+            raise ValueError("Missing the required parameter `model_id` when calling `models_store_create_model_version`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'model_id' in params:
+            path_params['modelId'] = params['model_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # 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(
+            '/v1/projects/{projectId}/models/{modelId}/versions', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ModelVersionArchive',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def models_store_create_multi_part_upload(self, body: 'VersionUploadsBody', project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1CreateMultiPartUploadResponse':  # noqa: E501
         """CreateMultiPartUpload initiates the multi-part upload of a file. Multiple requests can be sent to upload different files.  # noqa: E501
 
@@ -1280,6 +1392,7 @@ class ModelsStoreApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str name:
         :return: V1ListModelsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1301,12 +1414,13 @@ class ModelsStoreApi(object):
 
         :param async_req bool
         :param str project_id: (required)
+        :param str name:
         :return: V1ListModelsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id']  # noqa: E501
+        all_params = ['project_id', 'name']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1333,6 +1447,8 @@ class ModelsStoreApi(object):
             path_params['projectId'] = params['project_id']  # noqa: E501
 
         query_params = []
+        if 'name' in params:
+            query_params.append(('name', params['name']))  # noqa: E501
 
         header_params = {}