lightning-sdk 0.1.37__py3-none-any.whl → 0.1.38__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 (62) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +21 -23
  3. lightning_sdk/api/ai_hub_api.py +29 -4
  4. lightning_sdk/api/job_api.py +6 -2
  5. lightning_sdk/api/teamspace_api.py +18 -14
  6. lightning_sdk/api/utils.py +19 -0
  7. lightning_sdk/cli/ai_hub.py +1 -1
  8. lightning_sdk/cli/run.py +8 -0
  9. lightning_sdk/job/base.py +13 -0
  10. lightning_sdk/job/job.py +8 -0
  11. lightning_sdk/job/v1.py +5 -1
  12. lightning_sdk/job/v2.py +4 -0
  13. lightning_sdk/lightning_cloud/openapi/__init__.py +13 -2
  14. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +5 -1
  15. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +680 -62
  16. lightning_sdk/lightning_cloud/openapi/models/__init__.py +13 -2
  17. lightning_sdk/lightning_cloud/openapi/models/create.py +6 -32
  18. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +32 -6
  19. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +27 -1
  20. lightning_sdk/lightning_cloud/openapi/models/multimachinejobs_id_body.py +123 -0
  21. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +53 -1
  22. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  23. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +201 -0
  24. lightning_sdk/lightning_cloud/openapi/models/update.py +6 -32
  25. lightning_sdk/lightning_cloud/openapi/models/v1_api_pricing_spec.py +149 -0
  26. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_template_request.py +32 -6
  28. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +6 -32
  29. lightning_sdk/lightning_cloud/openapi/models/v1_delete_multi_machine_job_response.py +97 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py +43 -17
  31. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_performance.py +305 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template.py +32 -6
  33. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_type.py +1 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_summary.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/{v1_efs_data_connection.py → v1_efs_config.py} +22 -22
  37. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +53 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +125 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_jobs_response.py +123 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_machines_selector.py +149 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +6 -6
  43. lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py +6 -6
  44. lightning_sdk/lightning_cloud/openapi/models/v1_message_content_type.py +103 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +53 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +383 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +108 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +279 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +2 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_system_info.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +53 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +6 -32
  54. lightning_sdk/lightning_cloud/openapi/models/validate.py +6 -32
  55. lightning_sdk/teamspace.py +5 -0
  56. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.38.dist-info}/METADATA +1 -1
  57. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.38.dist-info}/RECORD +61 -50
  58. lightning_sdk/lightning_cloud/openapi/models/v1_efs_folder_data_connection.py +0 -201
  59. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.38.dist-info}/LICENSE +0 -0
  60. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.38.dist-info}/WHEEL +0 -0
  61. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.38.dist-info}/entry_points.txt +0 -0
  62. {lightning_sdk-0.1.37.dist-info → lightning_sdk-0.1.38.dist-info}/top_level.txt +0 -0
@@ -102,6 +102,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body impo
102
102
  from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
103
103
  from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
104
104
  from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
105
+ from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
105
106
  from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
106
107
  from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
107
108
  from lightning_sdk.lightning_cloud.openapi.models.org_id_roles_body import OrgIdRolesBody
@@ -121,6 +122,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import
121
122
  from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
122
123
  from lightning_sdk.lightning_cloud.openapi.models.project_id_memberships_body import ProjectIdMembershipsBody
123
124
  from lightning_sdk.lightning_cloud.openapi.models.project_id_models_body import ProjectIdModelsBody
125
+ from lightning_sdk.lightning_cloud.openapi.models.project_id_multimachinejobs_body import ProjectIdMultimachinejobsBody
124
126
  from lightning_sdk.lightning_cloud.openapi.models.project_id_projectclustersbindings_body import ProjectIdProjectclustersbindingsBody
125
127
  from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
126
128
  from lightning_sdk.lightning_cloud.openapi.models.project_id_serviceexecution_body import ProjectIdServiceexecutionBody
@@ -164,6 +166,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job import V1AgentJob
164
166
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V1AgentJobArtifact
165
167
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
166
168
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
169
+ from lightning_sdk.lightning_cloud.openapi.models.v1_api_pricing_spec import V1ApiPricingSpec
167
170
  from lightning_sdk.lightning_cloud.openapi.models.v1_app_type import V1AppType
