lightning-sdk 0.1.49__py3-none-any.whl → 2025.11.5__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_daily_usage.py

@@ -42,23 +42,38 @@ class V1DailyUsage(object):
     """
     swagger_types = {
         'day': 'datetime',
-        'total_cost': 'float'
+        'total_completion_tokens': 'str',
+        'total_cost': 'float',
+        'total_num_messages': 'str',
+        'total_prompt_tokens': 'str'
     }
 
     attribute_map = {
         'day': 'day',
-        'total_cost': 'totalCost'
+        'total_completion_tokens': 'totalCompletionTokens',
+        'total_cost': 'totalCost',
+        'total_num_messages': 'totalNumMessages',
+        'total_prompt_tokens': 'totalPromptTokens'
     }
 
-    def __init__(self, day: 'datetime' =None, total_cost: 'float' =None):  # noqa: E501
+    def __init__(self, day: 'datetime' =None, total_completion_tokens: 'str' =None, total_cost: 'float' =None, total_num_messages: 'str' =None, total_prompt_tokens: 'str' =None):  # noqa: E501
         """V1DailyUsage - a model defined in Swagger"""  # noqa: E501
         self._day = None
+        self._total_completion_tokens = None
         self._total_cost = None
+        self._total_num_messages = None
+        self._total_prompt_tokens = None
         self.discriminator = None
         if day is not None:
             self.day = day
+        if total_completion_tokens is not None:
+            self.total_completion_tokens = total_completion_tokens
         if total_cost is not None:
             self.total_cost = total_cost
+        if total_num_messages is not None:
+            self.total_num_messages = total_num_messages
+        if total_prompt_tokens is not None:
+            self.total_prompt_tokens = total_prompt_tokens
 
     @property
     def day(self) -> 'datetime':
@@ -81,6 +96,27 @@ class V1DailyUsage(object):
 
         self._day = day
 
+    @property
+    def total_completion_tokens(self) -> 'str':
+        """Gets the total_completion_tokens of this V1DailyUsage.  # noqa: E501
+
+
+        :return: The total_completion_tokens of this V1DailyUsage.  # noqa: E501
+        :rtype: str
+        """
+        return self._total_completion_tokens
+
+    @total_completion_tokens.setter
+    def total_completion_tokens(self, total_completion_tokens: 'str'):
+        """Sets the total_completion_tokens of this V1DailyUsage.
+
+
+        :param total_completion_tokens: The total_completion_tokens of this V1DailyUsage.  # noqa: E501
+        :type: str
+        """
+
+        self._total_completion_tokens = total_completion_tokens
+
     @property
     def total_cost(self) -> 'float':
         """Gets the total_cost of this V1DailyUsage.  # noqa: E501
@@ -102,6 +138,48 @@ class V1DailyUsage(object):
 
         self._total_cost = total_cost
 
+    @property
+    def total_num_messages(self) -> 'str':
+        """Gets the total_num_messages of this V1DailyUsage.  # noqa: E501
+
+
+        :return: The total_num_messages of this V1DailyUsage.  # noqa: E501
+        :rtype: str
+        """
+        return self._total_num_messages
+
+    @total_num_messages.setter
+    def total_num_messages(self, total_num_messages: 'str'):
+        """Sets the total_num_messages of this V1DailyUsage.
+
+
+        :param total_num_messages: The total_num_messages of this V1DailyUsage.  # noqa: E501
+        :type: str
+        """
+
+        self._total_num_messages = total_num_messages
+
+    @property
+    def total_prompt_tokens(self) -> 'str':
+        """Gets the total_prompt_tokens of this V1DailyUsage.  # noqa: E501
+
+
+        :return: The total_prompt_tokens of this V1DailyUsage.  # noqa: E501
+        :rtype: str
+        """
+        return self._total_prompt_tokens
+
+    @total_prompt_tokens.setter
+    def total_prompt_tokens(self, total_prompt_tokens: 'str'):
+        """Sets the total_prompt_tokens of this V1DailyUsage.
+
+
+        :param total_prompt_tokens: The total_prompt_tokens of this V1DailyUsage.  # noqa: E501
+        :type: str
+        """
+
+        self._total_prompt_tokens = total_prompt_tokens
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py

@@ -47,22 +47,32 @@ class V1DataConnection(object):
         'cluster_id': 'str',
         'created_at': 'datetime',
         'efs': 'V1EfsConfig',
