lightning-sdk 0.1.58__py3-none-any.whl → 0.2.1__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 (121) 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 +23 -3
  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 +22 -0
  16. lightning_sdk/cli/download.py +78 -113
  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 +128 -191
  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 -255
  29. lightning_sdk/deployment/__init__.py +2 -0
  30. lightning_sdk/deployment/deployment.py +33 -8
  31. lightning_sdk/lightning_cloud/openapi/__init__.py +23 -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 +674 -0
  38. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +303 -4
  39. lightning_sdk/lightning_cloud/openapi/models/__init__.py +22 -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 +461 -0
  50. lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py +227 -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 +53 -1
  75. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_mmt.py +53 -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_artifacts_type.py +103 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +53 -1
  80. lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py +27 -1
  81. lightning_sdk/lightning_cloud/openapi/models/v1_list_pipelines_response.py +123 -0
  82. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py +27 -1
  83. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +29 -1
  84. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +27 -1
  85. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +27 -1
  86. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +2 -0
  87. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +209 -1
  88. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +513 -0
  89. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_schedule.py +149 -0
  90. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_step.py +253 -0
  91. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_step_status.py +331 -0
  92. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline_step_type.py +104 -0
  93. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +157 -1
  94. lightning_sdk/lightning_cloud/openapi/models/v1_restart_timing.py +27 -1
  95. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  96. lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +201 -0
  97. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +27 -1
  98. lightning_sdk/lightning_cloud/openapi/models/v1_update_job_visibility_response.py +97 -0
  99. lightning_sdk/lightning_cloud/openapi/models/v1_upload_temporary_artifact_request.py +123 -0
  100. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +95 -355
  101. lightning_sdk/lightning_cloud/openapi/models/validate.py +27 -1
  102. lightning_sdk/lightning_cloud/rest_client.py +4 -2
  103. lightning_sdk/machine.py +25 -1
  104. lightning_sdk/models.py +18 -12
  105. lightning_sdk/pipeline/__init__.py +4 -0
  106. lightning_sdk/pipeline/pipeline.py +109 -0
  107. lightning_sdk/pipeline/types.py +268 -0
  108. lightning_sdk/pipeline/utils.py +69 -0
  109. lightning_sdk/plugin.py +9 -10
  110. lightning_sdk/serve.py +134 -0
  111. lightning_sdk/services/utilities.py +2 -2
  112. lightning_sdk/studio.py +5 -1
  113. lightning_sdk/teamspace.py +1 -1
  114. lightning_sdk/utils/resolve.py +12 -1
  115. {lightning_sdk-0.1.58.dist-info → lightning_sdk-0.2.1.dist-info}/METADATA +6 -8
  116. {lightning_sdk-0.1.58.dist-info → lightning_sdk-0.2.1.dist-info}/RECORD +120 -88
  117. lightning_sdk/cli/legacy.py +0 -135
  118. {lightning_sdk-0.1.58.dist-info → lightning_sdk-0.2.1.dist-info}/LICENSE +0 -0
  119. {lightning_sdk-0.1.58.dist-info → lightning_sdk-0.2.1.dist-info}/WHEEL +0 -0
  120. {lightning_sdk-0.1.58.dist-info → lightning_sdk-0.2.1.dist-info}/entry_points.txt +0 -0
  121. {lightning_sdk-0.1.58.dist-info → lightning_sdk-0.2.1.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,103 @@
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 V1JobArtifactsType(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
+ """
38
+ allowed enum values
39
+ """
40
+ V1 = "V1"
41
+ V2 = "V2"
42
+ """
43
+ Attributes:
44
+ swagger_types (dict): The key is attribute name
45
+ and the value is attribute type.
46
+ attribute_map (dict): The key is attribute name
47
+ and the value is json key in definition.
48
+ """
49
+ swagger_types = {
50
+ }
51
+
52
+ attribute_map = {
53
+ }
54
+
55
+ def __init__(self): # noqa: E501
56
+ """V1JobArtifactsType - a model defined in Swagger""" # noqa: E501
57
+ self.discriminator = None
58
+
59
+ def to_dict(self) -> dict:
60
+ """Returns the model properties as a dict"""
61
+ result = {}
62
+
63
+ for attr, _ in six.iteritems(self.swagger_types):
64
+ value = getattr(self, attr)
65
+ if isinstance(value, list):
66
+ result[attr] = list(map(
67
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
68
+ value
69
+ ))
70
+ elif hasattr(value, "to_dict"):
71
+ result[attr] = value.to_dict()
72
+ elif isinstance(value, dict):
73
+ result[attr] = dict(map(
74
+ lambda item: (item[0], item[1].to_dict())
75
+ if hasattr(item[1], "to_dict") else item,
76
+ value.items()
77
+ ))
78
+ else:
79
+ result[attr] = value
80
+ if issubclass(V1JobArtifactsType, dict):
81
+ for key, value in self.items():
82
+ result[key] = value
83
+
84
+ return result
85
+
86
+ def to_str(self) -> str:
87
+ """Returns the string representation of the model"""
88
+ return pprint.pformat(self.to_dict())
89
+
90
+ def __repr__(self) -> str:
91
+ """For `print` and `pprint`"""
92
+ return self.to_str()
93
+
94
+ def __eq__(self, other: 'V1JobArtifactsType') -> bool:
95
+ """Returns true if both objects are equal"""
96
+ if not isinstance(other, V1JobArtifactsType):
97
+ return False
98
+
99
+ return self.__dict__ == other.__dict__
100
+
101
+ def __ne__(self, other: 'V1JobArtifactsType') -> bool:
102
+ """Returns true if both objects are not equal"""
103
+ return not self == other
@@ -57,7 +57,9 @@ class V1JobSpec(object):
57
57
  'instance_type': 'str',
58
58
  'modified_volume': 'bool',
59
59
  'path_mappings': 'list[V1PathMapping]',
60
+ 'pipeline_id': 'str',
60
61
  'quantity': 'int',
62
+ 'rank': 'int',
61
63
  'readiness_probe': 'V1JobHealthCheckConfig',
62
64
  'regions': 'list[str]',
63
65
  'requested_run_duration_seconds': 'str',
@@ -85,7 +87,9 @@ class V1JobSpec(object):
85
87
  'instance_type': 'instanceType',
86
88
  'modified_volume': 'modifiedVolume',
87
89
  'path_mappings': 'pathMappings',
90
+ 'pipeline_id': 'pipelineId',
88
91
  'quantity': 'quantity',
92
+ 'rank': 'rank',
89
93
  'readiness_probe': 'readinessProbe',
90
94
  'regions': 'regions',
91
95
  'requested_run_duration_seconds': 'requestedRunDurationSeconds',
@@ -96,7 +100,7 @@ class V1JobSpec(object):
96
100
  'volumes': 'volumes'
97
101
  }
98
102
 
99
- def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, quantity: 'int' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None): # noqa: E501
103
+ def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, pipeline_id: 'str' =None, quantity: 'int' =None, rank: 'int' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None): # noqa: E501
100
104
  """V1JobSpec - a model defined in Swagger""" # noqa: E501
