lightning-sdk 0.1.57__py3-none-any.whl → 0.2.0__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 (118) hide show
  1. lightning_sdk/__init__.py +5 -3
  2. lightning_sdk/api/deployment_api.py +23 -11
  3. lightning_sdk/api/job_api.py +42 -7
  4. lightning_sdk/api/lit_container_api.py +88 -22
  5. lightning_sdk/api/mmt_api.py +46 -8
  6. lightning_sdk/api/pipeline_api.py +50 -0
  7. lightning_sdk/api/teamspace_api.py +2 -2
  8. lightning_sdk/api/utils.py +15 -5
  9. lightning_sdk/cli/ai_hub.py +30 -65
  10. lightning_sdk/cli/coloring.py +60 -0
  11. lightning_sdk/cli/configure.py +25 -40
  12. lightning_sdk/cli/connect.py +7 -20
  13. lightning_sdk/cli/create.py +83 -0
  14. lightning_sdk/cli/delete.py +72 -75
  15. lightning_sdk/cli/docker.py +77 -0
  16. lightning_sdk/cli/download.py +71 -111
  17. lightning_sdk/cli/entrypoint.py +44 -65
  18. lightning_sdk/cli/generate.py +28 -43
  19. lightning_sdk/cli/inspect.py +22 -50
  20. lightning_sdk/cli/list.py +281 -222
  21. lightning_sdk/cli/mmts_menu.py +1 -1
  22. lightning_sdk/cli/open.py +62 -0
  23. lightning_sdk/cli/run.py +430 -263
  24. lightning_sdk/cli/serve.py +162 -189
  25. lightning_sdk/cli/start.py +55 -36
  26. lightning_sdk/cli/stop.py +97 -55
  27. lightning_sdk/cli/switch.py +53 -36
  28. lightning_sdk/cli/upload.py +318 -245
  29. lightning_sdk/deployment/__init__.py +2 -0
  30. lightning_sdk/deployment/deployment.py +33 -8
  31. lightning_sdk/lightning_cloud/openapi/__init__.py +21 -0
  32. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  33. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +10 -6
  34. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +355 -4
  35. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +4 -4
  36. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +14 -2
  37. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +670 -0
  38. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +303 -4
  39. lightning_sdk/lightning_cloud/openapi/models/__init__.py +20 -0
  40. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +17 -69
  41. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/create.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +53 -1
  44. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +105 -1
  45. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body1.py +1 -27
  46. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body2.py +149 -0
  47. lightning_sdk/lightning_cloud/openapi/models/org_id_memberships_body.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +157 -1
  49. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +435 -0
  50. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +201 -0
  51. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +157 -1
  52. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +79 -1
  53. lightning_sdk/lightning_cloud/openapi/models/uploads_upload_id_body.py +1 -27
  54. lightning_sdk/lightning_cloud/openapi/models/uploads_upload_id_body1.py +175 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +79 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +17 -69
  57. lightning_sdk/lightning_cloud/openapi/models/v1_capacity_block_offering.py +27 -1
  58. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_artifact_event_type.py +1 -1
  59. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +131 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +79 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_complete_upload_temporary_artifact_request.py +175 -0
  63. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +461 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_template_request.py +27 -1
  65. lightning_sdk/lightning_cloud/openapi/models/v1_create_job_request.py +201 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_create_managed_endpoint_response.py +149 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_create_multi_machine_job_request.py +253 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_delete_pipeline_response.py +149 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +105 -1
  71. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_details.py +175 -0
  72. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template.py +53 -1
  73. lightning_sdk/lightning_cloud/openapi/models/v1_filestore_data_connection.py +201 -0
  74. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_job.py +27 -1
  75. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_mmt.py +27 -1
  76. lightning_sdk/lightning_cloud/openapi/models/v1_find_capacity_block_offering_response.py +29 -3
  77. lightning_sdk/lightning_cloud/openapi/models/v1_job.py +133 -3
  78. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +53 -1
  79. lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py +27 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_list_pipelines_response.py +123 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py +27 -1
  82. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +29 -1
  83. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +27 -1
  84. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +27 -1
  85. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +2 -0
  86. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +157 -1
  87. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +487 -0
  88. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_step.py +253 -0
  89. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_step_status.py +331 -0
  90. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_step_type.py +104 -0
  91. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +157 -1
  92. lightning_sdk/lightning_cloud/openapi/models/v1_restart_timing.py +27 -1
  93. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  94. lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +201 -0
  95. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +27 -1
  96. lightning_sdk/lightning_cloud/openapi/models/v1_update_job_visibility_response.py +97 -0
  97. lightning_sdk/lightning_cloud/openapi/models/v1_upload_temporary_artifact_request.py +123 -0
  98. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +95 -355
  99. lightning_sdk/lightning_cloud/openapi/models/validate.py +27 -1
  100. lightning_sdk/lightning_cloud/rest_client.py +4 -2
  101. lightning_sdk/machine.py +25 -1
  102. lightning_sdk/models.py +18 -12
  103. lightning_sdk/pipeline/__init__.py +4 -0
  104. lightning_sdk/pipeline/pipeline.py +109 -0
  105. lightning_sdk/pipeline/types.py +268 -0
  106. lightning_sdk/pipeline/utils.py +69 -0
  107. lightning_sdk/plugin.py +9 -10
  108. lightning_sdk/services/utilities.py +2 -2
  109. lightning_sdk/studio.py +5 -1
  110. lightning_sdk/teamspace.py +1 -1
  111. lightning_sdk/utils/resolve.py +12 -1
  112. {lightning_sdk-0.1.57.dist-info → lightning_sdk-0.2.0.dist-info}/METADATA +6 -8
  113. {lightning_sdk-0.1.57.dist-info → lightning_sdk-0.2.0.dist-info}/RECORD +117 -88
  114. lightning_sdk/cli/legacy.py +0 -135
  115. {lightning_sdk-0.1.57.dist-info → lightning_sdk-0.2.0.dist-info}/LICENSE +0 -0
  116. {lightning_sdk-0.1.57.dist-info → lightning_sdk-0.2.0.dist-info}/WHEEL +0 -0
  117. {lightning_sdk-0.1.57.dist-info → lightning_sdk-0.2.0.dist-info}/entry_points.txt +0 -0
  118. {lightning_sdk-0.1.57.dist-info → lightning_sdk-0.2.0.dist-info}/top_level.txt +0 -0
