lightning-sdk 2025.12.5__py3-none-any.whl → 2025.12.9__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_job_health_check_config.py

@@ -46,7 +46,9 @@ class V1JobHealthCheckConfig(object):
         'http_get': 'V1HealthCheckHttpGet',
         'initial_delay_seconds': 'int',
         'interval_seconds': 'int',
-        'timeout_seconds': 'int'
+        'restart_policy': 'str',
+        'timeout_seconds': 'int',
+        'use_default_check': 'bool'
     }
 
     attribute_map = {
@@ -55,17 +57,21 @@ class V1JobHealthCheckConfig(object):
         'http_get': 'httpGet',
         'initial_delay_seconds': 'initialDelaySeconds',
         'interval_seconds': 'intervalSeconds',
-        'timeout_seconds': 'timeoutSeconds'
+        'restart_policy': 'restartPolicy',
+        'timeout_seconds': 'timeoutSeconds',
+        'use_default_check': 'useDefaultCheck'
     }
 
-    def __init__(self, _exec: 'V1HealthCheckExec' =None, failure_threshold: 'int' =None, http_get: 'V1HealthCheckHttpGet' =None, initial_delay_seconds: 'int' =None, interval_seconds: 'int' =None, timeout_seconds: 'int' =None):  # noqa: E501
+    def __init__(self, _exec: 'V1HealthCheckExec' =None, failure_threshold: 'int' =None, http_get: 'V1HealthCheckHttpGet' =None, initial_delay_seconds: 'int' =None, interval_seconds: 'int' =None, restart_policy: 'str' =None, timeout_seconds: 'int' =None, use_default_check: 'bool' =None):  # noqa: E501
         """V1JobHealthCheckConfig - a model defined in Swagger"""  # noqa: E501
         self.__exec = None
         self._failure_threshold = None
         self._http_get = None
         self._initial_delay_seconds = None
         self._interval_seconds = None
+        self._restart_policy = None
         self._timeout_seconds = None
+        self._use_default_check = None
         self.discriminator = None
         if _exec is not None:
             self._exec = _exec
@@ -77,8 +83,12 @@ class V1JobHealthCheckConfig(object):
             self.initial_delay_seconds = initial_delay_seconds
         if interval_seconds is not None:
             self.interval_seconds = interval_seconds
+        if restart_policy is not None:
+            self.restart_policy = restart_policy
         if timeout_seconds is not None:
             self.timeout_seconds = timeout_seconds
+        if use_default_check is not None:
+            self.use_default_check = use_default_check
 
     @property
     def _exec(self) -> 'V1HealthCheckExec':
@@ -185,6 +195,27 @@ class V1JobHealthCheckConfig(object):
 
         self._interval_seconds = interval_seconds
 
