lightning-sdk 2025.7.17__py3-none-any.whl → 2025.7.30rc0__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py

@@ -42,17 +42,20 @@ class V1Schedule(object):
     """
     swagger_types = {
         'action_type': 'V1ScheduleActionType',
+        'command': 'str',
         'created_at': 'datetime',
         'cron_expression': 'str',
         'display_name': 'str',
         'id': 'str',
         'name': 'str',
         'next': 'datetime',
+        'parallel_runs': 'bool',
         'parent_resource_id': 'str',
         'project_id': 'str',
         'resource_id': 'str',
         'resource_type': 'V1ScheduleResourceType',
         'state': 'str',
+        'timezone': 'str',
         'total': 'int',
         'updated_at': 'datetime',
         'user_id': 'str'
@@ -60,42 +63,50 @@ class V1Schedule(object):
 
     attribute_map = {
         'action_type': 'actionType',
+        'command': 'command',
         'created_at': 'createdAt',
         'cron_expression': 'cronExpression',
         'display_name': 'displayName',
         'id': 'id',
         'name': 'name',
         'next': 'next',
+        'parallel_runs': 'parallelRuns',
         'parent_resource_id': 'parentResourceId',
         'project_id': 'projectId',
         'resource_id': 'resourceId',
         'resource_type': 'resourceType',
         'state': 'state',
+        'timezone': 'timezone',
         'total': 'total',
         'updated_at': 'updatedAt',
         'user_id': 'userId'
     }
 
-    def __init__(self, action_type: 'V1ScheduleActionType' =None, created_at: 'datetime' =None, cron_expression: 'str' =None, display_name: 'str' =None, id: 'str' =None, name: 'str' =None, next: 'datetime' =None, parent_resource_id: 'str' =None, project_id: 'str' =None, resource_id: 'str' =None, resource_type: 'V1ScheduleResourceType' =None, state: 'str' =None, total: 'int' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, action_type: 'V1ScheduleActionType' =None, command: 'str' =None, created_at: 'datetime' =None, cron_expression: 'str' =None, display_name: 'str' =None, id: 'str' =None, name: 'str' =None, next: 'datetime' =None, parallel_runs: 'bool' =None, parent_resource_id: 'str' =None, project_id: 'str' =None, resource_id: 'str' =None, resource_type: 'V1ScheduleResourceType' =None, state: 'str' =None, timezone: 'str' =None, total: 'int' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Schedule - a model defined in Swagger"""  # noqa: E501
         self._action_type = None
+        self._command = None
         self._created_at = None
         self._cron_expression = None
         self._display_name = None
         self._id = None
         self._name = None
         self._next = None
+        self._parallel_runs = None
         self._parent_resource_id = None
         self._project_id = None
         self._resource_id = None
         self._resource_type = None
         self._state = None
+        self._timezone = None
         self._total = None
         self._updated_at = None
         self._user_id = None
         self.discriminator = None
         if action_type is not None:
             self.action_type = action_type
+        if command is not None:
+            self.command = command
         if created_at is not None:
             self.created_at = created_at
         if cron_expression is not None:
@@ -108,6 +119,8 @@ class V1Schedule(object):
             self.name = name
         if next is not None:
             self.next = next
+        if parallel_runs is not None:
+            self.parallel_runs = parallel_runs
         if parent_resource_id is not None:
             self.parent_resource_id = parent_resource_id
         if project_id is not None:
@@ -118,6 +131,8 @@ class V1Schedule(object):
             self.resource_type = resource_type
         if state is not None:
             self.state = state
+        if timezone is not None:
+            self.timezone = timezone
         if total is not None:
             self.total = total
         if updated_at is not None:
@@ -146,6 +161,27 @@ class V1Schedule(object):
 
         self._action_type = action_type
 
