lightning-sdk 2025.7.17__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 (68) 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/mmt_api.py +9 -0
  6. lightning_sdk/api/pipeline_api.py +4 -3
  7. lightning_sdk/api/studio_api.py +19 -5
  8. lightning_sdk/cli/clusters_menu.py +3 -3
  9. lightning_sdk/cli/create.py +22 -10
  10. lightning_sdk/cli/deploy/_auth.py +19 -3
  11. lightning_sdk/cli/deploy/serve.py +18 -4
  12. lightning_sdk/cli/entrypoint.py +1 -1
  13. lightning_sdk/cli/start.py +37 -7
  14. lightning_sdk/deployment/deployment.py +8 -0
  15. lightning_sdk/job/base.py +27 -3
  16. lightning_sdk/job/job.py +28 -4
  17. lightning_sdk/job/v1.py +10 -1
  18. lightning_sdk/job/v2.py +15 -1
  19. lightning_sdk/lightning_cloud/openapi/__init__.py +9 -1
  20. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +335 -0
  21. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +153 -48
  22. lightning_sdk/lightning_cloud/openapi/models/__init__.py +9 -1
  23. lightning_sdk/lightning_cloud/openapi/models/blogposts_id_body.py +53 -1
  24. lightning_sdk/lightning_cloud/openapi/models/conversations_id_body1.py +123 -0
  25. lightning_sdk/lightning_cloud/openapi/models/messages_id_body.py +123 -0
  26. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/user_id_upgradetrigger_body.py +175 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_blog_post.py +53 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_create_billing_upgrade_trigger_record_response.py +97 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_create_blog_post_request.py +53 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py +27 -1
  34. lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_request.py +29 -3
  35. lightning_sdk/lightning_cloud/openapi/models/{v1_get_clickhouse_assistant_session_daily_aggregated_response.py → v1_get_assistant_session_daily_aggregated_response.py} +22 -22
  36. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +79 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_like_status.py +104 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_managed_endpoint_models_response.py +123 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +27 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_quote_subscription_response.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_like_response.py +149 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_message_like_response.py +149 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +1 -261
  45. lightning_sdk/llm/llm.py +29 -5
  46. lightning_sdk/machine.py +12 -0
  47. lightning_sdk/mmt/base.py +20 -2
  48. lightning_sdk/mmt/mmt.py +25 -3
  49. lightning_sdk/mmt/v1.py +7 -1
  50. lightning_sdk/mmt/v2.py +21 -2
  51. lightning_sdk/organization.py +4 -0
  52. lightning_sdk/pipeline/pipeline.py +16 -5
  53. lightning_sdk/pipeline/printer.py +5 -3
  54. lightning_sdk/pipeline/schedule.py +844 -1
  55. lightning_sdk/pipeline/steps.py +19 -4
  56. lightning_sdk/sandbox.py +4 -1
  57. lightning_sdk/serve.py +2 -0
  58. lightning_sdk/studio.py +79 -39
  59. lightning_sdk/teamspace.py +14 -8
  60. lightning_sdk/utils/resolve.py +29 -2
  61. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.22.dist-info}/METADATA +1 -1
  62. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.22.dist-info}/RECORD +67 -59
  63. lightning_sdk/api/cluster_api.py +0 -119
  64. /lightning_sdk/cli/{inspect.py → inspection.py} +0 -0
  65. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.22.dist-info}/LICENSE +0 -0
  66. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.22.dist-info}/WHEEL +0 -0
  67. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.22.dist-info}/entry_points.txt +0 -0
  68. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.22.dist-info}/top_level.txt +0 -0
@@ -1,76 +1,76 @@
1
1
  docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
2
- lightning_sdk/__init__.py,sha256=dt3P4n0TcWPp5qAuBHgSUYGD5wo8APYdybkIsMinXyc,1109
2
+ lightning_sdk/__init__.py,sha256=43ap-3VeIcI6zAoOtLA9X-JNOHqQCOgDX1u6C-q4fFc,1145
3
3
  lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
4
4
  lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
5
5
  lightning_sdk/base_studio.py,sha256=_Pwwl37R9GRd7t-f2kO5aQXiLNrP4sUtUNht2ZkP8LE,3678
6
6
  lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
7
7
  lightning_sdk/helpers.py,sha256=KWMWnORHItIIA3PGn71YPs-7RjzGi8IXa2kQ5Qo4U8M,2459
8
8
  lightning_sdk/lit_container.py,sha256=8ys49TXI9MO89jLTA7MwDrKrssTRARAIF9OVmolDLq0,5273
9
- lightning_sdk/machine.py,sha256=9BOcDXHzmmkLPW5r4l9l8ELFrIoMIT0WLwieQweY1kQ,4106
9
+ lightning_sdk/machine.py,sha256=k1UTIPfo9JG8pw6RJH3aJ4w3Zk2JpuqILPMkXwpFG-4,4341
10
10
  lightning_sdk/models.py,sha256=tPFiiDQiDsFhfStcVPH5_PcujbBEfKS2IB5FS9NfwZk,7838
11
- lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
11
+ lightning_sdk/organization.py,sha256=jizjG4oW_oxkP2Qz4vL5cle4Cg12xB_vf4eT-bnvLP0,1352
12
12
  lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
13
13
  lightning_sdk/plugin.py,sha256=f3P5-xZY6x-MX0Fs2z_Q2erSxPSiHZARO0BVkCezHw4,15192
14
- lightning_sdk/sandbox.py,sha256=mI5ygR1L9TKpS_sbS87rWl2aqGmTTt7LGfBbGbApuq4,5544
15
- lightning_sdk/serve.py,sha256=yjKXmbmR47nC7OB8y5dI_dRqLmeLFYnHukVoihQnTcA,11686
14
+ lightning_sdk/sandbox.py,sha256=_NvnWotEXW2rBiVFZZ4krKXxVjuAqfNh04qELSM0-Pg,5786
15
+ lightning_sdk/serve.py,sha256=uW7zLhQ3X90ifetpxzTb8FNxifv5vIs7qZlgfEjVKzk,11794
16
16
  lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
17
- lightning_sdk/studio.py,sha256=dMBBRohRtzEls1CeNvwNp7n26ygJw7gw58sHvrXHng8,24643
18
- lightning_sdk/teamspace.py,sha256=mZA7ryQl53lEh02FXWzQf3srwZ3RysPhVKgQJCiBR3I,19563
17
+ lightning_sdk/studio.py,sha256=IolSuWup1jrCcvmA4nSWypFzTAk9UTrYxpmzUibuh-0,26383
18
+ lightning_sdk/teamspace.py,sha256=W-LepkoIjsaZXZyNoZVkNGWpDnGb8UNuICntgByRbIg,19880
19
19
  lightning_sdk/user.py,sha256=vdn8pZqkAZO0-LoRsBdg0TckRKtd_H3QF4gpiZcl4iY,1130
20
20
  lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
21
21
  lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
22
22
  lightning_sdk/api/ai_hub_api.py,sha256=azqDZ-PzasVAcoQHno7k7OO_xFOHQ4NDozxF8jEh83Y,7864
23
23
  lightning_sdk/api/base_studio_api.py,sha256=3R8tucZX2e9yKHBcY2rWFRP4dxqLrC6H75vdBDkH0ck,3617
24
- lightning_sdk/api/cluster_api.py,sha256=kAZeJbcNYG9rsYE7if5k1HlcND1Z_6cKivZ7lY4S20U,4392
25
- lightning_sdk/api/deployment_api.py,sha256=KvBwGaG896q7WS1OPMOIjFUh05W2g7S1djS2XJMVieQ,24215
26
- lightning_sdk/api/job_api.py,sha256=_mMAI_BG_48i-BLwCP_U72zgmM5zYa2KUZ7u66HWkIc,13568
24
+ lightning_sdk/api/cloud_account_api.py,sha256=tWwlBKJf9zoSUJWMmHqTJ2inkLXCGbd1W7CImRhY9pk,6133
25
+ lightning_sdk/api/deployment_api.py,sha256=ys9HspgKMTehF_h8IIxXO51UNXRjqgPPVctSp0m1iAE,24748
26
+ lightning_sdk/api/job_api.py,sha256=NJWc-lSqDQ-uhW5sLQ286dmz8aClvipZCtSsgBH6jIo,13993
27
27
  lightning_sdk/api/license_api.py,sha256=XV3RhefyPQDYjwY9AaBZe4rByZTEAnsvLDxcdm9q0Wo,2438