+    @property
+    def restart_policy(self) -> 'str':
+        """Gets the restart_policy of this V1JobHealthCheckConfig.  # noqa: E501
+
+
+        :return: The restart_policy of this V1JobHealthCheckConfig.  # noqa: E501
+        :rtype: str
+        """
+        return self._restart_policy
+
+    @restart_policy.setter
+    def restart_policy(self, restart_policy: 'str'):
+        """Sets the restart_policy of this V1JobHealthCheckConfig.
+
+
+        :param restart_policy: The restart_policy of this V1JobHealthCheckConfig.  # noqa: E501
+        :type: str
+        """
+
+        self._restart_policy = restart_policy
+
     @property
     def timeout_seconds(self) -> 'int':
         """Gets the timeout_seconds of this V1JobHealthCheckConfig.  # noqa: E501
@@ -206,6 +237,27 @@ class V1JobHealthCheckConfig(object):
 
         self._timeout_seconds = timeout_seconds
 
+    @property
+    def use_default_check(self) -> 'bool':
+        """Gets the use_default_check of this V1JobHealthCheckConfig.  # noqa: E501
+
+
+        :return: The use_default_check of this V1JobHealthCheckConfig.  # noqa: E501
+        :rtype: bool
+        """
+        return self._use_default_check
+
+    @use_default_check.setter
+    def use_default_check(self, use_default_check: 'bool'):
+        """Sets the use_default_check of this V1JobHealthCheckConfig.
+
+
+        :param use_default_check: The use_default_check of this V1JobHealthCheckConfig.  # noqa: E501
+        :type: bool
+        """
+
+        self._use_default_check = use_default_check
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -68,6 +68,8 @@ class V1JobSpec(object):
         'resources': 'V1Resources',
         'restart_policy': 'str',
         'run_id': 'str',
+        'shutdown_delay_seconds': 'int',
+        'shutdown_timeout_seconds': 'int',
         'spot': 'bool',
         'use_tls': 'bool',
         'volumes': 'list[V1Volume]'
@@ -101,12 +103,14 @@ class V1JobSpec(object):
         'resources': 'resources',
         'restart_policy': 'restartPolicy',
         'run_id': 'runId',
+        'shutdown_delay_seconds': 'shutdownDelaySeconds',
+        'shutdown_timeout_seconds': 'shutdownTimeoutSeconds',
         'spot': 'spot',
         'use_tls': 'useTls',
         'volumes': 'volumes'
     }
 
-    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, disable_artifacts_persistence: 'bool' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, machine_image_version: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, pipeline_id: 'str' =None, quantity: 'int' =None, rank: 'int' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, use_tls: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
+    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, disable_artifacts_persistence: 'bool' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, machine_image_version: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, pipeline_id: 'str' =None, quantity: 'int' =None, rank: 'int' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, shutdown_delay_seconds: 'int' =None, shutdown_timeout_seconds: 'int' =None, spot: 'bool' =None, use_tls: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
         self._artifacts_destination = None
         self._artifacts_source = None
@@ -135,6 +139,8 @@ class V1JobSpec(object):
         self._resources = None
         self._restart_policy = None
         self._run_id = None
+        self._shutdown_delay_seconds = None
+        self._shutdown_timeout_seconds = None
         self._spot = None
         self._use_tls = None
         self._volumes = None
@@ -193,6 +199,10 @@ class V1JobSpec(object):
             self.restart_policy = restart_policy
         if run_id is not None:
             self.run_id = run_id
+        if shutdown_delay_seconds is not None:
+            self.shutdown_delay_seconds = shutdown_delay_seconds
+        if shutdown_timeout_seconds is not None:
+            self.shutdown_timeout_seconds = shutdown_timeout_seconds
         if spot is not None:
             self.spot = spot
         if use_tls is not None:
@@ -225,6 +235,7 @@ class V1JobSpec(object):
     def artifacts_source(self) -> 'str':
         """Gets the artifacts_source of this V1JobSpec.  # noqa: E501
 
+        Which folder artifacts should be persisted from  # noqa: E501
 
         :return: The artifacts_source of this V1JobSpec.  # noqa: E501
         :rtype: str
@@ -235,6 +246,7 @@ class V1JobSpec(object):
     def artifacts_source(self, artifacts_source: 'str'):
         """Sets the artifacts_source of this V1JobSpec.
 
+        Which folder artifacts should be persisted from  # noqa: E501
 
         :param artifacts_source: The artifacts_source of this V1JobSpec.  # noqa: E501
         :type: str
@@ -288,6 +300,7 @@ class V1JobSpec(object):
     def command(self) -> 'str':
         """Gets the command of this V1JobSpec.  # noqa: E501
 
+        Command to run on the machine  # noqa: E501
 
         :return: The command of this V1JobSpec.  # noqa: E501
         :rtype: str
@@ -298,6 +311,7 @@ class V1JobSpec(object):
     def command(self, command: 'str'):
         """Sets the command of this V1JobSpec.
 
+        Command to run on the machine  # noqa: E501
 
         :param command: The command of this V1JobSpec.  # noqa: E501
         :type: str
@@ -477,6 +491,7 @@ class V1JobSpec(object):
     def instance_name(self) -> 'str':
         """Gets the instance_name of this V1JobSpec.  # noqa: E501
 
