lightning-sdk 0.1.38__py3-none-any.whl → 0.1.39__py3-none-any.whl

lightning_sdk/__init__.py

@@ -27,5 +27,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.38"
+__version__ = "0.1.39"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/api/deployment_api.py

@@ -548,8 +548,6 @@ def to_spec(
         spot=spot,
         instance_name=_MACHINE_TO_COMPUTE_NAME[machine],
         readiness_probe=to_health_check(health_check),
-        skip_data_connections_setup=True,
-        skip_filesystem_setup=True,
     )
 
 

lightning_sdk/api/job_api.py

@@ -150,6 +150,8 @@ class JobApiV2:
         env: Optional[Dict[str, str]],
         image_credentials: Optional[str],
         cluster_auth: bool,
+        artifacts_local: Optional[str],
+        artifacts_remote: Optional[str],
     ) -> V1Job:
         env_vars = []
         if env is not None:
@@ -171,6 +173,8 @@ class JobApiV2:
             spot=interruptible,
             image_cluster_credentials=cluster_auth,
             image_secret_ref=image_credentials or "",
+            artifacts_source=artifacts_local or "",
+            artifacts_destination=artifacts_remote or "",
         )
         body = ProjectIdJobsBody(name=name, spec=spec)
 

lightning_sdk/api/teamspace_api.py

@@ -260,12 +260,14 @@ class TeamspaceApi:
         name: str,
         version: str,
         download_dir: Path,
