lightning-sdk 0.2.21rc0__py3-none-any.whl → 0.2.22__py3-none-any.whl

lightning_sdk/pipeline/utils.py

@@ -1,6 +1,6 @@
 from typing import List
 
-from lightning_sdk.lightning_cloud.openapi.models import V1PipelineStep, V1PipelineStepType
+from lightning_sdk.lightning_cloud.openapi.models import V1PipelineStep
 
 DEFAULT = "DEFAULT"
 
@@ -51,19 +51,4 @@ def prepare_steps(steps: List["V1PipelineStep"]) -> List["V1PipelineStep"]:
                 if name_to_idx[name] >= name_to_idx[current_step.name]:
                     raise ValueError("You can only reference prior steps")
 
-    print()
-    print("===== Generated Pipeline =====")
-    for step_idx, step in enumerate(steps):
-        step_type = ""
-        if step.type == V1PipelineStepType.DEPLOYMENT:
-            step_type = "Deployment"
-        elif step.type == V1PipelineStepType.JOB:
-            step_type = "Job"
-        else:
-            step_type = "MMT"
-        wait_for = "nothing" if len(step.wait_for) == 0 else step.wait_for
-        print(f"{step_idx} - {step_type}['{step.name}'] wait_for {wait_for}")
-    print("===== ================== =====")
-    print()
-
     return steps

{lightning_sdk-0.2.21rc0.dist-info → lightning_sdk-0.2.22.dist-info}/METADATA

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