+        User requested instance name (nickname such as cpu-4, etc.)  # noqa: E501
 
         :return: The instance_name of this V1JobSpec.  # noqa: E501
         :rtype: str
@@ -487,6 +502,7 @@ class V1JobSpec(object):
     def instance_name(self, instance_name: 'str'):
         """Sets the instance_name of this V1JobSpec.
 
+        User requested instance name (nickname such as cpu-4, etc.)  # noqa: E501
 
         :param instance_name: The instance_name of this V1JobSpec.  # noqa: E501
         :type: str
@@ -767,6 +783,48 @@ class V1JobSpec(object):
 
         self._run_id = run_id
 
+    @property
+    def shutdown_delay_seconds(self) -> 'int':
+        """Gets the shutdown_delay_seconds of this V1JobSpec.  # noqa: E501
+
+
+        :return: The shutdown_delay_seconds of this V1JobSpec.  # noqa: E501
+        :rtype: int
+        """
+        return self._shutdown_delay_seconds
+
+    @shutdown_delay_seconds.setter
+    def shutdown_delay_seconds(self, shutdown_delay_seconds: 'int'):
+        """Sets the shutdown_delay_seconds of this V1JobSpec.
+
+
+        :param shutdown_delay_seconds: The shutdown_delay_seconds of this V1JobSpec.  # noqa: E501
+        :type: int
+        """
+
+        self._shutdown_delay_seconds = shutdown_delay_seconds
+
+    @property
+    def shutdown_timeout_seconds(self) -> 'int':
+        """Gets the shutdown_timeout_seconds of this V1JobSpec.  # noqa: E501
+
+
+        :return: The shutdown_timeout_seconds of this V1JobSpec.  # noqa: E501
+        :rtype: int
+        """
+        return self._shutdown_timeout_seconds
+
+    @shutdown_timeout_seconds.setter
+    def shutdown_timeout_seconds(self, shutdown_timeout_seconds: 'int'):
+        """Sets the shutdown_timeout_seconds of this V1JobSpec.
+
+
+        :param shutdown_timeout_seconds: The shutdown_timeout_seconds of this V1JobSpec.  # noqa: E501
+        :type: int
+        """
+
+        self._shutdown_timeout_seconds = shutdown_timeout_seconds
+
     @property
     def spot(self) -> 'bool':
         """Gets the spot of this V1JobSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py

@@ -41,6 +41,7 @@ class V1KubernetesDirectV1(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'automate_agent_installation': 'bool',
         'aws': 'V1KubernetesAWSConfig',
         'grafana_dashboard_url': 'str',
         'grafana_namespace': 'str',
@@ -50,11 +51,13 @@ class V1KubernetesDirectV1(object):
         'incident_slack_notifiers': 'list[V1SlackNotifier]',
         'kubeconfig': 'str',
         'kubeconfig_elevated': 'str',
+        'prometheus_url': 'str',
         'settings': 'V1KubernetesDirectSettingsV1',
         'skip_user_label_injection': 'bool'
     }
 
     attribute_map = {
+        'automate_agent_installation': 'automateAgentInstallation',
         'aws': 'aws',
         'grafana_dashboard_url': 'grafanaDashboardUrl',
         'grafana_namespace': 'grafanaNamespace',
@@ -64,12 +67,14 @@ class V1KubernetesDirectV1(object):
         'incident_slack_notifiers': 'incidentSlackNotifiers',
         'kubeconfig': 'kubeconfig',
         'kubeconfig_elevated': 'kubeconfigElevated',
+        'prometheus_url': 'prometheusUrl',
         'settings': 'settings',
         'skip_user_label_injection': 'skipUserLabelInjection'
     }
 