168
171
  from lightning_sdk.lightning_cloud.openapi.models.v1_append_logger_metrics_response import V1AppendLoggerMetricsResponse
169
172
  from lightning_sdk.lightning_cloud.openapi.models.v1_approve_auto_join_domain_response import V1ApproveAutoJoinDomainResponse
@@ -304,6 +307,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_managed_model_respon
304
307
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_metrics_stream_response import V1DeleteMetricsStreamResponse
305
308
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_response import V1DeleteModelResponse
306
309
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_model_version_response import V1DeleteModelVersionResponse
310
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_multi_machine_job_response import V1DeleteMultiMachineJobResponse
307
311
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_response import V1DeleteOrgMembershipResponse
308
312
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_role_binding_response import V1DeleteOrgMembershipRoleBindingResponse
309
313
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_role_response import V1DeleteOrgRoleResponse
@@ -328,6 +332,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deploym
328
332
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event import V1DeploymentEvent
329
333
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event_type import V1DeploymentEventType
330
334
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_metrics import V1DeploymentMetrics
335
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_performance import V1DeploymentPerformance
331
336
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_release import V1DeploymentRelease
332
337
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_spec import V1DeploymentSpec
333
338
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_state import V1DeploymentState
@@ -352,9 +357,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_drive_status import V1Drive
352
357
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type import V1DriveType
353
358
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_spec import V1DriveTypeSpec
354
359
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_type_status import V1DriveTypeStatus
355
- from lightning_sdk.lightning_cloud.openapi.models.v1_efs_folder_data_connection import V1EFSFolderDataConnection
356
360
  from lightning_sdk.lightning_cloud.openapi.models.v1_ebs import V1Ebs
357
- from lightning_sdk.lightning_cloud.openapi.models.v1_efs_data_connection import V1EfsDataConnection
361
+ from lightning_sdk.lightning_cloud.openapi.models.v1_efs_config import V1EfsConfig
358
362
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint import V1Endpoint
359
363
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_auth import V1EndpointAuth
360
364
  from lightning_sdk.lightning_cloud.openapi.models.v1_endpoint_prewarm import V1EndpointPrewarm
@@ -442,6 +446,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_timing import V1JobTimi
442
446
  from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
443
447
  from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
444
448
  from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
449
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
445
450
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
446
451
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
447
452
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_basic_auth import V1LightningBasicAuth
@@ -516,6 +521,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response i
516
521
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_metrics_streams_response import V1ListMetricsStreamsResponse
517
522
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_response import V1ListModelVersionsResponse
518
523
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
524
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
519
525
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_new_features_for_user_response import V1ListNewFeaturesForUserResponse
520
526
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
521
527
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_membership_role_binding_response import V1ListOrgMembershipRoleBindingResponse
@@ -551,6 +557,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_login_response import V1Log
551
557
  from lightning_sdk.lightning_cloud.openapi.models.v1_logout_request import V1LogoutRequest
552
558
  from lightning_sdk.lightning_cloud.openapi.models.v1_logout_response import V1LogoutResponse
553
559
  from lightning_sdk.lightning_cloud.openapi.models.v1_logs_response import V1LogsResponse
560
+ from lightning_sdk.lightning_cloud.openapi.models.v1_machines_selector import V1MachinesSelector
554
561
  from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_request import V1MagicLinkLoginRequest
555
562
  from lightning_sdk.lightning_cloud.openapi.models.v1_magic_link_login_response import V1MagicLinkLoginResponse
556
563
  from lightning_sdk.lightning_cloud.openapi.models.v1_managed_endpoint import V1ManagedEndpoint
@@ -559,6 +566,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_membership import V1Members
559
566
  from lightning_sdk.lightning_cloud.openapi.models.v1_message import V1Message
560
567
  from lightning_sdk.lightning_cloud.openapi.models.v1_message_author import V1MessageAuthor
561
568
  from lightning_sdk.lightning_cloud.openapi.models.v1_message_content import V1MessageContent
569
+ from lightning_sdk.lightning_cloud.openapi.models.v1_message_content_type import V1MessageContentType
562
570
  from lightning_sdk.lightning_cloud.openapi.models.v1_metadata import V1Metadata
563
571
  from lightning_sdk.lightning_cloud.openapi.models.v1_metric_value import V1MetricValue
564
572
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics import V1Metrics
@@ -568,6 +576,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1Me
568
576
  from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