28
28
  lightning_sdk/api/lit_container_api.py,sha256=jCJVwd-3MNjejL9FyvH89pzt-SeG3G8qCJD16iTMJAQ,11454
29
29
  lightning_sdk/api/llm_api.py,sha256=mDGQJRQzDECYJ5c5BCU5qQETf5qvFCQJ_H2bc-BX6U4,9272
30
- lightning_sdk/api/mmt_api.py,sha256=-v7ATab-ThAM-HRClS92Ehxuu9MlBfdKWWFCGvVUHiM,8962
30
+ lightning_sdk/api/mmt_api.py,sha256=Z7IHNB4WDjGl9cSsYdq2OM2DepE6elSPlyaysULquiA,9380
31
31
  lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
32
- lightning_sdk/api/pipeline_api.py,sha256=oS0WIBB748SLD7bqkEQ8rnkixrsAML_M01OUunmLFSQ,4470
33
- lightning_sdk/api/studio_api.py,sha256=J-kp6IoXlQaVz1S4BtaK93VrLAbMOolQYDGJyrkPDRo,30702
32
+ lightning_sdk/api/pipeline_api.py,sha256=t8QiFxExXJB-LzsFmDTJB-IXjiNw_F77B33iYKpiGzk,4554
33
+ lightning_sdk/api/studio_api.py,sha256=Pev7vI3kr3QT_ETnfomDSpgJfvbZEXfHGQnb2UZzshY,31208
34
34
  lightning_sdk/api/teamspace_api.py,sha256=CsaaxmaLmTRIRwu37wtQ3quGYql62HJT3BZJ3Q-1d9c,16854
35
35
  lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
36
36
  lightning_sdk/api/utils.py,sha256=K0Uc6D1KxxjFElqjsuCXg85SYreQi3P7jJ1VzP0MpEQ,23966
37
37
  lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
38
38
  lightning_sdk/cli/ai_hub.py,sha256=8MxGQC7SZALAsIj0tZ9HMZCeVHESUcve0cNazRzIFbE,1812
39
- lightning_sdk/cli/clusters_menu.py,sha256=HX5A7yE-3JijEcF2ANeSUhE9gjq43h-5KDw0aC0op54,1767
39
+ lightning_sdk/cli/clusters_menu.py,sha256=AVlaWggNYU_-RZu9egagHB1JElDArx8WZMVH1DKPm2A,1801
40
40
  lightning_sdk/cli/coloring.py,sha256=YEb1LiYb6CekVfCRb83-npAmNgd-7c1LzXuNDo4GLSc,2415
41
41
  lightning_sdk/cli/configure.py,sha256=ZZkCYmnxrRSiZdpbNMLbnxLGy4i4ce7lEcV3NVvYfp8,4457
42
42
  lightning_sdk/cli/connect.py,sha256=LjCi_0ZJkTm8IbIzLyIL0SATEQ2bWdVb6aARXnYXcVw,988
43
- lightning_sdk/cli/create.py,sha256=HetFaK-022qdiI6T9I74EMbsskGa8HL2LiM4G7wgFus,3464
43
+ lightning_sdk/cli/create.py,sha256=4L7Kgw86ySsjNBx28LOo5dDG0Emn42Bdp9FmTHdZTeU,3996
44
44
  lightning_sdk/cli/delete.py,sha256=zHMAkLzt6RgF1cmM-XC6mz8TyBgvCul0FeQFK3cIOwY,3812
45
45
  lightning_sdk/cli/docker_cli.py,sha256=EozLC4qnvHhgukmnu35itfI5n7v-abCpwKPkPy3eOV4,997
46
46
  lightning_sdk/cli/download.py,sha256=_n-FZKsdmjLX0s7kt6ge5-EAPn1WnPqdDYiFMLn-raE,10655
47
- lightning_sdk/cli/entrypoint.py,sha256=lxyPRSgHvogoxv6F_XtqE4as4eK4YEqWh1MfJMsAadA,3574
47
+ lightning_sdk/cli/entrypoint.py,sha256=Na3u6rxlF3ILG_SYZGduSze5NXqjBB4MdB6DLulOi8Y,3577
48
48
  lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
49
49
  lightning_sdk/cli/generate.py,sha256=8mjm5tMYIXMoN2X_eEKpwBIUcCWaARiQAmreh-9LIBg,1575
50
- lightning_sdk/cli/inspect.py,sha256=GA3XDC3DSkqJ4B5NQR9qmGGvkP0K8Sk90xZhzTGzC1o,1614
50
+ lightning_sdk/cli/inspection.py,sha256=GA3XDC3DSkqJ4B5NQR9qmGGvkP0K8Sk90xZhzTGzC1o,1614
51
51
  lightning_sdk/cli/job_and_mmt_action.py,sha256=OV_i5K3g-NnMUF8Sq6QliQdltnhGVRdtoyZqJVXu4Ss,1659
52
52
  lightning_sdk/cli/jobs_menu.py,sha256=_2qsOBTdIKT5NCjwrBo9-AAn3fIvdN9IcVUVHF_KGQk,2162
53
53
  lightning_sdk/cli/list.py,sha256=bH_rX9J_Ubb9dkF2fpoZf0zMXH9YoEjcm63gk5v0zs4,10919
54
54
  lightning_sdk/cli/mmts_menu.py,sha256=HUXo3ZoZ3fWOCNWTQWoJgUlFXYq5uVm_6uFjAq7BDe8,2219
55
55
  lightning_sdk/cli/open.py,sha256=zdQ3E9Wbi5f5wOoJqZ3yFL45kWysGaza-rCPUinrOnA,2747
56
56
  lightning_sdk/cli/run.py,sha256=8JZiDrKwDhlaTOJd6qq2mCWJRqKm6shCWLzpbmFYIkE,13929
57
- lightning_sdk/cli/start.py,sha256=7PpvnJeg8siJBBad_64mpHKXlpr-8pfDqRMHvKIFBd4,2339
57
+ lightning_sdk/cli/start.py,sha256=BDi58pgY73yzI-ehC9RzxHqwNAF-deq5ftgpLQpSfxY,3016
58
58
  lightning_sdk/cli/stop.py,sha256=5nCrUe1BONpX1nKNhbSFqLaXXKaRhSO7PvM1BVYLgn4,2864
59
59
  lightning_sdk/cli/studios_menu.py,sha256=TA9rO6_fFHGMz0Nt4rJ6iV80X5pZE4xShrSiyXoU-oQ,4129
60
60
  lightning_sdk/cli/switch.py,sha256=qLvDoQRldCNgO1XvIGg-i-GyqnkHVb_U1-X4svqeNNU,1879
61
61
  lightning_sdk/cli/teamspace_menu.py,sha256=kjHLZwb66vt3MoIWOOn8rutZ8o4_uGd0j5fmu56cwuE,4097
62
62
  lightning_sdk/cli/upload.py,sha256=b_GbSJB9UP7E4w3QE0laNh6urbcAs6wJtHJJ1iVqVcM,13769
63
63
  lightning_sdk/cli/deploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
64
- lightning_sdk/cli/deploy/_auth.py,sha256=F3W2eU33LIsfFu7AQyBRN01BL800vJJFJCHwg0Bfp_U,7054
64
+ lightning_sdk/cli/deploy/_auth.py,sha256=ccIzl0W9hJtFmhdEAn1xcBGIHQfO4d_k8grTbq4ufIs,7711
65
65
  lightning_sdk/cli/deploy/devbox.py,sha256=HOQMWehwaveawVTyBDaZhNZs_NL5g4RCWs_lf0hIW5U,6800
66
- lightning_sdk/cli/deploy/serve.py,sha256=qqml7oYf4IPcoiL13P2ly247_PQ8IGv-8hM2NBWePJc,14458
66
+ lightning_sdk/cli/deploy/serve.py,sha256=GG2TtwB0Ec_WFdA3QkVo5L4kNQW-4ZOON02t3ftyYjw,15006
67
67
  lightning_sdk/deployment/__init__.py,sha256=dXsa4psDzFYFklsq3JC-2V_L4FQjGZnQAf-ZiVlqG9c,545