-    def __init__(self, aws: 'V1KubernetesAWSConfig' =None, grafana_dashboard_url: 'str' =None, grafana_namespace: 'str' =None, grafana_service_name: 'str' =None, grafana_service_port: 'str' =None, incident_historical_scan_complete: 'bool' =None, incident_slack_notifiers: 'list[V1SlackNotifier]' =None, kubeconfig: 'str' =None, kubeconfig_elevated: 'str' =None, settings: 'V1KubernetesDirectSettingsV1' =None, skip_user_label_injection: 'bool' =None):  # noqa: E501
+    def __init__(self, automate_agent_installation: 'bool' =None, aws: 'V1KubernetesAWSConfig' =None, grafana_dashboard_url: 'str' =None, grafana_namespace: 'str' =None, grafana_service_name: 'str' =None, grafana_service_port: 'str' =None, incident_historical_scan_complete: 'bool' =None, incident_slack_notifiers: 'list[V1SlackNotifier]' =None, kubeconfig: 'str' =None, kubeconfig_elevated: 'str' =None, prometheus_url: 'str' =None, settings: 'V1KubernetesDirectSettingsV1' =None, skip_user_label_injection: 'bool' =None):  # noqa: E501
         """V1KubernetesDirectV1 - a model defined in Swagger"""  # noqa: E501
+        self._automate_agent_installation = None
         self._aws = None
         self._grafana_dashboard_url = None
         self._grafana_namespace = None
@@ -79,9 +84,12 @@ class V1KubernetesDirectV1(object):
         self._incident_slack_notifiers = None
         self._kubeconfig = None
         self._kubeconfig_elevated = None
+        self._prometheus_url = None
         self._settings = None
         self._skip_user_label_injection = None
         self.discriminator = None
+        if automate_agent_installation is not None:
+            self.automate_agent_installation = automate_agent_installation
         if aws is not None:
             self.aws = aws
         if grafana_dashboard_url is not None:
@@ -100,11 +108,34 @@ class V1KubernetesDirectV1(object):
             self.kubeconfig = kubeconfig
         if kubeconfig_elevated is not None:
             self.kubeconfig_elevated = kubeconfig_elevated
+        if prometheus_url is not None:
+            self.prometheus_url = prometheus_url
         if settings is not None:
             self.settings = settings
         if skip_user_label_injection is not None:
             self.skip_user_label_injection = skip_user_label_injection
 
+    @property
+    def automate_agent_installation(self) -> 'bool':
+        """Gets the automate_agent_installation of this V1KubernetesDirectV1.  # noqa: E501
+
+
+        :return: The automate_agent_installation of this V1KubernetesDirectV1.  # noqa: E501
+        :rtype: bool
+        """
+        return self._automate_agent_installation
+
+    @automate_agent_installation.setter
+    def automate_agent_installation(self, automate_agent_installation: 'bool'):
+        """Sets the automate_agent_installation of this V1KubernetesDirectV1.
+
+
+        :param automate_agent_installation: The automate_agent_installation of this V1KubernetesDirectV1.  # noqa: E501
+        :type: bool
+        """
+
+        self._automate_agent_installation = automate_agent_installation
+
     @property
     def aws(self) -> 'V1KubernetesAWSConfig':
         """Gets the aws of this V1KubernetesDirectV1.  # noqa: E501
@@ -294,6 +325,27 @@ class V1KubernetesDirectV1(object):
 
         self._kubeconfig_elevated = kubeconfig_elevated
 
+    @property
+    def prometheus_url(self) -> 'str':
+        """Gets the prometheus_url of this V1KubernetesDirectV1.  # noqa: E501
+
+
+        :return: The prometheus_url of this V1KubernetesDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._prometheus_url
+
+    @prometheus_url.setter
+    def prometheus_url(self, prometheus_url: 'str'):
+        """Sets the prometheus_url of this V1KubernetesDirectV1.
+
+
+        :param prometheus_url: The prometheus_url of this V1KubernetesDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._prometheus_url = prometheus_url
+
     @property
     def settings(self) -> 'V1KubernetesDirectSettingsV1':
         """Gets the settings of this V1KubernetesDirectV1.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lit_dataset_version_archive.py