569
577
  from lightning_sdk.lightning_cloud.openapi.models.v1_model_version_archive import V1ModelVersionArchive
570
578
  from lightning_sdk.lightning_cloud.openapi.models.v1_mount_target import V1MountTarget
579
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job import V1MultiMachineJob
580
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_state import V1MultiMachineJobState
581
+ from lightning_sdk.lightning_cloud.openapi.models.v1_multi_machine_job_status import V1MultiMachineJobStatus
571
582
  from lightning_sdk.lightning_cloud.openapi.models.v1_named_get_logger_metrics import V1NamedGetLoggerMetrics
572
583
  from lightning_sdk.lightning_cloud.openapi.models.v1_network_config import V1NetworkConfig
573
584
  from lightning_sdk.lightning_cloud.openapi.models.v1_new_feature import V1NewFeature
@@ -46,8 +46,7 @@ class Create(object):
46
46
  'cluster_id': 'str',
47
47
  'create_index': 'bool',
48
48
  'create_resources': 'bool',
49
- 'efs': 'V1EfsDataConnection',
50
- 'efs_folder': 'V1EFSFolderDataConnection',
49
+ 'efs': 'V1EfsConfig',
51
50
  'force': 'bool',
52
51
  'gcp': 'V1GcpDataConnection',
53
52
  'gcs_folder': 'V1GCSFolderDataConnection',
@@ -66,7 +65,6 @@ class Create(object):
66
65
  'create_index': 'createIndex',
67
66
  'create_resources': 'createResources',
68
67
  'efs': 'efs',
69
- 'efs_folder': 'efsFolder',
70
68
  'force': 'force',
71
69
  'gcp': 'gcp',
72
70
  'gcs_folder': 'gcsFolder',
@@ -78,7 +76,7 @@ class Create(object):
78
76
  'writable': 'writable'
79
77
  }
80
78
 
81
- def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None): # noqa: E501
79
+ def __init__(self, access_cluster_ids: 'list[str]' =None, aws: 'V1AwsDataConnection' =None, cluster_id: 'str' =None, create_index: 'bool' =None, create_resources: 'bool' =None, efs: 'V1EfsConfig' =None, force: 'bool' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, id: 'str' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None): # noqa: E501
82
80
  """Create - a model defined in Swagger""" # noqa: E501
83
81
  self._access_cluster_ids = None
84
82
  self._aws = None
@@ -86,7 +84,6 @@ class Create(object):
86
84
  self._create_index = None
87
85
  self._create_resources = None
88
86
  self._efs = None
89
- self._efs_folder = None
90
87
  self._force = None
91
88
  self._gcp = None
92
89
  self._gcs_folder = None
@@ -109,8 +106,6 @@ class Create(object):
109
106
  self.create_resources = create_resources
110
107
  if efs is not None:
111
108
  self.efs = efs
112
- if efs_folder is not None:
113
- self.efs_folder = efs_folder
114
109
  if force is not None:
115
110
  self.force = force
116
111
  if gcp is not None:
@@ -236,47 +231,26 @@ class Create(object):
236
231
  self._create_resources = create_resources
237
232
 
238
233
  @property
239
- def efs(self) -> 'V1EfsDataConnection':
234
+ def efs(self) -> 'V1EfsConfig':
240
235
  """Gets the efs of this Create. # noqa: E501
241
236
 
242
237
 
243
238
  :return: The efs of this Create. # noqa: E501
244
- :rtype: V1EfsDataConnection
239
+ :rtype: V1EfsConfig
245
240
  """
246
241
  return self._efs
247
242
 
248
243
  @efs.setter
249
- def efs(self, efs: 'V1EfsDataConnection'):
244
+ def efs(self, efs: 'V1EfsConfig'):
250
245
  """Sets the efs of this Create.
251
246
 
252
247
 
253
248
  :param efs: The efs of this Create. # noqa: E501
254
- :type: V1EfsDataConnection
249
+ :type: V1EfsConfig
255
250
  """
256
251
 
257
252
  self._efs = efs
258
253
 
