lightning-sdk 2025.10.27__py3-none-any.whl → 2025.10.31__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_membership.py

@@ -42,6 +42,7 @@ class V1Membership(object):
     """
     swagger_types = {
         'balance': 'float',
+        'budgeting_enabled': 'bool',
         'created_at': 'datetime',
         'creator_id': 'str',
         'current_storage_bytes': 'str',
@@ -67,6 +68,7 @@ class V1Membership(object):
 
     attribute_map = {
         'balance': 'balance',
+        'budgeting_enabled': 'budgetingEnabled',
         'created_at': 'createdAt',
         'creator_id': 'creatorId',
         'current_storage_bytes': 'currentStorageBytes',
@@ -90,9 +92,10 @@ class V1Membership(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, current_storage_bytes: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, free_storage_bytes: 'str' =None, is_default: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, organization_billing_enabled: 'bool' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, balance: 'float' =None, budgeting_enabled: 'bool' =None, created_at: 'datetime' =None, creator_id: 'str' =None, current_storage_bytes: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, free_storage_bytes: 'str' =None, is_default: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, organization_billing_enabled: 'bool' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Membership - a model defined in Swagger"""  # noqa: E501
         self._balance = None
+        self._budgeting_enabled = None
         self._created_at = None
         self._creator_id = None
         self._current_storage_bytes = None
@@ -117,6 +120,8 @@ class V1Membership(object):
         self.discriminator = None
         if balance is not None:
             self.balance = balance
+        if budgeting_enabled is not None:
+            self.budgeting_enabled = budgeting_enabled
         if created_at is not None:
             self.created_at = created_at
         if creator_id is not None:
@@ -181,6 +186,27 @@ class V1Membership(object):
 
         self._balance = balance
 
+    @property
+    def budgeting_enabled(self) -> 'bool':
+        """Gets the budgeting_enabled of this V1Membership.  # noqa: E501
+
+
+        :return: The budgeting_enabled of this V1Membership.  # noqa: E501
+        :rtype: bool
+        """
+        return self._budgeting_enabled
+
+    @budgeting_enabled.setter
+    def budgeting_enabled(self, budgeting_enabled: 'bool'):
+        """Sets the budgeting_enabled of this V1Membership.
+
+
+        :param budgeting_enabled: The budgeting_enabled of this V1Membership.  # noqa: E501
+        :type: bool
+        """
+
+        self._budgeting_enabled = budgeting_enabled
+
     @property
     def created_at(self) -> 'datetime':
         """Gets the created_at of this V1Membership.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_message_action.py

@@ -0,0 +1,279 @@
+# 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 V1MessageAction(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 = {
+        'actioned_at': 'datetime',
+        'assistant_id': 'str',
+        'block_id': 'str',
+        'conversation_id': 'str',
+        'message_id': 'str',
+        'project_id': 'str',
+        'type': 'str'
+    }
+
+    attribute_map = {
+        'actioned_at': 'actionedAt',
+        'assistant_id': 'assistantId',
+        'block_id': 'blockId',
+        'conversation_id': 'conversationId',
+        'message_id': 'messageId',
+        'project_id': 'projectId',
+        'type': 'type'
+    }
+
+    def __init__(self, actioned_at: 'datetime' =None, assistant_id: 'str' =None, block_id: 'str' =None, conversation_id: 'str' =None, message_id: 'str' =None, project_id: 'str' =None, type: 'str' =None):  # noqa: E501
+        """V1MessageAction - a model defined in Swagger"""  # noqa: E501
+        self._actioned_at = None
+        self._assistant_id = None
+        self._block_id = None
+        self._conversation_id = None
+        self._message_id = None
+        self._project_id = None
+        self._type = None
+        self.discriminator = None
+        if actioned_at is not None:
+            self.actioned_at = actioned_at
+        if assistant_id is not None:
+            self.assistant_id = assistant_id
+        if block_id is not None:
+            self.block_id = block_id
+        if conversation_id is not None:
+            self.conversation_id = conversation_id
+        if message_id is not None:
+            self.message_id = message_id
+        if project_id is not None:
+            self.project_id = project_id
+        if type is not None:
+            self.type = type
+
+    @property
+    def actioned_at(self) -> 'datetime':
+        """Gets the actioned_at of this V1MessageAction.  # noqa: E501
+
+
+        :return: The actioned_at of this V1MessageAction.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._actioned_at
+
+    @actioned_at.setter
+    def actioned_at(self, actioned_at: 'datetime'):
+        """Sets the actioned_at of this V1MessageAction.
+
+
+        :param actioned_at: The actioned_at of this V1MessageAction.  # noqa: E501
+        :type: datetime
+        """
+
+        self._actioned_at = actioned_at
+
+    @property
+    def assistant_id(self) -> 'str':
+        """Gets the assistant_id of this V1MessageAction.  # noqa: E501
+
+
+        :return: The assistant_id of this V1MessageAction.  # noqa: E501
+        :rtype: str
+        """
+        return self._assistant_id
+
+    @assistant_id.setter
+    def assistant_id(self, assistant_id: 'str'):
+        """Sets the assistant_id of this V1MessageAction.
+
+
+        :param assistant_id: The assistant_id of this V1MessageAction.  # noqa: E501
+        :type: str
+        """
+
+        self._assistant_id = assistant_id
+
+    @property
+    def block_id(self) -> 'str':
+        """Gets the block_id of this V1MessageAction.  # noqa: E501
+
+
+        :return: The block_id of this V1MessageAction.  # noqa: E501
+        :rtype: str
+        """
+        return self._block_id
+
+    @block_id.setter
+    def block_id(self, block_id: 'str'):
+        """Sets the block_id of this V1MessageAction.
+
+
+        :param block_id: The block_id of this V1MessageAction.  # noqa: E501
+        :type: str
+        """
+
+        self._block_id = block_id
+
+    @property
+    def conversation_id(self) -> 'str':
+        """Gets the conversation_id of this V1MessageAction.  # noqa: E501
+
+
+        :return: The conversation_id of this V1MessageAction.  # noqa: E501
+        :rtype: str
+        """
+        return self._conversation_id
+
+    @conversation_id.setter
+    def conversation_id(self, conversation_id: 'str'):
+        """Sets the conversation_id of this V1MessageAction.
+
+
+        :param conversation_id: The conversation_id of this V1MessageAction.  # noqa: E501
+        :type: str
+        """
+
+        self._conversation_id = conversation_id
+
+    @property
+    def message_id(self) -> 'str':
+        """Gets the message_id of this V1MessageAction.  # noqa: E501
+
+
+        :return: The message_id of this V1MessageAction.  # noqa: E501
+        :rtype: str
+        """
+        return self._message_id
+
+    @message_id.setter
+    def message_id(self, message_id: 'str'):
+        """Sets the message_id of this V1MessageAction.
+
+
+        :param message_id: The message_id of this V1MessageAction.  # noqa: E501
+        :type: str
+        """
+
+        self._message_id = message_id
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1MessageAction.  # noqa: E501
+
+
+        :return: The project_id of this V1MessageAction.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1MessageAction.
+
+
+        :param project_id: The project_id of this V1MessageAction.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    @property
+    def type(self) -> 'str':
+        """Gets the type of this V1MessageAction.  # noqa: E501
+
+
+        :return: The type of this V1MessageAction.  # noqa: E501
+        :rtype: str
+        """
+        return self._type
+
+    @type.setter
+    def type(self, type: 'str'):
+        """Sets the type of this V1MessageAction.
+
+
+        :param type: The type of this V1MessageAction.  # noqa: E501
+        :type: str
+        """
+
+        self._type = type
+
+    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(V1MessageAction, 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: 'V1MessageAction') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1MessageAction):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1MessageAction') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py

