lightning-sdk 2025.7.10__py3-none-any.whl → 2025.7.22__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 (93) hide show
  1. lightning_sdk/__init__.py +3 -2
  2. lightning_sdk/api/cloud_account_api.py +154 -0
  3. lightning_sdk/api/deployment_api.py +11 -0
  4. lightning_sdk/api/job_api.py +9 -0
  5. lightning_sdk/api/llm_api.py +11 -6
  6. lightning_sdk/api/mmt_api.py +9 -0
  7. lightning_sdk/api/pipeline_api.py +4 -3
  8. lightning_sdk/api/studio_api.py +19 -5
  9. lightning_sdk/cli/clusters_menu.py +3 -3
  10. lightning_sdk/cli/create.py +22 -10
  11. lightning_sdk/cli/deploy/_auth.py +19 -3
  12. lightning_sdk/cli/deploy/serve.py +18 -4
  13. lightning_sdk/cli/entrypoint.py +1 -1
  14. lightning_sdk/cli/start.py +37 -7
  15. lightning_sdk/deployment/deployment.py +8 -0
  16. lightning_sdk/job/base.py +37 -5
  17. lightning_sdk/job/job.py +28 -4
  18. lightning_sdk/job/v1.py +10 -1
  19. lightning_sdk/job/v2.py +15 -1
  20. lightning_sdk/lightning_cloud/openapi/__init__.py +15 -1
  21. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +335 -0
  22. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +214 -0
  23. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +5 -1
  24. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +11 -11
  25. lightning_sdk/lightning_cloud/openapi/models/__init__.py +15 -1
  26. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/blogposts_id_body.py +53 -1
  28. lightning_sdk/lightning_cloud/openapi/models/{v1_list_new_features_for_user_response.py → conversations_id_body1.py} +23 -23
  29. lightning_sdk/lightning_cloud/openapi/models/messages_id_body.py +123 -0
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +29 -3
  31. lightning_sdk/lightning_cloud/openapi/models/project_id_storage_body.py +1 -27
  32. lightning_sdk/lightning_cloud/openapi/models/protobuf_null_value.py +102 -0
  33. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py +1 -27
  35. lightning_sdk/lightning_cloud/openapi/models/uploads_upload_id_body1.py +3 -55
  36. lightning_sdk/lightning_cloud/openapi/models/user_id_upgradetrigger_body.py +175 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_ai_pod_v1.py +123 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_artifact.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_session_daily_aggregated.py +357 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_blog_post.py +53 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_type.py +1 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_complete_upload.py +3 -55
  45. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_billing_upgrade_trigger_record_response.py +97 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_blog_post_request.py +53 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py +27 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_request.py +29 -3
  50. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_function_tool.py +175 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_get_artifacts_page_response.py +29 -3
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_assistant_session_daily_aggregated_response.py +201 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +79 -1
  55. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_instance_artifact.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_like_status.py +104 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_list_notification_dialogs_response.py +149 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_managed_endpoint_models_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +53 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_presigned_url.py +1 -53
  61. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_quote_subscription_response.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_tool.py +149 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_like_response.py +149 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_message_like_response.py +149 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +105 -261
  68. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +27 -1
  69. lightning_sdk/llm/llm.py +32 -5
  70. lightning_sdk/llm/public_assistants.json +3 -1
  71. lightning_sdk/machine.py +24 -1
  72. lightning_sdk/mmt/base.py +20 -2
  73. lightning_sdk/mmt/mmt.py +25 -3
  74. lightning_sdk/mmt/v1.py +7 -1
  75. lightning_sdk/mmt/v2.py +21 -2
  76. lightning_sdk/organization.py +4 -0
  77. lightning_sdk/pipeline/pipeline.py +16 -5
  78. lightning_sdk/pipeline/printer.py +5 -3
  79. lightning_sdk/pipeline/schedule.py +844 -1
  80. lightning_sdk/pipeline/steps.py +19 -4
  81. lightning_sdk/sandbox.py +4 -1
  82. lightning_sdk/serve.py +2 -0
  83. lightning_sdk/studio.py +79 -39
  84. lightning_sdk/teamspace.py +14 -8
  85. lightning_sdk/utils/resolve.py +29 -2
  86. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/METADATA +1 -1
  87. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/RECORD +92 -78
  88. lightning_sdk/api/cluster_api.py +0 -119
  89. /lightning_sdk/cli/{inspect.py → inspection.py} +0 -0
  90. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/LICENSE +0 -0
  91. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/WHEEL +0 -0
  92. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/entry_points.txt +0 -0
  93. {lightning_sdk-2025.7.10.dist-info → lightning_sdk-2025.7.22.dist-info}/top_level.txt +0 -0