68
- lightning_sdk/deployment/deployment.py,sha256=kD3P3uonV7Vr33BVnyD-HkxdUuQs0F2nt0GYoWb3Wx4,21057
68
+ lightning_sdk/deployment/deployment.py,sha256=qFmuFw96kcEp6dubvhDeaDc2n9ylpNCwLL1_ImtQ3O0,21526
69
69
  lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
70
- lightning_sdk/job/base.py,sha256=vyJiNCw34mfyH19AJVzGt7Kggw9mVBdoe0gsgCJSgOI,19099
71
- lightning_sdk/job/job.py,sha256=1Xf0ne4wwXpkb_GJgWD8mfueJZoKR8G4lC5T6OXijlw,13075
72
- lightning_sdk/job/v1.py,sha256=zgdZFnlyhzOBlbwdwx6Ob9YKBrzMnJflzLXy4qyb4So,9070
73
- lightning_sdk/job/v2.py,sha256=9O31CwIQjwm6qZSEZtFBWbZyBuCSmyFMsHCRbQvJdBU,10283
70
+ lightning_sdk/job/base.py,sha256=GAE7wvTH9e_F9bU8sogT88EMCP4D0d1deZMeVmBtWTE,20856
71
+ lightning_sdk/job/job.py,sha256=jPlW3u34lJ6tpszm-Q6EHzWrfMIpQfM7-z8UCFFVyaE,14780
72
+ lightning_sdk/job/v1.py,sha256=WISVvPs_j_g1oaNcbEOAua_JpaNO_9_BueJoPgeul6E,9618
73
+ lightning_sdk/job/v2.py,sha256=TYnMdXIgeFxzV3zRVY5TMk1xf-aQgfTPrayuYDj3ykc,11020
74
74
  lightning_sdk/job/work.py,sha256=Xm5byWTcOxYI8zZ3Sz6P8HxBTBAJ3S2KucndXA5KePQ,2332
75
75
  lightning_sdk/lightning_cloud/__init__.py,sha256=o91SMAlwr4Ke5ESe8fHjqXcj31_h7rT-MlFoXA-n2EI,173
76
76
  lightning_sdk/lightning_cloud/__version__.py,sha256=lOfmWHtjmiuSG28TbKQqd2B3nwmSGOlKVFwhaj_cRJk,23
@@ -79,16 +79,16 @@ lightning_sdk/lightning_cloud/login.py,sha256=k4iE5qLUMB8_VeQa1sNz7jPCWLOV5S7j9Q
79
79
  lightning_sdk/lightning_cloud/rest_client.py,sha256=K2J_QXihpbp5tVzs9PzffgMCQY1NeH5tRpuJBYj27vU,7125
80
80
  lightning_sdk/lightning_cloud/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
81
81
  lightning_sdk/lightning_cloud/cli/__main__.py,sha256=aHrigVV0qIp74MuYTdN6uO8a9Xp6E1aOwdR8w1fG-7c,688
82
- lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=UehxrDY3thJNaQG43c6C98H-aBQ6vTWYAIgTEonoWcs,109379
82
+ lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=v_h4Px6untfoIf07V_n1uwQvZIx6lHhoI6VsMUYgvKY,110316
83
83
  lightning_sdk/lightning_cloud/openapi/api_client.py,sha256=pUTQMNcZmH4BhpnuAXuT7wnegaxaX26bzdEWjdoLeTo,25630
84
84
  lightning_sdk/lightning_cloud/openapi/configuration.py,sha256=8KpFKEm-IKe3t0_o5H1Y0EJX06r94Gk5cIQerA0TNk0,8292
85
85
  lightning_sdk/lightning_cloud/openapi/rest.py,sha256=ZPPr6ZkBp6LtuAsiUU7D8Pz8Dt9ECbEM_26Ov74tdpw,13322
86
86
  lightning_sdk/lightning_cloud/openapi/api/__init__.py,sha256=tkHeWI-jhPo7VdUf5SbQhMG97r3AxWlD_1xIxSOoHpM,4227
87
87
  lightning_sdk/lightning_cloud/openapi/api/agent_service_api.py,sha256=Va0QcK6cO3rgHxV16xI-fnq08mgZe_eFQl8mgyaZols,56603
88
88
  lightning_sdk/lightning_cloud/openapi/api/analytics_service_api.py,sha256=0BccCXj-fpDmrWImCQzPlHz7T8Hrw-vuwPCY-ieWt8Q,5263
89
- lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py,sha256=P27zNbfKiEGeowfZKPOp8ShGqQO3kXyUKFnmcwZemIY,104422
89
+ lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py,sha256=Cy4q_gWhtISP0kCMWUdNBy0d7sy0o7Lzz8kbzrunjWE,120412
90
90
  lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py,sha256=NtdfsFZtQAltWyj1wYg3HNk0cxT2HWbNhAqQSLvMDgM,30948
91
- lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=UYYGBby1dI0PsXPoiScWTPZDR9s0d3lGcTx-0qbcfa8,73431
91
+ lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=MCVMWkeAn9rAF6XWE-MVouqmxrJOzvjeIaUUGUG5Um4,78117
92
92
  lightning_sdk/lightning_cloud/openapi/api/blog_posts_service_api.py,sha256=T6ogreuE2ZC3RPzwJ-LLdK3OCxxPlhRQaI9PPI1pFfQ,20832
93
93
  lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py,sha256=CNMj3srdarHVdPkvUyq3vVaPfxTx0kmjjzxHAJ_3DzY,28234
94
94
  lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=FWRF4vKthJDCG0I-fHCGnw2fvB70PFhHsU1TuArUcMw,418722
@@ -126,7 +126,7 @@ lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py,sha256=kFyE4kt5
126
126
  lightning_sdk/lightning_cloud/openapi/api/studio_jobs_service_api.py,sha256=VJ7VcSvdmGbPkgmQZhaOSR47olcNPMzRlmT5J5uKsng,27877
127
127
  lightning_sdk/lightning_cloud/openapi/api/user_service_api.py,sha256=bUCDZf1P16juI8-J_dpE-MvRt3x8x5HBJpluxix6Vzc,68363
128
128
  lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py,sha256=pg1aaSuYJUAlUDKLN07l8wC-poU_HFxZ_98QZHty74I,10344
129
- lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=QQn43S2Cb7Y_DN5AjGMRUYVwOOV7TPo2_HZHSaCWliA,103387
129
+ lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=fD7KMJ6W1QP3G6t3YU07R2KVuQxKYG2LhNVIECklhD4,104324
130
130
  lightning_sdk/lightning_cloud/openapi/models/affiliatelinks_id_body.py,sha256=X087AkCafKp2g8E2MR5ghU3pxCNNcoehZJttvuVfdC8,4274
131
131
  lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py,sha256=NzKKkTs2PtMSIJJYxeh3aDBj-djBrI_qATGz7n5Oqdk,9234
132
132
  lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py,sha256=jKg8Q5cJ3iY9DpDHvig6rgezR_qYYPIAr5Cez2pSifs,22374
@@ -142,7 +142,7 @@ lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py,
142
142
  lightning_sdk/lightning_cloud/openapi/models/billing_checkout_body.py,sha256=bMABc4XkRzK-t2RzWg1bd2YZSwZkqYZoUex9iWO1lPI,5379
143
143
  lightning_sdk/lightning_cloud/openapi/models/billing_transfer_body.py,sha256=MaovoJiJDzOXNhCskZAN5UNwnFEFP7ONF83DPRgfhgk,4510
144
144
  lightning_sdk/lightning_cloud/openapi/models/billing_transfer_body1.py,sha256=6zrWBAHmmKHJgqJtxahcM1mDrO3Z55WxaGK0RY10oOw,5275
145
- lightning_sdk/lightning_cloud/openapi/models/blogposts_id_body.py,sha256=h5cyywqY3y-CtyVBkPTusiI-C1GlnmfIj9AhCmgAUS8,7232
145
+ lightning_sdk/lightning_cloud/openapi/models/blogposts_id_body.py,sha256=DOLpgT5rWiX1MCFW_VIYarHjYSfQNIXC5NUUf5ljT6A,8895
146
146
  lightning_sdk/lightning_cloud/openapi/models/captures_id_body.py,sha256=NavbDzRYKsQ-8FYNHvtdb-T7JJZZAF7U282Wc4YbGOc,3621