101
105
  self._artifacts_destination = None
102
106
  self._artifacts_source = None
@@ -114,7 +118,9 @@ class V1JobSpec(object):
114
118
  self._instance_type = None
115
119
  self._modified_volume = None
116
120
  self._path_mappings = None
121
+ self._pipeline_id = None
117
122
  self._quantity = None
123
+ self._rank = None
118
124
  self._readiness_probe = None
119
125
  self._regions = None
120
126
  self._requested_run_duration_seconds = None
@@ -156,8 +162,12 @@ class V1JobSpec(object):
156
162
  self.modified_volume = modified_volume
157
163
  if path_mappings is not None:
158
164
  self.path_mappings = path_mappings
165
+ if pipeline_id is not None:
166
+ self.pipeline_id = pipeline_id
159
167
  if quantity is not None:
160
168
  self.quantity = quantity
169
+ if rank is not None:
170
+ self.rank = rank
161
171
  if readiness_probe is not None:
162
172
  self.readiness_probe = readiness_probe
163
173
  if regions is not None:
@@ -511,6 +521,27 @@ class V1JobSpec(object):
511
521
 
512
522
  self._path_mappings = path_mappings
513
523
 
524
+ @property
525
+ def pipeline_id(self) -> 'str':
526
+ """Gets the pipeline_id of this V1JobSpec. # noqa: E501
527
+
528
+
529
+ :return: The pipeline_id of this V1JobSpec. # noqa: E501
530
+ :rtype: str
531
+ """
532
+ return self._pipeline_id
533
+
534
+ @pipeline_id.setter
535
+ def pipeline_id(self, pipeline_id: 'str'):
536
+ """Sets the pipeline_id of this V1JobSpec.
537
+
538
+
539
+ :param pipeline_id: The pipeline_id of this V1JobSpec. # noqa: E501
540
+ :type: str
541
+ """
542
+
543
+ self._pipeline_id = pipeline_id
544
+
514
545
  @property
