lightning-sdk 2026.1.22__py3-none-any.whl → 2026.1.27__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (58) hide show
  1. lightning_sdk/__version__.py +1 -1
  2. lightning_sdk/api/studio_api.py +7 -14
  3. lightning_sdk/api/teamspace_api.py +155 -48
  4. lightning_sdk/api/utils.py +8 -0
  5. lightning_sdk/cli/cp/__init__.py +14 -11
  6. lightning_sdk/cli/cp/teamspace_uploads.py +93 -0
  7. lightning_sdk/cli/legacy/download.py +29 -98
  8. lightning_sdk/cli/legacy/upload.py +24 -31
  9. lightning_sdk/cli/studio/cp.py +8 -5
  10. lightning_sdk/cli/studio/ls.py +1 -1
  11. lightning_sdk/cli/studio/rm.py +1 -1
  12. lightning_sdk/cli/utils/{studio_filesystem.py → filesystem.py} +43 -5
  13. lightning_sdk/exceptions.py +27 -0
  14. lightning_sdk/lightning_cloud/openapi/__init__.py +14 -12
  15. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  16. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +5 -1
  17. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +163 -466
  18. lightning_sdk/lightning_cloud/openapi/api/container_registry_service_api.py +456 -0
  19. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +5 -1
  20. lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py +11 -11
  21. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +113 -0
  22. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +113 -0
  23. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +5 -1
  24. lightning_sdk/lightning_cloud/openapi/models/__init__.py +13 -12
  25. lightning_sdk/lightning_cloud/openapi/models/{v1_container_registry_integration.py → container_registry_config_ecr.py} +49 -23
  26. lightning_sdk/lightning_cloud/openapi/models/{v1_container_registry_status.py → container_registry_provider.py} +14 -10
  27. lightning_sdk/lightning_cloud/openapi/models/container_registry_service_create_container_registry_body.py +201 -0
  28. lightning_sdk/lightning_cloud/openapi/models/{v1_ecr_registry_config_input.py → container_registry_service_refresh_container_registry_credentials_body.py} +21 -21
  29. lightning_sdk/lightning_cloud/openapi/models/jobs_service_duplicate_deployment_body.py +175 -0
  30. lightning_sdk/lightning_cloud/openapi/models/organizations_service_update_org_role_body.py +175 -0
  31. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_container_registry.py +63 -89
  35. lightning_sdk/lightning_cloud/openapi/models/{cluster_service_add_container_registry_body.py → v1_container_registry_config.py} +16 -16
  36. lightning_sdk/lightning_cloud/openapi/models/{v1_validate_container_registry_response.py → v1_container_registry_scopes.py} +39 -39
  37. lightning_sdk/lightning_cloud/openapi/models/{cluster_service_validate_container_registry_body.py → v1_create_container_registry_response.py} +6 -6
  38. lightning_sdk/lightning_cloud/openapi/models/{cluster_service_refresh_container_registry_credentials_body.py → v1_delete_org_cluster_capacity_reservation_response.py} +6 -6
  39. lightning_sdk/lightning_cloud/openapi/models/v1_describe_org_cluster_capacity_reservation_response.py +201 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_generic_job_spec.py +79 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_job.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +1 -27
  43. lightning_sdk/lightning_cloud/openapi/models/v1_list_container_registries_response.py +6 -6
  44. lightning_sdk/lightning_cloud/openapi/models/{v1_ecr_registry_config.py → v1_mithril_direct_v1.py} +51 -51
  45. lightning_sdk/lightning_cloud/openapi/models/v1_refresh_container_registry_credentials_response.py +1 -27
  46. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +53 -53
  47. lightning_sdk/lightning_cloud/openapi/rest.py +2 -2
  48. lightning_sdk/teamspace.py +28 -7
  49. {lightning_sdk-2026.1.22.dist-info → lightning_sdk-2026.1.27.dist-info}/METADATA +1 -1
  50. {lightning_sdk-2026.1.22.dist-info → lightning_sdk-2026.1.27.dist-info}/RECORD +55 -52
  51. lightning_sdk/lightning_cloud/openapi/models/v1_add_container_registry_response.py +0 -123
  52. lightning_sdk/lightning_cloud/openapi/models/v1_container_registry_info.py +0 -281
  53. lightning_sdk/lightning_cloud/openapi/models/v1_ecr_registry_details.py +0 -201
  54. /lightning_sdk/lightning_cloud/openapi/models/{v1_list_filesystem_mmts_response.py → v1_list_filesystem_mm_ts_response.py} +0 -0
  55. {lightning_sdk-2026.1.22.dist-info → lightning_sdk-2026.1.27.dist-info}/LICENSE +0 -0
  56. {lightning_sdk-2026.1.22.dist-info → lightning_sdk-2026.1.27.dist-info}/WHEEL +0 -0
  57. {lightning_sdk-2026.1.22.dist-info → lightning_sdk-2026.1.27.dist-info}/entry_points.txt +0 -0
  58. {lightning_sdk-2026.1.22.dist-info → lightning_sdk-2026.1.27.dist-info}/top_level.txt +0 -0