147
147
  lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versionpublications_body.py,sha256=IbmdHzvly0fAu-59FWl7qj78xKZlT4MzZhsQkJ0aA2I,5659
148
148
  lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versionpublications_body1.py,sha256=SrLAqpWlTUhuHRgY2CLE9CIuSYQJX9xwAda7cNsoS9k,5677
@@ -161,6 +161,7 @@ lightning_sdk/lightning_cloud/openapi/models/clusters_id_body.py,sha256=oeXzYSM3
161
161
  lightning_sdk/lightning_cloud/openapi/models/clusters_id_body1.py,sha256=xcvOg6LUIBF_qxgoZw7A8c7V9aef9wh-My6o5lcCEBw,4264
162
162
  lightning_sdk/lightning_cloud/openapi/models/command_argument_command_argument_type.py,sha256=WRgeyIo7LXbhOV1ZPt5Z0TlBVNChFlXf7jXPv3LZY1U,3171
163
163
  lightning_sdk/lightning_cloud/openapi/models/conversations_id_body.py,sha256=blW66WBRE-qMCkzLCOrx4EqmM1vI42O3MvP9rIlx_tE,3651
164
+ lightning_sdk/lightning_cloud/openapi/models/conversations_id_body1.py,sha256=8HGNnTkyTt-J2WbTIcFM8J1SaxYzTSLjvFxZYVxBXtk,3695
164
165
  lightning_sdk/lightning_cloud/openapi/models/create.py,sha256=p4fNu_fsUPoiUbSIcwUH1rmp0DrsXqKImwxUELsxvvs,15612
165
166
  lightning_sdk/lightning_cloud/openapi/models/create_checkout_session_request_wallet_type.py,sha256=6mUpEXvoZxyrnOvfT9PfF34E3OGvRMzMfLFndUe-jKU,3217
166
167
  lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py,sha256=7nkUADBmLPlO4h-3c999Vnnv1OeJ8fPOI1KVMG2sQu4,19950
@@ -224,6 +225,7 @@ lightning_sdk/lightning_cloud/openapi/models/litloggermetrics_id_body.py,sha256=
224
225
  lightning_sdk/lightning_cloud/openapi/models/litpages_id_body.py,sha256=-0ImUYoTU73XiLZBDGcqGmnSj3ZO7mDltXeXoDtBwfs,9811
225
226
  lightning_sdk/lightning_cloud/openapi/models/litregistry_lit_repo_name_body.py,sha256=KZY9WeqcTUgOyyh6fzNB5cD3BlcQT_NcPHdUsSRIzmc,4019
226
227
  lightning_sdk/lightning_cloud/openapi/models/loggermetrics_id_body.py,sha256=snEf_CsjDcd0kt0cFDPqDHCdAkoAjYCiisrBk4mAZR0,3763
228
+ lightning_sdk/lightning_cloud/openapi/models/messages_id_body.py,sha256=2900Uq1_nyTOYqF4yCTIfI75bP6_hkaOlo8g34h2v-k,3635
227
229
  lightning_sdk/lightning_cloud/openapi/models/metrics_stream_id_loggerartifacts_body.py,sha256=MUMnHTkEmomIKq8f3fp8oP8jqQ1oE2i8K_TCxxtj3Q0,3781
228
230
  lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py,sha256=IY61t9MEfbrwpfeKVfrvB-eZUSIE1bbo4EkhbMcO9Ik,13111
229
231
  lightning_sdk/lightning_cloud/openapi/models/metricsstream_delete_body.py,sha256=9Ftuk903BxyRqdByfAfpGlSyhftspbw8qnRkvY3X9fU,3687
@@ -259,7 +261,7 @@ lightning_sdk/lightning_cloud/openapi/models/project_id_models_body.py,sha256=u5
259
261
  lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py,sha256=98YWVtnWAZfx8ydzWjmhm2qGOIcasXrNOO7LGRxZf0o,7029
260
262
  lightning_sdk/lightning_cloud/openapi/models/project_id_pipelines_body.py,sha256=hPMZiN9o_loufmw-hCcFfQ57z0S6lis9iNPjSrjTrek,7986
261
263
  lightning_sdk/lightning_cloud/openapi/models/project_id_projectclustersbindings_body.py,sha256=TcewvPRwyAZ5puCB06dAdLUGvdAdnJY77XQoejdQp1A,4656
262
- lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py,sha256=5SERCk2J0A06SlgiS4wAiyV5GhTMKmAyvXfRoitXIVE,8195
264
+ lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py,sha256=AdYyOeUDoebY8w4HVnIeljKL6yZ1qkA6NYvKzkQLkPE,8916
263
265
  lightning_sdk/lightning_cloud/openapi/models/project_id_secrets_body.py,sha256=jV7ztO03VzSC55pGFNGFsjRZEyIMOXvTIBcu4zNFXg8,4981
264
266
  lightning_sdk/lightning_cloud/openapi/models/project_id_serviceexecution_body.py,sha256=TLV2FM_8gld89PvYc9TpatyEEz7CIyNXy4ixAK9WOvY,5666
265
267
  lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py,sha256=GvWXA_I6HvvCszRlKefxVDk29Mbxk1Ka2VbVCVsbgw8,3840
@@ -270,7 +272,7 @@ lightning_sdk/lightning_cloud/openapi/models/protobuf_any.py,sha256=I4a5W5zqZprI
270
272
  lightning_sdk/lightning_cloud/openapi/models/protobuf_null_value.py,sha256=LOQFanGx6mzOYVjlDFd2olKTBfWkhF7vfnHUmZ8lCWI,3063
271
273
  lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py,sha256=N33Q2glfHsqcMS4cuFjygVx3hBV1cPt800SOocXErw8,5213
272
274
  lightning_sdk/lightning_cloud/openapi/models/rpc_status.py,sha256=H6kd_K1f4TVLpY6-PtPOnolIxU13G4d3xKO04gOOF8o,5865
273
- lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py,sha256=0IQBwYEnx0im2Gw3zxWgWL7k441Ex3aq-uSar29a5hY,12703
275
+ lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py,sha256=A2eeqiDHgpY5e0gUsp5TvCatznwOuQ4YSZWWqe2iH7w,13396
274
276
  lightning_sdk/lightning_cloud/openapi/models/secrets_id_body.py,sha256=KumAq6hfKkV6JHYYo66OimHvwssXBPpRfQwCK5qROP0,3591
275
277
  lightning_sdk/lightning_cloud/openapi/models/secrets_id_body1.py,sha256=buEAJmT2ZAc3CUYZg1aUds7ltjWPaTuzBeSQXriVF8c,3601
276
278
  lightning_sdk/lightning_cloud/openapi/models/server_id_alerts_body.py,sha256=q7eWFFwt5txQZS5DrQeCvXY5jWGlmtj5CmwMPadawV4,5938
@@ -299,6 +301,7 @@ lightning_sdk/lightning_cloud/openapi/models/usagerestrictions_id_body.py,sha256
299
301
  lightning_sdk/lightning_cloud/openapi/models/user_id_affiliatelinks_body.py,sha256=ZRkTVx2a8RTc0D6I1htp6FFZ72m4oByHFcfOWrwtiWk,4330
300
302
  lightning_sdk/lightning_cloud/openapi/models/user_id_membershiprolebindings_body.py,sha256=1qu7B4kRkyXXeDda4cCL_pm0V2Gm1vxWC65YJduFNSU,3820
301
303
  lightning_sdk/lightning_cloud/openapi/models/user_id_membershiprolebindings_body1.py,sha256=lIXUnz4RcH0DevFEWv5CRRwe6bcREFnPCH-Wja8cLcg,3830
304
+ lightning_sdk/lightning_cloud/openapi/models/user_id_upgradetrigger_body.py,sha256=HzHFQkbokEKIFucLDDdIWTyn4_AmQmVqTfs-F4-O21I,5237
302
305
  lightning_sdk/lightning_cloud/openapi/models/v1_accelerator_quota_info.py,sha256=zxHBo05HL4Pq8xDMC1IDKJvSE8Ak-VLTOXFj0fIE5Yg,5790
