lightning-sdk 2025.8.19.post0__py3-none-any.whl → 2025.8.21__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_project_cluster_binding.py

@@ -48,6 +48,7 @@ class V1ProjectClusterBinding(object):
         'is_cluster_healthy': 'bool',
         'project_id': 'str',
         'sa_key': 'str',
+        'sa_key_created_at': 'datetime',
         'updated_at': 'datetime'
     }
 
@@ -59,10 +60,11 @@ class V1ProjectClusterBinding(object):
         'is_cluster_healthy': 'isClusterHealthy',
         'project_id': 'projectId',
         'sa_key': 'saKey',
+        'sa_key_created_at': 'saKeyCreatedAt',
         'updated_at': 'updatedAt'
     }
 
-    def __init__(self, cluster_id: 'str' =None, cluster_name: 'str' =None, cluster_region: 'str' =None, created_at: 'datetime' =None, is_cluster_healthy: 'bool' =None, project_id: 'str' =None, sa_key: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, cluster_name: 'str' =None, cluster_region: 'str' =None, created_at: 'datetime' =None, is_cluster_healthy: 'bool' =None, project_id: 'str' =None, sa_key: 'str' =None, sa_key_created_at: 'datetime' =None, updated_at: 'datetime' =None):  # noqa: E501
         """V1ProjectClusterBinding - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
         self._cluster_name = None
@@ -71,6 +73,7 @@ class V1ProjectClusterBinding(object):
         self._is_cluster_healthy = None
         self._project_id = None
         self._sa_key = None
+        self._sa_key_created_at = None
         self._updated_at = None
         self.discriminator = None
         if cluster_id is not None:
@@ -87,6 +90,8 @@ class V1ProjectClusterBinding(object):
             self.project_id = project_id
         if sa_key is not None:
             self.sa_key = sa_key
+        if sa_key_created_at is not None:
+            self.sa_key_created_at = sa_key_created_at
         if updated_at is not None:
             self.updated_at = updated_at
 
@@ -237,6 +242,27 @@ class V1ProjectClusterBinding(object):
 
         self._sa_key = sa_key
 
+    @property
+    def sa_key_created_at(self) -> 'datetime':
+        """Gets the sa_key_created_at of this V1ProjectClusterBinding.  # noqa: E501
+
+
+        :return: The sa_key_created_at of this V1ProjectClusterBinding.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._sa_key_created_at
+
+    @sa_key_created_at.setter
+    def sa_key_created_at(self, sa_key_created_at: 'datetime'):
+        """Sets the sa_key_created_at of this V1ProjectClusterBinding.
+
+
+        :param sa_key_created_at: The sa_key_created_at of this V1ProjectClusterBinding.  # noqa: E501
+        :type: datetime
+        """
+
+        self._sa_key_created_at = sa_key_created_at
+
     @property
     def updated_at(self) -> 'datetime':
         """Gets the updated_at of this V1ProjectClusterBinding.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_quote_annual_upsell_response.py

@@ -0,0 +1,201 @@
+# 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 V1QuoteAnnualUpsellResponse(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 = {
+        'academic': 'bool',
+        'additional_months': 'int',
+        'annual_amount': 'int',
+        'discounted_amount': 'int'
+    }
+
+    attribute_map = {
+        'academic': 'academic',
+        'additional_months': 'additionalMonths',
+        'annual_amount': 'annualAmount',
+        'discounted_amount': 'discountedAmount'
+    }
+
+    def __init__(self, academic: 'bool' =None, additional_months: 'int' =None, annual_amount: 'int' =None, discounted_amount: 'int' =None):  # noqa: E501
+        """V1QuoteAnnualUpsellResponse - a model defined in Swagger"""  # noqa: E501
+        self._academic = None
+        self._additional_months = None
+        self._annual_amount = None
+        self._discounted_amount = None
+        self.discriminator = None
+        if academic is not None:
+            self.academic = academic
+        if additional_months is not None:
+            self.additional_months = additional_months
+        if annual_amount is not None:
+            self.annual_amount = annual_amount
+        if discounted_amount is not None:
+            self.discounted_amount = discounted_amount
+
+    @property
+    def academic(self) -> 'bool':
+        """Gets the academic of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+
+
+        :return: The academic of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :rtype: bool
+        """
+        return self._academic
+
+    @academic.setter
+    def academic(self, academic: 'bool'):
+        """Sets the academic of this V1QuoteAnnualUpsellResponse.
+
+
+        :param academic: The academic of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :type: bool
+        """
+
+        self._academic = academic
+
+    @property
+    def additional_months(self) -> 'int':
+        """Gets the additional_months of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+
+
+        :return: The additional_months of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :rtype: int
+        """
+        return self._additional_months
+
+    @additional_months.setter
+    def additional_months(self, additional_months: 'int'):
+        """Sets the additional_months of this V1QuoteAnnualUpsellResponse.
+
+
+        :param additional_months: The additional_months of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :type: int
+        """
+
+        self._additional_months = additional_months
+
+    @property
+    def annual_amount(self) -> 'int':
+        """Gets the annual_amount of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+
+
+        :return: The annual_amount of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :rtype: int
+        """
+        return self._annual_amount
+
+    @annual_amount.setter
+    def annual_amount(self, annual_amount: 'int'):
+        """Sets the annual_amount of this V1QuoteAnnualUpsellResponse.
+
+
+        :param annual_amount: The annual_amount of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :type: int
+        """
+
+        self._annual_amount = annual_amount
+
+    @property
+    def discounted_amount(self) -> 'int':
+        """Gets the discounted_amount of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+
+
+        :return: The discounted_amount of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :rtype: int
+        """
+        return self._discounted_amount
+
+    @discounted_amount.setter
+    def discounted_amount(self, discounted_amount: 'int'):
+        """Sets the discounted_amount of this V1QuoteAnnualUpsellResponse.
+
+
+        :param discounted_amount: The discounted_amount of this V1QuoteAnnualUpsellResponse.  # noqa: E501
+        :type: int
+        """
+
+        self._discounted_amount = discounted_amount
+
+    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(V1QuoteAnnualUpsellResponse, 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: 'V1QuoteAnnualUpsellResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1QuoteAnnualUpsellResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1QuoteAnnualUpsellResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -79,7 +79,6 @@ class V1UserFeatures(object):
         'featured_studios_admin': 'bool',
         'gcp_overprovisioning': 'bool',
         'gcs_connections_optimized': 'bool',
-        'gcs_folders': 'bool',
         'instant_capacity_reservation': 'bool',
         'job_artifacts_v2': 'bool',
         'kubernetes_cluster_ui': 'bool',
@@ -165,7 +164,6 @@ class V1UserFeatures(object):
         'featured_studios_admin': 'featuredStudiosAdmin',
         'gcp_overprovisioning': 'gcpOverprovisioning',
         'gcs_connections_optimized': 'gcsConnectionsOptimized',
-        'gcs_folders': 'gcsFolders',
         'instant_capacity_reservation': 'instantCapacityReservation',
         'job_artifacts_v2': 'jobArtifactsV2',
         'kubernetes_cluster_ui': 'kubernetesClusterUi',
@@ -212,7 +210,7 @@ class V1UserFeatures(object):
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, gcs_folders: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f237: 'bool' =None, f238: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f242: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcp_overprovisioning: 'bool' =None, gcs_connections_optimized: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, onboarding_v2: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, single_wallet: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._affiliate_links = None
         self._agents_v2 = None
@@ -252,7 +250,6 @@ class V1UserFeatures(object):
         self._featured_studios_admin = None
         self._gcp_overprovisioning = None
         self._gcs_connections_optimized = None
-        self._gcs_folders = None
         self._instant_capacity_reservation = None
         self._job_artifacts_v2 = None
         self._kubernetes_cluster_ui = None
@@ -374,8 +371,6 @@ class V1UserFeatures(object):
             self.gcp_overprovisioning = gcp_overprovisioning
         if gcs_connections_optimized is not None:
             self.gcs_connections_optimized = gcs_connections_optimized
-        if gcs_folders is not None:
-            self.gcs_folders = gcs_folders
         if instant_capacity_reservation is not None:
             self.instant_capacity_reservation = instant_capacity_reservation
         if job_artifacts_v2 is not None:
@@ -1263,27 +1258,6 @@ class V1UserFeatures(object):
 
         self._gcs_connections_optimized = gcs_connections_optimized
 
-    @property
-    def gcs_folders(self) -> 'bool':
-        """Gets the gcs_folders of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The gcs_folders of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._gcs_folders
-
-    @gcs_folders.setter
-    def gcs_folders(self, gcs_folders: 'bool'):
-        """Sets the gcs_folders of this V1UserFeatures.
-
-
-        :param gcs_folders: The gcs_folders of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._gcs_folders = gcs_folders
-
     @property
     def instant_capacity_reservation(self) -> 'bool':
         """Gets the instant_capacity_reservation of this V1UserFeatures.  # noqa: E501