@@ -371,6 +371,7 @@ class V1LitDatasetVersionArchive(object):
     def version(self) -> 'str':
         """Gets the version of this V1LitDatasetVersionArchive.  # noqa: E501
 
+        v1, v2, v3, etc.  # noqa: E501
 
         :return: The version of this V1LitDatasetVersionArchive.  # noqa: E501
         :rtype: str
@@ -381,6 +382,7 @@ class V1LitDatasetVersionArchive(object):
     def version(self, version: 'str'):
         """Sets the version of this V1LitDatasetVersionArchive.
 
+        v1, v2, v3, etc.  # noqa: E501
 
         :param version: The version of this V1LitDatasetVersionArchive.  # noqa: E501
         :type: str

lightning_sdk/lightning_cloud/openapi/models/v1_machine.py

@@ -42,13 +42,16 @@ class V1Machine(object):
     """
     swagger_types = {
         'address': 'str',
+        'cloud_space_name': 'str',
         'cluster_id': 'str',
         'created_at': 'datetime',
+        'display_name': 'str',
         'id': 'str',
         'instance_type': 'str',
         'management_api_url': 'str',
         'name': 'str',
         'org_id': 'str',
+        'project_id': 'str',
         'provider': 'str',
         'provider_instance_id': 'str',
         'provider_region': 'str',
@@ -65,13 +68,16 @@ class V1Machine(object):
 
     attribute_map = {
         'address': 'address',
+        'cloud_space_name': 'cloudSpaceName',
         'cluster_id': 'clusterId',
         'created_at': 'createdAt',
+        'display_name': 'displayName',
         'id': 'id',
         'instance_type': 'instanceType',
         'management_api_url': 'managementApiUrl',
         'name': 'name',
         'org_id': 'orgId',
+        'project_id': 'projectId',
         'provider': 'provider',
         'provider_instance_id': 'providerInstanceId',
         'provider_region': 'providerRegion',
@@ -86,16 +92,19 @@ class V1Machine(object):
         'warning_message': 'warningMessage'
     }
 
-    def __init__(self, address: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, instance_type: 'str' =None, management_api_url: 'str' =None, name: 'str' =None, org_id: 'str' =None, provider: 'str' =None, provider_instance_id: 'str' =None, provider_region: 'str' =None, provisioning_method: 'str' =None, ready_at: 'datetime' =None, resource_id: 'str' =None, resource_type: 'str' =None, resources: 'V1Resources' =None, status: 'str' =None, unschedulable: 'bool' =None, updated_at: 'datetime' =None, warning_message: 'str' =None):  # noqa: E501
+    def __init__(self, address: 'str' =None, cloud_space_name: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, display_name: 'str' =None, id: 'str' =None, instance_type: 'str' =None, management_api_url: 'str' =None, name: 'str' =None, org_id: 'str' =None, project_id: 'str' =None, provider: 'str' =None, provider_instance_id: 'str' =None, provider_region: 'str' =None, provisioning_method: 'str' =None, ready_at: 'datetime' =None, resource_id: 'str' =None, resource_type: 'str' =None, resources: 'V1Resources' =None, status: 'str' =None, unschedulable: 'bool' =None, updated_at: 'datetime' =None, warning_message: 'str' =None):  # noqa: E501
         """V1Machine - a model defined in Swagger"""  # noqa: E501
         self._address = None
+        self._cloud_space_name = None
         self._cluster_id = None
         self._created_at = None
+        self._display_name = None
         self._id = None
         self._instance_type = None
         self._management_api_url = None
         self._name = None
         self._org_id = None
+        self._project_id = None
         self._provider = None
         self._provider_instance_id = None
         self._provider_region = None
@@ -111,10 +120,14 @@ class V1Machine(object):
         self.discriminator = None
         if address is not None:
             self.address = address
+        if cloud_space_name is not None:
+            self.cloud_space_name = cloud_space_name
         if cluster_id is not None:
             self.cluster_id = cluster_id
         if created_at is not None:
             self.created_at = created_at
+        if display_name is not None:
+            self.display_name = display_name
         if id is not None:
             self.id = id
         if instance_type is not None:
@@ -125,6 +138,8 @@ class V1Machine(object):
             self.name = name
         if org_id is not None:
             self.org_id = org_id
+        if project_id is not None:
+            self.project_id = project_id
         if provider is not None:
             self.provider = provider
         if provider_instance_id is not None:
@@ -171,6 +186,27 @@ class V1Machine(object):
 
         self._address = address
 
+    @property
+    def cloud_space_name(self) -> 'str':
+        """Gets the cloud_space_name of this V1Machine.  # noqa: E501
+
+
+        :return: The cloud_space_name of this V1Machine.  # noqa: E501
+        :rtype: str
+        """
+        return self._cloud_space_name
+
+    @cloud_space_name.setter
+    def cloud_space_name(self, cloud_space_name: 'str'):
+        """Sets the cloud_space_name of this V1Machine.
+
+
+        :param cloud_space_name: The cloud_space_name of this V1Machine.  # noqa: E501
+        :type: str
+        """
+
+        self._cloud_space_name = cloud_space_name
+
     @property
     def cluster_id(self) -> 'str':
         """Gets the cluster_id of this V1Machine.  # noqa: E501