303
306
  lightning_sdk/lightning_cloud/openapi/models/v1_ack_user_storage_violation_response.py,sha256=GatfYbZcGir2v750_QkUAH_Ou7W6b21URlQvCfMUJGc,3088
304
307
  lightning_sdk/lightning_cloud/openapi/models/v1_add_job_timing_response.py,sha256=y9hY0fNb7oRsunjnpcqhCyh27qjpiwN99ILIk80I1iQ,3022
@@ -336,7 +339,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_billing_feature.py,sha256=vkQ7QN
336
339
  lightning_sdk/lightning_cloud/openapi/models/v1_billing_period.py,sha256=uTaXGz4Vincg0Z3s4MBRgqq5YTxNLuAI0lx1-v02nTU,3059
337
340
  lightning_sdk/lightning_cloud/openapi/models/v1_billing_subscription.py,sha256=WQ3MwKOTHQU6-xcCYSSYZRotxvUWwvmmcHbTz70HL3A,10290
338
341
  lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py,sha256=UTst9HW2U8iUxK14-7A-MEaI5eTVOAg8AYc_MFOgHMM,3157
339
- lightning_sdk/lightning_cloud/openapi/models/v1_blog_post.py,sha256=1f313iUJOTEJmap1xZ9skwaMUfirl3AT8sIGHo2ZDlo,10490
342
+ lightning_sdk/lightning_cloud/openapi/models/v1_blog_post.py,sha256=6lZcTVNA0rLUZh2JLlBpp-7HnjP7FQaQlsz-TXzmYGI,12113
340
343
  lightning_sdk/lightning_cloud/openapi/models/v1_body.py,sha256=dgBZ3hPV_Q_lfc6dNoNjzBpOJZMklBtM8IiQBcvbtEA,3580
341
344
  lightning_sdk/lightning_cloud/openapi/models/v1_build_spec.py,sha256=hqAjFwjZZeW-Jk3xWknetyoBA0J7xSrcUwl8rlkVUCA,6706
342
345
  lightning_sdk/lightning_cloud/openapi/models/v1_cancel_cloud_space_instance_switch_response.py,sha256=b7511arCBmJy8Lp7oWblcifG2r1xyV44q9nSW7yQzzM,3130
@@ -413,15 +416,16 @@ lightning_sdk/lightning_cloud/openapi/models/v1_completed_part.py,sha256=YEU2PZu
413
416
  lightning_sdk/lightning_cloud/openapi/models/v1_compute_config.py,sha256=YGkx66eiDeq6-CxY-AUTUToW11a9W6Nf-k1ufvd1Drc,8173
414
417
  lightning_sdk/lightning_cloud/openapi/models/v1_contact_assistant_owner_reason.py,sha256=9J2PuNGXlDXMcsmINbocUYGXVKhRlvqDdRwgtAH_qPU,3121
415
418
  lightning_sdk/lightning_cloud/openapi/models/v1_contact_assistant_owner_response.py,sha256=_FXW0hslMncLSPPxDmWEbhePRz-4tFc98KpqMf0M_6s,3076
416
- lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py,sha256=xINDLIVGuMEXSog0ryP3rQW1y_9PmEIMsL7P7LHzb_M,9220
419
+ lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py,sha256=w5Wwo3xrvEMgWyGZQIcgq8kAUMyGvJCbMNv_kY-Shs0,9883
417
420
  lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py,sha256=7JovH85aN8mwC_gUVdCBDyqUCwwNlsEAqkBGKMTmwMo,9149
418
421
  lightning_sdk/lightning_cloud/openapi/models/v1_count_metrics_streams_response.py,sha256=ovqawh81CI7PvAqcElhGYw0-NVwSzwbm4s2PFErdxHA,3751
419
422
  lightning_sdk/lightning_cloud/openapi/models/v1_cpu_system_metrics.py,sha256=Mgybx_EGelKpTVzfW9WXp5HiiVMLPoes8vdVq8We1TU,10120
420
423
  lightning_sdk/lightning_cloud/openapi/models/v1_create_agent_multipart_upload_response.py,sha256=Ov9pBANdLtAbfz0ed_N-UGBeUwZozYKuqbst0F8u2Cw,3900
421
424
  lightning_sdk/lightning_cloud/openapi/models/v1_create_billing_portal_session_request.py,sha256=A4JfCA8Ya1V3cBB3WzJ-ZhQSpPgZsClgB8mTQL7w3xI,5434
422
425
  lightning_sdk/lightning_cloud/openapi/models/v1_create_billing_portal_session_response.py,sha256=5dPJJNUNUt1owibUndL9YzrLD-Z4wSO4O-rtxfNtvw0,3781
423
- lightning_sdk/lightning_cloud/openapi/models/v1_create_blog_post_request.py,sha256=2h9Z1xP1shKCei1XDSgXv4QAqbtWZUXVq_FTUYXiUxs,7472
424
- lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py,sha256=D9Iv5eelLkJ_4lbBJVP4RZr0r-cu2g5RTb3JMUlwXPI,10165
426
+ lightning_sdk/lightning_cloud/openapi/models/v1_create_billing_upgrade_trigger_record_response.py,sha256=muy7Lfuy_WQLy6JJK17x0rnhBiRN629dj3MwkvpNrm8,3148
427
+ lightning_sdk/lightning_cloud/openapi/models/v1_create_blog_post_request.py,sha256=sMANKUOJ4pgfaoPz8gtpKCZlwtj8LxfEtDMZBRmd1Yw,9199
428
+ lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py,sha256=lQUIg2SvoxGWpEpGBuel3JfPaT3haPAVWSdfweylyPQ,10858
425
429
  lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_response.py,sha256=-Qd3G_DIiPxyPYfHsnO0jDX_HBdG7MVVsxhYqI1QQeA,3731
426
430
  lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_app_instance_response.py,sha256=hrhbbyDpX8OMGtiJrWxWFBZWWTyxm2jgiQ6ITvDOvsA,5198
427
431
  lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py,sha256=9EPib6PkH2dfp6kue6-3mXptlSZncVOlRlhyZQ_5l0g,14448
@@ -449,7 +453,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_req
449
453
  lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_response.py,sha256=tX9omq4WA6WXxeEY9dtMwIBKHI2qm313YKCsNYB6_G0,3751
450
454
  lightning_sdk/lightning_cloud/openapi/models/v1_create_snowflake_connection_response.py,sha256=b4biVVwhuLZ7FY9HKuksWhaF4ZzsliU4iC8aMpWlGjs,3970
451
455
  lightning_sdk/lightning_cloud/openapi/models/v1_create_ssh_public_key_request.py,sha256=YRXp5UnGhUCW8_ezbmqJRtpnxw9zrGAW8e8Q6ectLDc,5212
452
- lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_request.py,sha256=mdgZFNCrPGpuPiu8DkzWOJWSsMt9CveeVMzoQAp2Rk8,8571
456
+ lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_request.py,sha256=-xh6ezKBYfB3IWQhfUmbjls7iIChtA7JMvSTVTrD5Hg,9511
453
457
  lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_response.py,sha256=n3jJzoW0w_tISOxcK2fbfc42LAPpkXVKbmQufltELPc,3851
454
458
  lightning_sdk/lightning_cloud/openapi/models/v1_create_user_secret_request.py,sha256=RWDzrdrlk_BxC9gE230gvtIRmV9uPUwlhRUnuo-gS9k,4364
455
459
  lightning_sdk/lightning_cloud/openapi/models/v1_daily_usage.py,sha256=RlZANkpfFzATUZvbV3zdploDXrnOVVM0jOvV0L5bXDk,7060
@@ -601,7 +605,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_affiliate_link_response.py,s
601
605
  lightning_sdk/lightning_cloud/openapi/models/v1_get_agent_job_env_response.py,sha256=KpiIorM_b3t-DSynCtX0vTB6O3gFq7Bw1LDMgqhYc6A,4390
602
606
  lightning_sdk/lightning_cloud/openapi/models/v1_get_agent_job_logs_metadata_response.py,sha256=c_ZsP4V1UTFAxiq6g0gsu6PKD04SEBdY_M7x7edxoTU,5538