+    @property
+    def command(self) -> 'str':
+        """Gets the command of this V1Schedule.  # noqa: E501
+
+
+        :return: The command of this V1Schedule.  # noqa: E501
+        :rtype: str
+        """
+        return self._command
+
+    @command.setter
+    def command(self, command: 'str'):
+        """Sets the command of this V1Schedule.
+
+
+        :param command: The command of this V1Schedule.  # noqa: E501
+        :type: str
+        """
+
+        self._command = command
+
     @property
     def created_at(self) -> 'datetime':
         """Gets the created_at of this V1Schedule.  # noqa: E501
@@ -272,6 +308,27 @@ class V1Schedule(object):
 
         self._next = next
 
+    @property
+    def parallel_runs(self) -> 'bool':
+        """Gets the parallel_runs of this V1Schedule.  # noqa: E501
+
+
+        :return: The parallel_runs of this V1Schedule.  # noqa: E501
+        :rtype: bool
+        """
+        return self._parallel_runs
+
+    @parallel_runs.setter
+    def parallel_runs(self, parallel_runs: 'bool'):
+        """Sets the parallel_runs of this V1Schedule.
+
+
+        :param parallel_runs: The parallel_runs of this V1Schedule.  # noqa: E501
+        :type: bool
+        """
+
+        self._parallel_runs = parallel_runs
+
     @property
     def parent_resource_id(self) -> 'str':
         """Gets the parent_resource_id of this V1Schedule.  # noqa: E501
@@ -377,6 +434,27 @@ class V1Schedule(object):
 
         self._state = state
 
+    @property
+    def timezone(self) -> 'str':
+        """Gets the timezone of this V1Schedule.  # noqa: E501
+
+
+        :return: The timezone of this V1Schedule.  # noqa: E501
+        :rtype: str
+        """
+        return self._timezone
+
+    @timezone.setter
+    def timezone(self, timezone: 'str'):
+        """Sets the timezone of this V1Schedule.
+
+
+        :param timezone: The timezone of this V1Schedule.  # noqa: E501
+        :type: str
+        """
+
+        self._timezone = timezone
+
     @property
     def total(self) -> 'int':
         """Gets the total of this V1Schedule.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_service_health.py

@@ -41,25 +41,51 @@ class V1ServiceHealth(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'is_fatal': 'bool',
         'name': 'str',
         'status': 'ServiceHealthServiceStatus'
     }
 
     attribute_map = {
+        'is_fatal': 'isFatal',
         'name': 'name',
         'status': 'status'
     }
 
-    def __init__(self, name: 'str' =None, status: 'ServiceHealthServiceStatus' =None):  # noqa: E501
+    def __init__(self, is_fatal: 'bool' =None, name: 'str' =None, status: 'ServiceHealthServiceStatus' =None):  # noqa: E501
         """V1ServiceHealth - a model defined in Swagger"""  # noqa: E501
+        self._is_fatal = None
         self._name = None
         self._status = None
         self.discriminator = None
+        if is_fatal is not None:
+            self.is_fatal = is_fatal
         if name is not None:
             self.name = name
         if status is not None:
             self.status = status
 
+    @property
+    def is_fatal(self) -> 'bool':
+        """Gets the is_fatal of this V1ServiceHealth.  # noqa: E501
+
+
+        :return: The is_fatal of this V1ServiceHealth.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_fatal
+
+    @is_fatal.setter
+    def is_fatal(self, is_fatal: 'bool'):
+        """Sets the is_fatal of this V1ServiceHealth.
+
+
+        :param is_fatal: The is_fatal of this V1ServiceHealth.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_fatal = is_fatal
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1ServiceHealth.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1.py

@@ -41,20 +41,98 @@ class V1SlurmV1(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'login_node_ip': 'str',
+        'ssh_port': 'int',
+        'ssh_user': 'str',
         'work_dir': 'str'
     }
 
     attribute_map = {
+        'login_node_ip': 'loginNodeIp',
+        'ssh_port': 'sshPort',
+        'ssh_user': 'sshUser',
         'work_dir': 'workDir'
     }
 
-    def __init__(self, work_dir: 'str' =None):  # noqa: E501
+    def __init__(self, login_node_ip: 'str' =None, ssh_port: 'int' =None, ssh_user: 'str' =None, work_dir: 'str' =None):  # noqa: E501
         """V1SlurmV1 - a model defined in Swagger"""  # noqa: E501
+        self._login_node_ip = None
+        self._ssh_port = None
+        self._ssh_user = None
         self._work_dir = None
         self.discriminator = None
+        if login_node_ip is not None:
+            self.login_node_ip = login_node_ip
+        if ssh_port is not None:
+            self.ssh_port = ssh_port
+        if ssh_user is not None:
+            self.ssh_user = ssh_user
         if work_dir is not None:
             self.work_dir = work_dir
 
+    @property
+    def login_node_ip(self) -> 'str':
+        """Gets the login_node_ip of this V1SlurmV1.  # noqa: E501
+
+
+        :return: The login_node_ip of this V1SlurmV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._login_node_ip
+
+    @login_node_ip.setter
+    def login_node_ip(self, login_node_ip: 'str'):
+        """Sets the login_node_ip of this V1SlurmV1.
+
+
+        :param login_node_ip: The login_node_ip of this V1SlurmV1.  # noqa: E501
+        :type: str
+        """
+
+        self._login_node_ip = login_node_ip
+
+    @property
+    def ssh_port(self) -> 'int':
+        """Gets the ssh_port of this V1SlurmV1.  # noqa: E501
+
+
+        :return: The ssh_port of this V1SlurmV1.  # noqa: E501
+        :rtype: int
+        """
+        return self._ssh_port
+
+    @ssh_port.setter
+    def ssh_port(self, ssh_port: 'int'):
+        """Sets the ssh_port of this V1SlurmV1.
+
+
+        :param ssh_port: The ssh_port of this V1SlurmV1.  # noqa: E501
+        :type: int
+        """
+
+        self._ssh_port = ssh_port
+
+    @property
+    def ssh_user(self) -> 'str':
+        """Gets the ssh_user of this V1SlurmV1.  # noqa: E501
+
+
+        :return: The ssh_user of this V1SlurmV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._ssh_user
+
+    @ssh_user.setter
+    def ssh_user(self, ssh_user: 'str'):
+        """Sets the ssh_user of this V1SlurmV1.
+
+
+        :param ssh_user: The ssh_user of this V1SlurmV1.  # noqa: E501
+        :type: str
+        """
+
+        self._ssh_user = ssh_user
+
     @property
     def work_dir(self) -> 'str':
         """Gets the work_dir of this V1SlurmV1.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1_status.py

