lightning-sdk 0.2.20__py3-none-any.whl → 0.2.21rc1__py3-none-any.whl

lightning_sdk/pipeline/pipeline.py

@@ -63,7 +63,7 @@ class Pipeline:
         pipeline = None
 
         if name.startswith("pip_"):
-            pipeline = self._pipeline_api.get_pipeline_by_id(name, self._teamspace.id)
+            pipeline = self._pipeline_api.get_pipeline_by_id(self._teamspace.id, name)
 
         if pipeline:
             self._name = pipeline.name

lightning_sdk/services/utilities.py

@@ -35,7 +35,9 @@ def _get_project(client: LightningClient, project_name: Optional[str] = None) ->
     raise ValueError("No valid projects found. Please reach out to lightning.ai team to create a project")
 
 
-def _get_cluster(client: LightningClient, project_id: str, cluster_id: Optional[str] = None) -> V1ProjectClusterBinding:
+def _get_cluster(
+    client: LightningClient, project_id: str, cluster_id: Optional[str] = None, allow_neoclouds: bool = False
+) -> V1ProjectClusterBinding:
     """Get a project membership for the user from the backend."""
     clusters = client.projects_service_list_project_cluster_bindings(project_id=project_id)
     if cluster_id:
@@ -47,6 +49,18 @@ def _get_cluster(client: LightningClient, project_id: str, cluster_id: Optional[
             f"Found {[c.cluster_id for c in clusters.clusters]}."
         )
 
+    # filter neoclouds out
+    if not allow_neoclouds:
+        cluster_objs = client.cluster_service_list_clusters(project_id=project_id)
+        # filter for aws or gcp cluster
+        valid_clusters = filter(
+            lambda c: c.spec.aws_v1 is not None or c.spec.google_cloud_v1 is not None, cluster_objs.clusters
+        )
+        valid_clusters = {c.id for c in valid_clusters}
+
+        if len(valid_clusters):
+            clusters.clusters = list(filter(lambda c: c.cluster_id in valid_clusters, clusters.clusters))
+
     clusters = sorted(clusters.clusters, key=lambda x: x.created_at)
     if len(clusters):
         return clusters[0]

{lightning_sdk-0.2.20.dist-info → lightning_sdk-0.2.21rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.2.20
+Version: 0.2.21rc1
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License

{lightning_sdk-0.2.20.dist-info → lightning_sdk-0.2.21rc1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=WAHxXqjnCfJI2Lo82tot4qgg93_dNrw3xFJWzkoCGgc,1105
+lightning_sdk/__init__.py,sha256=rf_9sAIQvu29ycoVRuBKYsXAYo6IEB7sIlB6OYkdZc4,1109
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
 lightning_sdk/base_studio.py,sha256=-ZnnZl5fG4w3BOQAelQndg7wEao2jGxpaYrpujlB7OQ,3353
@@ -21,15 +21,15 @@ lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEM
 lightning_sdk/api/ai_hub_api.py,sha256=azqDZ-PzasVAcoQHno7k7OO_xFOHQ4NDozxF8jEh83Y,7864
 lightning_sdk/api/base_studio_api.py,sha256=cipTG4I5OvFLDVAbM1E0lvGjP9YhhoByfa9mQFF9ueI,3383
 lightning_sdk/api/cluster_api.py,sha256=mfHGKHy-RNpI7nKlhw9Zc7Jv6f4DCFq-qRIwOR3VptY,3972
-lightning_sdk/api/deployment_api.py,sha256=Dw4NaFiNTYhqz099N5hYZzF-K0dUoWnGK2vE--iUy1c,24104
+lightning_sdk/api/deployment_api.py,sha256=Bogjf0rqdxVQmi7rbwZLSvevSlLVk7eIFaXFi6R75EM,24213
 lightning_sdk/api/job_api.py,sha256=_mMAI_BG_48i-BLwCP_U72zgmM5zYa2KUZ7u66HWkIc,13568
-lightning_sdk/api/license_api.py,sha256=F5JervZmLqPe6WNFlTPjBlS1GtGHkUfiTTlFaDovNHk,2434
+lightning_sdk/api/license_api.py,sha256=IVaexSSXHyeu1RBQ5Cab1XqswdnVueDvntH9TMlZp-4,2406
 lightning_sdk/api/lit_container_api.py,sha256=jCJVwd-3MNjejL9FyvH89pzt-SeG3G8qCJD16iTMJAQ,11454
-lightning_sdk/api/llm_api.py,sha256=1mk2gbaQ3iinV7BsNc31Nf-iRdUwPWDsl8qf4PsAads,5208
+lightning_sdk/api/llm_api.py,sha256=GKQrk7Id_E4_Bp4j70ebSEgiB--iVrCd9SEA1D0VG2I,7180
 lightning_sdk/api/mmt_api.py,sha256=-v7ATab-ThAM-HRClS92Ehxuu9MlBfdKWWFCGvVUHiM,8962
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
 lightning_sdk/api/pipeline_api.py,sha256=P5P9C6qOpyBGU0t5N68h1LuFAsAKmPPgkac6uObrYKw,1676
-lightning_sdk/api/studio_api.py,sha256=vEzfu0lITyu9dAi6IPCDIqrhcwMWzDDOBf9B-kmXyEI,30997
+lightning_sdk/api/studio_api.py,sha256=X1M229YnGiE1EfnncQbBFXADVNnWAn14l5h5KBZJmVs,31192
 lightning_sdk/api/teamspace_api.py,sha256=CsaaxmaLmTRIRwu37wtQ3quGYql62HJT3BZJ3Q-1d9c,16854
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
 lightning_sdk/api/utils.py,sha256=K0Uc6D1KxxjFElqjsuCXg85SYreQi3P7jJ1VzP0MpEQ,23966
@@ -64,7 +64,7 @@ lightning_sdk/cli/deploy/_auth.py,sha256=F3W2eU33LIsfFu7AQyBRN01BL800vJJFJCHwg0B
 lightning_sdk/cli/deploy/devbox.py,sha256=HOQMWehwaveawVTyBDaZhNZs_NL5g4RCWs_lf0hIW5U,6800
 lightning_sdk/cli/deploy/serve.py,sha256=qqml7oYf4IPcoiL13P2ly247_PQ8IGv-8hM2NBWePJc,14458
 lightning_sdk/deployment/__init__.py,sha256=dXsa4psDzFYFklsq3JC-2V_L4FQjGZnQAf-ZiVlqG9c,545
-lightning_sdk/deployment/deployment.py,sha256=Bdgbc7pKke2BnDd2eqJNGT34Q2XngCJznQojfXS_vRc,20116
+lightning_sdk/deployment/deployment.py,sha256=BTAo--qbkLXuaUhTJIkWHiMmSJv4t5MNPEBB0T2wdkk,20627
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
 lightning_sdk/job/base.py,sha256=TS5KavtfBAFbLbNqqumEizY9edjO1joSmtUdcO5CThQ,17748
 lightning_sdk/job/job.py,sha256=1Xf0ne4wwXpkb_GJgWD8mfueJZoKR8G4lC5T6OXijlw,13075
@@ -78,7 +78,7 @@ lightning_sdk/lightning_cloud/login.py,sha256=k4iE5qLUMB8_VeQa1sNz7jPCWLOV5S7j9Q
 lightning_sdk/lightning_cloud/rest_client.py,sha256=K2J_QXihpbp5tVzs9PzffgMCQY1NeH5tRpuJBYj27vU,7125
 lightning_sdk/lightning_cloud/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/lightning_cloud/cli/__main__.py,sha256=aHrigVV0qIp74MuYTdN6uO8a9Xp6E1aOwdR8w1fG-7c,688
-lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=6R76ikKwd5KGuhj4EGFYu9TwJxz6vhFIPrNymnG5sPI,105787
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=KTNMCBDB1g61f00A52lS2YlK0BRHNNeDOCoDgZR0Ew4,105676
 lightning_sdk/lightning_cloud/openapi/api_client.py,sha256=pUTQMNcZmH4BhpnuAXuT7wnegaxaX26bzdEWjdoLeTo,25630
 lightning_sdk/lightning_cloud/openapi/configuration.py,sha256=8KpFKEm-IKe3t0_o5H1Y0EJX06r94Gk5cIQerA0TNk0,8292
 lightning_sdk/lightning_cloud/openapi/rest.py,sha256=ZPPr6ZkBp6LtuAsiUU7D8Pz8Dt9ECbEM_26Ov74tdpw,13322
@@ -95,7 +95,7 @@ lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=4RIioSmH
 lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py,sha256=4FRsl7HTfKfrM00tEB9BRz7p1heBucqqkb7TST58zfs,55075
 lightning_sdk/lightning_cloud/openapi/api/dataset_service_api.py,sha256=wCrcXOMuVs5rA4PHyrVJqpInO-Vy1hulddjCJUenPJc,23508
 lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py,sha256=VK4ZuwoE0pvWEfSWr2KZNoutRtjovJFH3-6NqFmQjNg,31887
-lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py,sha256=IgERziSMhxLxK5ATpsNpNs5Ybpu577mCH7f6yBCr7uw,113191
+lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py,sha256=9jd-VS-LdBWmx2rLQxBd1PWKRFSK3pVE0f4Flh22Vxg,113746
 lightning_sdk/lightning_cloud/openapi/api/experiments_service_api.py,sha256=lYQt6nvbaqFnRsWFf-jL3Fou-A7Hx6qgrpRgHI3eARQ,9844
 lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py,sha256=CuQjJKp7fS_PtYO8wFk1fM3kmKy9DDC6R1qjvWtI3kg,41970
 lightning_sdk/lightning_cloud/openapi/api/git_credentials_service_api.py,sha256=tz7xhdaeHoR9_KEx8vR0SyFLUsjElrcwl9hBaT_X_mc,19757
@@ -122,8 +122,8 @@ lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py,sha256=C-gEj9
 lightning_sdk/lightning_cloud/openapi/api/ssh_public_key_service_api.py,sha256=z-CQmVfinxsv8KDfghVokdL8kbeu95EAtX2kBA0apS8,24595
 lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py,sha256=kFyE4kt5NzGmqbAJW63VJOIREzZ0KMAGRMFIgMKEhxI,69459
 lightning_sdk/lightning_cloud/openapi/api/studio_jobs_service_api.py,sha256=VJ7VcSvdmGbPkgmQZhaOSR47olcNPMzRlmT5J5uKsng,27877
-lightning_sdk/lightning_cloud/openapi/api/user_service_api.py,sha256=MR0jvZY0I-fFEPNpcSqrN3vl9XjrEn_D7_gYhwYeMrc,66796
-lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=AuAfcab21ltBUAdcpW7JwKNJ-bKYaZhGhcqRTaCPXD0,99975
+lightning_sdk/lightning_cloud/openapi/api/user_service_api.py,sha256=ztdbmsD6MygMfKX8GhwvMCNhI8h3nTifjolaQSEA9qQ,63606
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=HnMxWyVyetdddEB93r4HRmPhcJWhfJs3e2A68Vvy0V0,99864
 lightning_sdk/lightning_cloud/openapi/models/affiliatelinks_id_body.py,sha256=X087AkCafKp2g8E2MR5ghU3pxCNNcoehZJttvuVfdC8,4274
 lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py,sha256=NzKKkTs2PtMSIJJYxeh3aDBj-djBrI_qATGz7n5Oqdk,9234
 lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py,sha256=vYRXPGrwpj8nsSVwlaf3c6d5ysuvpNOXrfeUOKZn0dM,21508
@@ -279,7 +279,7 @@ lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py,sha256=JUD
 lightning_sdk/lightning_cloud/openapi/models/stream_result_of_v1_conversation_response_chunk.py,sha256=4MA1NbHpLqP5uuzHIm8FqwMhzzYbCGGK-QoXrkmH0E4,4808
 lightning_sdk/lightning_cloud/openapi/models/stream_result_of_v1_get_long_running_command_in_cloud_space_response.py,sha256=btVXTDqnRONLLrql3PchTcVU-l6mTPSTPCpBNF4TH6Y,5128
 lightning_sdk/lightning_cloud/openapi/models/studioapp_jobs_body.py,sha256=UY2dYV5XKUmk3xP3uWviA9nWRNgxyU63cG9DCiez0Ic,8045
-lightning_sdk/lightning_cloud/openapi/models/update.py,sha256=Rq6ahzZoVMhfnNJIi1pYXa09nAVV09Vmz_ZioSNbqsQ,9811
+lightning_sdk/lightning_cloud/openapi/models/update.py,sha256=CeD6K8vvX47qSCFG8fcqcKvv2Iw-DNQG9PIOesBPL2Y,12099
 lightning_sdk/lightning_cloud/openapi/models/update1.py,sha256=L7Q9tJ9PpjRHG_ymnfUFhhfcA-oT9pn9rLNgmnBxwpc,9879
 lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body.py,sha256=TvPHj5LvRGCwMwIHIX7pmEsPT-XFEWjutaKzKtSoUo4,4482
 lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body1.py,sha256=0PD4kYCsvIxwA8EP6QZybO_2ZVcvDBqU8ZG6-JayO1o,4496
@@ -349,8 +349,8 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics.p
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_cold_start_metrics_stats.py,sha256=z1A7rGjP24c4QSX8oLPMiGwZy81RsGzHj61haQUo7Gg,12002
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_engagement_response.py,sha256=9qAqwfTU1L2GsLqIkUgW7oR8aHnsvvZL_Mb2jePKq7k,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py,sha256=cgM0EZkX306dkQCPcc1jD99T9JJrofltVSaE8B-clIs,4114
-lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py,sha256=DATyonxIx2vkEh-W-aaEFKvhakoXdcTKdvMPZofwHks,10715
-lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py,sha256=VyEfm9O3l2nFfJIADcxBOXIuP3WvRZY9NsiHQU2vmZo,8885
+lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py,sha256=_wZRm-X_Rvwt5lLVhvfB2XlChTehcfyO3pQ95nAjlV0,12209
+lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py,sha256=1YbSYs_6NktS1ASlturv4_zkRdM5gnRRafWFPlVNm0c,10003
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py,sha256=B6RPn6mn5rorzj-JQjjBHQpI4nAJpqSGDsRE3Va6dfA,3256
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_instance_collab_status.py,sha256=dtN2yV40wb-F_2ncCQl8n_vk0sp0ybDiP2VuH8Mx8Lk,6813
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_instance_config.py,sha256=kBpqxSiAudcFwBcmWOgMSda8CW1qr4ZgpC8FlLYQJWU,8916
@@ -369,7 +369,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py,sha256=XyoUjcO4
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py,sha256=UZqx7u-6dtbETKJ1huh5qmhPizvvSoD-7dQJ5UajQZ4,45251
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_availability.py,sha256=WrS0UpcTqqCu79ny5yOBJ16RNs5B4AQSXbniWE7iez0,7103
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py,sha256=vpwujOiKIrB1PKg7nzCFs7rvi3wL_Q1OBV5KyTTDNG0,15208
-lightning_sdk/lightning_cloud/openapi/models/v1_cluster_deletion_options.py,sha256=GyhAXDpVtfXfaUybBE0aRRvJ6d3DfD7VV3xH_fgQOlE,4859
+lightning_sdk/lightning_cloud/openapi/models/v1_cluster_deletion_options.py,sha256=68iaqDjJYOpZsvaOzGJCsBVF4ISrFe8K9-o3bwr2tDI,5793
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_encryption_key.py,sha256=Cf9AIJCmKjQO0Gp7UbXIXq2IXkxLUBgSgK3sGSYzg8U,6681
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_names.py,sha256=Xs3mhVEiuJVbkKwMarY7iB_NwKY9_sK0xGgXDMuo0cs,3637
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_proxy.py,sha256=4LbHvINs2CwQnuPg_uWw03GNNRT_wdqX42m_Zlqg2dY,5547
@@ -410,7 +410,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_create_blog_post_request.py,sha2
 lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py,sha256=D9Iv5eelLkJ_4lbBJVP4RZr0r-cu2g5RTb3JMUlwXPI,10165
 lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_response.py,sha256=-Qd3G_DIiPxyPYfHsnO0jDX_HBdG7MVVsxhYqI1QQeA,3731
 lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_app_instance_response.py,sha256=hrhbbyDpX8OMGtiJrWxWFBZWWTyxm2jgiQ6ITvDOvsA,5198
-lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py,sha256=qr9Ll7S_Tx61HpsAIITxvYcanZehZE5Oq-Fgi3XFv34,10592
+lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py,sha256=1a9IWpKgHud2DssNDkUmhlnkJbCX2PlAF6M12DTYpFA,13336
 lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_instance_metric_response.py,sha256=nlcwQOpHnExhbrU2JA4EllADm7E0eARkIUa693f-Q10,3130
 lightning_sdk/lightning_cloud/openapi/models/v1_create_cluster_capacity_reservation_response.py,sha256=_MiRFv5p8S-eIx9SxOpu4pnrep5DqeVYPTDAAloajps,7515
 lightning_sdk/lightning_cloud/openapi/models/v1_create_cluster_encryption_keys_request.py,sha256=SuCWcub0gfFpFohVz-nNcweY2slx6O3kIR2SryS3WEs,3895
@@ -429,7 +429,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_create_multi_machine_job_request
 lightning_sdk/lightning_cloud/openapi/models/v1_create_multi_part_upload_response.py,sha256=6wG2wImsg8-C8zWbVDxseNKWUblev9jmjt8CJiBnJxs,3850
 lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py,sha256=YRaiSv5-n_OKxg2HTJT5bdS5-NIeKhEherMGoudzIBQ,14249
 lightning_sdk/lightning_cloud/openapi/models/v1_create_pipeline_template_request.py,sha256=1yft4xwUCMqx-GGlZfeCIpZCS5nqs44NwWmTG_kHBXA,12087
-lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py,sha256=3MUMaITCxHLiQrd535Paomd7l8qDOJ0oehfOg2d-aNo,9810
+lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py,sha256=6Yrns9d7r3dUN2gsApVz0RhRgKR1bYyO3t8qEJUYTQI,12976
 lightning_sdk/lightning_cloud/openapi/models/v1_create_server_alert_response.py,sha256=xg8zP8ESSMaCQddNffS5k7U_0SfdYquXEXczZabTFcA,3052
 lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_request.py,sha256=avVOuNMWp_zAKdIKrzBTFCRy4FJGXeGvFPfgZBcxPx4,6624
 lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_response.py,sha256=tX9omq4WA6WXxeEY9dtMwIBKHI2qm313YKCsNYB6_G0,3751
@@ -620,7 +620,6 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_user_balance_response.py,sha
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_notification_preferences_response.py,sha256=9LzkmmAYYCcMHZX2a55wiwE87QitC_-MiKeyX3uJeDQ,4402
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py,sha256=4aguhh3Xiv2RK5cvdZmFcOKUo2UOgG_0oEOspfgRWFE,32165
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py,sha256=x8AVSyjvzRtGxcV_Drik6TCKm3eY1jQ9bwMe4Akzoo8,12088
-lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py,sha256=0k1scTY1IAeKr3p0r7eV_xWRE7Nx54doTGygiQQPkbM,6879
 lightning_sdk/lightning_cloud/openapi/models/v1_git_credentials.py,sha256=siIFxkpP0VVU53XVSj9ZHB3JWvUr9qOFliSDDx28i-g,6705
 lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=dcEf_ihjdYvz90tU5qcd3A0SIPhkZ6O4ZmZ2JAr-M1k,18619
 lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1_status.py,sha256=Md6rMiUCMY4oDVE2IYaALD_I0o2PQIKF-9628IGmTyY,3791
@@ -792,7 +791,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py,sha256=T0IXm
 lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py,sha256=WaN3PgJhZKTQO2FTeIdf6cWsFgnu1q3KKQ_xWMKYq3g,13940
 lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py,sha256=1of-AEIimGnUd_0fpS5U8apGFYdsIXogRpFGxed67s0,5892
 lightning_sdk/lightning_cloud/openapi/models/v1_membership.py,sha256=yRLmKJvrElPTkDU_T67oo_cyJzLJZKTbfeXLdgrAyrs,18936
-lightning_sdk/lightning_cloud/openapi/models/v1_message.py,sha256=v2JNnO12b2tMxHXmnxw-UrQJadRkXg4GSpPE3w4FIe0,11645
+lightning_sdk/lightning_cloud/openapi/models/v1_message.py,sha256=KacSnSdhY1c05GDQk2VIArGe4s1StHoq64fCEdw5SSU,13326
 lightning_sdk/lightning_cloud/openapi/models/v1_message_author.py,sha256=QRC1HofufQOPROrxwtLeaEfE8f1h_ehZ0tdqlYfkgyM,3591
 lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py,sha256=rlYtO8Llal6jISWpQMyOC8zMOvdVWaljXTwlHp_gzdc,4535
 lightning_sdk/lightning_cloud/openapi/models/v1_message_content_type.py,sha256=4umoW08b-M8UXpnhvUP7yb3VxjowY5F8RGqVVZrpB3M,3089
@@ -901,7 +900,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_restart_timing.py,sha256=8fx4cqA
 lightning_sdk/lightning_cloud/openapi/models/v1_restore_deployment_release_response.py,sha256=UkaXXGq-i5YDPzBOy5Zj1nJQZJeyBs1Z9VO5a2DYPkk,3094
 lightning_sdk/lightning_cloud/openapi/models/v1_role.py,sha256=-qebDpqvEYzWLAV9o0klUom-8zOFM6a7cJhrhrsdhhY,7554
 lightning_sdk/lightning_cloud/openapi/models/v1_rolling_update_strategy.py,sha256=Mxi5aq4dve439nGwv329293wxmNOg0ny0M04CdbR3hE,4634
-lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py,sha256=en-DOKlWlMziTtNWfFt262K5s0jeBjw0KSima64zg7k,7146
+lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py,sha256=CxmeADHkr62HxVy1SHdOG4yePy0boKFNDSoruwUEHhk,7910
 lightning_sdk/lightning_cloud/openapi/models/v1_rule.py,sha256=35UHmQVEVAiE2kF9qMW8lG1KCzz6nVp5t84Ops0ugZY,5850
 lightning_sdk/lightning_cloud/openapi/models/v1_rule_action.py,sha256=yebZDL5MKUiNwPysT6HzB9JrHPSCg3W4kVta8yjAgl4,3221
 lightning_sdk/lightning_cloud/openapi/models/v1_rule_condition.py,sha256=nE-VbhbyaLswY514acnjljeJRxnVICCTKWVe4cKFQ8w,5320
@@ -1002,7 +1001,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_upstream_open_ai.py,sha256=jt1qQ
 lightning_sdk/lightning_cloud/openapi/models/v1_usage.py,sha256=ozMzoGD9mfZGYy4J5j50Dps19Y6o8cn-aYW_oRMZMy8,16865
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py,sha256=U7qC698Xj5tb3D93ZskG6sDf3lTXE13UTlGeDTvtRU4,14062
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py,sha256=k9pDp9UIaOEEWz6bTNWF_KMfcNCOp-F67N-IZ9MO2Rs,8304
-lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=tWvfzea7iTbmDQTPcBJO-buBsuVsc_ya-P6UyPHDvwE,85234
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=vwPUOwJ9429d1D7_q7dRgOfO4WVb_D-VZl7pa8M-V74,76757
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py,sha256=0fnZpdhxNRXGNyILHtfm3rVztfHpSF4kXGc5kTt4zls,13960
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py,sha256=3WpZ-lf7xPwuYyQDMdP7Uc6-dh3vf5TaaUlcMfesfMk,5208
 lightning_sdk/lightning_cloud/openapi/models/v1_user_slurm_job_action_response.py,sha256=BdNzXH8Vsf5PHjl9Rd-TVkpAgx1YC9rf8LD0js-ba20,3058
@@ -1043,28 +1042,28 @@ lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgyd
 lightning_sdk/lightning_cloud/utils/name_generator.py,sha256=MkciuA10332V0mcE2PxLIiwWomWE0Fm_gNGK01vwRr4,58046
 lightning_sdk/lightning_cloud/utils/network.py,sha256=axPgl8rhyPcPjxiztDxyksfxax3VNg2OXL5F5Uc81b4,406
 lightning_sdk/llm/__init__.py,sha256=ErZva0HqN2iPtK_6hI6GN7A_HPGNrHo3wYh7vyFdO3Q,57
-lightning_sdk/llm/llm.py,sha256=kS6NSDywqbhkkluxZYOaKSgBjxGm6xvwtwPUp0m1SsQ,10667
+lightning_sdk/llm/llm.py,sha256=cb3oXqX2-eEq3WK1pUFWbSD8GYGNvHCQYwJyak58vfo,12410
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
 lightning_sdk/mmt/base.py,sha256=B3HC-c82bPHprEZh1mhLCPCrCE8BOKqwIhY7xCF9CXg,15152
 lightning_sdk/mmt/mmt.py,sha256=swdGP1DOM42a_QmmY1vg3--6ZBDiC4zToAzU9Eycv4U,13446
 lightning_sdk/mmt/v1.py,sha256=TxLtL0ssDoP8eyleDaFyYr4evkOKbLJcckLVIfalOno,8429
 lightning_sdk/mmt/v2.py,sha256=Em1XBoqViqUMKm-sshzdMcSH5UTtZZwbJcsgqY6-mw0,9625
 lightning_sdk/pipeline/__init__.py,sha256=uZS2v8HPGfNz2SrCCVrZ9EhihS3kd6aGtwJLbuaI3dA,167
-lightning_sdk/pipeline/pipeline.py,sha256=CCRcHUjL24yKm24NMB5xzK1sre6bQ5sY07tMifmpqI0,3762
+lightning_sdk/pipeline/pipeline.py,sha256=-zNPzr8r32R4ciAL2bOS8vGXElZDe5lOYhdkZuHRswY,3762
 lightning_sdk/pipeline/types.py,sha256=HvBu4rtQfKFewXy3_MxE9OjCmVBlrGZzvrZJKqMZJFg,10578
 lightning_sdk/pipeline/utils.py,sha256=qOxvrYYw8XMd6aknGi5_FFxLvcPwNUuGaHoanx7TZXs,2655
 lightning_sdk/services/__init__.py,sha256=wi1yv0SCnfJub5tOq8y9SblK3-CEseHJuvH-HmtAy7U,229
 lightning_sdk/services/file_endpoint.py,sha256=we5HC_o74J4Y6fSP_31jIizi_I_1FO_Rb2qblspD9eE,7855
 lightning_sdk/services/finetune_llm.py,sha256=5ewjo_bsk5hwvhGoLRdHBv1nP6LeoXDm5_0DOfzqTXQ,1091
 lightning_sdk/services/license.py,sha256=BMDCOjv81l55ZHl4V56EyiDVFBY2EYHaVBun9M_kAtM,16758
-lightning_sdk/services/utilities.py,sha256=zx4I1IWxHVIbmGMDZ9gAD4h-hXeNyj5xqyTTnSssSyg,4123
+lightning_sdk/services/utilities.py,sha256=WlzKDGqUrQFcT85lnLgHvexf7AB1ppxS4_RBkS9Dbg8,4686
 lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/resolve.py,sha256=6qlBUkOmcFTjhQx_CAGfnvWBbMYp6XrCV_sX_IqplLE,6748
-lightning_sdk-0.2.20.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.2.20.dist-info/METADATA,sha256=hn-_APjoxukmnfCogZQs55XTWF_n2tJj5b4TRxjeFc4,3992
-lightning_sdk-0.2.20.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-0.2.20.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.2.20.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.2.20.dist-info/RECORD,,
+lightning_sdk-0.2.21rc1.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.2.21rc1.dist-info/METADATA,sha256=24lp7VIkBYRgwZR9Gy999K2r-IdRZrd-AjN0ip87PUY,3995
+lightning_sdk-0.2.21rc1.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-0.2.21rc1.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.2.21rc1.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.2.21rc1.dist-info/RECORD,,

lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py

@@ -1,201 +0,0 @@
-# coding: utf-8
-
-"""
-    external/v1/auth_service.proto
-
-    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
-
-    OpenAPI spec version: version not set
-    
-    Generated by: https://github.com/swagger-api/swagger-codegen.git
-
-    NOTE
-    ----
-    standard swagger-codegen-cli for this python client has been modified
-    by custom templates. The purpose of these templates is to include
-    typing information in the API and Model code. Please refer to the
-    main grid repository for more info
-"""
-
-import pprint
-import re  # noqa: F401
-
-from typing import TYPE_CHECKING
-
-import six
-
-if TYPE_CHECKING:
-    from datetime import datetime
-    from lightning_sdk.lightning_cloud.openapi.models import *
-
-class V1GetUserStorageResponse(object):
-    """NOTE: This class is auto generated by the swagger code generator program.
-
-    Do not edit the class manually.
-    """
-    """
-    Attributes:
-      swagger_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
-    """
-    swagger_types = {
-        'acked_storage_violation': 'bool',
-        'in_storage_violation': 'bool',
-        'indexation_in_progress': 'bool',
-        'storage_bytes': 'str'
-    }
-
-    attribute_map = {
-        'acked_storage_violation': 'ackedStorageViolation',
-        'in_storage_violation': 'inStorageViolation',
-        'indexation_in_progress': 'indexationInProgress',
-        'storage_bytes': 'storageBytes'
-    }
-
-    def __init__(self, acked_storage_violation: 'bool' =None, in_storage_violation: 'bool' =None, indexation_in_progress: 'bool' =None, storage_bytes: 'str' =None):  # noqa: E501
-        """V1GetUserStorageResponse - a model defined in Swagger"""  # noqa: E501
-        self._acked_storage_violation = None
-        self._in_storage_violation = None
-        self._indexation_in_progress = None
-        self._storage_bytes = None
-        self.discriminator = None
-        if acked_storage_violation is not None:
-            self.acked_storage_violation = acked_storage_violation
-        if in_storage_violation is not None:
-            self.in_storage_violation = in_storage_violation
-        if indexation_in_progress is not None:
-            self.indexation_in_progress = indexation_in_progress
-        if storage_bytes is not None:
-            self.storage_bytes = storage_bytes
-
-    @property
-    def acked_storage_violation(self) -> 'bool':
-        """Gets the acked_storage_violation of this V1GetUserStorageResponse.  # noqa: E501
-
-
-        :return: The acked_storage_violation of this V1GetUserStorageResponse.  # noqa: E501
-        :rtype: bool
-        """
-        return self._acked_storage_violation
-
-    @acked_storage_violation.setter
-    def acked_storage_violation(self, acked_storage_violation: 'bool'):
-        """Sets the acked_storage_violation of this V1GetUserStorageResponse.
-
-
-        :param acked_storage_violation: The acked_storage_violation of this V1GetUserStorageResponse.  # noqa: E501
-        :type: bool
-        """
-
-        self._acked_storage_violation = acked_storage_violation
-
-    @property
-    def in_storage_violation(self) -> 'bool':
-        """Gets the in_storage_violation of this V1GetUserStorageResponse.  # noqa: E501
-
-
-        :return: The in_storage_violation of this V1GetUserStorageResponse.  # noqa: E501
-        :rtype: bool
-        """
-        return self._in_storage_violation
-
-    @in_storage_violation.setter
-    def in_storage_violation(self, in_storage_violation: 'bool'):
-        """Sets the in_storage_violation of this V1GetUserStorageResponse.
-
-
-        :param in_storage_violation: The in_storage_violation of this V1GetUserStorageResponse.  # noqa: E501
-        :type: bool
-        """
-
-        self._in_storage_violation = in_storage_violation
-
-    @property
-    def indexation_in_progress(self) -> 'bool':
-        """Gets the indexation_in_progress of this V1GetUserStorageResponse.  # noqa: E501
-
-
-        :return: The indexation_in_progress of this V1GetUserStorageResponse.  # noqa: E501
-        :rtype: bool
-        """
-        return self._indexation_in_progress
-
-    @indexation_in_progress.setter
-    def indexation_in_progress(self, indexation_in_progress: 'bool'):
-        """Sets the indexation_in_progress of this V1GetUserStorageResponse.
-
-
-        :param indexation_in_progress: The indexation_in_progress of this V1GetUserStorageResponse.  # noqa: E501
-        :type: bool
-        """
-
-        self._indexation_in_progress = indexation_in_progress
-
-    @property
-    def storage_bytes(self) -> 'str':
-        """Gets the storage_bytes of this V1GetUserStorageResponse.  # noqa: E501
-
-
-        :return: The storage_bytes of this V1GetUserStorageResponse.  # noqa: E501
-        :rtype: str
-        """
-        return self._storage_bytes
-
-    @storage_bytes.setter
-    def storage_bytes(self, storage_bytes: 'str'):
-        """Sets the storage_bytes of this V1GetUserStorageResponse.
-
-
-        :param storage_bytes: The storage_bytes of this V1GetUserStorageResponse.  # noqa: E501
-        :type: str
-        """
-
-        self._storage_bytes = storage_bytes
-
-    def to_dict(self) -> dict:
-        """Returns the model properties as a dict"""
-        result = {}
-
-        for attr, _ in six.iteritems(self.swagger_types):
-            value = getattr(self, attr)
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
-                    value
-                ))
-            elif hasattr(value, "to_dict"):
-                result[attr] = value.to_dict()
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], item[1].to_dict())
-                    if hasattr(item[1], "to_dict") else item,
-                    value.items()
-                ))
-            else:
-                result[attr] = value
-        if issubclass(V1GetUserStorageResponse, dict):
-            for key, value in self.items():
-                result[key] = value
-
-        return result
-
-    def to_str(self) -> str:
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self) -> str:
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other: 'V1GetUserStorageResponse') -> bool:
-        """Returns true if both objects are equal"""
-        if not isinstance(other, V1GetUserStorageResponse):
-            return False
-
-        return self.__dict__ == other.__dict__
-
-    def __ne__(self, other: 'V1GetUserStorageResponse') -> bool:
-        """Returns true if both objects are not equal"""
-        return not self == other