259
- @property
260
- def efs_folder(self) -> 'V1EFSFolderDataConnection':
261
- """Gets the efs_folder of this Create. # noqa: E501
262
-
263
-
264
- :return: The efs_folder of this Create. # noqa: E501
265
- :rtype: V1EFSFolderDataConnection
266
- """
267
- return self._efs_folder
268
-
269
- @efs_folder.setter
270
- def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
271
- """Sets the efs_folder of this Create.
272
-
273
-
274
- :param efs_folder: The efs_folder of this Create. # noqa: E501
275
- :type: V1EFSFolderDataConnection
276
- """
277
-
278
- self._efs_folder = efs_folder
279
-
280
254
  @property
281
255
  def force(self) -> 'bool':
282
256
  """Gets the force of this Create. # noqa: E501
@@ -46,10 +46,11 @@ class DeploymenttemplatesIdBody(object):
46
46
  'description': 'str',
47
47
  'featured': 'bool',
48
48
  'image_url': 'str',
49
- 'metrics': 'list[V1DeploymentMetrics]',
49
+ 'metrics': 'V1DeploymentMetrics',
50
50
  'name': 'str',
51
51
  'org_id': 'str',
52
52
  'parameter_spec': 'V1ParameterizationSpec',
53
+ 'pricing': 'V1ApiPricingSpec',
53
54
  'spec': 'str',
54
55
  'tags': 'list[V1ResourceTag]',
55
56
  'thumbnail': 'str',
@@ -67,6 +68,7 @@ class DeploymenttemplatesIdBody(object):
67
68
  'name': 'name',
68
69
  'org_id': 'orgId',
69
70
  'parameter_spec': 'parameterSpec',
71
+ 'pricing': 'pricing',
70
72
  'spec': 'spec',
71
73
  'tags': 'tags',
72
74
  'thumbnail': 'thumbnail',
@@ -74,7 +76,7 @@ class DeploymenttemplatesIdBody(object):
74
76
  'visibility': 'visibility'
75
77
  }
76
78
 
77
- def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, description: 'str' =None, featured: 'bool' =None, image_url: 'str' =None, metrics: 'list[V1DeploymentMetrics]' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, visibility: 'V1DeploymentTemplateType' =None): # noqa: E501
79
+ def __init__(self, about_page_content: 'str' =None, categories: 'list[str]' =None, description: 'str' =None, featured: 'bool' =None, image_url: 'str' =None, metrics: 'V1DeploymentMetrics' =None, name: 'str' =None, org_id: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, pricing: 'V1ApiPricingSpec' =None, spec: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, visibility: 'V1DeploymentTemplateType' =None): # noqa: E501
78
80
  """DeploymenttemplatesIdBody - a model defined in Swagger""" # noqa: E501
79
81
  self._about_page_content = None
80
82
  self._categories = None
@@ -85,6 +87,7 @@ class DeploymenttemplatesIdBody(object):
85
87
  self._name = None
86
88
  self._org_id = None
87
89
  self._parameter_spec = None
90
+ self._pricing = None
88
91
  self._spec = None
89
92
  self._tags = None
90
93
  self._thumbnail = None
@@ -109,6 +112,8 @@ class DeploymenttemplatesIdBody(object):
109
112
  self.org_id = org_id
110
113
  if parameter_spec is not None:
111
114
  self.parameter_spec = parameter_spec
115
+ if pricing is not None:
116
+ self.pricing = pricing
112
117
  if spec is not None:
113
118
  self.spec = spec
114
119
  if tags is not None:
@@ -226,22 +231,22 @@ class DeploymenttemplatesIdBody(object):
226
231
  self._image_url = image_url
227
232
 
228
233
  @property
229
- def metrics(self) -> 'list[V1DeploymentMetrics]':
234
+ def metrics(self) -> 'V1DeploymentMetrics':
230
235
  """Gets the metrics of this DeploymenttemplatesIdBody. # noqa: E501
231
236
 
232
237
 
233
238
  :return: The metrics of this DeploymenttemplatesIdBody. # noqa: E501
234
- :rtype: list[V1DeploymentMetrics]
239
+ :rtype: V1DeploymentMetrics
235
240
  """
236
241
  return self._metrics
237
242
 
238
243
  @metrics.setter
239
- def metrics(self, metrics: 'list[V1DeploymentMetrics]'):
244
+ def metrics(self, metrics: 'V1DeploymentMetrics'):
240
245
  """Sets the metrics of this DeploymenttemplatesIdBody.
241
246
 
242
247
 
243
248
  :param metrics: The metrics of this DeploymenttemplatesIdBody. # noqa: E501