@@ -213,6 +249,27 @@ class V1Machine(object):
 
         self._created_at = created_at
 
+    @property
+    def display_name(self) -> 'str':
+        """Gets the display_name of this V1Machine.  # noqa: E501
+
+
+        :return: The display_name of this V1Machine.  # noqa: E501
+        :rtype: str
+        """
+        return self._display_name
+
+    @display_name.setter
+    def display_name(self, display_name: 'str'):
+        """Sets the display_name of this V1Machine.
+
+
+        :param display_name: The display_name of this V1Machine.  # noqa: E501
+        :type: str
+        """
+
+        self._display_name = display_name
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1Machine.  # noqa: E501
@@ -318,6 +375,27 @@ class V1Machine(object):
 
         self._org_id = org_id
 
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1Machine.  # noqa: E501
+
+
+        :return: The project_id of this V1Machine.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1Machine.
+
+
+        :param project_id: The project_id of this V1Machine.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
     @property
     def provider(self) -> 'str':
         """Gets the provider of this V1Machine.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_machine_direct_v1.py

@@ -43,25 +43,30 @@ class V1MachineDirectV1(object):
     swagger_types = {
         'primary_region': 'str',
         'regions': 'list[str]',
+        'setup_infiniband': 'bool',
         'tailscale_auth_key_secret_id': 'str'
     }
 
     attribute_map = {
         'primary_region': 'primaryRegion',
         'regions': 'regions',
+        'setup_infiniband': 'setupInfiniband',
         'tailscale_auth_key_secret_id': 'tailscaleAuthKeySecretId'
     }
 
-    def __init__(self, primary_region: 'str' =None, regions: 'list[str]' =None, tailscale_auth_key_secret_id: 'str' =None):  # noqa: E501
+    def __init__(self, primary_region: 'str' =None, regions: 'list[str]' =None, setup_infiniband: 'bool' =None, tailscale_auth_key_secret_id: 'str' =None):  # noqa: E501
         """V1MachineDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._primary_region = None
         self._regions = None
+        self._setup_infiniband = None
         self._tailscale_auth_key_secret_id = None
         self.discriminator = None
         if primary_region is not None:
             self.primary_region = primary_region
         if regions is not None:
             self.regions = regions
+        if setup_infiniband is not None:
+            self.setup_infiniband = setup_infiniband
         if tailscale_auth_key_secret_id is not None:
             self.tailscale_auth_key_secret_id = tailscale_auth_key_secret_id
 
@@ -107,6 +112,27 @@ class V1MachineDirectV1(object):
 
         self._regions = regions
 
