lightning-sdk 0.2.20__py3-none-any.whl → 0.2.21rc1__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.20"
+__version__ = "0.2.21.rc1"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/deployment_api.py

@@ -282,10 +282,11 @@ class DeploymentApi:
 
         requires_release = False
         requires_release |= apply_change(deployment.spec, "image", image)
+
         requires_release |= apply_change(deployment.spec, "entrypoint", entrypoint)
         requires_release |= apply_change(deployment.spec, "command", command)
         requires_release |= apply_change(deployment.spec, "env", to_env(env))
-        requires_release |= apply_change(deployment.spec, "readiness_probe", to_health_check(health_check))
+        requires_release |= apply_change(deployment.spec, "readiness_probe", to_health_check(health_check, False))
         requires_release |= apply_change(deployment.spec, "cluster_id", cloud_account)
         requires_release |= apply_change(deployment.spec, "spot", spot)
         requires_release |= apply_change(deployment.spec, "quantity", quantity)
@@ -524,8 +525,12 @@ def to_endpoint(
 
 
 def to_health_check(
-    health_check: Optional[Union[HttpHealthCheck, ExecHealthCheck]] = None
+    health_check: Optional[Union[HttpHealthCheck, ExecHealthCheck]] = None,
+    use_default: bool = True,
 ) -> Optional[V1JobHealthCheckConfig]:
+    if health_check is None and not use_default:
+        return None
+
     # Use Default health check if none is provided
     if not health_check:
         return V1JobHealthCheckConfig(

lightning_sdk/api/license_api.py

@@ -5,13 +5,13 @@ from urllib.parse import urlencode
 from lightning_sdk.lightning_cloud import env
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
-LICENSE_CODE = os.environ.get("LICENSE_CODE", "we843fiji89")
+LICENSE_CODE = os.environ.get("LICENSE_CODE", "d9s79g79ss")
 # https://lightning.ai/home?settings=licenses
-LICENSE_SIGNING_URL = f"{env.LIGHTNING_CLOUD_URL}/ai-hub?settings=licenses"
+LICENSE_SIGNING_URL = f"{env.LIGHTNING_CLOUD_URL}?settings=licenses"
 
 
 def generate_url_user_settings(redirect_to: str = LICENSE_SIGNING_URL) -> str:
-    params = urlencode({"redirectTo": redirect_to, "mode": "licenses", "okbhrt": LICENSE_CODE})
+    params = urlencode({"redirectTo": redirect_to, "okbhrt": LICENSE_CODE})
     return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
 
 

lightning_sdk/api/llm_api.py

@@ -1,7 +1,8 @@
+import asyncio
 import base64
 import json
 import os
-from typing import Dict, Generator, List, Optional, Union
+from typing import AsyncGenerator, Dict, Generator, List, Optional, Union
 
 from pip._vendor.urllib3 import HTTPResponse
 
@@ -110,6 +111,57 @@ class LLMApi:
             return result.result
         return self._stream_chat_response(result)
 
+    async def async_start_conversation(
+        self,
+        prompt: str,
+        system_prompt: Optional[str],
+        max_completion_tokens: int,
+        assistant_id: str,
+        images: Optional[List[str]] = None,
+        conversation_id: Optional[str] = None,
+        billing_project_id: Optional[str] = None,
+        name: Optional[str] = None,
+        metadata: Optional[Dict[str, str]] = None,
+        stream: bool = False,
+    ) -> Union[V1ConversationResponseChunk, AsyncGenerator[V1ConversationResponseChunk, None]]:
+        is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
+        body = {
+            "message": {
+                "author": {"role": "user"},
+                "content": [
+                    {"contentType": "text", "parts": [prompt]},
+                ],
+            },
+            "max_completion_tokens": max_completion_tokens,
+            "conversation_id": conversation_id,
+            "billing_project_id": billing_project_id,
+            "name": name,
+            "stream": stream,
+            "metadata": metadata or {},
+            "internal_conversation": is_internal_conversation,
+        }
+        if images:
+            for image in images:
+                url = image
+                if not image.startswith("http"):
+                    url = self._encode_image_bytes_to_data_url(image)
+
+                body["message"]["content"].append(
+                    {
+                        "contentType": "image",
+                        "parts": [url],
+                    }
+                )
+
+        if not stream:
+            thread = await asyncio.to_thread(
+                self._client.assistants_service_start_conversation, body, assistant_id, async_req=True
+            )
+            result = await asyncio.to_thread(thread.get)
+            return result.result
+
+        raise NotImplementedError("Streaming is not supported in this client.")
+
     def list_conversations(self, assistant_id: str) -> List[str]:
         result = self._client.assistants_service_list_conversations(assistant_id)
         return result.conversations

lightning_sdk/api/studio_api.py

@@ -154,6 +154,11 @@ class StudioApi:
     @backoff.on_exception(backoff.expo, AttributeError, max_tries=10)
     def _check_code_status_top_up_restore_finished(self, studio_id: str, teamspace_id: str) -> bool:
         """Retries checking the top_up_restore_finished value of the code status when there's an AttributeError."""
+        if (
+            self.get_studio_status(studio_id, teamspace_id) is None
+            or self.get_studio_status(studio_id, teamspace_id).in_use is None
+        ):
+            return False
         startup_status = self.get_studio_status(studio_id, teamspace_id).in_use.startup_status
         return startup_status and startup_status.top_up_restore_finished
 

lightning_sdk/deployment/deployment.py

@@ -276,8 +276,8 @@ class Deployment:
             cloud_account=cloud_account,
             machine=machine,
             image=image,
-            entrypoint=entrypoint or "",
-            command=command or "",
+            entrypoint=entrypoint,
+            command=command,
             ports=ports,
             custom_domain=custom_domain,
             auth=auth,
@@ -340,7 +340,7 @@ class Deployment:
         return None
 
     @property
-    def readiness_probe(self) -> Optional[Union[HttpHealthCheck, ExecHealthCheck]]:
+    def health_check(self) -> Optional[Union[HttpHealthCheck, ExecHealthCheck]]:
         """The health check to validate the replicas are ready to receive traffic."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
@@ -467,6 +467,20 @@ class Deployment:
             return self._deployment.spec.image
         return None
 
+    @property
+    def entrypoint(self) -> Optional[str]:
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return self._deployment.spec.entrypoint
+        return None
+
+    @property
+    def command(self) -> Optional[str]:
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return self._deployment.spec.command
+        return None
+
     @property
     def _session(self) -> Any:
         if self._request_session is None:

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -563,7 +563,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_response import V1GetUserResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_breakdown_response import V1GetUserStorageBreakdownResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_response import V1GetUserStorageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status

lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py

@@ -1420,6 +1420,8 @@ class EndpointServiceApi(object):
         :param list[str] ids:
         :param bool active_cloudspaces:
         :param bool active_jobs:
+        :param list[str] cloudspace_ids:
+        :param list[str] job_ids:
         :return: V1ListEndpointsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1447,12 +1449,14 @@ class EndpointServiceApi(object):
         :param list[str] ids:
         :param bool active_cloudspaces:
         :param bool active_jobs:
+        :param list[str] cloudspace_ids:
+        :param list[str] job_ids:
         :return: V1ListEndpointsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['project_id', 'cloudspace_id', 'auto_start', 'cluster_id', 'ids', 'active_cloudspaces', 'active_jobs']  # noqa: E501
+        all_params = ['project_id', 'cloudspace_id', 'auto_start', 'cluster_id', 'ids', 'active_cloudspaces', 'active_jobs', 'cloudspace_ids', 'job_ids']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1492,6 +1496,12 @@ class EndpointServiceApi(object):
             query_params.append(('activeCloudspaces', params['active_cloudspaces']))  # noqa: E501
         if 'active_jobs' in params:
             query_params.append(('activeJobs', params['active_jobs']))  # noqa: E501
+        if 'cloudspace_ids' in params:
+            query_params.append(('cloudspaceIds', params['cloudspace_ids']))  # noqa: E501
+            collection_formats['cloudspaceIds'] = 'multi'  # noqa: E501
+        if 'job_ids' in params:
+            query_params.append(('jobIds', params['job_ids']))  # noqa: E501
+            collection_formats['jobIds'] = 'multi'  # noqa: E501
 
         header_params = {}
 

lightning_sdk/lightning_cloud/openapi/api/user_service_api.py

@@ -706,91 +706,6 @@ class UserServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def user_service_get_user_storage(self, **kwargs) -> 'V1GetUserStorageResponse':  # noqa: E501
-        """user_service_get_user_storage  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.user_service_get_user_storage(async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool
-        :return: V1GetUserStorageResponse
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-        kwargs['_return_http_data_only'] = True
-        if kwargs.get('async_req'):
-            return self.user_service_get_user_storage_with_http_info(**kwargs)  # noqa: E501
-        else:
-            (data) = self.user_service_get_user_storage_with_http_info(**kwargs)  # noqa: E501
-            return data
-
-    def user_service_get_user_storage_with_http_info(self, **kwargs) -> 'V1GetUserStorageResponse':  # noqa: E501
-        """user_service_get_user_storage  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.user_service_get_user_storage_with_http_info(async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool
-        :return: V1GetUserStorageResponse
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-
-        all_params = []  # 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 user_service_get_user_storage" % key
-                )
-            params[key] = val
-        del params['kwargs']
-
-        collection_formats = {}
-
-        path_params = {}
-
-        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/users/storage', 'GET',
-            path_params,
-            query_params,
-            header_params,
-            body=body_params,
-            post_params=form_params,
-            files=local_var_files,
-            response_type='V1GetUserStorageResponse',  # 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 user_service_get_user_storage_breakdown(self, **kwargs) -> 'V1GetUserStorageBreakdownResponse':  # noqa: E501
         """user_service_get_user_storage_breakdown  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -518,7 +518,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_response import V1GetUserResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_breakdown_response import V1GetUserStorageBreakdownResponse
-from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_response import V1GetUserStorageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status

lightning_sdk/lightning_cloud/openapi/models/update.py

@@ -43,8 +43,11 @@ class Update(object):
     swagger_types = {
         'allowed_machines': 'list[str]',
         'default_machine': 'str',
+        'description': 'str',
         'disabled': 'bool',
         'environment_type': 'V1CloudSpaceEnvironmentType',
+        'icon': 'str',
+        'initial_setup_script_text': 'str',
         'machine_image_version': 'str',
         'name': 'str',
         'org_id': 'str',
@@ -55,8 +58,11 @@ class Update(object):
     attribute_map = {
         'allowed_machines': 'allowedMachines',
         'default_machine': 'defaultMachine',
+        'description': 'description',
         'disabled': 'disabled',
         'environment_type': 'environmentType',
+        'icon': 'icon',
+        'initial_setup_script_text': 'initialSetupScriptText',
         'machine_image_version': 'machineImageVersion',
         'name': 'name',
         'org_id': 'orgId',
@@ -64,12 +70,15 @@ class Update(object):
         'setup_script_text': 'setupScriptText'
     }
 
-    def __init__(self, allowed_machines: 'list[str]' =None, default_machine: 'str' =None, disabled: 'bool' =None, environment_type: 'V1CloudSpaceEnvironmentType' =None, machine_image_version: 'str' =None, name: 'str' =None, org_id: 'str' =None, plugins: 'list[str]' =None, setup_script_text: 'str' =None):  # noqa: E501
+    def __init__(self, allowed_machines: 'list[str]' =None, default_machine: 'str' =None, description: 'str' =None, disabled: 'bool' =None, environment_type: 'V1CloudSpaceEnvironmentType' =None, icon: 'str' =None, initial_setup_script_text: 'str' =None, machine_image_version: 'str' =None, name: 'str' =None, org_id: 'str' =None, plugins: 'list[str]' =None, setup_script_text: 'str' =None):  # noqa: E501
         """Update - a model defined in Swagger"""  # noqa: E501
         self._allowed_machines = None
         self._default_machine = None
+        self._description = None
         self._disabled = None
         self._environment_type = None
+        self._icon = None
+        self._initial_setup_script_text = None
         self._machine_image_version = None
         self._name = None
         self._org_id = None
@@ -80,10 +89,16 @@ class Update(object):
             self.allowed_machines = allowed_machines
         if default_machine is not None:
             self.default_machine = default_machine
+        if description is not None:
+            self.description = description
         if disabled is not None:
             self.disabled = disabled
         if environment_type is not None:
             self.environment_type = environment_type
+        if icon is not None:
+            self.icon = icon
+        if initial_setup_script_text is not None:
+            self.initial_setup_script_text = initial_setup_script_text
         if machine_image_version is not None:
             self.machine_image_version = machine_image_version
         if name is not None:
@@ -137,6 +152,27 @@ class Update(object):
 
         self._default_machine = default_machine
 
+    @property
+    def description(self) -> 'str':
+        """Gets the description of this Update.  # noqa: E501
+
+
+        :return: The description of this Update.  # noqa: E501
+        :rtype: str
+        """
+        return self._description
+
+    @description.setter
+    def description(self, description: 'str'):
+        """Sets the description of this Update.
+
+
+        :param description: The description of this Update.  # noqa: E501
+        :type: str
+        """
+
+        self._description = description
+
     @property
     def disabled(self) -> 'bool':
         """Gets the disabled of this Update.  # noqa: E501
@@ -179,6 +215,48 @@ class Update(object):
 
         self._environment_type = environment_type
 
+    @property
+    def icon(self) -> 'str':
+        """Gets the icon of this Update.  # noqa: E501
+
+
+        :return: The icon of this Update.  # noqa: E501
+        :rtype: str
+        """
+        return self._icon
+
+    @icon.setter
+    def icon(self, icon: 'str'):
+        """Sets the icon of this Update.
+
+
+        :param icon: The icon of this Update.  # noqa: E501
+        :type: str
+        """
+
+        self._icon = icon
+
+    @property
+    def initial_setup_script_text(self) -> 'str':
+        """Gets the initial_setup_script_text of this Update.  # noqa: E501
+
+
+        :return: The initial_setup_script_text of this Update.  # noqa: E501
+        :rtype: str
+        """
+        return self._initial_setup_script_text
+
+    @initial_setup_script_text.setter
+    def initial_setup_script_text(self, initial_setup_script_text: 'str'):
+        """Sets the initial_setup_script_text of this Update.
+
+
+        :param initial_setup_script_text: The initial_setup_script_text of this Update.  # noqa: E501
+        :type: str
+        """
+
+        self._initial_setup_script_text = initial_setup_script_text
+
     @property
     def machine_image_version(self) -> 'str':
         """Gets the machine_image_version of this Update.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py

@@ -43,7 +43,9 @@ class V1CloudSpaceEnvironmentTemplate(object):
     swagger_types = {
         'config': 'V1CloudSpaceEnvironmentTemplateConfig',
         'created_at': 'datetime',
+        'description': 'str',
         'disabled': 'bool',
+        'icon': 'str',
         'id': 'str',
         'managed': 'bool',
         'managed_id': 'str',
@@ -56,7 +58,9 @@ class V1CloudSpaceEnvironmentTemplate(object):
     attribute_map = {
         'config': 'config',
         'created_at': 'createdAt',
+        'description': 'description',
         'disabled': 'disabled',
+        'icon': 'icon',
         'id': 'id',
         'managed': 'managed',
         'managed_id': 'managedId',
@@ -66,11 +70,13 @@ class V1CloudSpaceEnvironmentTemplate(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, config: 'V1CloudSpaceEnvironmentTemplateConfig' =None, created_at: 'datetime' =None, disabled: 'bool' =None, id: 'str' =None, managed: 'bool' =None, managed_id: 'str' =None, name: 'str' =None, org_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, config: 'V1CloudSpaceEnvironmentTemplateConfig' =None, created_at: 'datetime' =None, description: 'str' =None, disabled: 'bool' =None, icon: 'str' =None, id: 'str' =None, managed: 'bool' =None, managed_id: 'str' =None, name: 'str' =None, org_id: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1CloudSpaceEnvironmentTemplate - a model defined in Swagger"""  # noqa: E501
         self._config = None
         self._created_at = None
+        self._description = None
         self._disabled = None
+        self._icon = None
         self._id = None
         self._managed = None
         self._managed_id = None
@@ -83,8 +89,12 @@ class V1CloudSpaceEnvironmentTemplate(object):
             self.config = config
         if created_at is not None:
             self.created_at = created_at
+        if description is not None:
+            self.description = description
         if disabled is not None:
             self.disabled = disabled
+        if icon is not None:
+            self.icon = icon
         if id is not None:
             self.id = id
         if managed is not None:
@@ -142,6 +152,27 @@ class V1CloudSpaceEnvironmentTemplate(object):
 
         self._created_at = created_at
 
+    @property
+    def description(self) -> 'str':
+        """Gets the description of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501
+
+
+        :return: The description of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501
+        :rtype: str
+        """
+        return self._description
+
+    @description.setter
+    def description(self, description: 'str'):
+        """Sets the description of this V1CloudSpaceEnvironmentTemplate.
+
+
+        :param description: The description of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501
+        :type: str
+        """
+
+        self._description = description
+
     @property
     def disabled(self) -> 'bool':
         """Gets the disabled of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501
@@ -163,6 +194,27 @@ class V1CloudSpaceEnvironmentTemplate(object):
 
         self._disabled = disabled
 
+    @property
+    def icon(self) -> 'str':
+        """Gets the icon of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501
+
+
+        :return: The icon of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501
+        :rtype: str
+        """
+        return self._icon
+
+    @icon.setter
+    def icon(self, icon: 'str'):
+        """Sets the icon of this V1CloudSpaceEnvironmentTemplate.
+
+
+        :param icon: The icon of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501
+        :type: str
+        """
+
+        self._icon = icon
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1CloudSpaceEnvironmentTemplate.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py

@@ -44,6 +44,7 @@ class V1CloudSpaceEnvironmentTemplateConfig(object):
         'allowed_machines': 'list[str]',
         'default_machine': 'str',
         'environment_type': 'V1CloudSpaceEnvironmentType',
+        'initial_setup_script_text': 'str',
         'machine_image_version': 'str',
         'plugins': 'list[str]',
         'setup_script_text': 'str'
@@ -53,16 +54,18 @@ class V1CloudSpaceEnvironmentTemplateConfig(object):
         'allowed_machines': 'allowedMachines',
         'default_machine': 'defaultMachine',
         'environment_type': 'environmentType',
+        'initial_setup_script_text': 'initialSetupScriptText',
         'machine_image_version': 'machineImageVersion',
         'plugins': 'plugins',
         'setup_script_text': 'setupScriptText'
     }
 
-    def __init__(self, allowed_machines: 'list[str]' =None, default_machine: 'str' =None, environment_type: 'V1CloudSpaceEnvironmentType' =None, machine_image_version: 'str' =None, plugins: 'list[str]' =None, setup_script_text: 'str' =None):  # noqa: E501
+    def __init__(self, allowed_machines: 'list[str]' =None, default_machine: 'str' =None, environment_type: 'V1CloudSpaceEnvironmentType' =None, initial_setup_script_text: 'str' =None, machine_image_version: 'str' =None, plugins: 'list[str]' =None, setup_script_text: 'str' =None):  # noqa: E501
         """V1CloudSpaceEnvironmentTemplateConfig - a model defined in Swagger"""  # noqa: E501
         self._allowed_machines = None
         self._default_machine = None
         self._environment_type = None
+        self._initial_setup_script_text = None
         self._machine_image_version = None
         self._plugins = None
         self._setup_script_text = None
@@ -73,6 +76,8 @@ class V1CloudSpaceEnvironmentTemplateConfig(object):
             self.default_machine = default_machine
         if environment_type is not None:
             self.environment_type = environment_type
+        if initial_setup_script_text is not None:
+            self.initial_setup_script_text = initial_setup_script_text
         if machine_image_version is not None:
             self.machine_image_version = machine_image_version
         if plugins is not None:
@@ -143,6 +148,27 @@ class V1CloudSpaceEnvironmentTemplateConfig(object):
 
         self._environment_type = environment_type
 
+    @property
+    def initial_setup_script_text(self) -> 'str':
+        """Gets the initial_setup_script_text of this V1CloudSpaceEnvironmentTemplateConfig.  # noqa: E501
+
+
+        :return: The initial_setup_script_text of this V1CloudSpaceEnvironmentTemplateConfig.  # noqa: E501
+        :rtype: str
+        """
+        return self._initial_setup_script_text
+
+    @initial_setup_script_text.setter
+    def initial_setup_script_text(self, initial_setup_script_text: 'str'):
+        """Sets the initial_setup_script_text of this V1CloudSpaceEnvironmentTemplateConfig.
+
+
+        :param initial_setup_script_text: The initial_setup_script_text of this V1CloudSpaceEnvironmentTemplateConfig.  # noqa: E501
+        :type: str
+        """
+
+        self._initial_setup_script_text = initial_setup_script_text
+
     @property
     def machine_image_version(self) -> 'str':
         """Gets the machine_image_version of this V1CloudSpaceEnvironmentTemplateConfig.  # noqa: E501