244
- :type: list[V1DeploymentMetrics]
249
+ :type: V1DeploymentMetrics
245
250
  """
246
251
 
247
252
  self._metrics = metrics
@@ -309,6 +314,27 @@ class DeploymenttemplatesIdBody(object):
309
314
 
310
315
  self._parameter_spec = parameter_spec
311
316
 
317
+ @property
318
+ def pricing(self) -> 'V1ApiPricingSpec':
319
+ """Gets the pricing of this DeploymenttemplatesIdBody. # noqa: E501
320
+
321
+
322
+ :return: The pricing of this DeploymenttemplatesIdBody. # noqa: E501
323
+ :rtype: V1ApiPricingSpec
324
+ """
325
+ return self._pricing
326
+
327
+ @pricing.setter
328
+ def pricing(self, pricing: 'V1ApiPricingSpec'):
329
+ """Sets the pricing of this DeploymenttemplatesIdBody.
330
+
331
+
332
+ :param pricing: The pricing of this DeploymenttemplatesIdBody. # noqa: E501
333
+ :type: V1ApiPricingSpec
334
+ """
335
+
336
+ self._pricing = pricing
337
+
312
338
  @property
313
339
  def spec(self) -> 'str':
314
340
  """Gets the spec of this DeploymenttemplatesIdBody. # noqa: E501
@@ -68,6 +68,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
68
68
  'sync_eta_seconds': 'str',
69
69
  'sync_in_progress': 'bool',
70
70
  'sync_percentage': 'str',
71
+ 'termination_time': 'datetime',
71
72
  'vscode_url': 'str'
72
73
  }
73
74
 
@@ -99,10 +100,11 @@ class Externalv1CloudSpaceInstanceStatus(object):
99
100
  'sync_eta_seconds': 'syncEtaSeconds',
100
101
  'sync_in_progress': 'syncInProgress',
101
102
  'sync_percentage': 'syncPercentage',
103
+ 'termination_time': 'terminationTime',
102
104
  'vscode_url': 'vscodeUrl'
103
105
  }
104
106
 
105
- def __init__(self, app_url: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, price: 'float' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_eta_seconds: 'str' =None, sync_in_progress: 'bool' =None, sync_percentage: 'str' =None, vscode_url: 'str' =None): # noqa: E501
107
+ def __init__(self, app_url: 'str' =None, cloud_space_id: 'str' =None, cloud_space_instance_id: 'str' =None, compute_config: 'V1UserRequestedComputeConfig' =None, creation_timestamp: 'datetime' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, free: 'bool' =None, ide: 'str' =None, instance_id: 'str' =None, instance_region: 'str' =None, instance_url: 'str' =None, jupyterlab_url: 'str' =None, phase: 'V1CloudSpaceInstanceState' =None, price: 'float' =None, ssh_host: 'str' =None, ssh_port: 'int' =None, ssh_username: 'str' =None, start_timestamp: 'datetime' =None, startup_eta_seconds: 'str' =None, startup_percentage: 'str' =None, startup_phase: 'str' =None, startup_status: 'V1CloudSpaceInstanceStartupStatus' =None, status_message: 'str' =None, switched_at: 'datetime' =None, sync_eta_seconds: 'str' =None, sync_in_progress: 'bool' =None, sync_percentage: 'str' =None, termination_time: 'datetime' =None, vscode_url: 'str' =None): # noqa: E501
106
108
  """Externalv1CloudSpaceInstanceStatus - a model defined in Swagger""" # noqa: E501
107
109
  self._app_url = None
108
110
  self._cloud_space_id = None
@@ -131,6 +133,7 @@ class Externalv1CloudSpaceInstanceStatus(object):
131
133
  self._sync_eta_seconds = None
132
134
  self._sync_in_progress = None
133
135
  self._sync_percentage = None
136
+ self._termination_time = None
134
137
  self._vscode_url = None
135
138
  self.discriminator = None
136
139
  if app_url is not None:
@@ -187,6 +190,8 @@ class Externalv1CloudSpaceInstanceStatus(object):
187
190
  self.sync_in_progress = sync_in_progress
188
191
  if sync_percentage is not None:
189
192
  self.sync_percentage = sync_percentage
193
+ if termination_time is not None:
194
+ self.termination_time = termination_time
190
195
  if vscode_url is not None:
191
196
  self.vscode_url = vscode_url
192
197
 
@@ -759,6 +764,27 @@ class Externalv1CloudSpaceInstanceStatus(object):
759
764
 
760
765
  self._sync_percentage = sync_percentage
761
766
 
767
+ @property
768
+ def termination_time(self) -> 'datetime':
769
+ """Gets the termination_time of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
770
+
771
+
772
+ :return: The termination_time of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
773
+ :rtype: datetime
774
+ """
775
+ return self._termination_time
776
+
777
+ @termination_time.setter
778
+ def termination_time(self, termination_time: 'datetime'):
779
+ """Sets the termination_time of this Externalv1CloudSpaceInstanceStatus.
780
+
781
+
782
+ :param termination_time: The termination_time of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
783
+ :type: datetime
784
+ """
785
+
786
+ self._termination_time = termination_time
787
+
762
788
  @property
763
789
  def vscode_url(self) -> 'str':
764
790
  """Gets the vscode_url of this Externalv1CloudSpaceInstanceStatus. # noqa: E501
@@ -0,0 +1,123 @@
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 MultimachinejobsIdBody(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
+ 'desired_state': 'V1MultiMachineJobState'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'desired_state': 'desiredState'
49
+ }
50
+
51
+ def __init__(self, desired_state: 'V1MultiMachineJobState' =None): # noqa: E501
52
+ """MultimachinejobsIdBody - a model defined in Swagger""" # noqa: E501
53
+ self._desired_state = None
54
+ self.discriminator = None
55
+ if desired_state is not None:
56
+ self.desired_state = desired_state
57
+
58
+ @property
59
+ def desired_state(self) -> 'V1MultiMachineJobState':
60
+ """Gets the desired_state of this MultimachinejobsIdBody. # noqa: E501
61
+
62
+
63
+ :return: The desired_state of this MultimachinejobsIdBody. # noqa: E501
64
+ :rtype: V1MultiMachineJobState
65
+ """
66
+ return self._desired_state
67
+
68
+ @desired_state.setter
69
+ def desired_state(self, desired_state: 'V1MultiMachineJobState'):
70
+ """Sets the desired_state of this MultimachinejobsIdBody.
71
+
72
+
73
+ :param desired_state: The desired_state of this MultimachinejobsIdBody. # noqa: E501
74
+ :type: V1MultiMachineJobState
75
+ """
76
+
77
+ self._desired_state = desired_state
78
+
79
+ def to_dict(self) -> dict:
80
+ """Returns the model properties as a dict"""
81
+ result = {}
82
+
83
+ for attr, _ in six.iteritems(self.swagger_types):
84
+ value = getattr(self, attr)
85
+ if isinstance(value, list):
86
+ result[attr] = list(map(
87
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
88
+ value
89
+ ))
90
+ elif hasattr(value, "to_dict"):
91
+ result[attr] = value.to_dict()
92
+ elif isinstance(value, dict):
93
+ result[attr] = dict(map(
94
+ lambda item: (item[0], item[1].to_dict())
95
+ if hasattr(item[1], "to_dict") else item,
96
+ value.items()
97
+ ))
98
+ else:
99
+ result[attr] = value
100
+ if issubclass(MultimachinejobsIdBody, dict):
101
+ for key, value in self.items():
102
+ result[key] = value
103
+
104
+ return result
105
+
106
+ def to_str(self) -> str:
107
+ """Returns the string representation of the model"""
108
+ return pprint.pformat(self.to_dict())
109
+
110
+ def __repr__(self) -> str:
111
+ """For `print` and `pprint`"""
112
+ return self.to_str()
113
+
114
+ def __eq__(self, other: 'MultimachinejobsIdBody') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, MultimachinejobsIdBody):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'MultimachinejobsIdBody') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -50,10 +50,12 @@ class ProjectIdAgentsBody(object):
50
50
  'knowledge': 'str',
51
51
  'knowledge_configuration': 'V1KnowledgeConfiguration',
52
52
  'model': 'str',
53
+ 'model_provider': 'str',
53
54
  'name': 'str',
54
55
  'org_id': 'str',
55
56
  'prompt_suggestions': 'list[V1PromptSuggestion]',
56
57
  'prompt_template': 'str',
58
+ 'publish_status': 'str',
57
59
  'thumbnail_url': 'str'
58
60
  }
59
61
 
@@ -67,14 +69,16 @@ class ProjectIdAgentsBody(object):
67
69
  'knowledge': 'knowledge',
68
70
  'knowledge_configuration': 'knowledgeConfiguration',
69
71
  'model': 'model',
72
+ 'model_provider': 'modelProvider',
70
73
  'name': 'name',
71
74
  'org_id': 'orgId',
72
75
  'prompt_suggestions': 'promptSuggestions',
73
76
  'prompt_template': 'promptTemplate',
77
+ 'publish_status': 'publishStatus',
74
78
  'thumbnail_url': 'thumbnailUrl'
75
79
  }
76
80
 
77
- def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, description: 'str' =None, endpoint: 'V1Endpoint' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, model: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, thumbnail_url: 'str' =None): # noqa: E501
81
+ def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, description: 'str' =None, endpoint: 'V1Endpoint' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, model: 'str' =None, model_provider: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, publish_status: 'str' =None, thumbnail_url: 'str' =None): # noqa: E501
78
82
  """ProjectIdAgentsBody - a model defined in Swagger""" # noqa: E501
79
83
  self._cloudspace_id = None
80
84
  self._cluster_id = None
@@ -85,10 +89,12 @@ class ProjectIdAgentsBody(object):
85
89
  self._knowledge = None
86
90
  self._knowledge_configuration = None
87
91
  self._model = None
92
+ self._model_provider = None
88
93
  self._name = None
89
94
  self._org_id = None
90
95
  self._prompt_suggestions = None
91
96
  self._prompt_template = None
97
+ self._publish_status = None
92
98
  self._thumbnail_url = None
93
99
  self.discriminator = None
94
100
  if cloudspace_id is not None:
@@ -109,6 +115,8 @@ class ProjectIdAgentsBody(object):
109
115
  self.knowledge_configuration = knowledge_configuration
110
116
  if model is not None:
111
117
  self.model = model
118
+ if model_provider is not None:
119
+ self.model_provider = model_provider
112
120
  if name is not None:
113
121
  self.name = name
114
122
  if org_id is not None:
@@ -117,6 +125,8 @@ class ProjectIdAgentsBody(object):
117
125
  self.prompt_suggestions = prompt_suggestions
118
126
  if prompt_template is not None:
119
127
  self.prompt_template = prompt_template
128
+ if publish_status is not None:
129
+ self.publish_status = publish_status
120
130
  if thumbnail_url is not None:
121
131
  self.thumbnail_url = thumbnail_url
122
132
 
@@ -309,6 +319,27 @@ class ProjectIdAgentsBody(object):
309
319
 
310
320
  self._model = model
311
321
 
322
+ @property
323
+ def model_provider(self) -> 'str':
324
+ """Gets the model_provider of this ProjectIdAgentsBody. # noqa: E501
325
+
326
+
327
+ :return: The model_provider of this ProjectIdAgentsBody. # noqa: E501
328
+ :rtype: str
329
+ """
330
+ return self._model_provider
331
+
332
+ @model_provider.setter
333
+ def model_provider(self, model_provider: 'str'):
334
+ """Sets the model_provider of this ProjectIdAgentsBody.
335
+
336
+
337
+ :param model_provider: The model_provider of this ProjectIdAgentsBody. # noqa: E501
338
+ :type: str
339
+ """
340
+
341
+ self._model_provider = model_provider
342
+
312
343
  @property
313
344
  def name(self) -> 'str':
314
345
  """Gets the name of this ProjectIdAgentsBody. # noqa: E501