@@ -1950,6 +1950,119 @@ class OrganizationsServiceApi(object):
1950
1950
  _request_timeout=params.get('_request_timeout'),
1951
1951
  collection_formats=collection_formats)
1952
1952
 
1953
+ def organizations_service_update_org_role(self, body: 'OrganizationsServiceUpdateOrgRoleBody', org_id: 'str', id: 'str', **kwargs) -> 'V1OrgRole': # noqa: E501
1954
+ """organizations_service_update_org_role # noqa: E501
1955
+
1956
+ This method makes a synchronous HTTP request by default. To make an
1957
+ asynchronous HTTP request, please pass async_req=True
1958
+ >>> thread = api.organizations_service_update_org_role(body, org_id, id, async_req=True)
1959
+ >>> result = thread.get()
1960
+
1961
+ :param async_req bool
1962
+ :param OrganizationsServiceUpdateOrgRoleBody body: (required)
1963
+ :param str org_id: (required)
1964
+ :param str id: (required)
1965
+ :return: V1OrgRole
1966
+ If the method is called asynchronously,
1967
+ returns the request thread.
1968
+ """
1969
+ kwargs['_return_http_data_only'] = True
1970
+ if kwargs.get('async_req'):
1971
+ return self.organizations_service_update_org_role_with_http_info(body, org_id, id, **kwargs) # noqa: E501
1972
+ else:
1973
+ (data) = self.organizations_service_update_org_role_with_http_info(body, org_id, id, **kwargs) # noqa: E501
1974
+ return data
1975
+
1976
+ def organizations_service_update_org_role_with_http_info(self, body: 'OrganizationsServiceUpdateOrgRoleBody', org_id: 'str', id: 'str', **kwargs) -> 'V1OrgRole': # noqa: E501
1977
+ """organizations_service_update_org_role # noqa: E501
1978
+
1979
+ This method makes a synchronous HTTP request by default. To make an
1980
+ asynchronous HTTP request, please pass async_req=True
1981
+ >>> thread = api.organizations_service_update_org_role_with_http_info(body, org_id, id, async_req=True)
1982
+ >>> result = thread.get()
1983
+
1984
+ :param async_req bool
1985
+ :param OrganizationsServiceUpdateOrgRoleBody body: (required)
1986
+ :param str org_id: (required)
1987
+ :param str id: (required)
1988
+ :return: V1OrgRole
1989
+ If the method is called asynchronously,
1990
+ returns the request thread.
1991
+ """
1992
+
1993
+ all_params = ['body', 'org_id', 'id'] # noqa: E501
1994
+ all_params.append('async_req')
1995
+ all_params.append('_return_http_data_only')
1996
+ all_params.append('_preload_content')
1997
+ all_params.append('_request_timeout')
1998
+
1999
+ params = locals()
2000
+ for key, val in six.iteritems(params['kwargs']):
2001
+ if key not in all_params:
2002
+ raise TypeError(
2003
+ "Got an unexpected keyword argument '%s'"
2004
+ " to method organizations_service_update_org_role" % key
2005
+ )
2006
+ params[key] = val
2007
+ del params['kwargs']
2008
+ # verify the required parameter 'body' is set
2009
+ if ('body' not in params or
2010
+ params['body'] is None):
2011
+ raise ValueError("Missing the required parameter `body` when calling `organizations_service_update_org_role`") # noqa: E501
2012
+ # verify the required parameter 'org_id' is set
2013
+ if ('org_id' not in params or
2014
+ params['org_id'] is None):
2015
+ raise ValueError("Missing the required parameter `org_id` when calling `organizations_service_update_org_role`") # noqa: E501
2016
+ # verify the required parameter 'id' is set
2017
+ if ('id' not in params or
2018
+ params['id'] is None):
2019
+ raise ValueError("Missing the required parameter `id` when calling `organizations_service_update_org_role`") # noqa: E501
2020
+
2021
+ collection_formats = {}
2022
+
2023
+ path_params = {}
2024
+ if 'org_id' in params:
2025
+ path_params['orgId'] = params['org_id'] # noqa: E501
2026
+ if 'id' in params:
2027
+ path_params['id'] = params['id'] # noqa: E501
2028
+
2029
+ query_params = []
2030
+
2031
+ header_params = {}
2032
+
2033
+ form_params = []
2034
+ local_var_files = {}
2035
+
2036
+ body_params = None
2037
+ if 'body' in params:
2038
+ body_params = params['body']
2039
+ # HTTP header `Accept`
2040
+ header_params['Accept'] = self.api_client.select_header_accept(
2041
+ ['application/json']) # noqa: E501
2042
+
2043
+ # HTTP header `Content-Type`
2044
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
2045
+ ['application/json']) # noqa: E501
2046
+
2047
+ # Authentication setting
2048
+ auth_settings = [] # noqa: E501
2049
+
2050
+ return self.api_client.call_api(
2051
+ '/v1/orgs/{orgId}/roles/{id}', 'PUT',
2052
+ path_params,
2053
+ query_params,
2054
+ header_params,
2055
+ body=body_params,
2056
+ post_params=form_params,
2057
+ files=local_var_files,
2058
+ response_type='V1OrgRole', # noqa: E501
2059
+ auth_settings=auth_settings,
2060
+ async_req=params.get('async_req'),
2061
+ _return_http_data_only=params.get('_return_http_data_only'),
2062
+ _preload_content=params.get('_preload_content', True),
2063
+ _request_timeout=params.get('_request_timeout'),
2064
+ collection_formats=collection_formats)
2065
+
1953
2066
  def organizations_service_update_organization(self, body: 'OrganizationsServiceUpdateOrganizationBody', id: 'str', **kwargs) -> 'V1Organization': # noqa: E501