603
607
  lightning_sdk/lightning_cloud/openapi/models/v1_get_artifacts_page_response.py,sha256=GqpP8iq7PzRi8SSbQ1KDBr0kN9bOM7aN11SnDSJjwsA,4556
604
- lightning_sdk/lightning_cloud/openapi/models/v1_get_clickhouse_assistant_session_daily_aggregated_response.py,sha256=muh_WtKX-GLJPTeWF_taFb_7P8wOj_a-nbikTziFfAQ,7149
608
+ lightning_sdk/lightning_cloud/openapi/models/v1_get_assistant_session_daily_aggregated_response.py,sha256=ySjpb-qdyr1TDtECUtFU3NCfu0OSUfIH85pf5_oTugM,6929
605
609
  lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_cold_start_metrics_stats_response.py,sha256=MaawLnEDRXr0Hw4RpiWeIaZHuzCqWcDdOTSaEKVBmhA,4157
606
610
  lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_open_ports_response.py,sha256=nwkYDRlhEtRgCe_pz1H8aVcIVaYrlgwPJ6WWDsLkPgs,3897
607
611
  lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_status_response.py,sha256=mkmpn9psHLvC22e3HtkiL5ziWDAZ0UJDPShEfFcBmbY,7049
@@ -672,7 +676,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_job_type.py,sha256=Eds2FQFMTl41i
672
676
  lightning_sdk/lightning_cloud/openapi/models/v1_joinable_organization.py,sha256=IkuGZ89NWOMpe8JsxXJvkRKdcDUvV4F3akA_USvl5rs,9586
673
677
  lightning_sdk/lightning_cloud/openapi/models/v1_keep_alive_cloud_space_instance_response.py,sha256=NRrP0aqQbpd8ejQUApJ2L1vUoSx1UhNOlSh4JeQ2_es,3112
674
678
  lightning_sdk/lightning_cloud/openapi/models/v1_knowledge_configuration.py,sha256=zEm-VKsmvL6aJtICeq1w3nn-V-6GvYLRNYDbnvPGn-4,8841
675
- lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py,sha256=PEThtGUwU_yN527pg_B4Zcrko9Q-YU9DfsRODVZK1RY,4693
679
+ lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py,sha256=_KurqbmgOTXX7OXhw4jjF2V10i6TkcCCy81fd9VXlso,7566
676
680
  lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py,sha256=2D2DQbaciBzKf5QmXwvPg2wkaiMq5-4f9NOmEc1b6NI,4478
677
681
  lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py,sha256=B2XHls4lpI-ahiMHX61goAlkBB7MtAJDAgTMCbneQwM,5694
678
682
  lightning_sdk/lightning_cloud/openapi/models/v1_lightning_app_user.py,sha256=5jJlea7GP24fkRtNsTVl0AYrotiODCYe8-qyIxr3w0w,4352
@@ -693,6 +697,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_spec.py,sha256=5YP
693
697
  lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_state.py,sha256=rhE-WVVIDsrNl639D5o2K6eA4vxpz0UN4ss059FZKRw,3431
694
698
  lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_status.py,sha256=p2_tfwworAaIr-wdwPxbAfG-aEBzbOSeB0o-ey4EuZ8,14153
695
699
  lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_status_reason.py,sha256=uQETG72asM1WQnUQc4cNE0h4zmaLq-q3hPGOMeEEcak,3521
700
+ lightning_sdk/lightning_cloud/openapi/models/v1_like_status.py,sha256=YiBPGNzBOfn2hvI244ZlpkHMvIqitMfbJkaro1r0pA4,3063
696
701
  lightning_sdk/lightning_cloud/openapi/models/v1_list_affiliate_links_response.py,sha256=5TgibtzUcNlVTJ6jr6CI8KzBsya0t5XmIvnyb2Pz13g,3849
697
702
  lightning_sdk/lightning_cloud/openapi/models/v1_list_agent_job_artifacts_response.py,sha256=SUeGQPh69QORv0EWoUYEeRSMVZ-0G_kni3SpYn9lUxk,5847
698
703
  lightning_sdk/lightning_cloud/openapi/models/v1_list_agent_jobs_response.py,sha256=aOudJXodnAFiPCYxBh8kVYSbX3k2ebLCjiV6NLxauq4,3749
@@ -783,6 +788,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_list_project_memberships_respons
783
788
  lightning_sdk/lightning_cloud/openapi/models/v1_list_project_roles_response.py,sha256=J1u20GbdV87q2CQ29drebxnijIut1Np5WtXFogp3uG4,3775
784
789
  lightning_sdk/lightning_cloud/openapi/models/v1_list_published_cloud_spaces_response.py,sha256=nUGtzfLU8140FX96aZY-9V0ntTbJnhI07ArpdFZUZyg,5967
785
790
  lightning_sdk/lightning_cloud/openapi/models/v1_list_published_deployment_templates_response.py,sha256=InwtlpAJVFaWVQyox6IngoUObF5cLAodNhlLRgJc6DM,6067
791
+ lightning_sdk/lightning_cloud/openapi/models/v1_list_published_managed_endpoint_models_response.py,sha256=yaUrhlXv39wnYYD1dAzxBHQal6E4hYdILRK6Uggs6gI,4023
786
792
  lightning_sdk/lightning_cloud/openapi/models/v1_list_quests_response.py,sha256=YOAPh5y2CidpFRG0LTMOZkpBmK_Ic-SFWkNM8FpLGgA,3741
787
793
  lightning_sdk/lightning_cloud/openapi/models/v1_list_schedules_response.py,sha256=MXa_l9Pi5hUz1M5dEBMxhv9R_huK5w6kc-hy1I51CZk,3849
788
794
  lightning_sdk/lightning_cloud/openapi/models/v1_list_secrets_response.py,sha256=jclCKTTgv7HSHW_gMMNTJhXwO6dmY28AGTdT0q9hQzw,3777
@@ -816,7 +822,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py,sha256=T0IXm
816
822
  lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py,sha256=WaN3PgJhZKTQO2FTeIdf6cWsFgnu1q3KKQ_xWMKYq3g,13940
817
823
  lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py,sha256=1of-AEIimGnUd_0fpS5U8apGFYdsIXogRpFGxed67s0,5892
818
824
  lightning_sdk/lightning_cloud/openapi/models/v1_membership.py,sha256=yRLmKJvrElPTkDU_T67oo_cyJzLJZKTbfeXLdgrAyrs,18936
819
- lightning_sdk/lightning_cloud/openapi/models/v1_message.py,sha256=p_ZBY84txO3odGtWm-6QEjBm264_XCV4b9XKO-KH5GU,14034
825
+ lightning_sdk/lightning_cloud/openapi/models/v1_message.py,sha256=HrRrrgsMwiJX-jjNHdAnBM4nR0-Atp2O_S82-E_7IM4,14677
820
826
  lightning_sdk/lightning_cloud/openapi/models/v1_message_author.py,sha256=QRC1HofufQOPROrxwtLeaEfE8f1h_ehZ0tdqlYfkgyM,3591
821
827
  lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py,sha256=rlYtO8Llal6jISWpQMyOC8zMOvdVWaljXTwlHp_gzdc,4535
822
828
  lightning_sdk/lightning_cloud/openapi/models/v1_message_content_type.py,sha256=4umoW08b-M8UXpnhvUP7yb3VxjowY5F8RGqVVZrpB3M,3089
@@ -899,7 +905,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_quest.py,sha256=Sbf-ApnCz79mRC6n
899
905
  lightning_sdk/lightning_cloud/openapi/models/v1_quest_status.py,sha256=0iyYEDeFFy-om3nOezq1hVOwf-wynGpLmcnTLQ287Zo,3146
900
906
  lightning_sdk/lightning_cloud/openapi/models/v1_queue_server_type.py,sha256=4Gl6lgZsHDp8aLgnZnhR45yBieFGktnsOt6WeIcZ6Dc,3157
901
907
  lightning_sdk/lightning_cloud/openapi/models/v1_quotas.py,sha256=XA4nv7cWZ4D9GFXzqXOdPOJ0exrOiwrpRKy8vPOnxyk,11111