@@ -393,6 +424,27 @@ class ProjectIdAgentsBody(object):
393
424
 
394
425
  self._prompt_template = prompt_template
395
426
 
427
+ @property
428
+ def publish_status(self) -> 'str':
429
+ """Gets the publish_status of this ProjectIdAgentsBody. # noqa: E501
430
+
431
+
432
+ :return: The publish_status of this ProjectIdAgentsBody. # noqa: E501
433
+ :rtype: str
434
+ """
435
+ return self._publish_status
436
+
437
+ @publish_status.setter
438
+ def publish_status(self, publish_status: 'str'):
439
+ """Sets the publish_status of this ProjectIdAgentsBody.
440
+
441
+
442
+ :param publish_status: The publish_status of this ProjectIdAgentsBody. # noqa: E501
443
+ :type: str
444
+ """
445
+
446
+ self._publish_status = publish_status
447
+
396
448
  @property
397
449
  def thumbnail_url(self) -> 'str':
398
450
  """Gets the thumbnail_url of this ProjectIdAgentsBody. # noqa: E501
@@ -50,6 +50,7 @@ class ProjectIdCloudspacesBody(object):
50
50
  'disk_size': 'str',
51
51
  'display_name': 'str',
52
52
  'name': 'str',
53
+ 'requested_run_duration_seconds': 'str',
53
54
  'seed_files': 'list[V1CloudSpaceSeedFile]',
54
55
  'spot': 'bool'
55
56
  }
@@ -64,11 +65,12 @@ class ProjectIdCloudspacesBody(object):
64
65
  'disk_size': 'diskSize',
65
66
  'display_name': 'displayName',
66
67
  'name': 'name',
68
+ 'requested_run_duration_seconds': 'requestedRunDurationSeconds',
67
69
  'seed_files': 'seedFiles',
68
70
  'spot': 'spot'
69
71
  }
70
72
 
71
- def __init__(self, can_download_source_code: 'bool' =None, cloud_space_instance_cpu_image_override: 'str' =None, cloud_space_instance_gpu_image_override: 'str' =None, cluster_id: 'str' =None, compute_name: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disk_size: 'str' =None, display_name: 'str' =None, name: 'str' =None, seed_files: 'list[V1CloudSpaceSeedFile]' =None, spot: 'bool' =None): # noqa: E501
73
+ def __init__(self, can_download_source_code: 'bool' =None, cloud_space_instance_cpu_image_override: 'str' =None, cloud_space_instance_gpu_image_override: 'str' =None, cluster_id: 'str' =None, compute_name: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disk_size: 'str' =None, display_name: 'str' =None, name: 'str' =None, requested_run_duration_seconds: 'str' =None, seed_files: 'list[V1CloudSpaceSeedFile]' =None, spot: 'bool' =None): # noqa: E501
72
74
  """ProjectIdCloudspacesBody - a model defined in Swagger""" # noqa: E501
73
75
  self._can_download_source_code = None
74
76
  self._cloud_space_instance_cpu_image_override = None
@@ -79,6 +81,7 @@ class ProjectIdCloudspacesBody(object):
79
81
  self._disk_size = None
80
82
  self._display_name = None
81
83
  self._name = None
84
+ self._requested_run_duration_seconds = None
82
85
  self._seed_files = None
83
86
  self._spot = None
84
87
  self.discriminator = None
@@ -100,6 +103,8 @@ class ProjectIdCloudspacesBody(object):
100
103
  self.display_name = display_name
101
104
  if name is not None:
102
105
  self.name = name
106
+ if requested_run_duration_seconds is not None:
107
+ self.requested_run_duration_seconds = requested_run_duration_seconds
103
108
  if seed_files is not None:
104
109
  self.seed_files = seed_files
105
110
  if spot is not None:
@@ -294,6 +299,27 @@ class ProjectIdCloudspacesBody(object):
294
299
 
295
300
  self._name = name
296
301
 
302
+ @property
303
+ def requested_run_duration_seconds(self) -> 'str':
304
+ """Gets the requested_run_duration_seconds of this ProjectIdCloudspacesBody. # noqa: E501
305
+
306
+
307
+ :return: The requested_run_duration_seconds of this ProjectIdCloudspacesBody. # noqa: E501
308
+ :rtype: str
309
+ """
310
+ return self._requested_run_duration_seconds
311
+
312
+ @requested_run_duration_seconds.setter
313
+ def requested_run_duration_seconds(self, requested_run_duration_seconds: 'str'):
314
+ """Sets the requested_run_duration_seconds of this ProjectIdCloudspacesBody.
315
+
316
+
317
+ :param requested_run_duration_seconds: The requested_run_duration_seconds of this ProjectIdCloudspacesBody. # noqa: E501
318
+ :type: str
319
+ """
320
+
321
+ self._requested_run_duration_seconds = requested_run_duration_seconds
322
+
297
323
  @property
298
324
  def seed_files(self) -> 'list[V1CloudSpaceSeedFile]':
299
325
  """Gets the seed_files of this ProjectIdCloudspacesBody. # noqa: E501