-        teamspace_id: str,
+        teamspace_name: str,
+        teamspace_owner_name: str,
         progress_bar: bool = True,
     ) -> List[str]:
         return _download_model_files(
             client=self._client,
-            teamspace_id=teamspace_id,
+            teamspace_name=teamspace_name,
+            teamspace_owner_name=teamspace_owner_name,
             name=name,
             version=version,
             download_dir=download_dir,

lightning_sdk/api/utils.py

@@ -513,7 +513,8 @@ def _get_model_version(client: LightningClient, teamspace_id: str, name: str, ve
 
 def _download_model_files(
     client: LightningClient,
-    teamspace_id: str,
+    teamspace_name: str,
+    teamspace_owner_name: str,
     name: str,
     version: str,
     download_dir: Path,
@@ -521,7 +522,9 @@ def _download_model_files(
     num_workers: int = 20,
 ) -> List[str]:
     api = ModelsStoreApi(client.api_client)
-    response = api.models_store_get_model_files(project_id=teamspace_id, name=name, version=version)
+    response = api.models_store_get_model_files(
+        project_name=teamspace_name, project_owner_name=teamspace_owner_name, name=name, version=version
+    )
 
     pbar = None
     if progress_bar:
@@ -541,7 +544,7 @@ def _download_model_files(
             client=client,
             model_id=response.model_id,
             version=response.version,
-            teamspace_id=teamspace_id,
+            teamspace_id=response.project_id,
             remote_path=filepath,
             file_path=str(local_file),
             num_workers=num_workers,

lightning_sdk/cli/download.py

@@ -4,8 +4,8 @@ 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.models import download_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 
@@ -21,10 +21,8 @@ class _Downloads(_StudiosMenu):
             This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
           download_dir: 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_model(
+            name=name,
             download_dir=download_dir,
             progress_bar=True,
         )

lightning_sdk/cli/run.py

@@ -40,7 +40,19 @@ class _Run:
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cluster_auth: Whether to authenticate with the cluster to pull the image.
                 Required if the registry is part of a cluster provider (e.g. ECR).
+            artifacts_local: The path of inside the docker container, you want to persist images from.
+                CAUTION: When setting this to "/", it will effectively erase your container.
+                Only supported for jobs with a docker image compute environment.
+            artifacts_remote: The remote storage to persist your artifacts to.
+                Should be of format <CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>.
+                PATH_WITHIN_CONNECTION hereby is a path relative to the connection's root.
+                E.g. efs:data:some-path would result in an EFS connection named `data` and to the path `some-path`
+                within it.
+                Note that the connection needs to be added to the teamspace already in order for it to be found.
+                Only supported for jobs with a docker image compute environment.
         """
+        # TODO: the docstrings from artifacts_local and artifacts_remote don't show up completely,
+        # might need to switch to explicit cli definition
         self.job.__func__.__doc__ = docstr
 
     # TODO: sadly, fire displays both Optional[type] and Union[type, None] as Optional[Optional]
@@ -61,6 +73,8 @@ class _Run:
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> None:
         machine_enum = Machine(machine.upper())
         Job.run(
@@ -77,4 +91,6 @@ class _Run:
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )

lightning_sdk/cli/upload.py

@@ -9,8 +9,8 @@ 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.models import upload_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 
@@ -20,7 +20,7 @@ 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:
+    def model(self, name: str, path: str = ".", cloud_account: Optional[str] = None) -> None:
         """Upload a Model.
 
         Args:
@@ -29,14 +29,7 @@ class _Uploads(_StudiosMenu):
           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,
-        )
+        upload_model(name, path, cloud_account=cloud_account)
 
     def _resolve_studio(self, studio: Optional[str]) -> Studio:
         user = _get_authed_user()

lightning_sdk/job/base.py

@@ -52,6 +52,8 @@ class _BaseJob(ABC):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> "_BaseJob":
         from lightning_sdk.studio import Studio
 
@@ -89,12 +91,28 @@ class _BaseJob(ABC):
             if cluster_auth:
                 raise ValueError("cluster_auth is only supported when using a custom image")
 
+            if artifacts_local is not None or artifacts_remote is not None:
+                raise ValueError(
+                    "Specifying artifacts persistence is supported for docker images only. "
+                    "Other jobs will automatically persist artifacts to the teamspace distributed filesystem."
+                )
+
         else:
             if studio is not None:
                 raise RuntimeError(
                     "image and studio are mutually exclusive as both define the environment to run the job in"
                 )
 
+            # they either need to specified both or none of them
+            if bool(artifacts_local) != bool(artifacts_remote):
+                raise ValueError("Artifact persistence requires both artifacts_local and artifacts_remote to be set")
+
+            if artifacts_remote and len(artifacts_remote.split(":")) != 3:
+                raise ValueError(
+                    "Artifact persistence requires exactly three arguments separated by colon of kind "
+                    f"<CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>, got {artifacts_local}"
+                )
+
         inst = cls(name=name, teamspace=teamspace, org=org, user=user, _fetch_job=False)
         inst._submit(
             machine=machine,
@@ -106,6 +124,8 @@ class _BaseJob(ABC):
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
         return inst
 
@@ -121,6 +141,8 @@ class _BaseJob(ABC):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> None:
         """Submits a job and updates the internal _job attribute as well as the _name attribute."""
 

lightning_sdk/job/job.py

@@ -60,6 +60,8 @@ class Job(_BaseJob):
         interruptible: bool = False,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> "Job":
         ret_val = super().run(
             name=name,
@@ -75,6 +77,8 @@ class Job(_BaseJob):
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
         # required for typing with "Job"
         assert isinstance(ret_val, cls)
@@ -91,8 +95,10 @@ class Job(_BaseJob):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> None:
-        return self._internal_job._submit(
+        self._job = self._internal_job._submit(
             machine=machine,
             cluster=cluster,
             command=command,
@@ -102,7 +108,10 @@ class Job(_BaseJob):
             interruptible=interruptible,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
+        return self
 
     def stop(self) -> None:
         return self._internal_job.stop()

lightning_sdk/job/v1.py

@@ -69,6 +69,8 @@ class _JobV1(_BaseJob):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> None:
         if studio is None:
             raise ValueError("Studio is required for submitting jobs")
@@ -76,6 +78,9 @@ class _JobV1(_BaseJob):
         if image is not None or image_credentials is not None or cluster_auth:
             raise ValueError("Image is not supported for submitting jobs")
 
+        if artifacts_local is not None or artifacts_remote is not None:
+            raise ValueError("Specifying how to persist artifacts is not yet supported with jobs")
+
         if env is not None:
             raise ValueError("Environment variables are not supported for submitting jobs")
 

lightning_sdk/job/v2.py

@@ -36,6 +36,8 @@ class _JobV2(_BaseJob):
         cluster: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cluster_auth: bool = False,
+        artifacts_local: Optional[str] = None,
+        artifacts_remote: Optional[str] = None,
     ) -> None:
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -66,6 +68,8 @@ class _JobV2(_BaseJob):
             env=env,
             image_credentials=image_credentials,
             cluster_auth=cluster_auth,
+            artifacts_local=artifacts_local,
+            artifacts_remote=artifacts_remote,
         )
         self._job = submitted
         self._name = submitted.name
@@ -112,10 +116,18 @@ class _JobV2(_BaseJob):
 
     @property
     def artifact_path(self) -> Optional[str]:
+        if self._guaranteed_job.spec.image != "":
+            if self._guaranteed_job.spec.artifacts_destination != "":
+                splits = self._guaranteed_job.spec.artifacts_destination.split(":")
+                return f"/teamspace/{splits[0]}_connections/{splits[1]}/{splits[2]}"
+            return None
+
         return f"/teamspace/jobs/{self._guaranteed_job.name}/artifacts"
 
     @property
     def snapshot_path(self) -> Optional[str]:
+        if self._guaranteed_job.spec.image != "":
+            return None
         return f"/teamspace/jobs/{self._guaranteed_job.name}/snapshot"
 
     @property

lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py

@@ -716,6 +716,7 @@ class DataConnectionServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str cluster_id:
+        :param list[str] state:
         :return: V1ListDataConnectionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -738,12 +739,13 @@ class DataConnectionServiceApi(object):
         :param async_req bool
         :param str project_id: (required)
         :param str cluster_id:
+        :param list[str] state:
         :return: V1ListDataConnectionsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cluster_id']  # noqa: E501
+        all_params = ['project_id', 'cluster_id', 'state']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -772,6 +774,9 @@ class DataConnectionServiceApi(object):
         query_params = []
         if 'cluster_id' in params:
             query_params.append(('clusterId', params['cluster_id']))  # noqa: E501
+        if 'state' in params:
+            query_params.append(('state', params['state']))  # noqa: E501
+            collection_formats['state'] = 'multi'  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/models_store_api.py

@@ -1197,6 +1197,8 @@ class ModelsStoreApi(object):
         :param str name:
         :param str version:
         :param str project_id:
+        :param str project_name:
+        :param str project_owner_name:
         :return: V1GetModelFilesResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1220,12 +1222,14 @@ class ModelsStoreApi(object):
         :param str name:
         :param str version:
         :param str project_id:
+        :param str project_name:
+        :param str project_owner_name:
         :return: V1GetModelFilesResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['name', 'version', 'project_id']  # noqa: E501
+        all_params = ['name', 'version', 'project_id', 'project_name', 'project_owner_name']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1252,6 +1256,10 @@ class ModelsStoreApi(object):
             query_params.append(('version', params['version']))  # noqa: E501
         if 'project_id' in params:
             query_params.append(('projectId', params['project_id']))  # noqa: E501
+        if 'project_name' in params:
+            query_params.append(('projectName', params['project_name']))  # noqa: E501
+        if 'project_owner_name' in params:
+            query_params.append(('projectOwnerName', params['project_owner_name']))  # noqa: E501
 
         header_params = {}
 
@@ -1282,6 +1290,115 @@ class ModelsStoreApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def models_store_get_model_version(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """GetModelVersion used to get specific model version details  # 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_get_model_version(project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (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_get_model_version_with_http_info(project_id, model_id, version, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_get_model_version_with_http_info(project_id, model_id, version, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_get_model_version_with_http_info(self, project_id: 'str', model_id: 'str', version: 'str', **kwargs) -> 'V1ModelVersionArchive':  # noqa: E501
+        """GetModelVersion used to get specific model version details  # 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_get_model_version_with_http_info(project_id, model_id, version, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :param str version: (required)
+        :return: V1ModelVersionArchive
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'model_id', 'version']  # 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_get_model_version" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # 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_get_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_get_model_version`")  # noqa: E501
+        # verify the required parameter 'version' is set
+        if ('version' not in params or
+                params['version'] is None):
+            raise ValueError("Missing the required parameter `version` when calling `models_store_get_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
+        if 'version' in params:
+            path_params['version'] = params['version']  # 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(
+            '/v1/projects/{projectId}/models/{modelId}/versions/{version}', 'GET',
+            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_list_model_versions(self, project_id: 'str', model_id: 'str', **kwargs) -> 'V1ListModelVersionsResponse':  # noqa: E501
         """models_store_list_model_versions  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/models/id_start_body.py

@@ -41,19 +41,24 @@ class IdStartBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'compute_config': 'V1UserRequestedComputeConfig'
+        'compute_config': 'V1UserRequestedComputeConfig',
+        'is_forked': 'bool'
     }
 
     attribute_map = {
-        'compute_config': 'computeConfig'
+        'compute_config': 'computeConfig',
+        'is_forked': 'isForked'
     }
 
-    def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None):  # noqa: E501
+    def __init__(self, compute_config: 'V1UserRequestedComputeConfig' =None, is_forked: 'bool' =None):  # noqa: E501
         """IdStartBody - a model defined in Swagger"""  # noqa: E501
         self._compute_config = None
+        self._is_forked = None
         self.discriminator = None
         if compute_config is not None:
             self.compute_config = compute_config
+        if is_forked is not None:
+            self.is_forked = is_forked
 
     @property
     def compute_config(self) -> 'V1UserRequestedComputeConfig':
@@ -76,6 +81,27 @@ class IdStartBody(object):
 
         self._compute_config = compute_config
 
+    @property
+    def is_forked(self) -> 'bool':
+        """Gets the is_forked of this IdStartBody.  # noqa: E501
+
+
+        :return: The is_forked of this IdStartBody.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_forked
+
+    @is_forked.setter
+    def is_forked(self, is_forked: 'bool'):
+        """Sets the is_forked of this IdStartBody.
+
+
+        :param is_forked: The is_forked of this IdStartBody.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_forked = is_forked
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py

@@ -50,6 +50,7 @@ class ProjectIdCloudspacesBody(object):
         'disk_size': 'str',
         'display_name': 'str',
         'name': 'str',
+        'plugins': 'list[str]',
         'requested_run_duration_seconds': 'str',
         'seed_files': 'list[V1CloudSpaceSeedFile]',
         'spot': 'bool'
@@ -65,12 +66,13 @@ class ProjectIdCloudspacesBody(object):
         'disk_size': 'diskSize',
         'display_name': 'displayName',
         'name': 'name',
+        'plugins': 'plugins',
         'requested_run_duration_seconds': 'requestedRunDurationSeconds',
         'seed_files': 'seedFiles',
         'spot': 'spot'
     }
 
-    def __init__(self, can_download_source_code: 'bool' =None, cloud_space_instance_cpu_image_override: 'str' =None, cloud_space_instance_gpu_image_override: 'str' =None, cluster_id: 'str' =None, compute_name: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disk_size: 'str' =None, display_name: 'str' =None, name: 'str' =None, requested_run_duration_seconds: 'str' =None, seed_files: 'list[V1CloudSpaceSeedFile]' =None, spot: 'bool' =None):  # noqa: E501
+    def __init__(self, can_download_source_code: 'bool' =None, cloud_space_instance_cpu_image_override: 'str' =None, cloud_space_instance_gpu_image_override: 'str' =None, cluster_id: 'str' =None, compute_name: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disk_size: 'str' =None, display_name: 'str' =None, name: 'str' =None, plugins: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, seed_files: 'list[V1CloudSpaceSeedFile]' =None, spot: 'bool' =None):  # noqa: E501
         """ProjectIdCloudspacesBody - a model defined in Swagger"""  # noqa: E501
         self._can_download_source_code = None
         self._cloud_space_instance_cpu_image_override = None
@@ -81,6 +83,7 @@ class ProjectIdCloudspacesBody(object):
         self._disk_size = None
         self._display_name = None
         self._name = None
+        self._plugins = None
         self._requested_run_duration_seconds = None
         self._seed_files = None
         self._spot = None
@@ -103,6 +106,8 @@ class ProjectIdCloudspacesBody(object):
             self.display_name = display_name
         if name is not None:
             self.name = name
+        if plugins is not None:
+            self.plugins = plugins
         if requested_run_duration_seconds is not None:
             self.requested_run_duration_seconds = requested_run_duration_seconds
         if seed_files is not None:
@@ -299,6 +304,27 @@ class ProjectIdCloudspacesBody(object):
 
         self._name = name
 
+    @property
+    def plugins(self) -> 'list[str]':
+        """Gets the plugins of this ProjectIdCloudspacesBody.  # noqa: E501
+
+
+        :return: The plugins of this ProjectIdCloudspacesBody.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._plugins
+
+    @plugins.setter
+    def plugins(self, plugins: 'list[str]'):
+        """Sets the plugins of this ProjectIdCloudspacesBody.
+
+
+        :param plugins: The plugins of this ProjectIdCloudspacesBody.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._plugins = plugins
+
     @property
     def requested_run_duration_seconds(self) -> 'str':
         """Gets the requested_run_duration_seconds of this ProjectIdCloudspacesBody.  # noqa: E501