1954
2067
  """Update organization # noqa: E501
1955
2068
 
@@ -872,6 +872,7 @@ class StorageServiceApi(object):
872
872
  :param str cluster_id:
873
873
  :param bool local_index: if true, fetch from the local index
874
874
  :param bool include_folder_index: if true, include the folder index in the response
875
+ :param bool only_direct_children: if true, only include direct children of the prefix
875
876
  :return: V1GetArtifactsPageResponse
876
877
  If the method is called asynchronously,
877
878
  returns the request thread.
@@ -900,12 +901,13 @@ class StorageServiceApi(object):
900
901
  :param str cluster_id:
901
902
  :param bool local_index: if true, fetch from the local index
902
903
  :param bool include_folder_index: if true, include the folder index in the response
904
+ :param bool only_direct_children: if true, only include direct children of the prefix
903
905
  :return: V1GetArtifactsPageResponse
904
906
  If the method is called asynchronously,
905
907
  returns the request thread.
906
908
  """
907
909
 
908
- all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index', 'include_folder_index'] # noqa: E501
910
+ all_params = ['project_id', 'id', 'page_number', 'prefix', 'include_download_url', 'cluster_id', 'local_index', 'include_folder_index', 'only_direct_children'] # noqa: E501
909
911
  all_params.append('async_req')
910
912
  all_params.append('_return_http_data_only')
911
913
  all_params.append('_preload_content')
@@ -946,6 +948,8 @@ class StorageServiceApi(object):
946
948
  query_params.append(('localIndex', params['local_index'])) # noqa: E501
947
949
  if 'include_folder_index' in params:
948
950
  query_params.append(('includeFolderIndex', params['include_folder_index'])) # noqa: E501
951
+ if 'only_direct_children' in params:
952
+ query_params.append(('onlyDirectChildren', params['only_direct_children'])) # noqa: E501
949
953
 
950
954
  header_params = {}
951
955
 
@@ -81,7 +81,6 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_clo
81
81
  from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_version_publication_body import CloudSpaceServiceUpdateCloudSpaceVersionPublicationBody
82
82
  from lightning_sdk.lightning_cloud.openapi.models.cloud_space_service_update_cloud_space_visibility_body import CloudSpaceServiceUpdateCloudSpaceVisibilityBody
83
83
  from lightning_sdk.lightning_cloud.openapi.models.cloudy_service_update_user_cloudy_settings_body import CloudyServiceUpdateUserCloudySettingsBody
84
- from lightning_sdk.lightning_cloud.openapi.models.cluster_service_add_container_registry_body import ClusterServiceAddContainerRegistryBody
85
84
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_capacity_reservation_body import ClusterServiceCreateClusterCapacityReservationBody
86
85
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_proxy_body import ClusterServiceCreateClusterProxyBody
87
86
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_cluster_usage_restriction_body import ClusterServiceCreateClusterUsageRestrictionBody
@@ -91,14 +90,16 @@ from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_project
91
90
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_create_server_alert_body import ClusterServiceCreateServerAlertBody
92
91
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_interrupt_server_body import ClusterServiceInterruptServerBody
93
92
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_purchase_capacity_block_body import ClusterServicePurchaseCapacityBlockBody
94
- from lightning_sdk.lightning_cloud.openapi.models.cluster_service_refresh_container_registry_credentials_body import ClusterServiceRefreshContainerRegistryCredentialsBody
95
93
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_report_machine_system_metrics_body import ClusterServiceReportMachineSystemMetricsBody
96
94
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_server_check_in_body import ClusterServiceServerCheckInBody
97
95
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_sleep_server_body import ClusterServiceSleepServerBody
98
96
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_body import ClusterServiceUpdateClusterBody
99
97
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_cluster_usage_restriction_body import ClusterServiceUpdateClusterUsageRestrictionBody
100
98
  from lightning_sdk.lightning_cloud.openapi.models.cluster_service_update_project_cluster_body import ClusterServiceUpdateProjectClusterBody