@@ -42,21 +42,36 @@ class V1SlurmV1Status(object):
     """
     swagger_types = {
         'nodes': 'list[V1SlurmNode]',
+        'ssh_host': 'str',
+        'ssh_port': 'int',
+        'ssh_username': 'str',
         'version': 'str'
     }
 
     attribute_map = {
         'nodes': 'nodes',
+        'ssh_host': 'sshHost',
+        'ssh_port': 'sshPort',
+        'ssh_username': 'sshUsername',
         'version': 'version'
     }
 
-    def __init__(self, nodes: 'list[V1SlurmNode]' =None, version: 'str' =None):  # noqa: E501
+    def __init__(self, nodes: 'list[V1SlurmNode]' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, version: 'str' =None):  # noqa: E501
         """V1SlurmV1Status - a model defined in Swagger"""  # noqa: E501
         self._nodes = None
+        self._ssh_host = None
+        self._ssh_port = None
+        self._ssh_username = None
         self._version = None
         self.discriminator = None
         if nodes is not None:
             self.nodes = nodes
+        if ssh_host is not None:
+            self.ssh_host = ssh_host
+        if ssh_port is not None:
+            self.ssh_port = ssh_port
+        if ssh_username is not None:
+            self.ssh_username = ssh_username
         if version is not None:
             self.version = version
 
@@ -81,6 +96,69 @@ class V1SlurmV1Status(object):
 
         self._nodes = nodes
 
+    @property
+    def ssh_host(self) -> 'str':
+        """Gets the ssh_host of this V1SlurmV1Status.  # noqa: E501
+
+
+        :return: The ssh_host of this V1SlurmV1Status.  # noqa: E501
+        :rtype: str
+        """
+        return self._ssh_host
+
+    @ssh_host.setter
+    def ssh_host(self, ssh_host: 'str'):
+        """Sets the ssh_host of this V1SlurmV1Status.
+
+
+        :param ssh_host: The ssh_host of this V1SlurmV1Status.  # noqa: E501
+        :type: str
+        """
+
+        self._ssh_host = ssh_host
+
+    @property
+    def ssh_port(self) -> 'int':
+        """Gets the ssh_port of this V1SlurmV1Status.  # noqa: E501
+
+
+        :return: The ssh_port of this V1SlurmV1Status.  # noqa: E501
+        :rtype: int
+        """
+        return self._ssh_port
+
+    @ssh_port.setter
+    def ssh_port(self, ssh_port: 'int'):
+        """Sets the ssh_port of this V1SlurmV1Status.
+
+
+        :param ssh_port: The ssh_port of this V1SlurmV1Status.  # noqa: E501
+        :type: int
+        """
+
+        self._ssh_port = ssh_port
+
+    @property
+    def ssh_username(self) -> 'str':
+        """Gets the ssh_username of this V1SlurmV1Status.  # noqa: E501
+
+
+        :return: The ssh_username of this V1SlurmV1Status.  # noqa: E501
+        :rtype: str
+        """
+        return self._ssh_username
+
+    @ssh_username.setter
+    def ssh_username(self, ssh_username: 'str'):
+        """Sets the ssh_username of this V1SlurmV1Status.
+
+
+        :param ssh_username: The ssh_username of this V1SlurmV1Status.  # noqa: E501
+        :type: str
+        """
+
+        self._ssh_username = ssh_username
+
     @property
     def version(self) -> 'str':
         """Gets the version of this V1SlurmV1Status.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_tool_call.py