902
- lightning_sdk/lightning_cloud/openapi/models/v1_quote_subscription_response.py,sha256=f85VHsPrsXdmsWRYno2XNapgN1qmV1S2c5fpKxB8mhg,9665
908
+ lightning_sdk/lightning_cloud/openapi/models/v1_quote_subscription_response.py,sha256=yp-Xbfi13-eI7dk3p-n-hBBWvlCvBgTU6I23MXabXvo,10412
903
909
  lightning_sdk/lightning_cloud/openapi/models/v1_r2_data_connection.py,sha256=6PJpIK9tlhBVnJdhVNsO2HZkGsINDRq03iZ9vd3CLbE,8989
904
910
  lightning_sdk/lightning_cloud/openapi/models/v1_refresh_index_response.py,sha256=IU2rKiAWVuyUB3gSAlbRhSxjzucdUGt7uUS6szzl3vk,3022
905
911
  lightning_sdk/lightning_cloud/openapi/models/v1_refresh_path_response.py,sha256=JMb9WUz0B8QWFfcgc8CuI0-MkYeI_2MX4Nhs93TISac,3016
@@ -932,7 +938,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_rule_condition.py,sha256=nE-Vbhb
932
938
  lightning_sdk/lightning_cloud/openapi/models/v1_rule_effect.py,sha256=WqpSW3YL7UPkIXJDIusIk-PfUn6Vo3Jqv3CNFhJ_C48,3085
933
939
  lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py,sha256=yPJBwlZDUUqwjN_zaGAm_2-AvZV2jYPB-oxnw0GdvYo,4441
934
940
  lightning_sdk/lightning_cloud/openapi/models/v1_s3_folder_data_connection.py,sha256=VmokLg1mjerYS4aRU0dAtGel3wT-VwGDyMw6E1Ut2rY,3721
935
- lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py,sha256=QzUtBGXSNfUuQy-GWvh5Q4WdTYuoaKzfmx6yZ1FJPns,13678
941
+ lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py,sha256=0mVR965vieGQZZPan9DM9JzL25UjLjZM5hGfsr3rCDk,14351
936
942
  lightning_sdk/lightning_cloud/openapi/models/v1_schedule_action_type.py,sha256=bbAckQHtEu2Y2O2FBqoI_ap4IFYfbbzeIGCWMv24qZQ,3135
937
943
  lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py,sha256=Dd31UlMVr8GR2TobjVDoNYXa8TaXAI1V7QkCb0QXE0I,3165
938
944
  lightning_sdk/lightning_cloud/openapi/models/v1_search_job_logs_response.py,sha256=SIoZOP_UDvybzADuo8rCRcAtie7-GIbw-TyKJ00XKWQ,4591
@@ -1000,6 +1006,8 @@ lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_visibility_re
1000
1006
  lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_accelerators_request.py,sha256=Hz8tGrb6oRzTbtf3AC9QL7B2TlqSEPLzHqDfhosCiB0,4867
1001
1007
  lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_accelerators_response.py,sha256=Eo8HbgCNJkuIG5Ccyl7wPuOk72FhAhkaPGuAltr4tSw,3100
1002
1008
  lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_availability_request.py,sha256=E8htNmxp_YBjhqK3HLV42a3tdu5F07-Iv0wtEFTPRjg,6603
1009
+ lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_like_response.py,sha256=Y35TnUACI047cdpIcz-ow7vgdsMRkPTxjnGI4JmrxCU,4561
1010
+ lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_message_like_response.py,sha256=q8SU00NIRsHMnvGEW0T8DEyexiEdiVzY-tyuy5ElZ4s,4918
1003
1011
  lightning_sdk/lightning_cloud/openapi/models/v1_update_deployment_visibility_response.py,sha256=IFxkDJt17RkQowDTQjR3Cz4USqwpNxMKRbJc0m7U28Y,3106
1004
1012
  lightning_sdk/lightning_cloud/openapi/models/v1_update_index_response.py,sha256=ZZFoJkImB3Kv8aamaSr2NdL4382XnQzvy0z59EPMElg,3016
1005
1013
  lightning_sdk/lightning_cloud/openapi/models/v1_update_job_visibility_response.py,sha256=zWiq4NhhgdAEu8XIFxhwKAIQIT3dFwoHY61fx-KI4Ss,3064
@@ -1033,7 +1041,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_upstream_open_ai.py,sha256=jt1qQ
1033
1041
  lightning_sdk/lightning_cloud/openapi/models/v1_usage.py,sha256=ozMzoGD9mfZGYy4J5j50Dps19Y6o8cn-aYW_oRMZMy8,16865
1034
1042
  lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py,sha256=U7qC698Xj5tb3D93ZskG6sDf3lTXE13UTlGeDTvtRU4,14062
1035
1043
  lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py,sha256=k9pDp9UIaOEEWz6bTNWF_KMfcNCOp-F67N-IZ9MO2Rs,8304
1036
- lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=3MNsEPlggjQFOz8Kl35v_IIrBT6xw7nLK7g6c136sJE,84919
1044
+ lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=einKVMSSa8vf3Q2x1yVaMApwWibJfL6uFcdOqvCLQz8,76226
1037
1045
  lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py,sha256=0fnZpdhxNRXGNyILHtfm3rVztfHpSF4kXGc5kTt4zls,13960
1038
1046
  lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py,sha256=3WpZ-lf7xPwuYyQDMdP7Uc6-dh3vf5TaaUlcMfesfMk,5208
1039
1047
  lightning_sdk/lightning_cloud/openapi/models/v1_user_slurm_job_action_response.py,sha256=BdNzXH8Vsf5PHjl9Rd-TVkpAgx1YC9rf8LD0js-ba20,3058
@@ -1076,18 +1084,18 @@ lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgyd
1076
1084
  lightning_sdk/lightning_cloud/utils/name_generator.py,sha256=MkciuA10332V0mcE2PxLIiwWomWE0Fm_gNGK01vwRr4,58046
1077
1085
  lightning_sdk/lightning_cloud/utils/network.py,sha256=axPgl8rhyPcPjxiztDxyksfxax3VNg2OXL5F5Uc81b4,406
1078
1086
  lightning_sdk/llm/__init__.py,sha256=ErZva0HqN2iPtK_6hI6GN7A_HPGNrHo3wYh7vyFdO3Q,57
1079
- lightning_sdk/llm/llm.py,sha256=qp-Vj5Crz9dlSe2_FXCtbrJaRufOKPSxashuifZwGWs,13153
1087
+ lightning_sdk/llm/llm.py,sha256=WFVqBxNowVvbPVjJ2gcZBbGgy1L54X5AQDf_L0LwB9c,14131
1080
1088
  lightning_sdk/llm/public_assistants.json,sha256=sdRouOIQNcCUyh0blI5hf0MCTSAZrxGMU_I8YV5JZ_c,528
1081
1089
  lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
1082
- lightning_sdk/mmt/base.py,sha256=B3HC-c82bPHprEZh1mhLCPCrCE8BOKqwIhY7xCF9CXg,15152
1083
- lightning_sdk/mmt/mmt.py,sha256=swdGP1DOM42a_QmmY1vg3--6ZBDiC4zToAzU9Eycv4U,13446
1084
- lightning_sdk/mmt/v1.py,sha256=TxLtL0ssDoP8eyleDaFyYr4evkOKbLJcckLVIfalOno,8429
1085
- lightning_sdk/mmt/v2.py,sha256=Em1XBoqViqUMKm-sshzdMcSH5UTtZZwbJcsgqY6-mw0,9625
1090
+ lightning_sdk/mmt/base.py,sha256=iqVudKDxazomuezj6l7pa-m9I1EQnM82OKs4ZQbo4Ck,16434
1091
+ lightning_sdk/mmt/mmt.py,sha256=CZK0KopnW8j-oez464bYMxc0UbqzsYazsNTq1vBtdlg,14982
1092
+ lightning_sdk/mmt/v1.py,sha256=tIMp7Oy5243LEExbLEAm8frKqffdiNA5WjvTdr4NOio,8858
1093
+ lightning_sdk/mmt/v2.py,sha256=haFPizfIj3N84Vwq0Nl1LAXXaPTPdeeDiHXD8lAd79Y,10722
1086
1094
  lightning_sdk/pipeline/__init__.py,sha256=Sja_0NJ8vgh-2ThSVP3WDI9a9qghrWd21LkaQp4Zsp8,378