515
546
  def quantity(self) -> 'int':
516
547
  """Gets the quantity of this V1JobSpec. # noqa: E501
@@ -532,6 +563,27 @@ class V1JobSpec(object):
532
563
 
533
564
  self._quantity = quantity
534
565
 
566
+ @property
567
+ def rank(self) -> 'int':
568
+ """Gets the rank of this V1JobSpec. # noqa: E501
569
+
570
+
571
+ :return: The rank of this V1JobSpec. # noqa: E501
572
+ :rtype: int
573
+ """
574
+ return self._rank
575
+
576
+ @rank.setter
577
+ def rank(self, rank: 'int'):
578
+ """Sets the rank of this V1JobSpec.
579
+
580
+
581
+ :param rank: The rank of this V1JobSpec. # noqa: E501
582
+ :type: int
583
+ """
584
+
585
+ self._rank = rank
586
+
535
587
  @property
536
588
  def readiness_probe(self) -> 'V1JobHealthCheckConfig':
537
589
  """Gets the readiness_probe of this V1JobSpec. # noqa: E501
@@ -44,6 +44,7 @@ class V1JobTiming(object):
44
44
  'end': 'datetime',
45
45
  'name': 'str',
46
46
  'payload': 'str',
47
+ 'rank': 'int',
47
48
  'start': 'datetime'
48
49
  }
49
50
 
@@ -51,14 +52,16 @@ class V1JobTiming(object):
51
52
  'end': 'end',
52
53
  'name': 'name',
53
54
  'payload': 'payload',
55
+ 'rank': 'rank',
54
56
  'start': 'start'
55
57
  }
56
58
 
57
- def __init__(self, end: 'datetime' =None, name: 'str' =None, payload: 'str' =None, start: 'datetime' =None): # noqa: E501
59
+ def __init__(self, end: 'datetime' =None, name: 'str' =None, payload: 'str' =None, rank: 'int' =None, start: 'datetime' =None): # noqa: E501
58
60
  """V1JobTiming - a model defined in Swagger""" # noqa: E501
59
61
  self._end = None
60
62
  self._name = None
61
63
  self._payload = None
64
+ self._rank = None
62
65
  self._start = None
63
66
  self.discriminator = None
64
67
  if end is not None:
@@ -67,6 +70,8 @@ class V1JobTiming(object):
67
70
  self.name = name
68
71
  if payload is not None:
69
72
  self.payload = payload
73
+ if rank is not None:
74
+ self.rank = rank
70
75
  if start is not None:
71
76
  self.start = start
72
77
 
@@ -133,6 +138,27 @@ class V1JobTiming(object):
133
138
 
134
139
  self._payload = payload
135
140
 
141
+ @property
142
+ def rank(self) -> 'int':
143
+ """Gets the rank of this V1JobTiming. # noqa: E501
144
+
145
+
146
+ :return: The rank of this V1JobTiming. # noqa: E501
147
+ :rtype: int
148
+ """
149
+ return self._rank
150
+
151
+ @rank.setter
152
+ def rank(self, rank: 'int'):
153
+ """Sets the rank of this V1JobTiming.
154
+
155
+
156
+ :param rank: The rank of this V1JobTiming. # noqa: E501
157
+ :type: int
158
+ """
159
+
160
+ self._rank = rank
161
+
136
162
  @property
137
163
  def start(self) -> 'datetime':