@@ -0,0 +1,175 @@
+# 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 V1ToolCall(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 = {
+        'function': 'V1FunctionCall',
+        'id': 'str',
+        'type': 'str'
+    }
+
+    attribute_map = {
+        'function': 'function',
+        'id': 'id',
+        'type': 'type'
+    }
+
+    def __init__(self, function: 'V1FunctionCall' =None, id: 'str' =None, type: 'str' =None):  # noqa: E501
+        """V1ToolCall - a model defined in Swagger"""  # noqa: E501
+        self._function = None
+        self._id = None
+        self._type = None
+        self.discriminator = None
+        if function is not None:
+            self.function = function
+        if id is not None:
+            self.id = id
+        if type is not None:
+            self.type = type
+
+    @property
+    def function(self) -> 'V1FunctionCall':
+        """Gets the function of this V1ToolCall.  # noqa: E501
+
+
+        :return: The function of this V1ToolCall.  # noqa: E501
+        :rtype: V1FunctionCall
+        """
+        return self._function
+
+    @function.setter
+    def function(self, function: 'V1FunctionCall'):
+        """Sets the function of this V1ToolCall.
+
+
+        :param function: The function of this V1ToolCall.  # noqa: E501
+        :type: V1FunctionCall
+        """
+
+        self._function = function
+
+    @property
+    def id(self) -> 'str':
+        """Gets the id of this V1ToolCall.  # noqa: E501
+
+
+        :return: The id of this V1ToolCall.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id: 'str'):
+        """Sets the id of this V1ToolCall.
+
+
+        :param id: The id of this V1ToolCall.  # noqa: E501
+        :type: str
+        """
+
+        self._id = id
+
+    @property
+    def type(self) -> 'str':
+        """Gets the type of this V1ToolCall.  # noqa: E501
+
+
+        :return: The type of this V1ToolCall.  # noqa: E501
+        :rtype: str
+        """
+        return self._type
+
+    @type.setter
+    def type(self, type: 'str'):
+        """Sets the type of this V1ToolCall.
+
+
+        :param type: The type of this V1ToolCall.  # 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(V1ToolCall, 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: 'V1ToolCall') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ToolCall):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ToolCall') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other