101
- from lightning_sdk.lightning_cloud.openapi.models.cluster_service_validate_container_registry_body import ClusterServiceValidateContainerRegistryBody
99
+ from lightning_sdk.lightning_cloud.openapi.models.container_registry_config_ecr import ContainerRegistryConfigECR
100
+ from lightning_sdk.lightning_cloud.openapi.models.container_registry_provider import ContainerRegistryProvider
101
+ from lightning_sdk.lightning_cloud.openapi.models.container_registry_service_create_container_registry_body import ContainerRegistryServiceCreateContainerRegistryBody
102
+ from lightning_sdk.lightning_cloud.openapi.models.container_registry_service_refresh_container_registry_credentials_body import ContainerRegistryServiceRefreshContainerRegistryCredentialsBody
102
103
  from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
103
104
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
104
105
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_service_create_data_connection_body import DataConnectionServiceCreateDataConnectionBody
@@ -126,6 +127,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_deployment
126
127
  from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_deployment_body import JobsServiceCreateDeploymentBody
127
128
  from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_job_body import JobsServiceCreateJobBody
128
129
  from lightning_sdk.lightning_cloud.openapi.models.jobs_service_create_multi_machine_job_body import JobsServiceCreateMultiMachineJobBody
130
+ from lightning_sdk.lightning_cloud.openapi.models.jobs_service_duplicate_deployment_body import JobsServiceDuplicateDeploymentBody
129
131
  from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_deployment_routing_telemetry_body import JobsServiceReportDeploymentRoutingTelemetryBody
130
132
  from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_job_system_metrics_body import JobsServiceReportJobSystemMetricsBody
131
133
  from lightning_sdk.lightning_cloud.openapi.models.jobs_service_report_logs_activity_body import JobsServiceReportLogsActivityBody
@@ -190,6 +192,7 @@ from lightning_sdk.lightning_cloud.openapi.models.organizations_service_auto_joi
190
192
  from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_membership_body import OrganizationsServiceCreateOrgMembershipBody
191
193
  from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_membership_role_binding_body import OrganizationsServiceCreateOrgMembershipRoleBindingBody
192
194
  from lightning_sdk.lightning_cloud.openapi.models.organizations_service_create_org_role_body import OrganizationsServiceCreateOrgRoleBody
195
+ from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_org_role_body import OrganizationsServiceUpdateOrgRoleBody
193
196
  from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_organization_body import OrganizationsServiceUpdateOrganizationBody
194
197
  from lightning_sdk.lightning_cloud.openapi.models.organizations_service_update_organization_credits_auto_replenish_body import OrganizationsServiceUpdateOrganizationCreditsAutoReplenishBody
195
198
  from lightning_sdk.lightning_cloud.openapi.models.organizations_service_validate_auto_join_domain_body import OrganizationsServiceValidateAutoJoinDomainBody
@@ -251,7 +254,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import
251
254
  from lightning_sdk.lightning_cloud.openapi.models.v1_abort_storage_transfer_response import V1AbortStorageTransferResponse
252
255
  from lightning_sdk.lightning_cloud.openapi.models.v1_accelerator_quota_info import V1AcceleratorQuotaInfo
253
256
  from lightning_sdk.lightning_cloud.openapi.models.v1_ack_user_storage_violation_response import V1AckUserStorageViolationResponse
254
- from lightning_sdk.lightning_cloud.openapi.models.v1_add_container_registry_response import V1AddContainerRegistryResponse
255
257
  from lightning_sdk.lightning_cloud.openapi.models.v1_add_job_timing_response import V1AddJobTimingResponse
256
258
  from lightning_sdk.lightning_cloud.openapi.models.v1_affiliate_link import V1AffiliateLink
257
259
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_complete_part_response import V1AgentCompletePartResponse
@@ -370,9 +372,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_rea
370
372
  from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_response import V1ContactAssistantOwnerResponse
371
373
  from lightning_sdk.lightning_cloud.openapi.models.v1_container_metrics import V1ContainerMetrics
372
374
  from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry import V1ContainerRegistry