@@ -43,6 +43,7 @@ class V1ProjectMembership(object):
     swagger_types = {
         'avatar_url': 'str',
         'balance': 'float',
+        'budgeting_enabled': 'bool',
         'created_at': 'datetime',
         'creator_id': 'str',
         'current_storage_bytes': 'str',
@@ -76,6 +77,7 @@ class V1ProjectMembership(object):
     attribute_map = {
         'avatar_url': 'avatarUrl',
         'balance': 'balance',
+        'budgeting_enabled': 'budgetingEnabled',
         'created_at': 'createdAt',
         'creator_id': 'creatorId',
         'current_storage_bytes': 'currentStorageBytes',
@@ -106,10 +108,11 @@ class V1ProjectMembership(object):
         'username': 'username'
     }
 
-    def __init__(self, avatar_url: 'str' =None, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, current_storage_bytes: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, email: 'str' =None, first_name: 'str' =None, free_credits_enabled: 'bool' =None, free_storage_bytes: 'str' =None, inactive: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, job_title: 'str' =None, last_name: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, organization: 'str' =None, organization_billing_enabled: 'bool' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None, username: 'str' =None):  # noqa: E501
+    def __init__(self, avatar_url: 'str' =None, balance: 'float' =None, budgeting_enabled: 'bool' =None, created_at: 'datetime' =None, creator_id: 'str' =None, current_storage_bytes: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, email: 'str' =None, first_name: 'str' =None, free_credits_enabled: 'bool' =None, free_storage_bytes: 'str' =None, inactive: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, job_title: 'str' =None, last_name: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant: 'str' =None, organization: 'str' =None, organization_billing_enabled: 'bool' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None, username: 'str' =None):  # noqa: E501
         """V1ProjectMembership - a model defined in Swagger"""  # noqa: E501
         self._avatar_url = None
         self._balance = None
+        self._budgeting_enabled = None
         self._created_at = None
         self._creator_id = None
         self._current_storage_bytes = None
@@ -143,6 +146,8 @@ class V1ProjectMembership(object):
             self.avatar_url = avatar_url
         if balance is not None:
             self.balance = balance
+        if budgeting_enabled is not None:
+            self.budgeting_enabled = budgeting_enabled
         if created_at is not None:
             self.created_at = created_at
         if creator_id is not None:
@@ -242,6 +247,27 @@ class V1ProjectMembership(object):
 
         self._balance = balance
 
+    @property
+    def budgeting_enabled(self) -> 'bool':
+        """Gets the budgeting_enabled of this V1ProjectMembership.  # noqa: E501
+
+
+        :return: The budgeting_enabled of this V1ProjectMembership.  # noqa: E501
+        :rtype: bool
+        """
+        return self._budgeting_enabled
+
+    @budgeting_enabled.setter
+    def budgeting_enabled(self, budgeting_enabled: 'bool'):
+        """Sets the budgeting_enabled of this V1ProjectMembership.
+
+
+        :param budgeting_enabled: The budgeting_enabled of this V1ProjectMembership.  # noqa: E501
+        :type: bool
+        """
+
+        self._budgeting_enabled = budgeting_enabled
+
     @property
     def created_at(self) -> 'datetime':
         """Gets the created_at of this V1ProjectMembership.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py

@@ -54,6 +54,7 @@ class V1ProjectSettings(object):
         'auto_replenish_amount': 'float',
         'auto_replenish_threshold': 'float',
         'auto_switch_machine': 'bool',
+        'budgeting_enabled': 'bool',
         'default_machine_image_version': 'str',
         'default_machine_type': 'str',
         'organization_billing_enabled': 'bool',
@@ -79,6 +80,7 @@ class V1ProjectSettings(object):
         'auto_replenish_amount': 'autoReplenishAmount',
         'auto_replenish_threshold': 'autoReplenishThreshold',
         'auto_switch_machine': 'autoSwitchMachine',
+        'budgeting_enabled': 'budgetingEnabled',
         'default_machine_image_version': 'defaultMachineImageVersion',
         'default_machine_type': 'defaultMachineType',
         'organization_billing_enabled': 'organizationBillingEnabled',
@@ -90,7 +92,7 @@ class V1ProjectSettings(object):
         'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle'
     }
 
-    def __init__(self, allow_aws_saas: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_dgx_saas: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_lightning_saas: 'bool' =None, allow_nebius_saas: 'bool' =None, allow_voltage_park_saas: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, organization_billing_enabled: 'bool' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, same_compute_on_resume: 'bool' =None, start_studio_on_spot_instance: 'bool' =None, switch_to_default_machine_on_idle: 'bool' =None):  # noqa: E501
+    def __init__(self, allow_aws_saas: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_dgx_saas: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_lightning_saas: 'bool' =None, allow_nebius_saas: 'bool' =None, allow_voltage_park_saas: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, budgeting_enabled: 'bool' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, organization_billing_enabled: 'bool' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, same_compute_on_resume: 'bool' =None, start_studio_on_spot_instance: 'bool' =None, switch_to_default_machine_on_idle: 'bool' =None):  # noqa: E501
         """V1ProjectSettings - a model defined in Swagger"""  # noqa: E501
         self._allow_aws_saas = None
         self._allow_credits_auto_replenish = None
@@ -105,6 +107,7 @@ class V1ProjectSettings(object):
         self._auto_replenish_amount = None
         self._auto_replenish_threshold = None
         self._auto_switch_machine = None
+        self._budgeting_enabled = None
         self._default_machine_image_version = None
         self._default_machine_type = None
         self._organization_billing_enabled = None
@@ -141,6 +144,8 @@ class V1ProjectSettings(object):
             self.auto_replenish_threshold = auto_replenish_threshold
         if auto_switch_machine is not None:
             self.auto_switch_machine = auto_switch_machine
+        if budgeting_enabled is not None:
+            self.budgeting_enabled = budgeting_enabled
         if default_machine_image_version is not None:
             self.default_machine_image_version = default_machine_image_version
         if default_machine_type is not None:
@@ -435,6 +440,27 @@ class V1ProjectSettings(object):
 
         self._auto_switch_machine = auto_switch_machine
 
+    @property
+    def budgeting_enabled(self) -> 'bool':
+        """Gets the budgeting_enabled of this V1ProjectSettings.  # noqa: E501
+
+
+        :return: The budgeting_enabled of this V1ProjectSettings.  # noqa: E501
+        :rtype: bool
+        """
+        return self._budgeting_enabled
+
+    @budgeting_enabled.setter
+    def budgeting_enabled(self, budgeting_enabled: 'bool'):
+        """Sets the budgeting_enabled of this V1ProjectSettings.
+
+
+        :param budgeting_enabled: The budgeting_enabled of this V1ProjectSettings.  # noqa: E501
+        :type: bool
+        """
+
+        self._budgeting_enabled = budgeting_enabled
+
     @property
     def default_machine_image_version(self) -> 'str':
         """Gets the default_machine_image_version of this V1ProjectSettings.  # noqa: E501