138
164
  """Gets the start of this V1JobTiming. # 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 V1ListPipelinesResponse(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
+ 'pipelines': 'list[V1Pipeline]'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'pipelines': 'pipelines'
49
+ }
50
+
51
+ def __init__(self, pipelines: 'list[V1Pipeline]' =None): # noqa: E501
52
+ """V1ListPipelinesResponse - a model defined in Swagger""" # noqa: E501
53
+ self._pipelines = None
54
+ self.discriminator = None
55
+ if pipelines is not None:
56
+ self.pipelines = pipelines
57
+
58
+ @property
59
+ def pipelines(self) -> 'list[V1Pipeline]':
60
+ """Gets the pipelines of this V1ListPipelinesResponse. # noqa: E501
61
+
62
+
63
+ :return: The pipelines of this V1ListPipelinesResponse. # noqa: E501
64
+ :rtype: list[V1Pipeline]
65
+ """
66
+ return self._pipelines
67
+
68
+ @pipelines.setter
69
+ def pipelines(self, pipelines: 'list[V1Pipeline]'):
70
+ """Sets the pipelines of this V1ListPipelinesResponse.
71
+
72
+
73
+ :param pipelines: The pipelines of this V1ListPipelinesResponse. # noqa: E501
74
+ :type: list[V1Pipeline]
75
+ """
76
+
77
+ self._pipelines = pipelines
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(V1ListPipelinesResponse, 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: 'V1ListPipelinesResponse') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, V1ListPipelinesResponse):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'V1ListPipelinesResponse') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -41,6 +41,7 @@ class V1LitRegistryArtifact(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'exposed_ports': 'list[str]',
44
45
  'full_hash_digest': 'str',
45
46
  'location': 'str',
46
47
  'pull_time': 'datetime',
@@ -51,6 +52,7 @@ class V1LitRegistryArtifact(object):
51
52
  }
52
53
 
53
54
  attribute_map = {
55
+ 'exposed_ports': 'exposedPorts',
54
56
  'full_hash_digest': 'fullHashDigest',
55
57
  'location': 'location',
56
58
  'pull_time': 'pullTime',
@@ -60,8 +62,9 @@ class V1LitRegistryArtifact(object):
60
62
  'tag_name': 'tagName'
61
63
  }
62
64
 
63
- def __init__(self, full_hash_digest: 'str' =None, location: 'str' =None, pull_time: 'datetime' =None, push_time: 'datetime' =None, short_hash_digest: 'str' =None, size: 'str' =None, tag_name: 'str' =None): # noqa: E501
65
+ def __init__(self, exposed_ports: 'list[str]' =None, full_hash_digest: 'str' =None, location: 'str' =None, pull_time: 'datetime' =None, push_time: 'datetime' =None, short_hash_digest: 'str' =None, size: 'str' =None, tag_name: 'str' =None): # noqa: E501
64
66
  """V1LitRegistryArtifact - a model defined in Swagger""" # noqa: E501
67
+ self._exposed_ports = None
65
68
  self._full_hash_digest = None
66
69
  self._location = None
67
70
  self._pull_time = None
@@ -70,6 +73,8 @@ class V1LitRegistryArtifact(object):
70
73
  self._size = None
71
74
  self._tag_name = None
72
75
  self.discriminator = None
76
+ if exposed_ports is not None:
77
+ self.exposed_ports = exposed_ports
73
78
  if full_hash_digest is not None:
74
79
  self.full_hash_digest = full_hash_digest
75
80
  if location is not None:
@@ -85,6 +90,27 @@ class V1LitRegistryArtifact(object):
85
90
  if tag_name is not None:
86
91
  self.tag_name = tag_name
87
92
 
93
+ @property
94
+ def exposed_ports(self) -> 'list[str]':
95
+ """Gets the exposed_ports of this V1LitRegistryArtifact. # noqa: E501
96
+
97
+
98
+ :return: The exposed_ports of this V1LitRegistryArtifact. # noqa: E501
99
+ :rtype: list[str]
100
+ """
101
+ return self._exposed_ports
102
+
103
+ @exposed_ports.setter
104
+ def exposed_ports(self, exposed_ports: 'list[str]'):
105
+ """Sets the exposed_ports of this V1LitRegistryArtifact.
106
+
107
+
108
+ :param exposed_ports: The exposed_ports of this V1LitRegistryArtifact. # noqa: E501
109
+ :type: list[str]
110
+ """
111
+
112
+ self._exposed_ports = exposed_ports
113
+
88
114
  @property
89
115
  def full_hash_digest(self) -> 'str':
90
116
  """Gets the full_hash_digest of this V1LitRegistryArtifact. # noqa: E501