373
- from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_info import V1ContainerRegistryInfo
374
- from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_integration import V1ContainerRegistryIntegration
375
- from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_status import V1ContainerRegistryStatus
375
+ from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_config import V1ContainerRegistryConfig
376
+ from lightning_sdk.lightning_cloud.openapi.models.v1_container_registry_scopes import V1ContainerRegistryScopes
376
377
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation import V1Conversation
377
378
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
378
379
  from lightning_sdk.lightning_cloud.openapi.models.v1_count_metrics_streams_response import V1CountMetricsStreamsResponse
@@ -391,6 +392,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_k
391
392
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_keys_response import V1CreateClusterEncryptionKeysResponse
392
393
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_request import V1CreateClusterRequest
393
394
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_response import V1CreateClusterResponse
395
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_container_registry_response import V1CreateContainerRegistryResponse
394
396
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_request import V1CreateDeploymentRequest
395
397
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_template_request import V1CreateDeploymentTemplateRequest
396
398
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_git_credentials_request import V1CreateGitCredentialsRequest
@@ -477,6 +479,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_respo
477
479
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
478
480
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_version_response import V1DeleteModelVersionResponse
479
481
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_multi_machine_job_response import V1DeleteMultiMachineJobResponse
482
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_cluster_capacity_reservation_response import V1DeleteOrgClusterCapacityReservationResponse
480
483
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_response import V1DeleteOrgMembershipResponse
481
484
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_role_binding_response import V1DeleteOrgMembershipRoleBindingResponse
482
485
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_role_response import V1DeleteOrgRoleResponse
@@ -526,6 +529,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_paramet
526
529
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_parameter_type import V1DeploymentTemplateParameterType
527
530
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_summary import V1DeploymentTemplateSummary
528
531
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_type import V1DeploymentTemplateType
532
+ from lightning_sdk.lightning_cloud.openapi.models.v1_describe_org_cluster_capacity_reservation_response import V1DescribeOrgClusterCapacityReservationResponse
529
533
  from lightning_sdk.lightning_cloud.openapi.models.v1_download_job_logs_response import V1DownloadJobLogsResponse
530
534
  from lightning_sdk.lightning_cloud.openapi.models.v1_download_lightningapp_instance_logs_response import V1DownloadLightningappInstanceLogsResponse
531
535
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive import V1Drive
@@ -536,9 +540,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
536
540
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
537
541
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
538
542
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
539
- from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config import V1ECRRegistryConfig
540
- from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_config_input import V1ECRRegistryConfigInput
541
- from lightning_sdk.lightning_cloud.openapi.models.v1_ecr_registry_details import V1ECRRegistryDetails
542
543
  from lightning_sdk.lightning_cloud.openapi.models.v1_efs_config import V1EfsConfig
543
544
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
544
545
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_auth import V1EndpointAuth
@@ -759,7 +760,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_respon
759
760
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_spaces_response import V1ListFilesystemCloudSpacesResponse
760
761
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
761
762
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
762
- from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mmts_response import V1ListFilesystemMMTsResponse
763
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mm_ts_response import V1ListFilesystemMMTsResponse
763
764
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_metrics_response import V1ListFilesystemMetricsResponse
764
765
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_slurm_jobs_response import V1ListFilesystemSlurmJobsResponse
765
766
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_response import V1ListFilesystemSnowflakeResponse
@@ -876,6 +877,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics import V1Metrics
876
877
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_stream import V1MetricsStream
877
878
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tags import V1MetricsTags
878
879
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1MetricsTracker
880
+ from lightning_sdk.lightning_cloud.openapi.models.v1_mithril_direct_v1 import V1MithrilDirectV1
879
881
  from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
880
882
  from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
881
883
  from lightning_sdk.lightning_cloud.openapi.models.v1_model_metrics import V1ModelMetrics
@@ -1126,7 +1128,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_user_requested_flow_compute
1126
1128
  from lightning_sdk.lightning_cloud.openapi.models.v1_user_slurm_job_action_response import V1UserSLURMJobActionResponse
1127
1129
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_assistant_status_response import V1ValidateAssistantStatusResponse
1128
1130
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_auto_join_domain_response import V1ValidateAutoJoinDomainResponse
1129
- from lightning_sdk.lightning_cloud.openapi.models.v1_validate_container_registry_response import V1ValidateContainerRegistryResponse
1130
1131
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_data_connection_response import V1ValidateDataConnectionResponse
1131
1132
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_request import V1ValidateDeploymentImageRequest
1132
1133
  from lightning_sdk.lightning_cloud.openapi.models.v1_validate_deployment_image_response import V1ValidateDeploymentImageResponse
@@ -28,7 +28,7 @@ if TYPE_CHECKING:
28
28
  from datetime import datetime