@@ -41,9 +41,13 @@ class V1Organization(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'allow_aws_saas': 'bool',
45
+ 'allow_gcp_saas': 'bool',
44
46
  'allow_guest': 'bool',
47
+ 'allow_lambda_saas': 'bool',
45
48
  'allow_member_invitations': 'bool',
46
49
  'allow_member_teamspace_creation': 'bool',
50
+ 'allow_vultr_saas': 'bool',
47
51
  'auto_invite_by_domain': 'bool',
48
52
  'auto_join_domain_validations': 'dict(str, V1AutoJoinDomainValidation)',
49
53
  'auto_join_domains': 'list[str]',
@@ -57,6 +61,8 @@ class V1Organization(object):
57
61
  'location': 'str',
58
62
  'name': 'str',
59
63
  'preferred_cluster': 'str',
64
+ 'preferred_deployment_provider': 'str',
65
+ 'preferred_studio_provider': 'str',
60
66
  'start_studios_on_spot_instance': 'bool',
61
67
  'teamspace_default_credits': 'float',
62
68
  'twitter_username': 'str',
@@ -64,9 +70,13 @@ class V1Organization(object):
64
70
  }
65
71
 
66
72
  attribute_map = {
73
+ 'allow_aws_saas': 'allowAwsSaas',
74
+ 'allow_gcp_saas': 'allowGcpSaas',
67
75
  'allow_guest': 'allowGuest',
76
+ 'allow_lambda_saas': 'allowLambdaSaas',
68
77
  'allow_member_invitations': 'allowMemberInvitations',
69
78
  'allow_member_teamspace_creation': 'allowMemberTeamspaceCreation',
79
+ 'allow_vultr_saas': 'allowVultrSaas',
70
80
  'auto_invite_by_domain': 'autoInviteByDomain',
71
81
  'auto_join_domain_validations': 'autoJoinDomainValidations',
72
82
  'auto_join_domains': 'autoJoinDomains',
@@ -80,17 +90,23 @@ class V1Organization(object):
80
90
  'location': 'location',
81
91
  'name': 'name',
82
92
  'preferred_cluster': 'preferredCluster',
93
+ 'preferred_deployment_provider': 'preferredDeploymentProvider',
94
+ 'preferred_studio_provider': 'preferredStudioProvider',
83
95
  'start_studios_on_spot_instance': 'startStudiosOnSpotInstance',
84
96
  'teamspace_default_credits': 'teamspaceDefaultCredits',
85
97
  'twitter_username': 'twitterUsername',
86
98
  'updated_at': 'updatedAt'
87
99
  }
88
100
 