+        'error': 'str',
+        'filestore': 'V1FilestoreDataConnection',
         'gcp': 'V1GcpDataConnection',
         'gcs_folder': 'V1GCSFolderDataConnection',
         'id': 'str',
         'index': 'V1Index',
+        'is_billable_folder': 'bool',
+        'is_cache': 'bool',
         'is_managed': 'bool',
+        'is_public': 'bool',
         'lock_out': 'bool',
+        'lustre': 'V1LustreDataConnection',
         'name': 'str',
         'number_of_files': 'str',
         'project_id': 'str',
+        'r2': 'V1R2DataConnection',
         'run_cmds': 'list[str]',
         's3_folder': 'V1S3FolderDataConnection',
         'snowflake': 'V1SnowflakeDataConnection',
         'state': 'V1DataConnectionState',
+        'total_daily_read_bytes': 'str',
+        'total_daily_write_bytes': 'str',
         'total_size_bytes': 'str',
         'type': 'str',
         'updated_at': 'datetime',
+        'weka': 'V1WekaDataConnection',
         'writable': 'bool'
     }
 
@@ -73,26 +83,36 @@ class V1DataConnection(object):
         'cluster_id': 'clusterId',
         'created_at': 'createdAt',
         'efs': 'efs',
+        'error': 'error',
+        'filestore': 'filestore',
         'gcp': 'gcp',
         'gcs_folder': 'gcsFolder',
         'id': 'id',
         'index': 'index',
+        'is_billable_folder': 'isBillableFolder',
+        'is_cache': 'isCache',
         'is_managed': 'isManaged',
+        'is_public': 'isPublic',
         'lock_out': 'lockOut',
+        'lustre': 'lustre',
         'name': 'name',
         'number_of_files': 'numberOfFiles',
         'project_id': 'projectId',
+        'r2': 'r2',
         'run_cmds': 'runCmds',
         's3_folder': 's3Folder',
         'snowflake': 'snowflake',
         'state': 'state',
+        'total_daily_read_bytes': 'totalDailyReadBytes',
+        'total_daily_write_bytes': 'totalDailyWriteBytes',
         'total_size_bytes': 'totalSizeBytes',
         'type': 'type',
         'updated_at': 'updatedAt',
+        'weka': 'weka',
         'writable': 'writable'
     }
 
-    def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_managed: 'bool' =None, lock_out: 'bool' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, access_cluster_ids: 'list[str]' =None, accessible: 'bool' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, efs: 'V1EfsConfig' =None, error: 'str' =None, filestore: 'V1FilestoreDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, index: 'V1Index' =None, is_billable_folder: 'bool' =None, is_cache: 'bool' =None, is_managed: 'bool' =None, is_public: 'bool' =None, lock_out: 'bool' =None, lustre: 'V1LustreDataConnection' =None, name: 'str' =None, number_of_files: 'str' =None, project_id: 'str' =None, r2: 'V1R2DataConnection' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, state: 'V1DataConnectionState' =None, total_daily_read_bytes: 'str' =None, total_daily_write_bytes: 'str' =None, total_size_bytes: 'str' =None, type: 'str' =None, updated_at: 'datetime' =None, weka: 'V1WekaDataConnection' =None, writable: 'bool' =None):  # noqa: E501
         """V1DataConnection - a model defined in Swagger"""  # noqa: E501
         self._access_cluster_ids = None
         self._accessible = None
@@ -100,22 +120,32 @@ class V1DataConnection(object):
         self._cluster_id = None
         self._created_at = None
         self._efs = None
+        self._error = None
+        self._filestore = None
         self._gcp = None
         self._gcs_folder = None
         self._id = None
         self._index = None
+        self._is_billable_folder = None
+        self._is_cache = None
         self._is_managed = None
+        self._is_public = None
         self._lock_out = None
+        self._lustre = None
         self._name = None
         self._number_of_files = None
         self._project_id = None
+        self._r2 = None
         self._run_cmds = None
         self._s3_folder = None
         self._snowflake = None
         self._state = None
+        self._total_daily_read_bytes = None
+        self._total_daily_write_bytes = None
         self._total_size_bytes = None
         self._type = None
         self._updated_at = None
+        self._weka = None
         self._writable = None
         self.discriminator = None
         if access_cluster_ids is not None:
