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

lightning_sdk/lightning_cloud/openapi/api/models_store_api.py

@@ -1197,6 +1197,9 @@ class ModelsStoreApi(object):
         :param str name:
         :param str version:
         :param str project_id:
+        :param str project_name:
+        :param str project_owner_name:
+        :param str id:
         :return: V1GetModelFilesResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1220,12 +1223,15 @@ class ModelsStoreApi(object):
         :param str name:
         :param str version:
         :param str project_id:
+        :param str project_name:
+        :param str project_owner_name:
+        :param str id:
         :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', 'id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1252,6 +1258,12 @@ 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
+        if 'id' in params:
+            query_params.append(('id', params['id']))  # noqa: E501
 
         header_params = {}
 
@@ -1282,6 +1294,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
 
@@ -1592,3 +1713,116 @@ class ModelsStoreApi(object):
             _preload_content=params.get('_preload_content', True),
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
+
+    def models_store_update_model_visibility(self, body: 'ModelIdVisibilityBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1UpdateModelVisibilityResponse':  # noqa: E501
+        """models_store_update_model_visibility  # 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_update_model_visibility(body, project_id, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelIdVisibilityBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :return: V1UpdateModelVisibilityResponse
+                 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_update_model_visibility_with_http_info(body, project_id, model_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.models_store_update_model_visibility_with_http_info(body, project_id, model_id, **kwargs)  # noqa: E501
+            return data
+
+    def models_store_update_model_visibility_with_http_info(self, body: 'ModelIdVisibilityBody', project_id: 'str', model_id: 'str', **kwargs) -> 'V1UpdateModelVisibilityResponse':  # noqa: E501
+        """models_store_update_model_visibility  # 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_update_model_visibility_with_http_info(body, project_id, model_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param ModelIdVisibilityBody body: (required)
+        :param str project_id: (required)
+        :param str model_id: (required)
+        :return: V1UpdateModelVisibilityResponse
+                 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_update_model_visibility" % 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_update_model_visibility`")  # 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_update_model_visibility`")  # 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_update_model_visibility`")  # 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}/visibility', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1UpdateModelVisibilityResponse',  # 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)

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -101,6 +101,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body impo
 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.model_id_visibility_body import ModelIdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
@@ -110,7 +111,6 @@ from lightning_sdk.lightning_cloud.openapi.models.orgs_id_body import OrgsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_agentmanagedendpoints_body import ProjectIdAgentmanagedendpointsBody
-from lightning_sdk.lightning_cloud.openapi.models.project_id_agentmanagedmodels_body import ProjectIdAgentmanagedmodelsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_agents_body import ProjectIdAgentsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_cloudspaces_body import ProjectIdCloudspacesBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_clusters_body import ProjectIdClustersBody
@@ -186,6 +186,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_billing_feature import V1Bi
 from lightning_sdk.lightning_cloud.openapi.models.v1_billing_period import V1BillingPeriod
 from lightning_sdk.lightning_cloud.openapi.models.v1_billing_subscription import V1BillingSubscription
 from lightning_sdk.lightning_cloud.openapi.models.v1_billing_tier import V1BillingTier
+from lightning_sdk.lightning_cloud.openapi.models.v1_body import V1Body
 from lightning_sdk.lightning_cloud.openapi.models.v1_build_spec import V1BuildSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_cpu_system_metrics import V1CPUSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance_switch_response import V1CancelCloudSpaceInstanceSwitchResponse
@@ -303,7 +304,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_logger_artifact_response import V1DeleteLoggerArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_endpoint_response import V1DeleteManagedEndpointResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_model_response import V1DeleteManagedModelResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_version_response import V1DeleteModelVersionResponse
@@ -329,6 +329,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_user_slurm_job_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state import V1DependencyCacheState
 from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import V1DependencyFileInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_api import V1DeploymentAPI
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event import V1DeploymentEvent
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event_type import V1DeploymentEventType
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_metrics import V1DeploymentMetrics
@@ -516,7 +517,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_endpoints_response import V1ListManagedEndpointsResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_models_response import V1ListManagedModelsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response import V1ListMembershipsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_response import V1ListModelVersionsResponse
@@ -562,6 +562,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_request im
 from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_response import V1MagicLinkLoginResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_managed_endpoint import V1ManagedEndpoint
 from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model import V1ManagedModel