89
- def __init__(self, allow_guest: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, created_at: 'datetime' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, id: 'str' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None): # noqa: E501
101
+ def __init__(self, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, created_at: 'datetime' =None, description: 'str' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, id: 'str' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None): # noqa: E501
90
102
  """V1Organization - a model defined in Swagger""" # noqa: E501
103
+ self._allow_aws_saas = None
104
+ self._allow_gcp_saas = None
91
105
  self._allow_guest = None
106
+ self._allow_lambda_saas = None
92
107
  self._allow_member_invitations = None
93
108
  self._allow_member_teamspace_creation = None
109
+ self._allow_vultr_saas = None
94
110
  self._auto_invite_by_domain = None
95
111
  self._auto_join_domain_validations = None
96
112
  self._auto_join_domains = None
@@ -104,17 +120,27 @@ class V1Organization(object):
104
120
  self._location = None
105
121
  self._name = None
106
122
  self._preferred_cluster = None
123
+ self._preferred_deployment_provider = None
124
+ self._preferred_studio_provider = None
107
125
  self._start_studios_on_spot_instance = None
108
126
  self._teamspace_default_credits = None
109
127
  self._twitter_username = None
110
128
  self._updated_at = None
111
129
  self.discriminator = None
130
+ if allow_aws_saas is not None:
131
+ self.allow_aws_saas = allow_aws_saas
132
+ if allow_gcp_saas is not None:
133
+ self.allow_gcp_saas = allow_gcp_saas
112
134
  if allow_guest is not None:
113
135
  self.allow_guest = allow_guest
136
+ if allow_lambda_saas is not None:
137
+ self.allow_lambda_saas = allow_lambda_saas
114
138
  if allow_member_invitations is not None:
115
139
  self.allow_member_invitations = allow_member_invitations
116
140
  if allow_member_teamspace_creation is not None:
117
141
  self.allow_member_teamspace_creation = allow_member_teamspace_creation
142
+ if allow_vultr_saas is not None:
143
+ self.allow_vultr_saas = allow_vultr_saas
118
144
  if auto_invite_by_domain is not None:
119
145
  self.auto_invite_by_domain = auto_invite_by_domain
120
146
  if auto_join_domain_validations is not None:
@@ -141,6 +167,10 @@ class V1Organization(object):
141
167
  self.name = name
142
168
  if preferred_cluster is not None:
143
169
  self.preferred_cluster = preferred_cluster
170
+ if preferred_deployment_provider is not None:
171
+ self.preferred_deployment_provider = preferred_deployment_provider
172
+ if preferred_studio_provider is not None:
173
+ self.preferred_studio_provider = preferred_studio_provider
144
174
  if start_studios_on_spot_instance is not None:
145
175
  self.start_studios_on_spot_instance = start_studios_on_spot_instance
146
176
  if teamspace_default_credits is not None:
@@ -150,6 +180,48 @@ class V1Organization(object):
150
180
  if updated_at is not None:
151
181
  self.updated_at = updated_at
152
182
 
183
+ @property
184
+ def allow_aws_saas(self) -> 'bool':
185
+ """Gets the allow_aws_saas of this V1Organization. # noqa: E501
186
+
187
+
188
+ :return: The allow_aws_saas of this V1Organization. # noqa: E501
189
+ :rtype: bool
190
+ """
191
+ return self._allow_aws_saas
192
+
193
+ @allow_aws_saas.setter
194
+ def allow_aws_saas(self, allow_aws_saas: 'bool'):
195
+ """Sets the allow_aws_saas of this V1Organization.
196
+
197
+
198
+ :param allow_aws_saas: The allow_aws_saas of this V1Organization. # noqa: E501
199
+ :type: bool
200
+ """
201
+
202
+ self._allow_aws_saas = allow_aws_saas
203
+
204
+ @property
205
+ def allow_gcp_saas(self) -> 'bool':
206
+ """Gets the allow_gcp_saas of this V1Organization. # noqa: E501
207
+
208
+
209
+ :return: The allow_gcp_saas of this V1Organization. # noqa: E501
210
+ :rtype: bool
211
+ """
212
+ return self._allow_gcp_saas
213
+
214
+ @allow_gcp_saas.setter
215
+ def allow_gcp_saas(self, allow_gcp_saas: 'bool'):
216
+ """Sets the allow_gcp_saas of this V1Organization.
217
+
218
+
219
+ :param allow_gcp_saas: The allow_gcp_saas of this V1Organization. # noqa: E501
220
+ :type: bool
221
+ """
222
+
223
+ self._allow_gcp_saas = allow_gcp_saas
224
+
153
225
  @property