@@ -42,6 +42,7 @@ class V1LitRepository(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'artifact_count': 'str',
45
+ 'artifacts': 'list[V1LitRegistryArtifact]',
45
46
  'creation_time': 'datetime',
46
47
  'description': 'str',
47
48
  'id': 'str',
@@ -53,6 +54,7 @@ class V1LitRepository(object):
53
54
 
54
55
  attribute_map = {
55
56
  'artifact_count': 'artifactCount',
57
+ 'artifacts': 'artifacts',
56
58
  'creation_time': 'creationTime',
57
59
  'description': 'description',
58
60
  'id': 'id',
@@ -62,9 +64,10 @@ class V1LitRepository(object):
62
64
  'update_time': 'updateTime'
63
65
  }
64
66
 
65
- def __init__(self, artifact_count: 'str' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None): # noqa: E501
67
+ def __init__(self, artifact_count: 'str' =None, artifacts: 'list[V1LitRegistryArtifact]' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None): # noqa: E501
66
68
  """V1LitRepository - a model defined in Swagger""" # noqa: E501
67
69
  self._artifact_count = None
70
+ self._artifacts = None
68
71
  self._creation_time = None
69
72
  self._description = None
70
73
  self._id = None
@@ -75,6 +78,8 @@ class V1LitRepository(object):
75
78
  self.discriminator = None
76
79
  if artifact_count is not None:
77
80
  self.artifact_count = artifact_count
81
+ if artifacts is not None:
82
+ self.artifacts = artifacts
78
83
  if creation_time is not None:
79
84
  self.creation_time = creation_time
80
85
  if description is not None:
@@ -111,6 +116,29 @@ class V1LitRepository(object):
111
116
 
112
117
  self._artifact_count = artifact_count
113
118
 
119
+ @property
120
+ def artifacts(self) -> 'list[V1LitRegistryArtifact]':
121
+ """Gets the artifacts of this V1LitRepository. # noqa: E501
122
+
123
+ All artifacts for this repo, this plus lit registry are coming from behind the scenes listLitRegistryRepositoryImageArtifactVersions. We'll use these for a complete view of all registry container artifacts vs the UI having to make the calls separately. # noqa: E501
124
+
125
+ :return: The artifacts of this V1LitRepository. # noqa: E501
126
+ :rtype: list[V1LitRegistryArtifact]
127
+ """
128
+ return self._artifacts
129
+
130
+ @artifacts.setter
131
+ def artifacts(self, artifacts: 'list[V1LitRegistryArtifact]'):
132
+ """Sets the artifacts of this V1LitRepository.
133
+
134
+ All artifacts for this repo, this plus lit registry are coming from behind the scenes listLitRegistryRepositoryImageArtifactVersions. We'll use these for a complete view of all registry container artifacts vs the UI having to make the calls separately. # noqa: E501
135
+
136
+ :param artifacts: The artifacts of this V1LitRepository. # noqa: E501
137
+ :type: list[V1LitRegistryArtifact]
138
+ """
139
+
140
+ self._artifacts = artifacts
141
+
114
142
  @property
115
143
  def creation_time(self) -> 'datetime':
116
144
  """Gets the creation_time of this V1LitRepository. # noqa: E501
@@ -43,6 +43,7 @@ class V1ManagedModel(object):
43
43
  swagger_types = {
44
44
  'abilities': 'V1ManagedModelAbilities',
45
45
  'completion_token_price': 'float',
46
+ 'deployment_details': 'V1DeploymentDetails',
46
47
  'description': 'str',
47
48
  'endpoint_id': 'str',
48
49
  'id': 'str',
@@ -54,6 +55,7 @@ class V1ManagedModel(object):
54
55
  attribute_map = {
55
56
  'abilities': 'abilities',
56
57
  'completion_token_price': 'completionTokenPrice',
58
+ 'deployment_details': 'deploymentDetails',
57
59
  'description': 'description',
58
60
  'endpoint_id': 'endpointId',
59
61
  'id': 'id',
@@ -62,10 +64,11 @@ class V1ManagedModel(object):
62
64
  'status': 'status'
63
65
  }
64
66
 
65
- def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, description: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, status: 'V1AssistantModelStatus' =None): # noqa: E501
67
+ def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, status: 'V1AssistantModelStatus' =None): # noqa: E501
66
68
  """V1ManagedModel - a model defined in Swagger""" # noqa: E501
67
69
  self._abilities = None
68
70
  self._completion_token_price = None
71
+ self._deployment_details = None
69
72
  self._description = None
70
73
  self._endpoint_id = None
71
74
  self._id = None
@@ -77,6 +80,8 @@ class V1ManagedModel(object):
77
80
  self.abilities = abilities
78
81
  if completion_token_price is not None:
79
82
  self.completion_token_price = completion_token_price
83
+ if deployment_details is not None:
84
+ self.deployment_details = deployment_details
80
85
  if description is not None:
81
86
  self.description = description
82
87
  if endpoint_id is not None:
@@ -132,6 +137,27 @@ class V1ManagedModel(object):
132
137
 
133
138
  self._completion_token_price = completion_token_price
134
139
 
140
+ @property
141
+ def deployment_details(self) -> 'V1DeploymentDetails':
142
+ """Gets the deployment_details of this V1ManagedModel. # noqa: E501
143
+
144
+
145
+ :return: The deployment_details of this V1ManagedModel. # noqa: E501
146
+ :rtype: V1DeploymentDetails
147
+ """
148
+ return self._deployment_details
149
+
150
+ @deployment_details.setter
151
+ def deployment_details(self, deployment_details: 'V1DeploymentDetails'):
152
+ """Sets the deployment_details of this V1ManagedModel.
153
+
154
+
155
+ :param deployment_details: The deployment_details of this V1ManagedModel. # noqa: E501
156
+ :type: V1DeploymentDetails
157
+ """
158
+
159
+ self._deployment_details = deployment_details
160
+
135
161
  @property
136
162
  def description(self) -> 'str':
137
163
  """Gets the description of this V1ManagedModel. # noqa: E501
@@ -49,6 +49,7 @@ class V1MultiMachineJob(object):
49
49
  'machines': 'int',
50
50
  'message': 'str',
51
51
  'name': 'str',
52
+ 'pipeline_id': 'str',
52
53
  'project_id': 'str',
53
54
  'spec': 'V1JobSpec',
54
55
  'state': 'V1MultiMachineJobState',
@@ -67,6 +68,7 @@ class V1MultiMachineJob(object):
67
68
  'machines': 'machines',
68
69
  'message': 'message',
69
70
  'name': 'name',
71
+ 'pipeline_id': 'pipelineId',
70
72
  'project_id': 'projectId',
71
73
  'spec': 'spec',
72
74
  'state': 'state',
@@ -76,7 +78,7 @@ class V1MultiMachineJob(object):
76
78
  'user_id': 'userId'
77
79
  }