@@ -44,6 +44,7 @@ class V1Conversation(object):
44
44
  'assistant_id': 'str',
45
45
  'created_at': 'datetime',
46
46
  'id': 'str',
47
+ 'like': 'V1LikeStatus',
47
48
  'messages': 'list[V1Message]',
48
49
  'name': 'str',
49
50
  'project_id': 'str',
@@ -56,6 +57,7 @@ class V1Conversation(object):
56
57
  'assistant_id': 'assistantId',
57
58
  'created_at': 'createdAt',
58
59
  'id': 'id',
60
+ 'like': 'like',
59
61
  'messages': 'messages',
60
62
  'name': 'name',
61
63
  'project_id': 'projectId',
@@ -64,11 +66,12 @@ class V1Conversation(object):
64
66
  'user_id': 'userId'
65
67
  }
66
68
 
67
- def __init__(self, assistant_id: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, messages: 'list[V1Message]' =None, name: 'str' =None, project_id: 'str' =None, title: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
69
+ def __init__(self, assistant_id: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, like: 'V1LikeStatus' =None, messages: 'list[V1Message]' =None, name: 'str' =None, project_id: 'str' =None, title: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
68
70
  """V1Conversation - a model defined in Swagger""" # noqa: E501
69
71
  self._assistant_id = None
70
72
  self._created_at = None
71
73
  self._id = None
74
+ self._like = None
72
75
  self._messages = None
73
76
  self._name = None
74
77
  self._project_id = None
@@ -82,6 +85,8 @@ class V1Conversation(object):
82
85
  self.created_at = created_at
83
86
  if id is not None:
84
87
  self.id = id
88
+ if like is not None:
89
+ self.like = like
85
90
  if messages is not None:
86
91
  self.messages = messages
87
92
  if name is not None:
@@ -158,6 +163,27 @@ class V1Conversation(object):
158
163
 
159
164
  self._id = id
160
165
 
166
+ @property
167
+ def like(self) -> 'V1LikeStatus':
168
+ """Gets the like of this V1Conversation. # noqa: E501
169
+
170
+
171
+ :return: The like of this V1Conversation. # noqa: E501
172
+ :rtype: V1LikeStatus
173
+ """
174
+ return self._like
175
+
176
+ @like.setter
177
+ def like(self, like: 'V1LikeStatus'):
178
+ """Sets the like of this V1Conversation.
179
+
180
+
181
+ :param like: The like of this V1Conversation. # noqa: E501
182
+ :type: V1LikeStatus
183
+ """
184
+
185
+ self._like = like
186
+
161
187
  @property
162
188
  def messages(self) -> 'list[V1Message]':
163
189
  """Gets the messages of this V1Conversation. # noqa: E501
@@ -0,0 +1,97 @@
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 V1CreateBillingUpgradeTriggerRecordResponse(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
+ }
45
+
46
+ attribute_map = {
47
+ }
48
+
49
+ def __init__(self): # noqa: E501
50
+ """V1CreateBillingUpgradeTriggerRecordResponse - a model defined in Swagger""" # noqa: E501
51
+ self.discriminator = None
52
+
53
+ def to_dict(self) -> dict:
54
+ """Returns the model properties as a dict"""
55
+ result = {}
56
+
57
+ for attr, _ in six.iteritems(self.swagger_types):
58
+ value = getattr(self, attr)
59
+ if isinstance(value, list):
60
+ result[attr] = list(map(
61
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
62
+ value
63
+ ))
64
+ elif hasattr(value, "to_dict"):
65
+ result[attr] = value.to_dict()
66
+ elif isinstance(value, dict):
67
+ result[attr] = dict(map(
68
+ lambda item: (item[0], item[1].to_dict())
69
+ if hasattr(item[1], "to_dict") else item,
70
+ value.items()
71
+ ))
72
+ else:
73
+ result[attr] = value
74
+ if issubclass(V1CreateBillingUpgradeTriggerRecordResponse, dict):
75
+ for key, value in self.items():
76
+ result[key] = value
77
+
78
+ return result
79
+
80
+ def to_str(self) -> str:
81
+ """Returns the string representation of the model"""
82
+ return pprint.pformat(self.to_dict())
83
+
84
+ def __repr__(self) -> str:
85
+ """For `print` and `pprint`"""
86
+ return self.to_str()
87
+
88
+ def __eq__(self, other: 'V1CreateBillingUpgradeTriggerRecordResponse') -> bool:
89
+ """Returns true if both objects are equal"""
90
+ if not isinstance(other, V1CreateBillingUpgradeTriggerRecordResponse):
91
+ return False
92
+
93
+ return self.__dict__ == other.__dict__
94
+
95
+ def __ne__(self, other: 'V1CreateBillingUpgradeTriggerRecordResponse') -> bool:
96
+ """Returns true if both objects are not equal"""
97
+ return not self == other
@@ -43,6 +43,8 @@ class V1CreateBlogPostRequest(object):
43
43
  swagger_types = {
44
44
  'author_id': 'str',
45
45
  'category': 'str',
46
+ 'customer_logo_url': 'str',
47
+ 'customer_name': 'str',
46
48
  'description': 'str',
47
49
  'image_url': 'str',
48
50
  'lit_page_id': 'str',
@@ -52,16 +54,20 @@ class V1CreateBlogPostRequest(object):
52
54
  attribute_map = {
53
55
  'author_id': 'authorId',
54
56
  'category': 'category',
57
+ 'customer_logo_url': 'customerLogoUrl',
58
+ 'customer_name': 'customerName',
55
59
  'description': 'description',
56
60
  'image_url': 'imageUrl',
57
61
  'lit_page_id': 'litPageId',
58
62
  'title': 'title'
59
63
  }
60
64
 
61
- def __init__(self, author_id: 'str' =None, category: 'str' =None, description: 'str' =None, image_url: 'str' =None, lit_page_id: 'str' =None, title: 'str' =None): # noqa: E501
65
+ def __init__(self, author_id: 'str' =None, category: 'str' =None, customer_logo_url: 'str' =None, customer_name: 'str' =None, description: 'str' =None, image_url: 'str' =None, lit_page_id: 'str' =None, title: 'str' =None): # noqa: E501
62
66
  """V1CreateBlogPostRequest - a model defined in Swagger""" # noqa: E501
63
67
  self._author_id = None
64
68
  self._category = None
69
+ self._customer_logo_url = None
70
+ self._customer_name = None
65
71
  self._description = None
66
72
  self._image_url = None
67
73
  self._lit_page_id = None
@@ -71,6 +77,10 @@ class V1CreateBlogPostRequest(object):
71
77
  self.author_id = author_id
72
78
  if category is not None:
73
79
  self.category = category
80
+ if customer_logo_url is not None:
81
+ self.customer_logo_url = customer_logo_url
82
+ if customer_name is not None:
83
+ self.customer_name = customer_name
74
84
  if description is not None:
75
85
  self.description = description
76
86
  if image_url is not None:
@@ -122,6 +132,48 @@ class V1CreateBlogPostRequest(object):
122
132
 
123
133
  self._category = category
124
134
 
135
+ @property
136
+ def customer_logo_url(self) -> 'str':
137
+ """Gets the customer_logo_url of this V1CreateBlogPostRequest. # noqa: E501
138
+
139
+
140
+ :return: The customer_logo_url of this V1CreateBlogPostRequest. # noqa: E501
141
+ :rtype: str
142
+ """
143
+ return self._customer_logo_url
144
+
145
+ @customer_logo_url.setter
146
+ def customer_logo_url(self, customer_logo_url: 'str'):
147
+ """Sets the customer_logo_url of this V1CreateBlogPostRequest.
148
+
149
+
150
+ :param customer_logo_url: The customer_logo_url of this V1CreateBlogPostRequest. # noqa: E501
151
+ :type: str
152
+ """
153
+
154
+ self._customer_logo_url = customer_logo_url
155
+
156
+ @property
157
+ def customer_name(self) -> 'str':
158
+ """Gets the customer_name of this V1CreateBlogPostRequest. # noqa: E501
159
+
160
+
161
+ :return: The customer_name of this V1CreateBlogPostRequest. # noqa: E501
162
+ :rtype: str
163
+ """
164
+ return self._customer_name
165
+
166
+ @customer_name.setter
167
+ def customer_name(self, customer_name: 'str'):
168
+ """Sets the customer_name of this V1CreateBlogPostRequest.
169
+
170
+
171
+ :param customer_name: The customer_name of this V1CreateBlogPostRequest. # noqa: E501
172
+ :type: str
173
+ """
174
+
175
+ self._customer_name = customer_name
176
+
125
177
  @property
126
178
  def description(self) -> 'str':
127
179
  """Gets the description of this V1CreateBlogPostRequest. # noqa: E501
@@ -43,6 +43,7 @@ class V1CreateCheckoutSessionRequest(object):
43
43
  swagger_types = {
44
44
  'amount': 'float',
45
45
  'capture_method': 'str',
46
+ 'gclid': 'str',
46
47
  'org_id': 'str',
47
48
  'project_id': 'str',
48
49
  'redirect_url_cancelled': 'str',
@@ -54,6 +55,7 @@ class V1CreateCheckoutSessionRequest(object):
54
55
  attribute_map = {
55
56
  'amount': 'amount',
56
57
  'capture_method': 'captureMethod',
58
+ 'gclid': 'gclid',
57
59
  'org_id': 'orgId',
58
60
  'project_id': 'projectId',
59
61
  'redirect_url_cancelled': 'redirectUrlCancelled',
@@ -62,10 +64,11 @@ class V1CreateCheckoutSessionRequest(object):
62
64
  'wallet_type': 'walletType'
63
65
  }
64
66
 
65
- def __init__(self, amount: 'float' =None, capture_method: 'str' =None, org_id: 'str' =None, project_id: 'str' =None, redirect_url_cancelled: 'str' =None, redirect_url_succeeded: 'str' =None, transaction_type: 'str' =None, wallet_type: 'CreateCheckoutSessionRequestWalletType' =None): # noqa: E501
67
+ def __init__(self, amount: 'float' =None, capture_method: 'str' =None, gclid: 'str' =None, org_id: 'str' =None, project_id: 'str' =None, redirect_url_cancelled: 'str' =None, redirect_url_succeeded: 'str' =None, transaction_type: 'str' =None, wallet_type: 'CreateCheckoutSessionRequestWalletType' =None): # noqa: E501
66
68
  """V1CreateCheckoutSessionRequest - a model defined in Swagger""" # noqa: E501
67
69
  self._amount = None
68
70
  self._capture_method = None
71
+ self._gclid = None
69
72
  self._org_id = None
70
73
  self._project_id = None
71
74
  self._redirect_url_cancelled = None
@@ -77,6 +80,8 @@ class V1CreateCheckoutSessionRequest(object):
77
80
  self.amount = amount
78
81
  if capture_method is not None:
79
82
  self.capture_method = capture_method
83
+ if gclid is not None:
84
+ self.gclid = gclid
80
85
  if org_id is not None:
81
86
  self.org_id = org_id
82
87
  if project_id is not None:
@@ -132,6 +137,27 @@ class V1CreateCheckoutSessionRequest(object):
132
137
 
133
138
  self._capture_method = capture_method
134
139
 
140
+ @property
141
+ def gclid(self) -> 'str':
142
+ """Gets the gclid of this V1CreateCheckoutSessionRequest. # noqa: E501
143
+
144
+
145
+ :return: The gclid of this V1CreateCheckoutSessionRequest. # noqa: E501
146
+ :rtype: str
147
+ """
148
+ return self._gclid
149
+
150
+ @gclid.setter
151
+ def gclid(self, gclid: 'str'):
152
+ """Sets the gclid of this V1CreateCheckoutSessionRequest.
153
+
154
+
155
+ :param gclid: The gclid of this V1CreateCheckoutSessionRequest. # noqa: E501
156
+ :type: str
157
+ """
158
+
159
+ self._gclid = gclid
160
+
135
161
  @property
136
162
  def org_id(self) -> 'str':
137
163
  """Gets the org_id of this V1CreateCheckoutSessionRequest. # noqa: E501
@@ -46,7 +46,8 @@ class V1CreateSubscriptionCheckoutSessionRequest(object):
46
46
  'product_id': 'str',
47
47
  'redirect_url_failed': 'str',
48
48
  'redirect_url_succeeded': 'str',
49
- 'seats': 'int'
49
+ 'seats': 'int',
50
+ 'upgrade_feature': 'str'
50
51
  }
51
52
 
52
53
  attribute_map = {
@@ -55,10 +56,11 @@ class V1CreateSubscriptionCheckoutSessionRequest(object):
55
56
  'product_id': 'productId',
56
57
  'redirect_url_failed': 'redirectUrlFailed',
57
58
  'redirect_url_succeeded': 'redirectUrlSucceeded',
58
- 'seats': 'seats'
59
+ 'seats': 'seats',
60
+ 'upgrade_feature': 'upgradeFeature'
59
61
  }
60
62
 
61
- def __init__(self, billing_period: 'V1BillingPeriod' =None, org_id: 'str' =None, product_id: 'str' =None, redirect_url_failed: 'str' =None, redirect_url_succeeded: 'str' =None, seats: 'int' =None): # noqa: E501
63
+ def __init__(self, billing_period: 'V1BillingPeriod' =None, org_id: 'str' =None, product_id: 'str' =None, redirect_url_failed: 'str' =None, redirect_url_succeeded: 'str' =None, seats: 'int' =None, upgrade_feature: 'str' =None): # noqa: E501
62
64
  """V1CreateSubscriptionCheckoutSessionRequest - a model defined in Swagger""" # noqa: E501
63
65
  self._billing_period = None
64
66
  self._org_id = None
@@ -66,6 +68,7 @@ class V1CreateSubscriptionCheckoutSessionRequest(object):
66
68
  self._redirect_url_failed = None
67
69
  self._redirect_url_succeeded = None
68
70
  self._seats = None
71
+ self._upgrade_feature = None
69
72
  self.discriminator = None
70
73
  if billing_period is not None:
71
74
  self.billing_period = billing_period
@@ -79,6 +82,8 @@ class V1CreateSubscriptionCheckoutSessionRequest(object):
79
82
  self.redirect_url_succeeded = redirect_url_succeeded
80
83
  if seats is not None:
81
84
  self.seats = seats
85
+ if upgrade_feature is not None:
86
+ self.upgrade_feature = upgrade_feature
82
87
 
83
88
  @property
84
89
  def billing_period(self) -> 'V1BillingPeriod':
@@ -206,6 +211,27 @@ class V1CreateSubscriptionCheckoutSessionRequest(object):
206
211
 
207
212
  self._seats = seats
208
213
 
214
+ @property
215
+ def upgrade_feature(self) -> 'str':
216
+ """Gets the upgrade_feature of this V1CreateSubscriptionCheckoutSessionRequest. # noqa: E501
217
+
218
+
219
+ :return: The upgrade_feature of this V1CreateSubscriptionCheckoutSessionRequest. # noqa: E501
220
+ :rtype: str
221
+ """
222
+ return self._upgrade_feature
223
+
224
+ @upgrade_feature.setter
225
+ def upgrade_feature(self, upgrade_feature: 'str'):
226
+ """Sets the upgrade_feature of this V1CreateSubscriptionCheckoutSessionRequest.
227
+
228
+
229
+ :param upgrade_feature: The upgrade_feature of this V1CreateSubscriptionCheckoutSessionRequest. # noqa: E501
230
+ :type: str
231
+ """
232
+
233
+ self._upgrade_feature = upgrade_feature
234
+
209
235
  def to_dict(self) -> dict:
210
236
  """Returns the model properties as a dict"""
211
237
  result = {}
@@ -41,6 +41,7 @@ class V1ExternalClusterSpec(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'ai_pod_v1': 'V1AiPodV1',
44
45
  'auth_token': 'str',
45
46
  'available_accelerators': 'list[str]',
46
47
  'aws_v1': 'V1AWSDirectV1',
@@ -73,6 +74,7 @@ class V1ExternalClusterSpec(object):
73
74
  }
74
75
 
75
76
  attribute_map = {
77
+ 'ai_pod_v1': 'aiPodV1',
76
78
  'auth_token': 'authToken',
77
79
  'available_accelerators': 'availableAccelerators',
78
80
  'aws_v1': 'awsV1',
@@ -104,8 +106,9 @@ class V1ExternalClusterSpec(object):
104
106
  'vultr_v1': 'vultrV1'
105
107
  }
106
108
 
107
- def __init__(self, auth_token: 'str' =None, available_accelerators: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cloudflare_v1: 'V1CloudflareV1' =None, cluster_type: 'V1ClusterType' =None, compute_cluster_ids: 'list[str]' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, driver: 'V1CloudProvider' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, kubernetes_v1: 'V1KubernetesDirectV1' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, locked_zones: 'list[str]' =None, monitor_deletion_disabled: 'bool' =None, nebius_v1: 'V1NebiusDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, parent_cluster_id: 'str' =None, parent_cluster_type: 'str' =None, pause_automation: 'bool' =None, reservation_details: 'V1ReservationDetails' =None, reserved_capacity_provider: 'bool' =None, reserved_instances_only: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, voltage_park_v1: 'V1VoltageParkDirectV1' =None, vultr_v1: 'V1VultrDirectV1' =None): # noqa: E501
109
+ def __init__(self, ai_pod_v1: 'V1AiPodV1' =None, auth_token: 'str' =None, available_accelerators: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cloudflare_v1: 'V1CloudflareV1' =None, cluster_type: 'V1ClusterType' =None, compute_cluster_ids: 'list[str]' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, driver: 'V1CloudProvider' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, kubernetes_v1: 'V1KubernetesDirectV1' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, locked_zones: 'list[str]' =None, monitor_deletion_disabled: 'bool' =None, nebius_v1: 'V1NebiusDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, parent_cluster_id: 'str' =None, parent_cluster_type: 'str' =None, pause_automation: 'bool' =None, reservation_details: 'V1ReservationDetails' =None, reserved_capacity_provider: 'bool' =None, reserved_instances_only: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, voltage_park_v1: 'V1VoltageParkDirectV1' =None, vultr_v1: 'V1VultrDirectV1' =None): # noqa: E501
108
110
  """V1ExternalClusterSpec - a model defined in Swagger""" # noqa: E501
111
+ self._ai_pod_v1 = None
109
112
  self._auth_token = None
110
113
  self._available_accelerators = None
111
114
  self._aws_v1 = None
@@ -136,6 +139,8 @@ class V1ExternalClusterSpec(object):
136
139
  self._voltage_park_v1 = None
137
140
  self._vultr_v1 = None
138
141
  self.discriminator = None
142
+ if ai_pod_v1 is not None:
143
+ self.ai_pod_v1 = ai_pod_v1
139
144
  if auth_token is not None:
140
145
  self.auth_token = auth_token
141
146
  if available_accelerators is not None:
@@ -195,6 +200,27 @@ class V1ExternalClusterSpec(object):
195
200
  if vultr_v1 is not None:
196
201
  self.vultr_v1 = vultr_v1
197
202
 
203
+ @property
204
+ def ai_pod_v1(self) -> 'V1AiPodV1':
205
+ """Gets the ai_pod_v1 of this V1ExternalClusterSpec. # noqa: E501
206
+
207
+
208
+ :return: The ai_pod_v1 of this V1ExternalClusterSpec. # noqa: E501
209
+ :rtype: V1AiPodV1
210
+ """
211
+ return self._ai_pod_v1
212
+
213
+ @ai_pod_v1.setter
214
+ def ai_pod_v1(self, ai_pod_v1: 'V1AiPodV1'):
215
+ """Sets the ai_pod_v1 of this V1ExternalClusterSpec.
216
+
217
+
218
+ :param ai_pod_v1: The ai_pod_v1 of this V1ExternalClusterSpec. # noqa: E501
219
+ :type: V1AiPodV1
220
+ """
221
+
222
+ self._ai_pod_v1 = ai_pod_v1
223
+
198
224
  @property
199
225
  def auth_token(self) -> 'str':
200
226
  """Gets the auth_token of this V1ExternalClusterSpec. # noqa: E501
@@ -0,0 +1,175 @@
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 V1FunctionTool(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
+ 'description': 'str',
45
+ 'name': 'str',
46
+ 'parameters': 'object'
47
+ }
48
+
49
+ attribute_map = {
50
+ 'description': 'description',
51
+ 'name': 'name',
52
+ 'parameters': 'parameters'
53
+ }
54
+
55
+ def __init__(self, description: 'str' =None, name: 'str' =None, parameters: 'object' =None): # noqa: E501
56
+ """V1FunctionTool - a model defined in Swagger""" # noqa: E501
57
+ self._description = None
58
+ self._name = None
59
+ self._parameters = None
60
+ self.discriminator = None
61
+ if description is not None:
62
+ self.description = description
63
+ if name is not None:
64
+ self.name = name
65
+ if parameters is not None:
66
+ self.parameters = parameters
67
+
68
+ @property
69
+ def description(self) -> 'str':
70
+ """Gets the description of this V1FunctionTool. # noqa: E501
71
+
72
+
73
+ :return: The description of this V1FunctionTool. # noqa: E501
74
+ :rtype: str
75
+ """
76
+ return self._description
77
+
78
+ @description.setter
79
+ def description(self, description: 'str'):
80
+ """Sets the description of this V1FunctionTool.
81
+
82
+
83
+ :param description: The description of this V1FunctionTool. # noqa: E501
84
+ :type: str
85
+ """
86
+
87
+ self._description = description
88
+
89
+ @property
90
+ def name(self) -> 'str':
91
+ """Gets the name of this V1FunctionTool. # noqa: E501
92
+
93
+
94
+ :return: The name of this V1FunctionTool. # noqa: E501
95
+ :rtype: str
96
+ """
97
+ return self._name
98
+
99
+ @name.setter
100
+ def name(self, name: 'str'):
101
+ """Sets the name of this V1FunctionTool.
102
+
103
+
104
+ :param name: The name of this V1FunctionTool. # noqa: E501
105
+ :type: str
106
+ """
107
+
108
+ self._name = name
109
+
110
+ @property
111
+ def parameters(self) -> 'object':
112
+ """Gets the parameters of this V1FunctionTool. # noqa: E501
113
+
114
+
115
+ :return: The parameters of this V1FunctionTool. # noqa: E501
116
+ :rtype: object
117
+ """
118
+ return self._parameters
119
+
120
+ @parameters.setter
121
+ def parameters(self, parameters: 'object'):
122
+ """Sets the parameters of this V1FunctionTool.
123
+
124
+
125
+ :param parameters: The parameters of this V1FunctionTool. # noqa: E501
126
+ :type: object
127
+ """
128
+
129
+ self._parameters = parameters
130
+
131
+ def to_dict(self) -> dict:
132
+ """Returns the model properties as a dict"""
133
+ result = {}
134
+
135
+ for attr, _ in six.iteritems(self.swagger_types):
136
+ value = getattr(self, attr)
137
+ if isinstance(value, list):
138
+ result[attr] = list(map(
139
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
140
+ value
141
+ ))
142
+ elif hasattr(value, "to_dict"):
143
+ result[attr] = value.to_dict()
144
+ elif isinstance(value, dict):
145
+ result[attr] = dict(map(
146
+ lambda item: (item[0], item[1].to_dict())
147
+ if hasattr(item[1], "to_dict") else item,
148
+ value.items()
149
+ ))
150
+ else:
151
+ result[attr] = value
152
+ if issubclass(V1FunctionTool, dict):
153
+ for key, value in self.items():
154
+ result[key] = value
155
+
156
+ return result
157
+
158
+ def to_str(self) -> str:
159
+ """Returns the string representation of the model"""
160
+ return pprint.pformat(self.to_dict())
161
+
162
+ def __repr__(self) -> str:
163
+ """For `print` and `pprint`"""
164
+ return self.to_str()
165
+
166
+ def __eq__(self, other: 'V1FunctionTool') -> bool:
167
+ """Returns true if both objects are equal"""
168
+ if not isinstance(other, V1FunctionTool):
169
+ return False
170
+
171
+ return self.__dict__ == other.__dict__
172
+
173
+ def __ne__(self, other: 'V1FunctionTool') -> bool:
174
+ """Returns true if both objects are not equal"""
175
+ return not self == other
@@ -41,19 +41,24 @@ class V1GetArtifactsPageResponse(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'artifacts': 'list[V1Artifact]'
44
+ 'artifacts': 'list[V1Artifact]',
45
+ 'count': 'str'
45
46
  }
46
47
 
47
48
  attribute_map = {
48
- 'artifacts': 'artifacts'
49
+ 'artifacts': 'artifacts',
50
+ 'count': 'count'
49
51
  }
50
52
 
51
- def __init__(self, artifacts: 'list[V1Artifact]' =None): # noqa: E501
53
+ def __init__(self, artifacts: 'list[V1Artifact]' =None, count: 'str' =None): # noqa: E501
52
54
  """V1GetArtifactsPageResponse - a model defined in Swagger""" # noqa: E501
53
55
  self._artifacts = None
56
+ self._count = None
54
57
  self.discriminator = None
55
58
  if artifacts is not None:
56
59
  self.artifacts = artifacts
60
+ if count is not None:
61
+ self.count = count
57
62
 
58
63
  @property
59
64
  def artifacts(self) -> 'list[V1Artifact]':
@@ -76,6 +81,27 @@ class V1GetArtifactsPageResponse(object):
76
81
 
77
82
  self._artifacts = artifacts
78
83
 
84
+ @property
85
+ def count(self) -> 'str':
86
+ """Gets the count of this V1GetArtifactsPageResponse. # noqa: E501
87
+
88
+
89
+ :return: The count of this V1GetArtifactsPageResponse. # noqa: E501
90
+ :rtype: str
91
+ """
92
+ return self._count
93
+
94
+ @count.setter
95
+ def count(self, count: 'str'):
96
+ """Sets the count of this V1GetArtifactsPageResponse.
97
+
98
+
99
+ :param count: The count of this V1GetArtifactsPageResponse. # noqa: E501
100
+ :type: str
101
+ """
102
+
103
+ self._count = count
104
+
79
105
  def to_dict(self) -> dict:
80
106
  """Returns the model properties as a dict"""
81
107
  result = {}