{lightning_sdk-0.2.21rc0.dist-info → lightning_sdk-0.2.22.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=iYr_Snl9t-FCjzSekdEUcAwh9Thp-bA_WsbMQQ4C25k,1109
+lightning_sdk/__init__.py,sha256=PGFLgjq7LSucuuIAOnJhYd-XPUZmUdUv8EAwbxQOnBE,1105
 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,14 +21,14 @@ 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=7pdff1MwwcRL7SJmDoIrqmJNUewXQYiVLFx8e5uc5eY,2426
+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=GKQrk7Id_E4_Bp4j70ebSEgiB--iVrCd9SEA1D0VG2I,7180
+lightning_sdk/api/llm_api.py,sha256=bsSOzfKhDByFMMQv1S5ZjrA1bT_5J7Y4HOJW1knSO9I,8433
 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/pipeline_api.py,sha256=A_So4S-zSRj37di0AXSqrjR3eH7OmtpIcqugjTotyH8,3248
 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
@@ -37,12 +37,12 @@ lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 lightning_sdk/cli/ai_hub.py,sha256=8MxGQC7SZALAsIj0tZ9HMZCeVHESUcve0cNazRzIFbE,1812
 lightning_sdk/cli/clusters_menu.py,sha256=HX5A7yE-3JijEcF2ANeSUhE9gjq43h-5KDw0aC0op54,1767
 lightning_sdk/cli/coloring.py,sha256=YEb1LiYb6CekVfCRb83-npAmNgd-7c1LzXuNDo4GLSc,2415
-lightning_sdk/cli/configure.py,sha256=B84BtYU0ccKvc1qfK1kqn7sbnHaKW_5Po54k-m7lKHc,4174
-lightning_sdk/cli/connect.py,sha256=qsKb_NMvaA4xrESjMEApM5lyfuHJApe3eCeeI8ttrmQ,975
+lightning_sdk/cli/configure.py,sha256=ZZkCYmnxrRSiZdpbNMLbnxLGy4i4ce7lEcV3NVvYfp8,4457
+lightning_sdk/cli/connect.py,sha256=LjCi_0ZJkTm8IbIzLyIL0SATEQ2bWdVb6aARXnYXcVw,988
 lightning_sdk/cli/create.py,sha256=HetFaK-022qdiI6T9I74EMbsskGa8HL2LiM4G7wgFus,3464
 lightning_sdk/cli/delete.py,sha256=zHMAkLzt6RgF1cmM-XC6mz8TyBgvCul0FeQFK3cIOwY,3812
 lightning_sdk/cli/docker_cli.py,sha256=EozLC4qnvHhgukmnu35itfI5n7v-abCpwKPkPy3eOV4,997
-lightning_sdk/cli/download.py,sha256=fnoAb_0-ONwz1RM0FJPR3yZS5hQW0o7Hw17kSXrmFQA,9505
+lightning_sdk/cli/download.py,sha256=_n-FZKsdmjLX0s7kt6ge5-EAPn1WnPqdDYiFMLn-raE,10655
 lightning_sdk/cli/entrypoint.py,sha256=9oH14WeyDHUDujzT3EKcVXOP_E7QnHi-BWMLoWIOTrw,3443
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
 lightning_sdk/cli/generate.py,sha256=8mjm5tMYIXMoN2X_eEKpwBIUcCWaARiQAmreh-9LIBg,1575
@@ -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,9 +78,9 @@ 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=KTNMCBDB1g61f00A52lS2YlK0BRHNNeDOCoDgZR0Ew4,105676
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=ej5oXbg2_OW5m_m-4LtGRODe9dWNeGiAGjYdo7yt1TI,105875
 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/configuration.py,sha256=H4PTkRz1QAonxKHbx3LG-gDEtdQPAxMO-bVfMfbUfJg,8306
 lightning_sdk/lightning_cloud/openapi/rest.py,sha256=ZPPr6ZkBp6LtuAsiUU7D8Pz8Dt9ECbEM_26Ov74tdpw,13322
 lightning_sdk/lightning_cloud/openapi/api/__init__.py,sha256=u_CPAbeWOto2OKXNiNQhIGBHD6PulgVm7P1JFPuhqdk,4047
 lightning_sdk/lightning_cloud/openapi/api/agent_service_api.py,sha256=Va0QcK6cO3rgHxV16xI-fnq08mgZe_eFQl8mgyaZols,56603
@@ -91,7 +91,7 @@ lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=N1He6aL9
 lightning_sdk/lightning_cloud/openapi/api/blog_posts_service_api.py,sha256=T6ogreuE2ZC3RPzwJ-LLdK3OCxxPlhRQaI9PPI1pFfQ,20832
 lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py,sha256=CNMj3srdarHVdPkvUyq3vVaPfxTx0kmjjzxHAJ_3DzY,28234
 lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=89TZB-WHY5yveVzKj0gr1qOi5tG3IPWwFVk1xsMQi9c,409288
-lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=4RIioSmHNrrNeam5Krcav3ajIrhgd3337IWhJyOxmLs,179295
+lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=iMBpSG03h786s8q0UPjoHtRsSPGvX48es8k3FSuU6c4,179069
 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
@@ -106,7 +106,7 @@ lightning_sdk/lightning_cloud/openapi/api/lightningwork_service_api.py,sha256=dX
 lightning_sdk/lightning_cloud/openapi/api/lit_dataset_service_api.py,sha256=s_1rrbdF9djkUqri_1FJznbQg_B8LIssoQ6T22EcPI4,94370
 lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py,sha256=BkTIX1af3Sn4MTxLLSCcgHQjBnyKuNc9wh1ghsVgXXI,76635
 lightning_sdk/lightning_cloud/openapi/api/lit_page_service_api.py,sha256=xf48UmWpzjs3woKVfFeyyqCLN13NA_qDepIwMw2SRUg,20457
-lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=gR33z3g5hi2OrUfqQCPU8BsAJGJQQH6Jk4ms8rS-Szg,37135
+lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=0RpkeN81WoD2-DMv6-fhFG3O_tQqpPIXxb0tXW5GoNo,43319
 lightning_sdk/lightning_cloud/openapi/api/models_store_api.py,sha256=28TZ5nmyx-X-XYGOz1un0dbH1t2qgm73fqRY7iFYQuI,104663
 lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py,sha256=8xYTFSC4yDs32CTcEoWR1ycXX8chLQWRCsKAXJlOCls,88781
 lightning_sdk/lightning_cloud/openapi/api/pipeline_templates_service_api.py,sha256=-xDf9hCQvcttBH5RVjLIExhd8N9ZgaByoVw_xlBShC0,13758
@@ -115,7 +115,7 @@ lightning_sdk/lightning_cloud/openapi/api/product_license_service_api.py,sha256=
 lightning_sdk/lightning_cloud/openapi/api/profiler_service_api.py,sha256=hM2vTawBBuOYZrxN1SjZ6mU5ldvB_jUXRcLb0as0F4M,28286
 lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py,sha256=E7PM1ZlmuXMLNnIQ2aEc_AUi2ssSOi-eWA9IS2hbRbI,84822
 lightning_sdk/lightning_cloud/openapi/api/quest_service_api.py,sha256=IQppAcHoy8J99xdnZVDfyr-yzmFIkolMN3caF-wm7D0,16318
-lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py,sha256=eJcqgfSe6jQkwwXHtW6BNKbIlBNrq7euZnlvoOr1vQ4,23661
+lightning_sdk/lightning_cloud/openapi/api/schedules_service_api.py,sha256=lRG8s1o0sCZ-SuygA3pCUQfDWLQYJ5VRtkaaqD3w72o,23902
 lightning_sdk/lightning_cloud/openapi/api/secret_service_api.py,sha256=-oTJJ_XbfGAKle82anvK2XKSYE9OxRedCmUfJFjB9O0,41339
 lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py,sha256=Xw6x9q-nUSnid9vSTKnIColQD6U_cbbi2IzIxNVu_1E,45586
 lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py,sha256=C-gEj9pz_Rp3Yo9KnSEsOqNUi2FwzKMsCaPJugpIyLM,29981
@@ -123,7 +123,7 @@ lightning_sdk/lightning_cloud/openapi/api/ssh_public_key_service_api.py,sha256=z
 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=ztdbmsD6MygMfKX8GhwvMCNhI8h3nTifjolaQSEA9qQ,63606
-lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=HnMxWyVyetdddEB93r4HRmPhcJWhfJs3e2A68Vvy0V0,99864
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=giI2gZf92N5WI0DxNvjLPyj7M4u1yb8FiEn8b7qkMto,100063
 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
@@ -250,9 +250,9 @@ lightning_sdk/lightning_cloud/openapi/models/project_id_litregistry_body.py,sha2
 lightning_sdk/lightning_cloud/openapi/models/project_id_memberships_body.py,sha256=h9ROMHHPwk2zScssMD93JIDHkurxhzbz3hyg627H_S4,5946
 lightning_sdk/lightning_cloud/openapi/models/project_id_models_body.py,sha256=u5msnl06gwYwCBOpwJ92EsbyO4WPaAHitLEG88inqlE,6663
 lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py,sha256=98YWVtnWAZfx8ydzWjmhm2qGOIcasXrNOO7LGRxZf0o,7029
-lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py,sha256=Caua1Wj4J6ShKDAzmZSz7ZoRUYY9zLydleYNVOepvvg,7067
+lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py,sha256=hPMZiN9o_loufmw-hCcFfQ57z0S6lis9iNPjSrjTrek,7986
 lightning_sdk/lightning_cloud/openapi/models/project_id_projectclustersbindings_body.py,sha256=TcewvPRwyAZ5puCB06dAdLUGvdAdnJY77XQoejdQp1A,4656
-lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py,sha256=OPXiTVYFq95JCpmvnJHAp5VDLfMXkxivkObbmyHITMQ,6394
+lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py,sha256=5oCaS8j16Trh-IGUSX0CryvqaxHrJtlfEUtw-roodcY,7313
 lightning_sdk/lightning_cloud/openapi/models/project_id_secrets_body.py,sha256=jV7ztO03VzSC55pGFNGFsjRZEyIMOXvTIBcu4zNFXg8,4981
 lightning_sdk/lightning_cloud/openapi/models/project_id_serviceexecution_body.py,sha256=TLV2FM_8gld89PvYc9TpatyEEz7CIyNXy4ixAK9WOvY,5666
 lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py,sha256=GvWXA_I6HvvCszRlKefxVDk29Mbxk1Ka2VbVCVsbgw8,3840
@@ -262,7 +262,7 @@ lightning_sdk/lightning_cloud/openapi/models/projects_project_id_body.py,sha256=
 lightning_sdk/lightning_cloud/openapi/models/protobuf_any.py,sha256=I4a5W5zqZprIP8eTbU-LdYvIj7b2r2B4qPMNIgSCEwk,7143
 lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py,sha256=N33Q2glfHsqcMS4cuFjygVx3hBV1cPt800SOocXErw8,5213
 lightning_sdk/lightning_cloud/openapi/models/rpc_status.py,sha256=H6kd_K1f4TVLpY6-PtPOnolIxU13G4d3xKO04gOOF8o,5865
-lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py,sha256=mSXiBh26shQCJ0grUDkFdpuCgqGX8qP456KPT6W5L0E,10958
+lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py,sha256=8_eoUCDyRwLCbLRBIiA54DA6qgh8Cg_e49rzekNeQ9k,11849
 lightning_sdk/lightning_cloud/openapi/models/secrets_id_body.py,sha256=KumAq6hfKkV6JHYYo66OimHvwssXBPpRfQwCK5qROP0,3591
 lightning_sdk/lightning_cloud/openapi/models/secrets_id_body1.py,sha256=buEAJmT2ZAc3CUYZg1aUds7ltjWPaTuzBeSQXriVF8c,3601
 lightning_sdk/lightning_cloud/openapi/models/server_id_alerts_body.py,sha256=q7eWFFwt5txQZS5DrQeCvXY5jWGlmtj5CmwMPadawV4,5938
@@ -478,6 +478,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_lightningwork_response.py
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_response.py,sha256=y2m4YHWYBBGv3MUKFCj83LzqIvo5GVDjrH6hAgazYCM,3046
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_dataset_version_response.py,sha256=fDdkfWv0fbfYWK93592MfmXw5RTVTMyEjFKEXVkXFXQ,3088
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_page_response.py,sha256=dPBEV8PAH6ar5spg3cCa_hPMlajJcJ9ourbdoCku4XQ,3028
+lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_registry_repository_image_artifact_version_by_digest_response.py,sha256=QZy8K0fpESsKGpVZiPMyYnsfC2aGvCey6XeMYIPjpI8,3280
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_lit_repository_response.py,sha256=iYXqZlUrb-Xljvl4awlPvOmYusGZGdFJDXg56ec30i4,3064
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_logger_artifact_response.py,sha256=bUf_tyaYQjyhQ2Wiq8D0dZWU6yl9knq_uQDJcHQ0ImM,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_managed_endpoint_response.py,sha256=McRms2Ag2FZOUa39fJErUihEQXSia8RYIpPrGm5mZMw,3076
@@ -611,15 +612,15 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_organization_storage_metadat
 lightning_sdk/lightning_cloud/openapi/models/v1_get_project_artifact_response.py,sha256=rEdK3p6nxpkzhTNawWAtJnVx47ZRtQ2EisXb9u2KHoI,3701
 lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py,sha256=cTGzfhKQM32KoGDji3s9HzwPma-uie0n_kEvoDwvxNY,6260
 lightning_sdk/lightning_cloud/openapi/models/v1_get_project_compute_usage_response.py,sha256=vg_GbMoOXHa0AaU1ivnH2e7LKu2HL_EiuZfxw4GEcI0,5356
-lightning_sdk/lightning_cloud/openapi/models/v1_get_project_storage_metadata_response.py,sha256=2yqSf2YCHN_AEDVNcweVEoshxMaTKk71BCiuTAys77E,22811
+lightning_sdk/lightning_cloud/openapi/models/v1_get_project_storage_metadata_response.py,sha256=HtR9cSxDa8sxypqIQ8t0Ps_RZaomJdO4XTezs0Cq4gM,23703
 lightning_sdk/lightning_cloud/openapi/models/v1_get_service_execution_status_response.py,sha256=gjYCDHcnneoLAP-6KwfJo5amQeojVkGd1FBEkcCnRvE,5563
 lightning_sdk/lightning_cloud/openapi/models/v1_get_settings_response.py,sha256=sBvgmxKi2pAUqu62tRdXHtYW2VM-nOjhbIjqD_uqMHo,6623
 lightning_sdk/lightning_cloud/openapi/models/v1_get_snowflake_query_response.py,sha256=aq384_6_5s2v8f3am5db5c1k7ZikXZgpkTaDFFOhVCQ,4471
 lightning_sdk/lightning_cloud/openapi/models/v1_get_usage_details_response.py,sha256=FBuN_OgpDuk0uvF4hre_1RbfOkH8Nefv7FRwudif6mQ,3853
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_balance_response.py,sha256=c38qYUoYTtR5droDQ7oaesHXhFPR0S0lUE26xmO0YP4,6226
 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_response.py,sha256=WGVh-RUIRp9Mzh7j2Xzdoa6X7OQRW6CVc7RroqBYvJk,35271
+lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py,sha256=SoaFtCl7HWHvkSk97sx0f4LjoYteug1lVX9u5Kd6Lvw,14014
 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
@@ -907,7 +908,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_rule_condition.py,sha256=nE-Vbhb
 lightning_sdk/lightning_cloud/openapi/models/v1_rule_effect.py,sha256=WqpSW3YL7UPkIXJDIusIk-PfUn6Vo3Jqv3CNFhJ_C48,3085
 lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py,sha256=htTE306fh6Mjtf7tvpP7tj9MfdUCzLWs1YeC6QlL65E,4397
 lightning_sdk/lightning_cloud/openapi/models/v1_s3_folder_data_connection.py,sha256=VmokLg1mjerYS4aRU0dAtGel3wT-VwGDyMw6E1Ut2rY,3721
-lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py,sha256=cBvz61UD8RYcL0VA0xAwvrMEj2oIAMj0CE2z5z_33hk,11973
+lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py,sha256=Lm82RXNF5QMiSiS1zqu3p7b7zU1yyApKel9_JLiUezE,12844
 lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py,sha256=ZT6X6xqo4oSMD5tGTbZg5jy4JhoJo0bp1sSWGzuMAsI,3121
 lightning_sdk/lightning_cloud/openapi/models/v1_search_job_logs_response.py,sha256=SIoZOP_UDvybzADuo8rCRcAtie7-GIbw-TyKJ00XKWQ,4591
 lightning_sdk/lightning_cloud/openapi/models/v1_search_user.py,sha256=L2omKao6RHEWdKmLyTWHkJJB-PrichTJ8SqSJE68MAM,11623
@@ -982,7 +983,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_update_model_visibility_response
 lightning_sdk/lightning_cloud/openapi/models/v1_update_project_cluster_accelerators_response.py,sha256=9NujcwHJzLFuj66q_AZVdyyZrFiy7_5pwOx89eFdS_U,3142
 lightning_sdk/lightning_cloud/openapi/models/v1_update_shared_metrics_stream_response.py,sha256=mQaPeGjhJ6EohqmF9MFqRR1NKqvznWmvzBAXgMKlW98,3100
 lightning_sdk/lightning_cloud/openapi/models/v1_update_snowflake_query_response.py,sha256=NzTD_--qCS5GsJTcnYO2UNRlaQfJyJxB2BApVNGHoEk,3070
-lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py,sha256=6j9J2k3LNEDRbynDDi-zvQXH3gLyLrwJNe05_M7Qpvs,24520
+lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py,sha256=CT2VelDYHB3-b5O-5yOcq_JJOk_p2O7nlpanlb7Cuqo,27650
 lightning_sdk/lightning_cloud/openapi/models/v1_update_user_viewed_new_features_request.py,sha256=YufHsBeL0FyVYprYKDxxanjQWq8fMyGAsu_Zrml7TCo,3976
 lightning_sdk/lightning_cloud/openapi/models/v1_update_user_viewed_new_features_response.py,sha256=ozJy1abKfgovkFbWqvbqxUgbfWDNd6IP4f1fY_b-pXg,3112
 lightning_sdk/lightning_cloud/openapi/models/v1_upload_agent_job_artifact_response.py,sha256=tDVK4Fx1ukJLXj_x8mBKVWvSZsFQyOfaA8_ANPVzmxU,3741
@@ -1001,7 +1002,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=vwPUOwJ9429d1D7_q7dRgOfO4WVb_D-VZl7pa8M-V74,76757
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=O00Ygvc-Ogk5mNd3jKNyLQ0Ye8W4cL8fyjOR5Dvb1Ps,73244
 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
@@ -1041,18 +1042,19 @@ lightning_sdk/lightning_cloud/utils/data_connection.py,sha256=esKg01F9vdb05fLUf0
 lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgydm78kJzhOIx3C9qKls,2011
 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=nTdp5J7hoCsfqKla18Ta9ixAVfl_XAFyXKMuTkWgRmY,117
-lightning_sdk/llm/asyncllm.py,sha256=ERNm5PzdzuzwDWfkpd77B-C5TBpMqjedt9XPIoT2puI,1946
-lightning_sdk/llm/llm.py,sha256=kS6NSDywqbhkkluxZYOaKSgBjxGm6xvwtwPUp0m1SsQ,10667
+lightning_sdk/llm/__init__.py,sha256=ErZva0HqN2iPtK_6hI6GN7A_HPGNrHo3wYh7vyFdO3Q,57
+lightning_sdk/llm/llm.py,sha256=HYh7oEN1roti9X5Kv6q7Ov6FsgF_faHZuEt9zvCGiPc,12605
 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/types.py,sha256=HvBu4rtQfKFewXy3_MxE9OjCmVBlrGZzvrZJKqMZJFg,10578
-lightning_sdk/pipeline/utils.py,sha256=qOxvrYYw8XMd6aknGi5_FFxLvcPwNUuGaHoanx7TZXs,2655
+lightning_sdk/pipeline/__init__.py,sha256=48eW-rRoaZxrYSkhOBYqs2n4ytHctIkcn95d1SB549M,232
+lightning_sdk/pipeline/pipeline.py,sha256=x7m4RXDc-TFlGUHrHt2O2bPyMUMTDH7q9_3b3_MaSxQ,4372
+lightning_sdk/pipeline/printer.py,sha256=MVM-jHMWrLpnpUB7WMV064IDBE7WBg2pBOnjFsJbmPc,3750
+lightning_sdk/pipeline/schedule.py,sha256=0NYWzTRD_WXAVpwF3g5cCRAhn4bBYQVbidncmeKxoaM,102
+lightning_sdk/pipeline/types.py,sha256=PiWnlelTczbNkW8BwwAuiAxBNxcrVr0K-ukpCnDZGjg,10604
+lightning_sdk/pipeline/utils.py,sha256=roh7H66is6Gfcph3XWdc35XNbHWcFt_E-Bwg2H4m6V0,2086
 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
@@ -1062,9 +1064,9 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 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.21rc0.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.2.21rc0.dist-info/METADATA,sha256=r0pcAiTb5R12dc_B9wX9XtzUoOSmDhUSCLaGRhS6apY,3995
-lightning_sdk-0.2.21rc0.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-0.2.21rc0.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.2.21rc0.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.2.21rc0.dist-info/RECORD,,
+lightning_sdk-0.2.22.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.2.22.dist-info/METADATA,sha256=syK3luB8XU0fRuY9eVDzRLLVMCqn8VSyCZIXXHDmK7E,3992
+lightning_sdk-0.2.22.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-0.2.22.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.2.22.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.2.22.dist-info/RECORD,,

lightning_sdk/llm/asyncllm.py

@@ -1,48 +0,0 @@
-import os
-from typing import AsyncGenerator, Dict, List, Optional, Union
-
-from lightning_sdk.llm.llm import LLM
-
-
-class AsyncLLM(LLM):
-    async def chat(
-        self,
-        prompt: str,
-        system_prompt: Optional[str] = None,
-        max_completion_tokens: Optional[int] = 500,
-        images: Optional[Union[List[str], str]] = None,
-        conversation: Optional[str] = None,
-        metadata: Optional[Dict[str, str]] = None,
-        stream: bool = False,
-        upload_local_images: bool = False,
-    ) -> Union[str, AsyncGenerator[str, None]]:
-        if conversation and conversation not in self._conversations:
-            self._get_conversations()
-
-        if images:
-            if isinstance(images, str):
-                images = [images]
-            for image in images:
-                if not isinstance(image, str):
-                    raise NotImplementedError(f"Image type {type(image)} are not supported yet.")
-                if not image.startswith("http") and upload_local_images:
-                    self._teamspace.upload_file(file_path=image, remote_path=f"images/{os.path.basename(image)}")
-
-        conversation_id = self._conversations.get(conversation) if conversation else None
-        output = await self._llm_api.async_start_conversation(
-            prompt=prompt,
-            system_prompt=system_prompt,
-            max_completion_tokens=max_completion_tokens,
-            images=images,
-            assistant_id=self._model.id,
-            conversation_id=conversation_id,
-            billing_project_id=self._teamspace.id,
-            metadata=metadata,
-            name=conversation,
-            stream=stream,
-        )
-        if not stream:
-            if conversation and not conversation_id:
-                self._conversations[conversation] = output.conversation_id
-            return output.choices[0].delta.content
-        return self._stream_chat_response(output, conversation=conversation)