1087
- lightning_sdk/pipeline/pipeline.py,sha256=yF8ebZlhOa0pWxLUpmKqW5oy4VEMT7NFSXpUhdpWbak,5425
1088
- lightning_sdk/pipeline/printer.py,sha256=uNi3JD3RjQMVflJxik9o4CDfifROP70aV_vt6mms5zg,4671
1089
- lightning_sdk/pipeline/schedule.py,sha256=r8L4M_NH5zoXoCf13jpn2icNIJ3hUgUFZrrq-ArnTWA,147
1090
- lightning_sdk/pipeline/steps.py,sha256=5IM7BNiH42Rxq0sJNuwIN64qs-NCa81_XeP8pg_dCL8,12294
1095
+ lightning_sdk/pipeline/pipeline.py,sha256=A11Q_bRhiQm89n-VcdaXBJsvK8Yb_EY_IijOGnBmPoI,6163
1096
+ lightning_sdk/pipeline/printer.py,sha256=j8Upm9xpq94pl95oRcLiyd-_EEAH6JQAtI-n0gn4Zfg,4775
1097
+ lightning_sdk/pipeline/schedule.py,sha256=lC_r3s46bEfY_TLeHvCGdy-Ivc7vyJmtbyA9hdzFp8g,19755
1098
+ lightning_sdk/pipeline/steps.py,sha256=H2emV6ZURRAlK7TeHvos9KXvbphYOi7U6PyBoqigZoU,13013
1091
1099
  lightning_sdk/pipeline/utils.py,sha256=3L7UgY78w2VAVSQmelQNnz_X9zJdFo1NJBtez221ge0,3923
1092
1100
  lightning_sdk/services/__init__.py,sha256=wi1yv0SCnfJub5tOq8y9SblK3-CEseHJuvH-HmtAy7U,229
1093
1101
  lightning_sdk/services/file_endpoint.py,sha256=we5HC_o74J4Y6fSP_31jIizi_I_1FO_Rb2qblspD9eE,7855
@@ -1097,10 +1105,10 @@ lightning_sdk/services/utilities.py,sha256=WlzKDGqUrQFcT85lnLgHvexf7AB1ppxS4_RBk
1097
1105
  lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1098
1106
  lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
1099
1107
  lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
1100
- lightning_sdk/utils/resolve.py,sha256=6qlBUkOmcFTjhQx_CAGfnvWBbMYp6XrCV_sX_IqplLE,6748
1101
- lightning_sdk-2025.7.17.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
1102
- lightning_sdk-2025.7.17.dist-info/METADATA,sha256=k8Pp4TwROVhigYERAb4-6WeUbPPdiuNsOCzsvsfA4vU,3995
1103
- lightning_sdk-2025.7.17.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
1104
- lightning_sdk-2025.7.17.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
1105
- lightning_sdk-2025.7.17.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
1106
- lightning_sdk-2025.7.17.dist-info/RECORD,,
1108
+ lightning_sdk/utils/resolve.py,sha256=AwEutw17KAS2CMI_ds-G-_7IgNaX8CmhdO32I6VzIFw,7889
1109
+ lightning_sdk-2025.7.22.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
1110
+ lightning_sdk-2025.7.22.dist-info/METADATA,sha256=Ocy01WJJmO4SiS6BzPu631gZTELhnZT_dYo9gDseTFk,3995
1111
+ lightning_sdk-2025.7.22.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
1112
+ lightning_sdk-2025.7.22.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
1113
+ lightning_sdk-2025.7.22.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
1114
+ lightning_sdk-2025.7.22.dist-info/RECORD,,
@@ -1,119 +0,0 @@
1
- from typing import Dict, List, Optional
2
-
3
- from lightning_sdk.lightning_cloud.openapi import (
4
- Externalv1Cluster,
5
- V1CloudProvider,
6
- V1ClusterType,
7
- V1ExternalCluster,
8
- V1ListClusterAcceleratorsResponse,
9
- )
10
- from lightning_sdk.lightning_cloud.rest_client import LightningClient
11
-
12
-
13
- class ClusterApi:
14
- """Internal API client for API requests to cluster endpoints."""
15
-
16
- def __init__(self) -> None:
17
- self._client = LightningClient(max_tries=7)
18
-
19
- def get_cluster(self, cluster_id: str, project_id: str, org_id: str) -> Externalv1Cluster:
20
- """Gets the cluster from given params cluster_id, project_id and owner.
21
-
22
- :param cluster_id: cluster ID test
23
- :param project_id: the project the cluster is supposed to be associated with
24
- :param org_id: The owning org of this cluster
25
- :return:
26
- """
27
- res = self._client.cluster_service_get_cluster(id=cluster_id, org_id=org_id, project_id=project_id)
28
- if not res:
29
- raise ValueError(f"Cluster {cluster_id} does not exist")
30
- return res
31
-
32
- def list_clusters(self, project_id: str) -> List[V1ExternalCluster]:
33
- """Lists the clusters for a given project.
34
-
35
- Args:
36
- project_id: The project to list clusters for
37
-
38
- Returns:
39
- A list of clusters
40
- """
41
- res = self._client.cluster_service_list_project_clusters(
42
- project_id=project_id,
43
- )
44
- return res.clusters
45
-
46
- def list_cluster_accelerators(self, cluster_id: str, org_id: str) -> V1ListClusterAcceleratorsResponse:
47
- """Lists the accelerators for a given cluster.
48
-
49
- :param cluster_id: cluster ID test
50
- :param project_id: the project the cluster is supposed to be associated with
51
- :param org_id: The owning org of this cluster
52
- """
53
- res = self._client.cluster_service_list_cluster_accelerators(
54
- id=cluster_id,
55
- org_id=org_id,
56
- )
57
- if not res:
58
- raise ValueError(f"Cluster {cluster_id} does not exist")
59
- return res
60
-
61
- def list_global_clusters(self, project_id: str, org_id: str) -> List[Externalv1Cluster]:
62
- """Lists the accelerators for a given project.
63
-
64
- :param project_id: project ID test
65
- :param org_id: The owning org of this project
66
- """
67
- res = self._client.cluster_service_list_clusters(
68
- project_id=project_id,
69
- org_id=org_id,
70
- )
71
- if not res:
72
- raise ValueError(f"Project {project_id} does not exist")
73
- filtered_clusters = filter(lambda x: x.spec.cluster_type == V1ClusterType.GLOBAL, res.clusters)
74
- return list(filtered_clusters)
75
-
76
- def get_cluster_provider_mapping(self, project_id: str, org_id: str) -> Dict[V1CloudProvider, str]:
77
- """Gets the cluster provider mapping."""
78
- res = self.list_global_clusters(
79
- project_id=project_id,
80
- org_id=org_id,
81
- )
82
- return {self._get_cluster_provider(cluster): cluster.id for cluster in res}
83
-
84
- def _get_cluster_provider(self, cluster: Optional[Externalv1Cluster]) -> V1CloudProvider:
85
- """Determines the cloud provider based on the cluster configuration.
86
-
87
- Args:
88
- cluster: An optional Externalv1Cluster object containing cluster specifications
89
-
90
- Returns:
91
- V1CloudProvider: The determined cloud provider, defaults to AWS if no match is found
92
- """
93
- if not cluster:
94
- return V1CloudProvider.AWS
95
-
96
- if (
97
- cluster.spec
98
- and cluster.spec.driver
99
- and cluster.spec.driver in [V1CloudProvider.LIGHTNING, V1CloudProvider.DGX]
100
- ):
101
- return cluster.spec.driver
102
-
103
- if cluster.spec:
104
- if cluster.spec.aws_v1:
105
- return V1CloudProvider.AWS
106
- if cluster.spec.google_cloud_v1:
107
- return V1CloudProvider.GCP
108
- if cluster.spec.lambda_labs_v1:
109
- return V1CloudProvider.LAMBDA_LABS
110
- if cluster.spec.vultr_v1:
111
- return V1CloudProvider.VULTR
112
- if cluster.spec.slurm_v1:
113
- return V1CloudProvider.SLURM
114
- if cluster.spec.voltage_park_v1:
115
- return V1CloudProvider.VOLTAGE_PARK
116
- if cluster.spec.nebius_v1:
117
- return V1CloudProvider.NEBIUS
118
-
119
- return V1CloudProvider.AWS
File without changes