lightning-sdk 0.2.22__py3-none-any.whl → 0.2.24rc0__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.
- lightning_sdk/__init__.py +1 -1
- lightning_sdk/api/base_studio_api.py +9 -2
- lightning_sdk/api/deployment_api.py +9 -9
- lightning_sdk/api/license_api.py +2 -2
- lightning_sdk/api/llm_api.py +7 -11
- lightning_sdk/api/pipeline_api.py +31 -10
- lightning_sdk/api/studio_api.py +6 -0
- lightning_sdk/base_studio.py +22 -6
- lightning_sdk/cli/entrypoint.py +15 -13
- lightning_sdk/cli/start.py +5 -2
- lightning_sdk/deployment/deployment.py +17 -7
- lightning_sdk/lightning_cloud/openapi/__init__.py +20 -0
- lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
- lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
- lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +206 -0
- lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
- lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +97 -0
- lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
- lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +118 -1
- lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
- lightning_sdk/lightning_cloud/openapi/configuration.py +1 -1
- lightning_sdk/lightning_cloud/openapi/models/__init__.py +19 -0
- lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +81 -3
- lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_visibility_body.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +79 -1
- lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +53 -1
- lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_request.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +2 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +79 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +105 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
- lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster.py +253 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +853 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +29 -27
- lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +53 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_list_clusters_response.py +6 -6
- lightning_sdk/lightning_cloud/openapi/models/v1_list_project_clusters_response.py +6 -6
- lightning_sdk/lightning_cloud/openapi/models/v1_lite_published_cloud_space_response.py +513 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +29 -3
- lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +131 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_visibility_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
- lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
- lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +234 -104
- lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +78 -104
- lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +104 -0
- lightning_sdk/llm/llm.py +113 -115
- lightning_sdk/llm/public_assistants.json +8 -0
- lightning_sdk/pipeline/__init__.py +11 -2
- lightning_sdk/pipeline/pipeline.py +54 -14
- lightning_sdk/pipeline/printer.py +36 -16
- lightning_sdk/pipeline/schedule.py +2 -1
- lightning_sdk/pipeline/{types.py → steps.py} +77 -56
- lightning_sdk/pipeline/utils.py +65 -3
- lightning_sdk/sandbox.py +157 -0
- lightning_sdk/services/license.py +12 -6
- lightning_sdk/studio.py +10 -1
- {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/METADATA +1 -1
- {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/RECORD +101 -79
- {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/LICENSE +0 -0
- {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/WHEEL +0 -0
- {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/entry_points.txt +0 -0
- {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/top_level.txt +0 -0
|
@@ -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 V1ListCloudyExpertsResponse(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
|
+
'experts': 'list[V1CloudyExpert]'
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
attribute_map = {
|
|
48
|
+
'experts': 'experts'
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
def __init__(self, experts: 'list[V1CloudyExpert]' =None): # noqa: E501
|
|
52
|
+
"""V1ListCloudyExpertsResponse - a model defined in Swagger""" # noqa: E501
|
|
53
|
+
self._experts = None
|
|
54
|
+
self.discriminator = None
|
|
55
|
+
if experts is not None:
|
|
56
|
+
self.experts = experts
|
|
57
|
+
|
|
58
|
+
@property
|
|
59
|
+
def experts(self) -> 'list[V1CloudyExpert]':
|
|
60
|
+
"""Gets the experts of this V1ListCloudyExpertsResponse. # noqa: E501
|
|
61
|
+
|
|
62
|
+
|
|
63
|
+
:return: The experts of this V1ListCloudyExpertsResponse. # noqa: E501
|
|
64
|
+
:rtype: list[V1CloudyExpert]
|
|
65
|
+
"""
|
|
66
|
+
return self._experts
|
|
67
|
+
|
|
68
|
+
@experts.setter
|
|
69
|
+
def experts(self, experts: 'list[V1CloudyExpert]'):
|
|
70
|
+
"""Sets the experts of this V1ListCloudyExpertsResponse.
|
|
71
|
+
|
|
72
|
+
|
|
73
|
+
:param experts: The experts of this V1ListCloudyExpertsResponse. # noqa: E501
|
|
74
|
+
:type: list[V1CloudyExpert]
|
|
75
|
+
"""
|
|
76
|
+
|
|
77
|
+
self._experts = experts
|
|
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(V1ListCloudyExpertsResponse, 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: 'V1ListCloudyExpertsResponse') -> bool:
|
|
115
|
+
"""Returns true if both objects are equal"""
|
|
116
|
+
if not isinstance(other, V1ListCloudyExpertsResponse):
|
|
117
|
+
return False
|
|
118
|
+
|
|
119
|
+
return self.__dict__ == other.__dict__
|
|
120
|
+
|
|
121
|
+
def __ne__(self, other: 'V1ListCloudyExpertsResponse') -> bool:
|
|
122
|
+
"""Returns true if both objects are not equal"""
|
|
123
|
+
return not self == other
|
|
@@ -41,7 +41,7 @@ class V1ListClustersResponse(object):
|
|
|
41
41
|
and the value is json key in definition.
|
|
42
42
|
"""
|
|
43
43
|
swagger_types = {
|
|
44
|
-
'clusters': 'list[
|
|
44
|
+
'clusters': 'list[V1ExternalCluster]',
|
|
45
45
|
'default_cluster': 'str'
|
|
46
46
|
}
|
|
47
47
|
|
|
@@ -50,7 +50,7 @@ class V1ListClustersResponse(object):
|
|
|
50
50
|
'default_cluster': 'defaultCluster'
|
|
51
51
|
}
|
|
52
52
|
|
|
53
|
-
def __init__(self, clusters: 'list[
|
|
53
|
+
def __init__(self, clusters: 'list[V1ExternalCluster]' =None, default_cluster: 'str' =None): # noqa: E501
|
|
54
54
|
"""V1ListClustersResponse - a model defined in Swagger""" # noqa: E501
|
|
55
55
|
self._clusters = None
|
|
56
56
|
self._default_cluster = None
|
|
@@ -61,22 +61,22 @@ class V1ListClustersResponse(object):
|
|
|
61
61
|
self.default_cluster = default_cluster
|
|
62
62
|
|
|
63
63
|
@property
|
|
64
|
-
def clusters(self) -> 'list[
|
|
64
|
+
def clusters(self) -> 'list[V1ExternalCluster]':
|
|
65
65
|
"""Gets the clusters of this V1ListClustersResponse. # noqa: E501
|
|
66
66
|
|
|
67
67
|
|
|
68
68
|
:return: The clusters of this V1ListClustersResponse. # noqa: E501
|
|
69
|
-
:rtype: list[
|
|
69
|
+
:rtype: list[V1ExternalCluster]
|
|
70
70
|
"""
|
|
71
71
|
return self._clusters
|
|
72
72
|
|
|
73
73
|
@clusters.setter
|
|
74
|
-
def clusters(self, clusters: 'list[
|
|
74
|
+
def clusters(self, clusters: 'list[V1ExternalCluster]'):
|
|
75
75
|
"""Sets the clusters of this V1ListClustersResponse.
|
|
76
76
|
|
|
77
77
|
|
|
78
78
|
:param clusters: The clusters of this V1ListClustersResponse. # noqa: E501
|
|
79
|
-
:type: list[
|
|
79
|
+
:type: list[V1ExternalCluster]
|
|
80
80
|
"""
|
|
81
81
|
|
|
82
82
|
self._clusters = clusters
|
|
@@ -41,14 +41,14 @@ class V1ListProjectClustersResponse(object):
|
|
|
41
41
|
and the value is json key in definition.
|
|
42
42
|
"""
|
|
43
43
|
swagger_types = {
|
|
44
|
-
'clusters': 'list[
|
|
44
|
+
'clusters': 'list[V1ExternalCluster]'
|
|
45
45
|
}
|
|
46
46
|
|
|
47
47
|
attribute_map = {
|
|
48
48
|
'clusters': 'clusters'
|
|
49
49
|
}
|
|
50
50
|
|
|
51
|
-
def __init__(self, clusters: 'list[
|
|
51
|
+
def __init__(self, clusters: 'list[V1ExternalCluster]' =None): # noqa: E501
|
|
52
52
|
"""V1ListProjectClustersResponse - a model defined in Swagger""" # noqa: E501
|
|
53
53
|
self._clusters = None
|
|
54
54
|
self.discriminator = None
|
|
@@ -56,22 +56,22 @@ class V1ListProjectClustersResponse(object):
|
|
|
56
56
|
self.clusters = clusters
|
|
57
57
|
|
|
58
58
|
@property
|
|
59
|
-
def clusters(self) -> 'list[
|
|
59
|
+
def clusters(self) -> 'list[V1ExternalCluster]':
|
|
60
60
|
"""Gets the clusters of this V1ListProjectClustersResponse. # noqa: E501
|
|
61
61
|
|
|
62
62
|
|
|
63
63
|
:return: The clusters of this V1ListProjectClustersResponse. # noqa: E501
|
|
64
|
-
:rtype: list[
|
|
64
|
+
:rtype: list[V1ExternalCluster]
|
|
65
65
|
"""
|
|
66
66
|
return self._clusters
|
|
67
67
|
|
|
68
68
|
@clusters.setter
|
|
69
|
-
def clusters(self, clusters: 'list[
|
|
69
|
+
def clusters(self, clusters: 'list[V1ExternalCluster]'):
|
|
70
70
|
"""Sets the clusters of this V1ListProjectClustersResponse.
|
|
71
71
|
|
|
72
72
|
|
|
73
73
|
:param clusters: The clusters of this V1ListProjectClustersResponse. # noqa: E501
|
|
74
|
-
:type: list[
|
|
74
|
+
:type: list[V1ExternalCluster]
|
|
75
75
|
"""
|
|
76
76
|
|
|
77
77
|
self._clusters = clusters
|