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

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_storage_asset.py

@@ -45,10 +45,14 @@ class V1StorageAsset(object):
         'created_by': 'str',
         'id': 'str',
         'name': 'str',
+        'read_cost_daily': 'float',
         'storage_bytes': 'str',
         'storage_cost_daily': 'float',
         'sub_type': 'str',
-        'type': 'V1StorageAssetType'
+        'total_daily_read_bytes': 'str',
+        'total_daily_write_bytes': 'str',
+        'type': 'V1StorageAssetType',
+        'write_cost_daily': 'float'
     }
 
     attribute_map = {
@@ -56,22 +60,30 @@ class V1StorageAsset(object):
         'created_by': 'createdBy',
         'id': 'id',
         'name': 'name',
+        'read_cost_daily': 'readCostDaily',
         'storage_bytes': 'storageBytes',
         'storage_cost_daily': 'storageCostDaily',
         'sub_type': 'subType',
-        'type': 'type'
+        'total_daily_read_bytes': 'totalDailyReadBytes',
+        'total_daily_write_bytes': 'totalDailyWriteBytes',
+        'type': 'type',
+        'write_cost_daily': 'writeCostDaily'
     }
 
-    def __init__(self, created_at: 'datetime' =None, created_by: 'str' =None, id: 'str' =None, name: 'str' =None, storage_bytes: 'str' =None, storage_cost_daily: 'float' =None, sub_type: 'str' =None, type: 'V1StorageAssetType' =None):  # noqa: E501
+    def __init__(self, created_at: 'datetime' =None, created_by: 'str' =None, id: 'str' =None, name: 'str' =None, read_cost_daily: 'float' =None, storage_bytes: 'str' =None, storage_cost_daily: 'float' =None, sub_type: 'str' =None, total_daily_read_bytes: 'str' =None, total_daily_write_bytes: 'str' =None, type: 'V1StorageAssetType' =None, write_cost_daily: 'float' =None):  # noqa: E501
         """V1StorageAsset - a model defined in Swagger"""  # noqa: E501
         self._created_at = None
         self._created_by = None
         self._id = None
         self._name = None
+        self._read_cost_daily = None
         self._storage_bytes = None
         self._storage_cost_daily = None
         self._sub_type = None
+        self._total_daily_read_bytes = None
+        self._total_daily_write_bytes = None
         self._type = None
+        self._write_cost_daily = None
         self.discriminator = None
         if created_at is not None:
             self.created_at = created_at
@@ -81,14 +93,22 @@ class V1StorageAsset(object):
             self.id = id
         if name is not None:
             self.name = name
+        if read_cost_daily is not None:
+            self.read_cost_daily = read_cost_daily
         if storage_bytes is not None:
             self.storage_bytes = storage_bytes
         if storage_cost_daily is not None:
             self.storage_cost_daily = storage_cost_daily
         if sub_type is not None:
             self.sub_type = sub_type
+        if total_daily_read_bytes is not None:
+            self.total_daily_read_bytes = total_daily_read_bytes
+        if total_daily_write_bytes is not None:
+            self.total_daily_write_bytes = total_daily_write_bytes
         if type is not None:
             self.type = type
+        if write_cost_daily is not None:
+            self.write_cost_daily = write_cost_daily
 
     @property
     def created_at(self) -> 'datetime':
@@ -174,6 +194,27 @@ class V1StorageAsset(object):
 
         self._name = name
 
+    @property
+    def read_cost_daily(self) -> 'float':
+        """Gets the read_cost_daily of this V1StorageAsset.  # noqa: E501
+
+
+        :return: The read_cost_daily of this V1StorageAsset.  # noqa: E501
+        :rtype: float
+        """
+        return self._read_cost_daily
+
+    @read_cost_daily.setter
+    def read_cost_daily(self, read_cost_daily: 'float'):
+        """Sets the read_cost_daily of this V1StorageAsset.
+
+
+        :param read_cost_daily: The read_cost_daily of this V1StorageAsset.  # noqa: E501
+        :type: float
+        """
+
+        self._read_cost_daily = read_cost_daily
+
     @property
     def storage_bytes(self) -> 'str':
         """Gets the storage_bytes of this V1StorageAsset.  # noqa: E501
@@ -237,6 +278,48 @@ class V1StorageAsset(object):
 
         self._sub_type = sub_type
 
+    @property
+    def total_daily_read_bytes(self) -> 'str':
+        """Gets the total_daily_read_bytes of this V1StorageAsset.  # noqa: E501
+
+
+        :return: The total_daily_read_bytes of this V1StorageAsset.  # noqa: E501
+        :rtype: str
+        """
+        return self._total_daily_read_bytes
+
+    @total_daily_read_bytes.setter
+    def total_daily_read_bytes(self, total_daily_read_bytes: 'str'):
+        """Sets the total_daily_read_bytes of this V1StorageAsset.
+
+
+        :param total_daily_read_bytes: The total_daily_read_bytes of this V1StorageAsset.  # noqa: E501
+        :type: str
+        """
+
+        self._total_daily_read_bytes = total_daily_read_bytes
+
+    @property
+    def total_daily_write_bytes(self) -> 'str':
+        """Gets the total_daily_write_bytes of this V1StorageAsset.  # noqa: E501
+
+
+        :return: The total_daily_write_bytes of this V1StorageAsset.  # noqa: E501
+        :rtype: str
+        """
+        return self._total_daily_write_bytes
+
+    @total_daily_write_bytes.setter
+    def total_daily_write_bytes(self, total_daily_write_bytes: 'str'):
+        """Sets the total_daily_write_bytes of this V1StorageAsset.
+
+
+        :param total_daily_write_bytes: The total_daily_write_bytes of this V1StorageAsset.  # noqa: E501
+        :type: str
+        """
+
+        self._total_daily_write_bytes = total_daily_write_bytes
+
     @property
     def type(self) -> 'V1StorageAssetType':
         """Gets the type of this V1StorageAsset.  # noqa: E501
@@ -258,6 +341,27 @@ class V1StorageAsset(object):
 
         self._type = type
 
+    @property
+    def write_cost_daily(self) -> 'float':
+        """Gets the write_cost_daily of this V1StorageAsset.  # noqa: E501
+
+
+        :return: The write_cost_daily of this V1StorageAsset.  # noqa: E501
+        :rtype: float
+        """
+        return self._write_cost_daily
+
+    @write_cost_daily.setter
+    def write_cost_daily(self, write_cost_daily: 'float'):
+        """Sets the write_cost_daily of this V1StorageAsset.
+
+
+        :param write_cost_daily: The write_cost_daily of this V1StorageAsset.  # noqa: E501
+        :type: float
+        """
+
+        self._write_cost_daily = write_cost_daily
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

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/llm/public_assistants.py

@@ -27,6 +27,10 @@ PUBLIC_MODELS = {
         "id": "ast_01jz3thxskg4fcdk4xhkjkym5b",
         "context_length": 8000
     },
+    "lightning-ai/DeepSeek-V3.1": {
+        "id": "ast_01k378z15k0msktaxjpfrfwqqh",
+        "context_length": 163840
+    },
     "lightning-ai/gpt-oss-20b": {
         "id": "ast_01k1y2ywfh9zzexjzrva6y96ms",
         "context_length": 128000,