+    @property
+    def setup_infiniband(self) -> 'bool':
+        """Gets the setup_infiniband of this V1MachineDirectV1.  # noqa: E501
+
+
+        :return: The setup_infiniband of this V1MachineDirectV1.  # noqa: E501
+        :rtype: bool
+        """
+        return self._setup_infiniband
+
+    @setup_infiniband.setter
+    def setup_infiniband(self, setup_infiniband: 'bool'):
+        """Sets the setup_infiniband of this V1MachineDirectV1.
+
+
+        :param setup_infiniband: The setup_infiniband of this V1MachineDirectV1.  # noqa: E501
+        :type: bool
+        """
+
+        self._setup_infiniband = setup_infiniband
+
     @property
     def tailscale_auth_key_secret_id(self) -> 'str':
         """Gets the tailscale_auth_key_secret_id of this V1MachineDirectV1.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_message_action.py

@@ -241,6 +241,7 @@ class V1MessageAction(object):
     def type(self) -> 'str':
         """Gets the type of this V1MessageAction.  # noqa: E501
 
+        eg: \"save\", \"run\", \"copy\"...  # noqa: E501
 
         :return: The type of this V1MessageAction.  # noqa: E501
         :rtype: str
@@ -251,6 +252,7 @@ class V1MessageAction(object):
     def type(self, type: 'str'):
         """Sets the type of this V1MessageAction.
 
+        eg: \"save\", \"run\", \"copy\"...  # noqa: E501
 
         :param type: The type of this V1MessageAction.  # noqa: E501
         :type: str

lightning_sdk/lightning_cloud/openapi/models/v1_model_version_archive.py

@@ -371,6 +371,7 @@ class V1ModelVersionArchive(object):
     def version(self) -> 'str':
         """Gets the version of this V1ModelVersionArchive.  # noqa: E501
 
+        v1, v2, v3, etc.  # noqa: E501
 
         :return: The version of this V1ModelVersionArchive.  # noqa: E501
         :rtype: str
@@ -381,6 +382,7 @@ class V1ModelVersionArchive(object):
     def version(self, version: 'str'):
         """Sets the version of this V1ModelVersionArchive.
 
+        v1, v2, v3, etc.  # noqa: E501
 
         :param version: The version of this V1ModelVersionArchive.  # noqa: E501
         :type: str

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py

@@ -225,6 +225,7 @@ class V1MultiMachineJobEvent(object):
     def reason(self) -> 'str':
         """Gets the reason of this V1MultiMachineJobEvent.  # noqa: E501
 
+        Reason is why the action was taken. It is human-readable.  # noqa: E501
 
         :return: The reason of this V1MultiMachineJobEvent.  # noqa: E501
         :rtype: str
@@ -235,6 +236,7 @@ class V1MultiMachineJobEvent(object):
     def reason(self, reason: 'str'):
         """Sets the reason of this V1MultiMachineJobEvent.
 
+        Reason is why the action was taken. It is human-readable.  # noqa: E501
 
         :param reason: The reason of this V1MultiMachineJobEvent.  # noqa: E501
         :type: str

lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_step_status.py

@@ -225,6 +225,7 @@ class V1PipelineStepStatus(object):
     def state(self) -> 'str':
         """Gets the state of this V1PipelineStepStatus.  # noqa: E501
 
+        pending, running, stopped, completed, failed  # noqa: E501
 
         :return: The state of this V1PipelineStepStatus.  # noqa: E501
         :rtype: str
@@ -235,6 +236,7 @@ class V1PipelineStepStatus(object):
     def state(self, state: 'str'):
         """Sets the state of this V1PipelineStepStatus.
 
+        pending, running, stopped, completed, failed  # noqa: E501
 
         :param state: The state of this V1PipelineStepStatus.  # noqa: E501
         :type: str

lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_severity.py

@@ -39,6 +39,7 @@ class V1ServerAlertSeverity(object):
     """
     ERROR = "SERVER_ALERT_SEVERITY_ERROR"
     WARNING = "SERVER_ALERT_SEVERITY_WARNING"
+    INFO = "SERVER_ALERT_SEVERITY_INFO"
     """
     Attributes:
       swagger_types (dict): The key is attribute name