@@ -130,6 +160,10 @@ class V1DataConnection(object):
             self.created_at = created_at
         if efs is not None:
             self.efs = efs
+        if error is not None:
+            self.error = error
+        if filestore is not None:
+            self.filestore = filestore
         if gcp is not None:
             self.gcp = gcp
         if gcs_folder is not None:
@@ -138,16 +172,26 @@ class V1DataConnection(object):
             self.id = id
         if index is not None:
             self.index = index
+        if is_billable_folder is not None:
+            self.is_billable_folder = is_billable_folder
+        if is_cache is not None:
+            self.is_cache = is_cache
         if is_managed is not None:
             self.is_managed = is_managed
+        if is_public is not None:
+            self.is_public = is_public
         if lock_out is not None:
             self.lock_out = lock_out
+        if lustre is not None:
+            self.lustre = lustre
         if name is not None:
             self.name = name
         if number_of_files is not None:
             self.number_of_files = number_of_files
         if project_id is not None:
             self.project_id = project_id
+        if r2 is not None:
+            self.r2 = r2
         if run_cmds is not None:
             self.run_cmds = run_cmds
         if s3_folder is not None:
@@ -156,12 +200,18 @@ class V1DataConnection(object):
             self.snowflake = snowflake
         if state is not None:
             self.state = state
+        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 total_size_bytes is not None:
             self.total_size_bytes = total_size_bytes
         if type is not None:
             self.type = type
         if updated_at is not None:
             self.updated_at = updated_at
+        if weka is not None:
+            self.weka = weka
         if writable is not None:
             self.writable = writable
 
@@ -291,6 +341,48 @@ class V1DataConnection(object):
 
         self._efs = efs
 
+    @property
+    def error(self) -> 'str':
+        """Gets the error of this V1DataConnection.  # noqa: E501
+
+
+        :return: The error of this V1DataConnection.  # noqa: E501
+        :rtype: str
+        """
+        return self._error
+
+    @error.setter
+    def error(self, error: 'str'):
+        """Sets the error of this V1DataConnection.
+
+
+        :param error: The error of this V1DataConnection.  # noqa: E501
+        :type: str
+        """
+
+        self._error = error
+
+    @property
+    def filestore(self) -> 'V1FilestoreDataConnection':
+        """Gets the filestore of this V1DataConnection.  # noqa: E501
+
+
+        :return: The filestore of this V1DataConnection.  # noqa: E501
+        :rtype: V1FilestoreDataConnection
+        """
+        return self._filestore
+
+    @filestore.setter
+    def filestore(self, filestore: 'V1FilestoreDataConnection'):
+        """Sets the filestore of this V1DataConnection.
+
+
+        :param filestore: The filestore of this V1DataConnection.  # noqa: E501
+        :type: V1FilestoreDataConnection
+        """
+
+        self._filestore = filestore
+
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this V1DataConnection.  # noqa: E501
@@ -375,6 +467,48 @@ class V1DataConnection(object):
 
         self._index = index
 
+    @property
+    def is_billable_folder(self) -> 'bool':
+        """Gets the is_billable_folder of this V1DataConnection.  # noqa: E501
+
+
+        :return: The is_billable_folder of this V1DataConnection.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_billable_folder
+
+    @is_billable_folder.setter
+    def is_billable_folder(self, is_billable_folder: 'bool'):
+        """Sets the is_billable_folder of this V1DataConnection.
+
+
+        :param is_billable_folder: The is_billable_folder of this V1DataConnection.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_billable_folder = is_billable_folder
+
+    @property
+    def is_cache(self) -> 'bool':
+        """Gets the is_cache of this V1DataConnection.  # noqa: E501
+
+
+        :return: The is_cache of this V1DataConnection.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_cache
+
+    @is_cache.setter
+    def is_cache(self, is_cache: 'bool'):
+        """Sets the is_cache of this V1DataConnection.
+
+
+        :param is_cache: The is_cache of this V1DataConnection.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_cache = is_cache
+
     @property
     def is_managed(self) -> 'bool':
         """Gets the is_managed of this V1DataConnection.  # noqa: E501
@@ -396,6 +530,27 @@ class V1DataConnection(object):
 
         self._is_managed = is_managed
 
+    @property
+    def is_public(self) -> 'bool':
+        """Gets the is_public of this V1DataConnection.  # noqa: E501
+
+
+        :return: The is_public of this V1DataConnection.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_public
+
+    @is_public.setter
+    def is_public(self, is_public: 'bool'):
+        """Sets the is_public of this V1DataConnection.
+
+
+        :param is_public: The is_public of this V1DataConnection.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_public = is_public
+
     @property
     def lock_out(self) -> 'bool':
         """Gets the lock_out of this V1DataConnection.  # noqa: E501