78
80
 
79
- def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, state: 'V1MultiMachineJobState' =None, status: 'V1MultiMachineJobStatus' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
81
+ def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, pipeline_id: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, state: 'V1MultiMachineJobState' =None, status: 'V1MultiMachineJobStatus' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
80
82
  """V1MultiMachineJob - a model defined in Swagger""" # noqa: E501
81
83
  self._cloudspace_id = None
82
84
  self._created_at = None
@@ -86,6 +88,7 @@ class V1MultiMachineJob(object):
86
88
  self._machines = None
87
89
  self._message = None
88
90
  self._name = None
91
+ self._pipeline_id = None
89
92
  self._project_id = None
90
93
  self._spec = None
91
94
  self._state = None
@@ -110,6 +113,8 @@ class V1MultiMachineJob(object):
110
113
  self.message = message
111
114
  if name is not None:
112
115
  self.name = name
116
+ if pipeline_id is not None:
117
+ self.pipeline_id = pipeline_id
113
118
  if project_id is not None:
114
119
  self.project_id = project_id
115
120
  if spec is not None:
@@ -293,6 +298,27 @@ class V1MultiMachineJob(object):
293
298
 
294
299
  self._name = name
295
300
 
301
+ @property
302
+ def pipeline_id(self) -> 'str':
303
+ """Gets the pipeline_id of this V1MultiMachineJob. # noqa: E501
304
+
305
+
306
+ :return: The pipeline_id of this V1MultiMachineJob. # noqa: E501
307
+ :rtype: str
308
+ """
309
+ return self._pipeline_id
310
+
311
+ @pipeline_id.setter
312
+ def pipeline_id(self, pipeline_id: 'str'):
313
+ """Sets the pipeline_id of this V1MultiMachineJob.
314
+
315
+
316
+ :param pipeline_id: The pipeline_id of this V1MultiMachineJob. # noqa: E501
317
+ :type: str
318
+ """
319
+
320
+ self._pipeline_id = pipeline_id
321
+
296
322
  @property
297
323
  def project_id(self) -> 'str':
298
324
  """Gets the project_id of this V1MultiMachineJob. # noqa: E501
@@ -44,6 +44,8 @@ class V1MultiMachineJobState(object):
44
44
  DELETED = "MultiMachineJob_STATE_DELETED"
45
45
  FAILED = "MultiMachineJob_STATE_FAILED"
46
46
  COMPLETED = "MultiMachineJob_STATE_COMPLETED"
47
+ STOP = "MultiMachineJob_STATE_STOP"
48
+ DELETE = "MultiMachineJob_STATE_DELETE"
47
49
  """
48
50
  Attributes:
49
51
  swagger_types (dict): The key is attribute name