29
29
  from lightning_sdk.lightning_cloud.openapi.models import *
30
30
 
31
- class V1ContainerRegistryIntegration(object):
31
+ class ContainerRegistryConfigECR(object):
32
32
  """NOTE: This class is auto generated by the swagger code generator program.
33
33
 
34
34
  Do not edit the class manually.
@@ -41,40 +41,66 @@ class V1ContainerRegistryIntegration(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'registries': 'list[V1ContainerRegistry]'
44
+ 'last_refresh_at': 'datetime',
45
+ 'token_refresh_seconds': 'str'
45
46
  }
46
47
 
47
48
  attribute_map = {
48
- 'registries': 'registries'
49
+ 'last_refresh_at': 'lastRefreshAt',
50
+ 'token_refresh_seconds': 'tokenRefreshSeconds'
49
51
  }
50
52
 
51
- def __init__(self, registries: 'list[V1ContainerRegistry]' =None): # noqa: E501
52
- """V1ContainerRegistryIntegration - a model defined in Swagger""" # noqa: E501
53
- self._registries = None
53
+ def __init__(self, last_refresh_at: 'datetime' =None, token_refresh_seconds: 'str' =None): # noqa: E501
54
+ """ContainerRegistryConfigECR - a model defined in Swagger""" # noqa: E501
55
+ self._last_refresh_at = None
56
+ self._token_refresh_seconds = None
54
57
  self.discriminator = None
55
- if registries is not None:
56
- self.registries = registries
58
+ if last_refresh_at is not None:
59
+ self.last_refresh_at = last_refresh_at
60
+ if token_refresh_seconds is not None:
61
+ self.token_refresh_seconds = token_refresh_seconds
57
62
 
58
63
  @property
59
- def registries(self) -> 'list[V1ContainerRegistry]':
60
- """Gets the registries of this V1ContainerRegistryIntegration. # noqa: E501
64
+ def last_refresh_at(self) -> 'datetime':
65
+ """Gets the last_refresh_at of this ContainerRegistryConfigECR. # noqa: E501
61
66
 
62
67
 
63
- :return: The registries of this V1ContainerRegistryIntegration. # noqa: E501
64
- :rtype: list[V1ContainerRegistry]
68
+ :return: The last_refresh_at of this ContainerRegistryConfigECR. # noqa: E501
69
+ :rtype: datetime
65
70
  """
66
- return self._registries
71
+ return self._last_refresh_at
67
72
 
68
- @registries.setter
69
- def registries(self, registries: 'list[V1ContainerRegistry]'):
70
- """Sets the registries of this V1ContainerRegistryIntegration.
73
+ @last_refresh_at.setter
74
+ def last_refresh_at(self, last_refresh_at: 'datetime'):
75
+ """Sets the last_refresh_at of this ContainerRegistryConfigECR.
71
76
 
72
77
 
73
- :param registries: The registries of this V1ContainerRegistryIntegration. # noqa: E501
74
- :type: list[V1ContainerRegistry]
78
+ :param last_refresh_at: The last_refresh_at of this ContainerRegistryConfigECR. # noqa: E501
79
+ :type: datetime
75
80
  """
76
81
 
77
- self._registries = registries
82
+ self._last_refresh_at = last_refresh_at
83
+
84
+ @property
85
+ def token_refresh_seconds(self) -> 'str':
86
+ """Gets the token_refresh_seconds of this ContainerRegistryConfigECR. # noqa: E501
87
+
88
+
89
+ :return: The token_refresh_seconds of this ContainerRegistryConfigECR. # noqa: E501
90
+ :rtype: str
91
+ """
92
+ return self._token_refresh_seconds
93
+
94
+ @token_refresh_seconds.setter
95
+ def token_refresh_seconds(self, token_refresh_seconds: 'str'):
96
+ """Sets the token_refresh_seconds of this ContainerRegistryConfigECR.
97
+
98
+
99
+ :param token_refresh_seconds: The token_refresh_seconds of this ContainerRegistryConfigECR. # noqa: E501
100
+ :type: str
101
+ """
102
+
103
+ self._token_refresh_seconds = token_refresh_seconds
78
104
 
79
105
  def to_dict(self) -> dict:
80
106
  """Returns the model properties as a dict"""
@@ -97,7 +123,7 @@ class V1ContainerRegistryIntegration(object):
97
123
  ))
98
124
  else:
99
125
  result[attr] = value
100
- if issubclass(V1ContainerRegistryIntegration, dict):
126
+ if issubclass(ContainerRegistryConfigECR, dict):
101
127
  for key, value in self.items():
102
128
  result[key] = value
103
129
 
@@ -111,13 +137,13 @@ class V1ContainerRegistryIntegration(object):
111
137
  """For `print` and `pprint`"""
112
138
  return self.to_str()
113
139
 
114
- def __eq__(self, other: 'V1ContainerRegistryIntegration') -> bool:
140
+ def __eq__(self, other: 'ContainerRegistryConfigECR') -> bool:
115
141
  """Returns true if both objects are equal"""
116
- if not isinstance(other, V1ContainerRegistryIntegration):
142
+ if not isinstance(other, ContainerRegistryConfigECR):
117
143
  return False
118
144
 
119
145
  return self.__dict__ == other.__dict__
120
146
 
121
- def __ne__(self, other: 'V1ContainerRegistryIntegration') -> bool:
147
+ def __ne__(self, other: 'ContainerRegistryConfigECR') -> bool:
122
148
  """Returns true if both objects are not equal"""
123
149
  return not self == other
@@ -28,7 +28,7 @@ if TYPE_CHECKING:
28
28
  from datetime import datetime
29
29
  from lightning_sdk.lightning_cloud.openapi.models import *
30
30
 
31
- class V1ContainerRegistryStatus(object):
31
+ class ContainerRegistryProvider(object):
32
32
  """NOTE: This class is auto generated by the swagger code generator program.