lightning_sdk/llm/llm.py

@@ -285,7 +285,7 @@ class LLM:
         self,
         prompt: str,
         system_prompt: Optional[str] = None,
-        max_completion_tokens: Optional[int] = 500,
+        max_completion_tokens: Optional[int] = None,
         images: Optional[Union[List[str], str]] = None,
         conversation: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
@@ -319,7 +319,7 @@ class LLM:
         self,
         prompt: str,
         system_prompt: Optional[str] = None,
-        max_completion_tokens: Optional[int] = 500,
+        max_completion_tokens: Optional[int] = None,
         images: Optional[Union[List[str], str]] = None,
         conversation: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,

lightning_sdk/studio.py

@@ -59,6 +59,9 @@ class Studio:
     _skip_init = False
     _skip_setup = False
 
+    # whether to show progress bars during operations
+    show_progress = False
+
     def __init__(
         self,
         name: Optional[str] = None,
@@ -254,9 +257,26 @@ class Studio:
 
         if status != Status.Stopped:
             raise RuntimeError(f"Cannot start a studio that is not stopped. Studio {self.name} is {status}.")
-        self._studio_api.start_studio(
-            self._studio.id, self._teamspace.id, machine, interruptible=interruptible, max_runtime=max_runtime
-        )
+
+        # Show progress bar during startup
+        if self.show_progress:
+            from lightning_sdk.utils.progress import StudioProgressTracker
+
+            with StudioProgressTracker("start", show_progress=True) as progress:
+                # Start the studio without blocking
+                self._studio_api.start_studio_async(
+                    self._studio.id, self._teamspace.id, machine, interruptible=interruptible, max_runtime=max_runtime
+                )
+
+                # Track progress through completion
+                progress.track_startup_phases(
+                    lambda: self._studio_api.get_studio_status(self._studio.id, self._teamspace.id)
+                )
+        else:
+            # Use the blocking version if no progress is needed
+            self._studio_api.start_studio(
+                self._studio.id, self._teamspace.id, machine, interruptible=interruptible, max_runtime=max_runtime
+            )
 
         self._setup()
 
@@ -324,9 +344,22 @@ class Studio:
             raise RuntimeError(
                 f"Cannot switch machine on a studio that is not running. Studio {self.name} is {status}."
             )
-        self._studio_api.switch_studio_machine(
-            self._studio.id, self._teamspace.id, machine, interruptible=interruptible
-        )
+
+        if self.show_progress:
+            from lightning_sdk.utils.progress import StudioProgressTracker
+
+            with StudioProgressTracker("switch", show_progress=True) as progress:
+                # Update progress before starting the switch
+                progress.update_progress(5, "Initiating machine switch...")
+
+                # Start the switch operation with progress tracking
+                self._studio_api.switch_studio_machine_with_progress(
+                    self._studio.id, self._teamspace.id, machine, interruptible=interruptible, progress=progress
+                )
+        else:
+            self._studio_api.switch_studio_machine(
+                self._studio.id, self._teamspace.id, machine, interruptible=interruptible
+            )
 
     def run_and_detach(self, *commands: str, timeout: float = 10, check_interval: float = 1) -> str:
         """Runs given commands on the Studio and returns immediately.