lightning-sdk 2025.8.6rc2__py3-none-any.whl → 2025.8.8__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py

@@ -148,6 +148,127 @@ class SchedulesServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def schedules_service_create_schedule_run(self, body: 'object', project_id: 'str', cloudspace_id: 'str', schedule_id: 'str', **kwargs) -> 'V1ScheduleRun':  # noqa: E501
+        """schedules_service_create_schedule_run  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.schedules_service_create_schedule_run(body, project_id, cloudspace_id, schedule_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param object body: (required)
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str schedule_id: (required)
+        :return: V1ScheduleRun
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.schedules_service_create_schedule_run_with_http_info(body, project_id, cloudspace_id, schedule_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.schedules_service_create_schedule_run_with_http_info(body, project_id, cloudspace_id, schedule_id, **kwargs)  # noqa: E501
+            return data
+
+    def schedules_service_create_schedule_run_with_http_info(self, body: 'object', project_id: 'str', cloudspace_id: 'str', schedule_id: 'str', **kwargs) -> 'V1ScheduleRun':  # noqa: E501
+        """schedules_service_create_schedule_run  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.schedules_service_create_schedule_run_with_http_info(body, project_id, cloudspace_id, schedule_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param object body: (required)
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str schedule_id: (required)
+        :return: V1ScheduleRun
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['body', 'project_id', 'cloudspace_id', 'schedule_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method schedules_service_create_schedule_run" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'body' is set
+        if ('body' not in params or
+                params['body'] is None):
+            raise ValueError("Missing the required parameter `body` when calling `schedules_service_create_schedule_run`")  # noqa: E501
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `schedules_service_create_schedule_run`")  # noqa: E501
+        # verify the required parameter 'cloudspace_id' is set
+        if ('cloudspace_id' not in params or
+                params['cloudspace_id'] is None):
+            raise ValueError("Missing the required parameter `cloudspace_id` when calling `schedules_service_create_schedule_run`")  # noqa: E501
+        # verify the required parameter 'schedule_id' is set
+        if ('schedule_id' not in params or
+                params['schedule_id'] is None):
+            raise ValueError("Missing the required parameter `schedule_id` when calling `schedules_service_create_schedule_run`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'cloudspace_id' in params:
+            path_params['cloudspaceId'] = params['cloudspace_id']  # noqa: E501
+        if 'schedule_id' in params:
+            path_params['scheduleId'] = params['schedule_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'body' in params:
+            body_params = params['body']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/cloudspaces/{cloudspaceId}/schedules/{scheduleId}/runs', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ScheduleRun',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def schedules_service_delete_schedule(self, project_id: 'str', id: 'str', **kwargs) -> 'V1DeleteScheduleResponse':  # noqa: E501
         """schedules_service_delete_schedule  # noqa: E501
 
@@ -350,6 +471,232 @@ class SchedulesServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def schedules_service_get_schedule_run(self, project_id: 'str', cloudspace_id: 'str', schedule_id: 'str', run_id: 'str', **kwargs) -> 'V1ScheduleRun':  # noqa: E501
+        """schedules_service_get_schedule_run  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.schedules_service_get_schedule_run(project_id, cloudspace_id, schedule_id, run_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str schedule_id: (required)
+        :param str run_id: (required)
+        :return: V1ScheduleRun
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.schedules_service_get_schedule_run_with_http_info(project_id, cloudspace_id, schedule_id, run_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.schedules_service_get_schedule_run_with_http_info(project_id, cloudspace_id, schedule_id, run_id, **kwargs)  # noqa: E501
+            return data
+
+    def schedules_service_get_schedule_run_with_http_info(self, project_id: 'str', cloudspace_id: 'str', schedule_id: 'str', run_id: 'str', **kwargs) -> 'V1ScheduleRun':  # noqa: E501
+        """schedules_service_get_schedule_run  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.schedules_service_get_schedule_run_with_http_info(project_id, cloudspace_id, schedule_id, run_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str schedule_id: (required)
+        :param str run_id: (required)
+        :return: V1ScheduleRun
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cloudspace_id', 'schedule_id', 'run_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method schedules_service_get_schedule_run" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `schedules_service_get_schedule_run`")  # noqa: E501
+        # verify the required parameter 'cloudspace_id' is set
+        if ('cloudspace_id' not in params or
+                params['cloudspace_id'] is None):
+            raise ValueError("Missing the required parameter `cloudspace_id` when calling `schedules_service_get_schedule_run`")  # noqa: E501
+        # verify the required parameter 'schedule_id' is set
+        if ('schedule_id' not in params or
+                params['schedule_id'] is None):
+            raise ValueError("Missing the required parameter `schedule_id` when calling `schedules_service_get_schedule_run`")  # noqa: E501
+        # verify the required parameter 'run_id' is set
+        if ('run_id' not in params or
+                params['run_id'] is None):
+            raise ValueError("Missing the required parameter `run_id` when calling `schedules_service_get_schedule_run`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'cloudspace_id' in params:
+            path_params['cloudspaceId'] = params['cloudspace_id']  # noqa: E501
+        if 'schedule_id' in params:
+            path_params['scheduleId'] = params['schedule_id']  # noqa: E501
+        if 'run_id' in params:
+            path_params['runId'] = params['run_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/cloudspaces/{cloudspaceId}/schedules/{scheduleId}/runs/{runId}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ScheduleRun',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
+    def schedules_service_list_schedule_runs(self, project_id: 'str', cloudspace_id: 'str', schedule_id: 'str', **kwargs) -> 'V1ListScheduleRunsResponse':  # noqa: E501
+        """schedules_service_list_schedule_runs  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.schedules_service_list_schedule_runs(project_id, cloudspace_id, schedule_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str schedule_id: (required)
+        :return: V1ListScheduleRunsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.schedules_service_list_schedule_runs_with_http_info(project_id, cloudspace_id, schedule_id, **kwargs)  # noqa: E501
+        else:
+            (data) = self.schedules_service_list_schedule_runs_with_http_info(project_id, cloudspace_id, schedule_id, **kwargs)  # noqa: E501
+            return data
+
+    def schedules_service_list_schedule_runs_with_http_info(self, project_id: 'str', cloudspace_id: 'str', schedule_id: 'str', **kwargs) -> 'V1ListScheduleRunsResponse':  # noqa: E501
+        """schedules_service_list_schedule_runs  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.schedules_service_list_schedule_runs_with_http_info(project_id, cloudspace_id, schedule_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str project_id: (required)
+        :param str cloudspace_id: (required)
+        :param str schedule_id: (required)
+        :return: V1ListScheduleRunsResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['project_id', 'cloudspace_id', 'schedule_id']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method schedules_service_list_schedule_runs" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'project_id' is set
+        if ('project_id' not in params or
+                params['project_id'] is None):
+            raise ValueError("Missing the required parameter `project_id` when calling `schedules_service_list_schedule_runs`")  # noqa: E501
+        # verify the required parameter 'cloudspace_id' is set
+        if ('cloudspace_id' not in params or
+                params['cloudspace_id'] is None):
+            raise ValueError("Missing the required parameter `cloudspace_id` when calling `schedules_service_list_schedule_runs`")  # noqa: E501
+        # verify the required parameter 'schedule_id' is set
+        if ('schedule_id' not in params or
+                params['schedule_id'] is None):
+            raise ValueError("Missing the required parameter `schedule_id` when calling `schedules_service_list_schedule_runs`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'project_id' in params:
+            path_params['projectId'] = params['project_id']  # noqa: E501
+        if 'cloudspace_id' in params:
+            path_params['cloudspaceId'] = params['cloudspace_id']  # noqa: E501
+        if 'schedule_id' in params:
+            path_params['scheduleId'] = params['schedule_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/projects/{projectId}/cloudspaces/{cloudspaceId}/schedules/{scheduleId}/runs', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1ListScheduleRunsResponse',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def schedules_service_list_schedules(self, project_id: 'str', **kwargs) -> 'V1ListSchedulesResponse':  # noqa: E501
         """schedules_service_list_schedules  # noqa: E501
 

lightning_sdk/lightning_cloud/openapi/configuration.py

@@ -1,22 +1,5 @@
 # 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
-"""
-
 from __future__ import absolute_import
 
 import copy
@@ -100,8 +83,9 @@ class Configuration(six.with_metaclass(TypeWithDefault, object)):
         # per pool. urllib3 uses 1 connection as default value, but this is
         # not the best value when you are making a lot of possibly parallel
         # requests to the same host, which is often the case here.
-        # cpu_count * 5 is used as default value to increase performance.
-        self.connection_pool_maxsize = multiprocessing.cpu_count() * 5
+        # the max of cpu_count * 5 and 500 is used as default value to increase performance.
+        # each re-used connection can use memory somewhere in the range of 50–100 KB by default.
+        self.connection_pool_maxsize = max(multiprocessing.cpu_count() * 5, 500)
 
         # Proxy URL
         self.proxy = None

lightning_sdk/lightning_cloud/openapi/models/__init__.py

@@ -278,6 +278,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_deletion_options import V1ClusterDeletionOptions
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_encryption_key import V1ClusterEncryptionKey
+from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_metrics import V1ClusterMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_names import V1ClusterNames
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_proxy import V1ClusterProxy
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_resource_tag import V1ClusterResourceTag
@@ -682,6 +683,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_managed_endp
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_quests_response import V1ListQuestsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_slurm_cluster_users_response import V1ListSLURMClusterUsersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_response import V1ListSSHPublicKeysResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedule_runs_response import V1ListScheduleRunsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedules_response import V1ListSchedulesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response import V1ListSecretsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response import V1ListStudioJobsResponse
@@ -836,6 +838,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSH
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule import V1Schedule
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_action_type import V1ScheduleActionType
 from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_resource_type import V1ScheduleResourceType
+from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_run import V1ScheduleRun
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_job_logs_response import V1SearchJobLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_user import V1SearchUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_users_response import V1SearchUsersResponse

lightning_sdk/lightning_cloud/openapi/models/cluster_id_metrics_body.py

@@ -41,23 +41,28 @@ class ClusterIdMetricsBody(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'cluster_metrics': 'V1ClusterMetrics',
         'container_metrics': 'list[V1ContainerMetrics]',
         'node_metrics': 'list[V1NodeMetrics]',
         'pod_metrics': 'list[V1PodMetrics]'
     }
 
     attribute_map = {
+        'cluster_metrics': 'clusterMetrics',
         'container_metrics': 'containerMetrics',
         'node_metrics': 'nodeMetrics',
         'pod_metrics': 'podMetrics'
     }
 
-    def __init__(self, container_metrics: 'list[V1ContainerMetrics]' =None, node_metrics: 'list[V1NodeMetrics]' =None, pod_metrics: 'list[V1PodMetrics]' =None):  # noqa: E501
+    def __init__(self, cluster_metrics: 'V1ClusterMetrics' =None, container_metrics: 'list[V1ContainerMetrics]' =None, node_metrics: 'list[V1NodeMetrics]' =None, pod_metrics: 'list[V1PodMetrics]' =None):  # noqa: E501
         """ClusterIdMetricsBody - a model defined in Swagger"""  # noqa: E501
+        self._cluster_metrics = None
         self._container_metrics = None
         self._node_metrics = None
         self._pod_metrics = None
         self.discriminator = None
+        if cluster_metrics is not None:
+            self.cluster_metrics = cluster_metrics
         if container_metrics is not None:
             self.container_metrics = container_metrics
         if node_metrics is not None:
@@ -65,6 +70,27 @@ class ClusterIdMetricsBody(object):
         if pod_metrics is not None:
             self.pod_metrics = pod_metrics
 
+    @property
+    def cluster_metrics(self) -> 'V1ClusterMetrics':
+        """Gets the cluster_metrics of this ClusterIdMetricsBody.  # noqa: E501
+
+
+        :return: The cluster_metrics of this ClusterIdMetricsBody.  # noqa: E501
+        :rtype: V1ClusterMetrics
+        """
+        return self._cluster_metrics
+
+    @cluster_metrics.setter
+    def cluster_metrics(self, cluster_metrics: 'V1ClusterMetrics'):
+        """Sets the cluster_metrics of this ClusterIdMetricsBody.
+
+
+        :param cluster_metrics: The cluster_metrics of this ClusterIdMetricsBody.  # noqa: E501
+        :type: V1ClusterMetrics
+        """
+
+        self._cluster_metrics = cluster_metrics
+
     @property
     def container_metrics(self) -> 'list[V1ContainerMetrics]':
         """Gets the container_metrics of this ClusterIdMetricsBody.  # noqa: E501