+from lightning_sdk.lightning_cloud.openapi.models.v1_managed_model_abilities import V1ManagedModelAbilities
 from lightning_sdk.lightning_cloud.openapi.models.v1_membership import V1Membership
 from lightning_sdk.lightning_cloud.openapi.models.v1_message import V1Message
 from lightning_sdk.lightning_cloud.openapi.models.v1_message_author import V1MessageAuthor
@@ -617,6 +618,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_publish_cloud_space_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_published_cloud_space_response import V1PublishedCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_purchase_capacity_block_response import V1PurchaseCapacityBlockResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_python_dependency_info import V1PythonDependencyInfo
+from lightning_sdk.lightning_cloud.openapi.models.v1_query_param import V1QueryParam
 from lightning_sdk.lightning_cloud.openapi.models.v1_query_result import V1QueryResult
 from lightning_sdk.lightning_cloud.openapi.models.v1_query_result_row import V1QueryResultRow
 from lightning_sdk.lightning_cloud.openapi.models.v1_quest import V1Quest
@@ -635,6 +637,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_requ
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
+from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
 from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources
 from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice import V1ResponseChoice
 from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta import V1ResponseChoiceDelta
@@ -700,6 +703,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_snowflake_query_response import V1UpdateSnowflakeQueryResponse

lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py

@@ -41,6 +41,7 @@ class DeploymentsIdBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'apis': 'list[V1DeploymentAPI]',
         'autoscaling': 'V1AutoscalingSpec',
         'cloudspace_id': 'str',
         'created_at': 'datetime',
@@ -60,6 +61,7 @@ class DeploymentsIdBody(object):
     }
 
     attribute_map = {
+        'apis': 'apis',
         'autoscaling': 'autoscaling',
         'cloudspace_id': 'cloudspaceId',
         'created_at': 'createdAt',
@@ -78,8 +80,9 @@ class DeploymentsIdBody(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """DeploymentsIdBody - a model defined in Swagger"""  # noqa: E501
+        self._apis = None
         self._autoscaling = None
         self._cloudspace_id = None
         self._created_at = None
@@ -97,6 +100,8 @@ class DeploymentsIdBody(object):
         self._updated_at = None
         self._user_id = None
         self.discriminator = None
+        if apis is not None:
+            self.apis = apis
         if autoscaling is not None:
             self.autoscaling = autoscaling
         if cloudspace_id is not None:
@@ -130,6 +135,27 @@ class DeploymentsIdBody(object):
         if user_id is not None:
             self.user_id = user_id
 
+    @property
+    def apis(self) -> 'list[V1DeploymentAPI]':
+        """Gets the apis of this DeploymentsIdBody.  # noqa: E501
+
+
+        :return: The apis of this DeploymentsIdBody.  # noqa: E501
+        :rtype: list[V1DeploymentAPI]
+        """
+        return self._apis
+
+    @apis.setter
+    def apis(self, apis: 'list[V1DeploymentAPI]'):
+        """Sets the apis of this DeploymentsIdBody.
+
+
+        :param apis: The apis of this DeploymentsIdBody.  # noqa: E501
+        :type: list[V1DeploymentAPI]
+        """
+
+        self._apis = apis
+
     @property
     def autoscaling(self) -> 'V1AutoscalingSpec':
         """Gets the autoscaling of this DeploymentsIdBody.  # 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/model_id_visibility_body.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class ModelIdVisibilityBody(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'visibility': 'V1ResourceVisibility'
+    }
+
+    attribute_map = {
+        'visibility': 'visibility'
+    }
+
+    def __init__(self, visibility: 'V1ResourceVisibility' =None):  # noqa: E501
+        """ModelIdVisibilityBody - a model defined in Swagger"""  # noqa: E501
+        self._visibility = None
+        self.discriminator = None
+        if visibility is not None:
+            self.visibility = visibility
+
+    @property
+    def visibility(self) -> 'V1ResourceVisibility':
+        """Gets the visibility of this ModelIdVisibilityBody.  # noqa: E501
+
+
+        :return: The visibility of this ModelIdVisibilityBody.  # noqa: E501
+        :rtype: V1ResourceVisibility
+        """
+        return self._visibility
+
+    @visibility.setter
+    def visibility(self, visibility: 'V1ResourceVisibility'):
+        """Sets the visibility of this ModelIdVisibilityBody.
+
+
+        :param visibility: The visibility of this ModelIdVisibilityBody.  # noqa: E501
+        :type: V1ResourceVisibility
+        """
+
+        self._visibility = visibility
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(ModelIdVisibilityBody, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'ModelIdVisibilityBody') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, ModelIdVisibilityBody):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'ModelIdVisibilityBody') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

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