33
33
 
34
34
  Do not edit the class manually.
@@ -37,10 +37,14 @@ class V1ContainerRegistryStatus(object):
37
37
  """
38
38
  allowed enum values
39
39
  """
40
- UNSPECIFIED = "CONTAINER_REGISTRY_STATUS_UNSPECIFIED"
41
- PENDING = "CONTAINER_REGISTRY_STATUS_PENDING"
42
- VALIDATED = "CONTAINER_REGISTRY_STATUS_VALIDATED"
43
- ERROR = "CONTAINER_REGISTRY_STATUS_ERROR"
40
+ UNSPECIFIED = "Unspecified"
41
+ ECR = "ECR"
42
+ LITCR = "LitCR"
43
+ GHCR = "GHCR"
44
+ DOCKERHUB = "DockerHub"
45
+ GCR = "GCR"
46
+ GAR = "GAR"
47
+ ACR = "ACR"
44
48
  """
45
49
  Attributes:
46
50
  swagger_types (dict): The key is attribute name
@@ -55,7 +59,7 @@ class V1ContainerRegistryStatus(object):
55
59
  }
56
60
 
57
61
  def __init__(self): # noqa: E501
58
- """V1ContainerRegistryStatus - a model defined in Swagger""" # noqa: E501
62
+ """ContainerRegistryProvider - a model defined in Swagger""" # noqa: E501
59
63
  self.discriminator = None
60
64
 
61
65
  def to_dict(self) -> dict:
@@ -79,7 +83,7 @@ class V1ContainerRegistryStatus(object):
79
83
  ))
80
84
  else:
81
85
  result[attr] = value
82
- if issubclass(V1ContainerRegistryStatus, dict):
86
+ if issubclass(ContainerRegistryProvider, dict):
83
87
  for key, value in self.items():
84
88
  result[key] = value
85
89
 
@@ -93,13 +97,13 @@ class V1ContainerRegistryStatus(object):
93
97
  """For `print` and `pprint`"""
94
98
  return self.to_str()
95
99
 
96
- def __eq__(self, other: 'V1ContainerRegistryStatus') -> bool:
100
+ def __eq__(self, other: 'ContainerRegistryProvider') -> bool:
97
101
  """Returns true if both objects are equal"""
98
- if not isinstance(other, V1ContainerRegistryStatus):
102
+ if not isinstance(other, ContainerRegistryProvider):
99
103
  return False
100
104
 
101
105
  return self.__dict__ == other.__dict__
102
106
 
103
- def __ne__(self, other: 'V1ContainerRegistryStatus') -> bool:
107
+ def __ne__(self, other: 'ContainerRegistryProvider') -> bool:
104
108
  """Returns true if both objects are not equal"""
105
109
  return not self == other
@@ -0,0 +1,201 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class ContainerRegistryServiceCreateContainerRegistryBody(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'org_id': 'str',
45
+ 'provider': 'ContainerRegistryProvider',
46
+ 'scopes': 'V1ContainerRegistryScopes',
47
+ 'url': 'str'
48
+ }
49
+
50
+ attribute_map = {
51
+ 'org_id': 'orgId',
52
+ 'provider': 'provider',
53
+ 'scopes': 'scopes',
54
+ 'url': 'url'
55
+ }
56
+
57
+ def __init__(self, org_id: 'str' =None, provider: 'ContainerRegistryProvider' =None, scopes: 'V1ContainerRegistryScopes' =None, url: 'str' =None): # noqa: E501
58
+ """ContainerRegistryServiceCreateContainerRegistryBody - a model defined in Swagger""" # noqa: E501
59
+ self._org_id = None
60
+ self._provider = None
61
+ self._scopes = None
62
+ self._url = None
63
+ self.discriminator = None
64
+ if org_id is not None:
65
+ self.org_id = org_id
66
+ if provider is not None:
67
+ self.provider = provider
68
+ if scopes is not None:
69
+ self.scopes = scopes
70
+ if url is not None:
71
+ self.url = url
72
+
73
+ @property
74
+ def org_id(self) -> 'str':
75
+ """Gets the org_id of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
76
+
77
+
78
+ :return: The org_id of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
79
+ :rtype: str
80
+ """
81
+ return self._org_id
82
+
83
+ @org_id.setter
84
+ def org_id(self, org_id: 'str'):
85
+ """Sets the org_id of this ContainerRegistryServiceCreateContainerRegistryBody.
86
+
87
+
88
+ :param org_id: The org_id of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
89
+ :type: str
90
+ """
91
+
92
+ self._org_id = org_id
93
+
94
+ @property
95
+ def provider(self) -> 'ContainerRegistryProvider':
96
+ """Gets the provider of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
97
+
98
+
99
+ :return: The provider of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
100
+ :rtype: ContainerRegistryProvider
101
+ """
102
+ return self._provider
103
+
104
+ @provider.setter
105
+ def provider(self, provider: 'ContainerRegistryProvider'):
106
+ """Sets the provider of this ContainerRegistryServiceCreateContainerRegistryBody.
107
+
108
+
109
+ :param provider: The provider of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
110
+ :type: ContainerRegistryProvider
111
+ """
112
+
113
+ self._provider = provider
114
+
115
+ @property
116
+ def scopes(self) -> 'V1ContainerRegistryScopes':
117
+ """Gets the scopes of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
118
+
119
+
120
+ :return: The scopes of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
121
+ :rtype: V1ContainerRegistryScopes
122
+ """
123
+ return self._scopes
124
+
125
+ @scopes.setter
126
+ def scopes(self, scopes: 'V1ContainerRegistryScopes'):
127
+ """Sets the scopes of this ContainerRegistryServiceCreateContainerRegistryBody.
128
+
129
+
130
+ :param scopes: The scopes of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
131
+ :type: V1ContainerRegistryScopes
132
+ """
133
+
134
+ self._scopes = scopes
135
+
136
+ @property
137
+ def url(self) -> 'str':
138
+ """Gets the url of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
139
+
140
+
141
+ :return: The url of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
142
+ :rtype: str
143
+ """
144
+ return self._url
145
+
146
+ @url.setter
147
+ def url(self, url: 'str'):
148
+ """Sets the url of this ContainerRegistryServiceCreateContainerRegistryBody.
149
+
150
+
151
+ :param url: The url of this ContainerRegistryServiceCreateContainerRegistryBody. # noqa: E501
152
+ :type: str
153
+ """
154
+
155
+ self._url = url
156
+
157
+ def to_dict(self) -> dict:
158
+ """Returns the model properties as a dict"""
159
+ result = {}
160
+
161
+ for attr, _ in six.iteritems(self.swagger_types):
162
+ value = getattr(self, attr)
163
+ if isinstance(value, list):
164
+ result[attr] = list(map(
165
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
166
+ value
167
+ ))
168
+ elif hasattr(value, "to_dict"):
169
+ result[attr] = value.to_dict()
170
+ elif isinstance(value, dict):
171
+ result[attr] = dict(map(
172
+ lambda item: (item[0], item[1].to_dict())
173
+ if hasattr(item[1], "to_dict") else item,
174
+ value.items()
175
+ ))
176
+ else:
177
+ result[attr] = value
178
+ if issubclass(ContainerRegistryServiceCreateContainerRegistryBody, dict):
179
+ for key, value in self.items():
180
+ result[key] = value
181
+
182
+ return result
183
+
184
+ def to_str(self) -> str:
185
+ """Returns the string representation of the model"""
186
+ return pprint.pformat(self.to_dict())
187
+
188
+ def __repr__(self) -> str:
189
+ """For `print` and `pprint`"""
190
+ return self.to_str()
191
+
192
+ def __eq__(self, other: 'ContainerRegistryServiceCreateContainerRegistryBody') -> bool:
193
+ """Returns true if both objects are equal"""
194
+ if not isinstance(other, ContainerRegistryServiceCreateContainerRegistryBody):
195
+ return False
196
+
197
+ return self.__dict__ == other.__dict__
198
+
199
+ def __ne__(self, other: 'ContainerRegistryServiceCreateContainerRegistryBody') -> bool:
200
+ """Returns true if both objects are not equal"""
201
+ return not self == other