154
226
  def allow_guest(self) -> 'bool':
155
227
  """Gets the allow_guest of this V1Organization. # noqa: E501
@@ -171,6 +243,27 @@ class V1Organization(object):
171
243
 
172
244
  self._allow_guest = allow_guest
173
245
 
246
+ @property
247
+ def allow_lambda_saas(self) -> 'bool':
248
+ """Gets the allow_lambda_saas of this V1Organization. # noqa: E501
249
+
250
+
251
+ :return: The allow_lambda_saas of this V1Organization. # noqa: E501
252
+ :rtype: bool
253
+ """
254
+ return self._allow_lambda_saas
255
+
256
+ @allow_lambda_saas.setter
257
+ def allow_lambda_saas(self, allow_lambda_saas: 'bool'):
258
+ """Sets the allow_lambda_saas of this V1Organization.
259
+
260
+
261
+ :param allow_lambda_saas: The allow_lambda_saas of this V1Organization. # noqa: E501
262
+ :type: bool
263
+ """
264
+
265
+ self._allow_lambda_saas = allow_lambda_saas
266
+
174
267
  @property
175
268
  def allow_member_invitations(self) -> 'bool':
176
269
  """Gets the allow_member_invitations of this V1Organization. # noqa: E501
@@ -213,6 +306,27 @@ class V1Organization(object):
213
306
 
214
307
  self._allow_member_teamspace_creation = allow_member_teamspace_creation
215
308
 
309
+ @property
310
+ def allow_vultr_saas(self) -> 'bool':
311
+ """Gets the allow_vultr_saas of this V1Organization. # noqa: E501
312
+
313
+
314
+ :return: The allow_vultr_saas of this V1Organization. # noqa: E501
315
+ :rtype: bool
316
+ """
317
+ return self._allow_vultr_saas
318
+
319
+ @allow_vultr_saas.setter
320
+ def allow_vultr_saas(self, allow_vultr_saas: 'bool'):
321
+ """Sets the allow_vultr_saas of this V1Organization.
322
+
323
+
324
+ :param allow_vultr_saas: The allow_vultr_saas of this V1Organization. # noqa: E501
325
+ :type: bool
326
+ """
327
+
328
+ self._allow_vultr_saas = allow_vultr_saas
329
+
216
330
  @property
217
331
  def auto_invite_by_domain(self) -> 'bool':
218
332
  """Gets the auto_invite_by_domain of this V1Organization. # noqa: E501
@@ -486,6 +600,48 @@ class V1Organization(object):
486
600
 
487
601
  self._preferred_cluster = preferred_cluster
488
602
 
603
+ @property
604
+ def preferred_deployment_provider(self) -> 'str':
605
+ """Gets the preferred_deployment_provider of this V1Organization. # noqa: E501
606
+
607
+
608
+ :return: The preferred_deployment_provider of this V1Organization. # noqa: E501
609
+ :rtype: str
610
+ """
611
+ return self._preferred_deployment_provider
612
+
613
+ @preferred_deployment_provider.setter
614
+ def preferred_deployment_provider(self, preferred_deployment_provider: 'str'):
615
+ """Sets the preferred_deployment_provider of this V1Organization.
616
+
617
+
618
+ :param preferred_deployment_provider: The preferred_deployment_provider of this V1Organization. # noqa: E501
619
+ :type: str
620
+ """
621
+
622
+ self._preferred_deployment_provider = preferred_deployment_provider
623
+
624
+ @property
625
+ def preferred_studio_provider(self) -> 'str':
626
+ """Gets the preferred_studio_provider of this V1Organization. # noqa: E501
627
+
628
+
629
+ :return: The preferred_studio_provider of this V1Organization. # noqa: E501
630
+ :rtype: str
631
+ """
632
+ return self._preferred_studio_provider
633
+
634
+ @preferred_studio_provider.setter
635
+ def preferred_studio_provider(self, preferred_studio_provider: 'str'):
636
+ """Sets the preferred_studio_provider of this V1Organization.
637
+
638
+
639
+ :param preferred_studio_provider: The preferred_studio_provider of this V1Organization. # noqa: E501
640
+ :type: str
641
+ """
642
+
643
+ self._preferred_studio_provider = preferred_studio_provider
644
+
489
645
  @property
490
646
  def start_studios_on_spot_instance(self) -> 'bool':
491
647
  """Gets the start_studios_on_spot_instance of this V1Organization. # noqa: E501