@@ -417,6 +572,27 @@ class V1DataConnection(object):
 
         self._lock_out = lock_out
 
+    @property
+    def lustre(self) -> 'V1LustreDataConnection':
+        """Gets the lustre of this V1DataConnection.  # noqa: E501
+
+
+        :return: The lustre of this V1DataConnection.  # noqa: E501
+        :rtype: V1LustreDataConnection
+        """
+        return self._lustre
+
+    @lustre.setter
+    def lustre(self, lustre: 'V1LustreDataConnection'):
+        """Sets the lustre of this V1DataConnection.
+
+
+        :param lustre: The lustre of this V1DataConnection.  # noqa: E501
+        :type: V1LustreDataConnection
+        """
+
+        self._lustre = lustre
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1DataConnection.  # noqa: E501
@@ -480,6 +656,27 @@ class V1DataConnection(object):
 
         self._project_id = project_id
 
+    @property
+    def r2(self) -> 'V1R2DataConnection':
+        """Gets the r2 of this V1DataConnection.  # noqa: E501
+
+
+        :return: The r2 of this V1DataConnection.  # noqa: E501
+        :rtype: V1R2DataConnection
+        """
+        return self._r2
+
+    @r2.setter
+    def r2(self, r2: 'V1R2DataConnection'):
+        """Sets the r2 of this V1DataConnection.
+
+
+        :param r2: The r2 of this V1DataConnection.  # noqa: E501
+        :type: V1R2DataConnection
+        """
+
+        self._r2 = r2
+
     @property
     def run_cmds(self) -> 'list[str]':
         """Gets the run_cmds of this V1DataConnection.  # noqa: E501
@@ -564,6 +761,48 @@ class V1DataConnection(object):
 
         self._state = state
 
+    @property
+    def total_daily_read_bytes(self) -> 'str':
+        """Gets the total_daily_read_bytes of this V1DataConnection.  # noqa: E501
+
+
+        :return: The total_daily_read_bytes of this V1DataConnection.  # 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 V1DataConnection.
+
+
+        :param total_daily_read_bytes: The total_daily_read_bytes of this V1DataConnection.  # 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 V1DataConnection.  # noqa: E501
+
+
+        :return: The total_daily_write_bytes of this V1DataConnection.  # 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 V1DataConnection.
+
+
+        :param total_daily_write_bytes: The total_daily_write_bytes of this V1DataConnection.  # noqa: E501
+        :type: str
+        """
+
+        self._total_daily_write_bytes = total_daily_write_bytes
+
     @property
     def total_size_bytes(self) -> 'str':
         """Gets the total_size_bytes of this V1DataConnection.  # noqa: E501
@@ -627,6 +866,27 @@ class V1DataConnection(object):
 
         self._updated_at = updated_at
 
+    @property
+    def weka(self) -> 'V1WekaDataConnection':
+        """Gets the weka of this V1DataConnection.  # noqa: E501
+
+
+        :return: The weka of this V1DataConnection.  # noqa: E501
+        :rtype: V1WekaDataConnection
+        """
+        return self._weka
+
+    @weka.setter
+    def weka(self, weka: 'V1WekaDataConnection'):
+        """Sets the weka of this V1DataConnection.
+
+
+        :param weka: The weka of this V1DataConnection.  # noqa: E501
+        :type: V1WekaDataConnection
+        """
+
+        self._weka = weka
+
     @property
     def writable(self) -> 'bool':
         """Gets the writable of this V1DataConnection.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_tier.py

@@ -0,0 +1,103 @@
+# 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 V1DataConnectionTier(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    SSD = "DATA_CONNECTION_TIER_SSD"
+    HDD = "DATA_CONNECTION_TIER_HDD"
+    """
+    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 = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1DataConnectionTier - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    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(V1DataConnectionTier, 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: 'V1DataConnectionTier') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1DataConnectionTier):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1DataConnectionTier') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other