lightning-sdk 0.1.3__py3-none-any.whl → 0.1.46__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 (498) hide show
  1. lightning_sdk/__init__.py +12 -2
  2. lightning_sdk/agents.py +46 -0
  3. lightning_sdk/ai_hub.py +185 -0
  4. lightning_sdk/api/__init__.py +4 -0
  5. lightning_sdk/api/agents_api.py +107 -0
  6. lightning_sdk/api/ai_hub_api.py +130 -0
  7. lightning_sdk/api/deployment_api.py +574 -0
  8. lightning_sdk/api/job_api.py +308 -0
  9. lightning_sdk/api/mmt_api.py +188 -0
  10. lightning_sdk/api/org_api.py +1 -3
  11. lightning_sdk/api/studio_api.py +172 -81
  12. lightning_sdk/api/teamspace_api.py +219 -20
  13. lightning_sdk/api/user_api.py +24 -9
  14. lightning_sdk/api/utils.py +429 -48
  15. lightning_sdk/cli/ai_hub.py +49 -0
  16. lightning_sdk/cli/download.py +132 -0
  17. lightning_sdk/cli/entrypoint.py +11 -3
  18. lightning_sdk/cli/run.py +206 -0
  19. lightning_sdk/cli/serve.py +218 -0
  20. lightning_sdk/cli/studios_menu.py +78 -0
  21. lightning_sdk/cli/upload.py +79 -89
  22. lightning_sdk/constants.py +29 -1
  23. lightning_sdk/deployment/__init__.py +25 -0
  24. lightning_sdk/deployment/deployment.py +389 -0
  25. lightning_sdk/helpers.py +49 -0
  26. lightning_sdk/job/__init__.py +5 -0
  27. lightning_sdk/job/base.py +295 -0
  28. lightning_sdk/job/job.py +266 -0
  29. lightning_sdk/job/v1.py +241 -0
  30. lightning_sdk/job/v2.py +193 -0
  31. lightning_sdk/job/work.py +72 -0
  32. lightning_sdk/lightning_cloud/__version__.py +1 -1
  33. lightning_sdk/lightning_cloud/cli/__main__.py +15 -13
  34. lightning_sdk/lightning_cloud/env.py +1 -0
  35. lightning_sdk/lightning_cloud/login.py +12 -8
  36. lightning_sdk/lightning_cloud/openapi/__init__.py +288 -42
  37. lightning_sdk/lightning_cloud/openapi/api/__init__.py +10 -0
  38. lightning_sdk/lightning_cloud/openapi/api/analytics_service_api.py +141 -0
  39. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +1075 -227
  40. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +9 -1
  41. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +992 -233
  42. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +883 -120
  43. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +10 -5
  44. lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py +756 -0
  45. lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py +422 -1
  46. lightning_sdk/lightning_cloud/openapi/api/experiments_service_api.py +242 -0
  47. lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py +674 -0
  48. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +3823 -0
  49. lightning_sdk/lightning_cloud/openapi/api/lightningapp_instance_service_api.py +158 -594
  50. lightning_sdk/lightning_cloud/openapi/api/lightningapp_v2_service_api.py +0 -1086
  51. lightning_sdk/lightning_cloud/openapi/api/lightningwork_service_api.py +113 -0
  52. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +1753 -0
  53. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +242 -0
  54. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +1423 -108
  55. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +421 -1
  56. lightning_sdk/lightning_cloud/openapi/api/profiler_service_api.py +663 -0
  57. lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py +5 -1
  58. lightning_sdk/lightning_cloud/openapi/api/secret_service_api.py +478 -1
  59. lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py +9 -5
  60. lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py +686 -0
  61. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +1094 -0
  62. lightning_sdk/lightning_cloud/openapi/api/studio_jobs_service_api.py +4 -4
  63. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +1081 -34
  64. lightning_sdk/lightning_cloud/openapi/models/__init__.py +278 -40
  65. lightning_sdk/lightning_cloud/openapi/models/affiliatelinks_id_body.py +149 -0
  66. lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py +305 -0
  67. lightning_sdk/lightning_cloud/openapi/models/{assistants_id_body.py → agents_id_body.py} +201 -71
  68. lightning_sdk/lightning_cloud/openapi/models/app_id_works_body.py +149 -0
  69. lightning_sdk/lightning_cloud/openapi/models/approveautojoindomain_domain_body.py +123 -0
  70. lightning_sdk/lightning_cloud/openapi/models/apps_id_body1.py +107 -3
  71. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +1 -27
  72. lightning_sdk/lightning_cloud/openapi/models/{v1_get_cluster_health_response.py → captures_id_body.py} +16 -16
  73. lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versionpublications_body1.py +27 -1
  74. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_runs_body.py +27 -1
  75. lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py +58 -6
  76. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +253 -0
  77. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py +55 -3
  78. lightning_sdk/lightning_cloud/openapi/models/cluster_id_proxies_body.py +123 -0
  79. lightning_sdk/lightning_cloud/openapi/models/create.py +157 -1
  80. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +383 -0
  81. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +565 -0
  82. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +513 -0
  83. lightning_sdk/lightning_cloud/openapi/models/endpoints_id_body.py +43 -17
  84. lightning_sdk/lightning_cloud/openapi/models/experiment_name_variant_name_body.py +123 -0
  85. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +107 -1
  86. lightning_sdk/lightning_cloud/openapi/models/externalv1_lightningapp_instance.py +27 -1
  87. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +27 -1
  88. lightning_sdk/lightning_cloud/openapi/models/fileendpoints_id_body.py +79 -1
  89. lightning_sdk/lightning_cloud/openapi/models/id_codeconfig_body.py +29 -55
  90. lightning_sdk/lightning_cloud/openapi/models/id_engage_body.py +3 -29
  91. lightning_sdk/lightning_cloud/openapi/models/id_engage_body1.py +149 -0
  92. lightning_sdk/lightning_cloud/openapi/models/id_execute_body.py +3 -55
  93. lightning_sdk/lightning_cloud/openapi/models/id_execute_body1.py +175 -0
  94. lightning_sdk/lightning_cloud/openapi/models/id_get_body.py +133 -3
  95. lightning_sdk/lightning_cloud/openapi/models/id_index_body.py +67 -15
  96. lightning_sdk/lightning_cloud/openapi/models/id_index_body2.py +123 -0
  97. lightning_sdk/lightning_cloud/openapi/models/id_index_body3.py +175 -0
  98. lightning_sdk/lightning_cloud/openapi/models/id_reportlogsactivity_body.py +123 -0
  99. lightning_sdk/lightning_cloud/openapi/models/id_start_body.py +29 -3
  100. lightning_sdk/lightning_cloud/openapi/models/id_storage_body.py +52 -26
  101. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body.py +123 -0
  102. lightning_sdk/lightning_cloud/openapi/models/jobs_id_body1.py +27 -1
  103. lightning_sdk/lightning_cloud/openapi/models/jobs_id_body2.py +17 -43
  104. lightning_sdk/lightning_cloud/openapi/models/jobs_id_body3.py +149 -0
  105. lightning_sdk/lightning_cloud/openapi/models/litloggermetrics_id_body.py +175 -0
  106. lightning_sdk/lightning_cloud/openapi/models/litpages_id_body.py +27 -1
  107. lightning_sdk/lightning_cloud/openapi/models/loggermetrics_id_body.py +123 -0
  108. lightning_sdk/lightning_cloud/openapi/models/metrics_stream_id_loggerartifacts_body.py +123 -0
  109. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +383 -0
  110. lightning_sdk/lightning_cloud/openapi/models/{v1_cloud_space_id_list.py → metricsstream_delete_body.py} +10 -10
  111. lightning_sdk/lightning_cloud/openapi/models/metricsstream_id_body.py +175 -0
  112. lightning_sdk/lightning_cloud/openapi/models/{v1_upload_model_response.py → model_id_versions_body.py} +25 -51
  113. lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py +123 -0
  114. lightning_sdk/lightning_cloud/openapi/models/models_model_id_body.py +149 -0
  115. lightning_sdk/lightning_cloud/openapi/models/multimachinejobs_id_body.py +123 -0
  116. lightning_sdk/lightning_cloud/openapi/models/org_id_memberships_body.py +27 -1
  117. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +183 -1
  118. lightning_sdk/lightning_cloud/openapi/models/profiler_captures_body.py +279 -0
  119. lightning_sdk/lightning_cloud/openapi/models/profiler_enabled_body.py +149 -0
  120. lightning_sdk/lightning_cloud/openapi/models/project_id_agentmanagedendpoints_body.py +149 -0
  121. lightning_sdk/lightning_cloud/openapi/models/{project_id_assistants_body.py → project_id_agents_body.py} +159 -55
  122. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +71 -19
  123. lightning_sdk/lightning_cloud/openapi/models/project_id_fileendpoints_body.py +27 -1
  124. lightning_sdk/lightning_cloud/openapi/models/project_id_jobs_body.py +175 -0
  125. lightning_sdk/lightning_cloud/openapi/models/project_id_litregistry_body.py +123 -0
  126. lightning_sdk/lightning_cloud/openapi/models/project_id_memberships_body.py +53 -1
  127. lightning_sdk/lightning_cloud/openapi/models/{v1_upload_model_request.py → project_id_models_body.py} +70 -70
  128. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +227 -0
  129. lightning_sdk/lightning_cloud/openapi/models/project_id_secrets_body.py +27 -1
  130. lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py +123 -0
  131. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +29 -3
  132. lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py +175 -0
  133. lightning_sdk/lightning_cloud/openapi/models/secrets_id_body1.py +123 -0
  134. lightning_sdk/lightning_cloud/openapi/models/servers_server_id_body.py +123 -0
  135. lightning_sdk/lightning_cloud/openapi/models/service_artifact_artifact_kind.py +104 -0
  136. lightning_sdk/lightning_cloud/openapi/models/serviceexecution_id_body.py +43 -43
  137. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +79 -1
  138. lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py +305 -0
  139. lightning_sdk/lightning_cloud/openapi/models/{v1_download_model_response.py → snowflake_query_body.py} +51 -51
  140. lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py +27 -1
  141. lightning_sdk/lightning_cloud/openapi/models/update.py +105 -1
  142. lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body.py +149 -0
  143. lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body.py +149 -0
  144. lightning_sdk/lightning_cloud/openapi/models/user_id_affiliatelinks_body.py +149 -0
  145. lightning_sdk/lightning_cloud/openapi/models/v1_accelerator_quota_info.py +201 -0
  146. lightning_sdk/lightning_cloud/openapi/models/v1_ack_user_storage_violation_response.py +97 -0
  147. lightning_sdk/lightning_cloud/openapi/models/v1_add_job_timing_response.py +97 -0
  148. lightning_sdk/lightning_cloud/openapi/models/v1_affiliate_link.py +435 -0
  149. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +131 -1
  150. lightning_sdk/lightning_cloud/openapi/models/v1_api_pricing_spec.py +149 -0
  151. lightning_sdk/lightning_cloud/openapi/models/v1_app_type.py +104 -0
  152. lightning_sdk/lightning_cloud/openapi/models/v1_append_logger_metrics_response.py +97 -0
  153. lightning_sdk/lightning_cloud/openapi/models/v1_approve_auto_join_domain_response.py +123 -0
  154. lightning_sdk/lightning_cloud/openapi/models/v1_assign_variant_response.py +97 -0
  155. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +131 -1
  156. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_knowledge_item_status.py +253 -0
  157. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_knowledge_status.py +123 -0
  158. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +104 -0
  159. lightning_sdk/lightning_cloud/openapi/models/v1_auto_join_domain_validation.py +175 -0
  160. lightning_sdk/lightning_cloud/openapi/models/v1_auto_join_org_response.py +149 -0
  161. lightning_sdk/lightning_cloud/openapi/models/v1_autoscaling_spec.py +305 -0
  162. lightning_sdk/lightning_cloud/openapi/models/v1_autoscaling_target_metric.py +149 -0
  163. lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +133 -3
  164. lightning_sdk/lightning_cloud/openapi/models/v1_batch_update_lightningwork_response.py +97 -0
  165. lightning_sdk/lightning_cloud/openapi/models/v1_body.py +123 -0
  166. lightning_sdk/lightning_cloud/openapi/models/v1_cancellation_metadata.py +149 -0
  167. lightning_sdk/lightning_cloud/openapi/models/v1_capacity_block_offering.py +383 -0
  168. lightning_sdk/lightning_cloud/openapi/models/v1_check_snowflake_connection_response.py +123 -0
  169. lightning_sdk/lightning_cloud/openapi/models/v1_checkbox.py +201 -0
  170. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +136 -6
  171. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_code_version.py +27 -1
  172. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_instance_config.py +1 -53
  173. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_instance_startup_status.py +79 -1
  174. lightning_sdk/lightning_cloud/openapi/models/{v1_cluster_log_service.py → v1_cloud_space_session.py} +49 -49
  175. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version.py +53 -1
  176. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +445 -3
  177. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +55 -3
  178. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_names.py +123 -0
  179. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_proxy.py +201 -0
  180. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_resource_tag.py +149 -0
  181. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +357 -0
  182. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +157 -107
  183. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_state.py +1 -0
  184. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +17 -43
  185. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py +175 -0
  186. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_type.py +0 -1
  187. lightning_sdk/lightning_cloud/openapi/models/v1_command_argument.py +79 -1
  188. lightning_sdk/lightning_cloud/openapi/models/v1_complete_model_upload_response.py +97 -0
  189. lightning_sdk/lightning_cloud/openapi/models/v1_complete_multi_part_upload_response.py +97 -0
  190. lightning_sdk/lightning_cloud/openapi/models/v1_completed_part.py +149 -0
  191. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +15 -15
  192. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +26 -26
  193. lightning_sdk/lightning_cloud/openapi/models/v1_count_metrics_streams_response.py +123 -0
  194. lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py +53 -1
  195. lightning_sdk/lightning_cloud/openapi/models/v1_create_cluster_request.py +27 -1
  196. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_template_request.py +539 -0
  197. lightning_sdk/lightning_cloud/openapi/models/v1_create_multi_part_upload_response.py +123 -0
  198. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +159 -3
  199. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  200. lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_request.py +201 -0
  201. lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_response.py +123 -0
  202. lightning_sdk/lightning_cloud/openapi/models/v1_create_snowflake_connection_response.py +123 -0
  203. lightning_sdk/lightning_cloud/openapi/models/v1_create_user_secret_request.py +149 -0
  204. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +365 -1
  205. lightning_sdk/lightning_cloud/openapi/models/{v1_cluster_performance_profile.py → v1_data_connection_state.py} +11 -9
  206. lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py +201 -0
  207. lightning_sdk/lightning_cloud/openapi/models/v1_delete_affiliate_link_response.py +97 -0
  208. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cloud_space_session_response.py +97 -0
  209. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_proxy_response.py +97 -0
  210. lightning_sdk/lightning_cloud/openapi/models/v1_delete_deployment_release_response.py +97 -0
  211. lightning_sdk/lightning_cloud/openapi/models/{v1_azure_cluster_driver_status.py → v1_delete_deployment_response.py} +6 -6
  212. lightning_sdk/lightning_cloud/openapi/models/v1_delete_index_response.py +97 -0
  213. lightning_sdk/lightning_cloud/openapi/models/{v1_byom_cluster_driver.py → v1_delete_job_response.py} +6 -6
  214. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_lightningapp_v2_response.py → v1_delete_logger_artifact_response.py} +6 -6
  215. lightning_sdk/lightning_cloud/openapi/models/v1_delete_managed_endpoint_response.py +97 -0
  216. lightning_sdk/lightning_cloud/openapi/models/v1_delete_metrics_stream_response.py +97 -0
  217. lightning_sdk/lightning_cloud/openapi/models/v1_delete_multi_machine_job_response.py +97 -0
  218. lightning_sdk/lightning_cloud/openapi/models/v1_delete_profiler_capture_response.py +97 -0
  219. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_lightningapp_release_response.py → v1_delete_shared_metrics_stream_response.py} +6 -6
  220. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +617 -0
  221. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +227 -0
  222. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event.py +357 -0
  223. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event_type.py +104 -0
  224. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py +149 -0
  225. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_performance.py +305 -0
  226. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_release.py +331 -0
  227. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_spec.py +201 -0
  228. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py +105 -0
  229. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +279 -0
  230. lightning_sdk/lightning_cloud/openapi/models/{appsv2_id_body.py → v1_deployment_strategy.py} +39 -39
  231. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template.py +721 -0
  232. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_engagement_response.py +97 -0
  233. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_gallery_response.py +591 -0
  234. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter.py +435 -0
  235. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_placement.py +106 -0
  236. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_type.py +106 -0
  237. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_summary.py +591 -0
  238. lightning_sdk/lightning_cloud/openapi/models/{instance_type_availability.py → v1_deployment_template_type.py} +10 -10
  239. lightning_sdk/lightning_cloud/openapi/models/{v1_get_lightningapp_source_code_download_url_response.py → v1_download_job_logs_response.py} +10 -10
  240. lightning_sdk/lightning_cloud/openapi/models/v1_download_service_execution_artifact_response.py +175 -0
  241. lightning_sdk/lightning_cloud/openapi/models/v1_ebs.py +279 -0
  242. lightning_sdk/lightning_cloud/openapi/models/v1_efs_config.py +201 -0
  243. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint.py +53 -27
  244. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint_auth.py +27 -1
  245. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint_type.py +104 -0
  246. lightning_sdk/lightning_cloud/openapi/models/v1_execute_cloud_space_command_response.py +29 -3
  247. lightning_sdk/lightning_cloud/openapi/models/v1_execute_in_cloud_space_session_response.py +97 -0
  248. lightning_sdk/lightning_cloud/openapi/models/v1_execute_snowflake_query_response.py +149 -0
  249. lightning_sdk/lightning_cloud/openapi/models/v1_experiment.py +409 -0
  250. lightning_sdk/lightning_cloud/openapi/models/v1_export_snowflake_query_response.py +123 -0
  251. lightning_sdk/lightning_cloud/openapi/models/v1_file_endpoint.py +79 -1
  252. lightning_sdk/lightning_cloud/openapi/models/{v1_instance_spec.py → v1_filesystem_app.py} +85 -59
  253. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_cloud_space.py +149 -0
  254. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_dataset.py +123 -0
  255. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_job.py +175 -0
  256. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_slurm_job.py +149 -0
  257. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_snowflake_connection.py +123 -0
  258. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_work.py +149 -0
  259. lightning_sdk/lightning_cloud/openapi/models/v1_find_capacity_block_offering_response.py +123 -0
  260. lightning_sdk/lightning_cloud/openapi/models/v1_gcp_data_connection.py +27 -1
  261. lightning_sdk/lightning_cloud/openapi/models/v1_gcs_folder_data_connection.py +123 -0
  262. lightning_sdk/lightning_cloud/openapi/models/v1_get_affiliate_link_response.py +123 -0
  263. lightning_sdk/lightning_cloud/openapi/models/{v1_list_cluster_instance_types_response.py → v1_get_deployment_routing_telemetry_aggregated_response.py} +23 -23
  264. lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_response.py +123 -0
  265. lightning_sdk/lightning_cloud/openapi/models/v1_get_folder_index_response.py +27 -1
  266. lightning_sdk/lightning_cloud/openapi/models/{v1_cluster_driver_status.py → v1_get_job_stats_response.py} +39 -39
  267. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_system_metrics_response.py +123 -0
  268. lightning_sdk/lightning_cloud/openapi/models/v1_get_lightningapp_instance_open_ports_response.py +123 -0
  269. lightning_sdk/lightning_cloud/openapi/models/v1_get_logger_metrics_response.py +123 -0
  270. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_upload_urls_response.py +123 -0
  271. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_url_response.py +175 -0
  272. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +279 -0
  273. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py +149 -0
  274. lightning_sdk/lightning_cloud/openapi/models/{v1_lightningwork_cluster_driver.py → v1_get_project_artifact_response.py} +33 -14
  275. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +27 -1
  276. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_storage_metadata_response.py +383 -0
  277. lightning_sdk/lightning_cloud/openapi/models/v1_get_service_execution_status_response.py +67 -15
  278. lightning_sdk/lightning_cloud/openapi/models/v1_get_snowflake_query_response.py +149 -0
  279. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_balance_response.py +27 -1
  280. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +105 -1
  281. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py +279 -0
  282. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py +201 -0
  283. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +163 -3
  284. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1_status.py +123 -0
  285. lightning_sdk/lightning_cloud/openapi/models/v1_header.py +175 -0
  286. lightning_sdk/lightning_cloud/openapi/models/{v1_external_kubeconfig.py → v1_health_check_exec.py} +21 -21
  287. lightning_sdk/lightning_cloud/openapi/models/v1_health_check_http_get.py +149 -0
  288. lightning_sdk/lightning_cloud/openapi/models/v1_ids_logger_metrics.py +123 -0
  289. lightning_sdk/lightning_cloud/openapi/models/v1_input.py +175 -0
  290. lightning_sdk/lightning_cloud/openapi/models/v1_interrupt_server_response.py +97 -0
  291. lightning_sdk/lightning_cloud/openapi/models/v1_job.py +697 -0
  292. lightning_sdk/lightning_cloud/openapi/models/v1_job_health_check_config.py +253 -0
  293. lightning_sdk/lightning_cloud/openapi/models/v1_job_log_entry.py +175 -0
  294. lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_page.py +227 -0
  295. lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_response.py +149 -0
  296. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +669 -0
  297. lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py +201 -0
  298. lightning_sdk/lightning_cloud/openapi/models/v1_joinable_organization.py +331 -0
  299. lightning_sdk/lightning_cloud/openapi/models/v1_knowledge_configuration.py +279 -0
  300. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +125 -0
  301. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +27 -1
  302. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_instance_spec.py +105 -27
  303. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_instance_status.py +53 -1
  304. lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_spec.py +27 -27
  305. lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_status.py +81 -3
  306. lightning_sdk/lightning_cloud/openapi/models/v1_list_affiliate_links_response.py +123 -0
  307. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_python_versions_response.py +123 -0
  308. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_sessions_response.py +123 -0
  309. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_tags_response.py +6 -6
  310. lightning_sdk/lightning_cloud/openapi/models/v1_list_cluster_availabilities_response.py +123 -0
  311. lightning_sdk/lightning_cloud/openapi/models/v1_list_cluster_proxies_response.py +123 -0
  312. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_events_response.py +123 -0
  313. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_releases_response.py +123 -0
  314. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_tags_response.py +123 -0
  315. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_templates_response.py +175 -0
  316. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployments_response.py +123 -0
  317. lightning_sdk/lightning_cloud/openapi/models/v1_list_experiments_response.py +149 -0
  318. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_apps_response.py +123 -0
  319. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_cloud_spaces_response.py +123 -0
  320. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_datasets_response.py +123 -0
  321. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_jobs_response.py +123 -0
  322. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_slurm_jobs_response.py +123 -0
  323. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_snowflake_response.py +123 -0
  324. lightning_sdk/lightning_cloud/openapi/models/{v1_list_lightningapps_v2_response.py → v1_list_jobs_response.py} +31 -31
  325. lightning_sdk/lightning_cloud/openapi/models/v1_list_joinable_organizations_response.py +123 -0
  326. lightning_sdk/lightning_cloud/openapi/models/v1_list_logger_artifact_response.py +123 -0
  327. lightning_sdk/lightning_cloud/openapi/models/v1_list_metrics_streams_response.py +123 -0
  328. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_job_events_response.py +123 -0
  329. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_jobs_response.py +123 -0
  330. lightning_sdk/lightning_cloud/openapi/models/v1_list_new_features_for_user_response.py +123 -0
  331. lightning_sdk/lightning_cloud/openapi/models/v1_list_profiler_captures_response.py +123 -0
  332. lightning_sdk/lightning_cloud/openapi/models/v1_list_project_locked_resources_response.py +123 -0
  333. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_cloud_spaces_response.py +3 -29
  334. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_deployment_templates_response.py +175 -0
  335. lightning_sdk/lightning_cloud/openapi/models/v1_list_service_execution_lightningapp_instances_response.py +175 -0
  336. lightning_sdk/lightning_cloud/openapi/models/v1_lit_page.py +27 -1
  337. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +227 -0
  338. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +279 -0
  339. lightning_sdk/lightning_cloud/openapi/models/v1_locked_resource.py +227 -0
  340. lightning_sdk/lightning_cloud/openapi/models/v1_logger_artifact.py +227 -0
  341. lightning_sdk/lightning_cloud/openapi/models/v1_machines_selector.py +149 -0
  342. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +81 -3
  343. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +53 -1
  344. lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py +175 -19
  345. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +305 -0
  346. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py +175 -0
  347. lightning_sdk/lightning_cloud/openapi/models/v1_membership.py +79 -1
  348. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +138 -8
  349. lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py +6 -6
  350. lightning_sdk/lightning_cloud/openapi/models/v1_message_content_type.py +103 -0
  351. lightning_sdk/lightning_cloud/openapi/models/v1_metadata.py +131 -1
  352. lightning_sdk/lightning_cloud/openapi/models/v1_metric_value.py +175 -0
  353. lightning_sdk/lightning_cloud/openapi/models/v1_metrics.py +175 -0
  354. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +799 -0
  355. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_tags.py +201 -0
  356. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_tracker.py +383 -0
  357. lightning_sdk/lightning_cloud/openapi/models/v1_model.py +95 -17
  358. lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py +175 -0
  359. lightning_sdk/lightning_cloud/openapi/models/v1_model_version_archive.py +131 -1
  360. lightning_sdk/lightning_cloud/openapi/models/v1_mount_target.py +201 -0
  361. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +487 -0
  362. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py +331 -0
  363. lightning_sdk/lightning_cloud/openapi/models/{get_cluster_health_response_health_status.py → v1_multi_machine_job_event_type.py} +9 -9
  364. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance.py +149 -0
  365. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance_strategy.py +105 -0
  366. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +108 -0
  367. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +305 -0
  368. lightning_sdk/lightning_cloud/openapi/models/v1_named_get_logger_metrics.py +123 -0
  369. lightning_sdk/lightning_cloud/openapi/models/v1_new_feature.py +383 -0
  370. lightning_sdk/lightning_cloud/openapi/models/v1_onboarding_event_request.py +175 -0
  371. lightning_sdk/lightning_cloud/openapi/models/v1_onboarding_event_response.py +97 -0
  372. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +209 -1
  373. lightning_sdk/lightning_cloud/openapi/models/v1_parameterization_spec.py +227 -0
  374. lightning_sdk/lightning_cloud/openapi/models/v1_path_telemetry.py +123 -0
  375. lightning_sdk/lightning_cloud/openapi/models/v1_phase_type.py +104 -0
  376. lightning_sdk/lightning_cloud/openapi/models/v1_profiler_capture.py +357 -0
  377. lightning_sdk/lightning_cloud/openapi/models/v1_profiler_enabled_response.py +123 -0
  378. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +131 -1
  379. lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py +263 -3
  380. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +29 -3
  381. lightning_sdk/lightning_cloud/openapi/models/v1_project_storage.py +279 -0
  382. lightning_sdk/lightning_cloud/openapi/models/v1_published_cloud_space_response.py +188 -6
  383. lightning_sdk/lightning_cloud/openapi/models/v1_purchase_capacity_block_response.py +175 -0
  384. lightning_sdk/lightning_cloud/openapi/models/v1_query_param.py +175 -0
  385. lightning_sdk/lightning_cloud/openapi/models/v1_query_result.py +123 -0
  386. lightning_sdk/lightning_cloud/openapi/models/v1_query_result_row.py +123 -0
  387. lightning_sdk/lightning_cloud/openapi/models/v1_refresh_path_response.py +97 -0
  388. lightning_sdk/lightning_cloud/openapi/models/v1_regional_load_balancer.py +149 -0
  389. lightning_sdk/lightning_cloud/openapi/models/v1_report_logs_activity_response.py +97 -0
  390. lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_request.py +175 -0
  391. lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_response.py +97 -0
  392. lightning_sdk/lightning_cloud/openapi/models/{v1_cloud_space_tag.py → v1_resource_tag.py} +18 -18
  393. lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py +123 -0
  394. lightning_sdk/lightning_cloud/openapi/models/v1_resources.py +37 -11
  395. lightning_sdk/lightning_cloud/openapi/models/v1_restore_deployment_release_response.py +97 -0
  396. lightning_sdk/lightning_cloud/openapi/models/v1_rolling_update_strategy.py +149 -0
  397. lightning_sdk/lightning_cloud/openapi/models/v1_rule_condition.py +29 -3
  398. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +11 -0
  399. lightning_sdk/lightning_cloud/openapi/models/v1_s3_folder_data_connection.py +123 -0
  400. lightning_sdk/lightning_cloud/openapi/models/v1_search_job_logs_response.py +149 -0
  401. lightning_sdk/lightning_cloud/openapi/models/v1_secret.py +107 -3
  402. lightning_sdk/lightning_cloud/openapi/models/{v1_lightningapp_cluster_driver.py → v1_secret_type.py} +10 -9
  403. lightning_sdk/lightning_cloud/openapi/models/v1_select.py +149 -0
  404. lightning_sdk/lightning_cloud/openapi/models/v1_server_check_in_response.py +97 -0
  405. lightning_sdk/lightning_cloud/openapi/models/v1_service_artifact.py +201 -0
  406. lightning_sdk/lightning_cloud/openapi/models/v1_service_execution.py +43 -43
  407. lightning_sdk/lightning_cloud/openapi/models/v1_should_start_syncing_response.py +123 -0
  408. lightning_sdk/lightning_cloud/openapi/models/v1_signed_url.py +149 -0
  409. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +131 -1
  410. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +53 -1
  411. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1.py +27 -1
  412. lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py +253 -0
  413. lightning_sdk/lightning_cloud/openapi/models/v1_status_code_telemetry.py +123 -0
  414. lightning_sdk/lightning_cloud/openapi/models/v1_storage_asset.py +253 -0
  415. lightning_sdk/lightning_cloud/openapi/models/v1_storage_asset_type.py +105 -0
  416. lightning_sdk/lightning_cloud/openapi/models/v1_studio_job_app.py +1 -0
  417. lightning_sdk/lightning_cloud/openapi/models/v1_system_info.py +617 -0
  418. lightning_sdk/lightning_cloud/openapi/models/v1_telemetry.py +331 -0
  419. lightning_sdk/lightning_cloud/openapi/models/v1_timestamp_code_telemetry.py +123 -0
  420. lightning_sdk/lightning_cloud/openapi/models/v1_transaction.py +27 -1
  421. lightning_sdk/lightning_cloud/openapi/models/v1_transfer_user_balance_request.py +27 -1
  422. lightning_sdk/lightning_cloud/openapi/models/v1_update_billing_subscription_request.py +27 -1
  423. lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_accelerators_request.py +149 -0
  424. lightning_sdk/lightning_cloud/openapi/models/v1_update_index_response.py +97 -0
  425. lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py +97 -0
  426. lightning_sdk/lightning_cloud/openapi/models/v1_update_model_visibility_response.py +97 -0
  427. lightning_sdk/lightning_cloud/openapi/models/v1_update_shared_metrics_stream_response.py +97 -0
  428. lightning_sdk/lightning_cloud/openapi/models/v1_update_snowflake_query_response.py +97 -0
  429. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +53 -1
  430. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_viewed_new_features_request.py +123 -0
  431. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_viewed_new_features_response.py +97 -0
  432. lightning_sdk/lightning_cloud/openapi/models/v1_upstream_cloud_space.py +97 -19
  433. lightning_sdk/lightning_cloud/openapi/models/v1_upstream_job.py +227 -0
  434. lightning_sdk/lightning_cloud/openapi/models/v1_usage.py +27 -1
  435. lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py +107 -3
  436. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +1006 -330
  437. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +45 -45
  438. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py +29 -29
  439. lightning_sdk/lightning_cloud/openapi/models/v1_validate_assistant_status_response.py +149 -0
  440. lightning_sdk/lightning_cloud/openapi/models/v1_validate_auto_join_domain_response.py +97 -0
  441. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +107 -3
  442. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +149 -0
  443. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py +97 -0
  444. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_request.py +175 -0
  445. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +123 -0
  446. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_model_response.py +123 -0
  447. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +175 -0
  448. lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py +125 -0
  449. lightning_sdk/lightning_cloud/openapi/models/validate.py +97 -19
  450. lightning_sdk/lightning_cloud/openapi/models/validateautojoindomain_domain_body.py +123 -0
  451. lightning_sdk/lightning_cloud/openapi/models/version_uploads_body.py +123 -0
  452. lightning_sdk/lightning_cloud/rest_client.py +47 -23
  453. lightning_sdk/lightning_cloud/source_code/logs_socket_api.py +1 -1
  454. lightning_sdk/lightning_cloud/source_code/tar.py +1 -3
  455. lightning_sdk/lightning_cloud/utils/data_connection.py +145 -7
  456. lightning_sdk/machine.py +17 -4
  457. lightning_sdk/mmt/__init__.py +4 -0
  458. lightning_sdk/mmt/base.py +288 -0
  459. lightning_sdk/mmt/mmt.py +282 -0
  460. lightning_sdk/mmt/v1.py +185 -0
  461. lightning_sdk/mmt/v2.py +193 -0
  462. lightning_sdk/models.py +153 -0
  463. lightning_sdk/organization.py +1 -1
  464. lightning_sdk/plugin.py +207 -41
  465. lightning_sdk/services/__init__.py +2 -1
  466. lightning_sdk/services/file_endpoint.py +116 -213
  467. lightning_sdk/services/finetune/__init__.py +13 -15
  468. lightning_sdk/services/utilities.py +99 -26
  469. lightning_sdk/status.py +2 -1
  470. lightning_sdk/studio.py +90 -17
  471. lightning_sdk/teamspace.py +189 -11
  472. lightning_sdk/user.py +1 -1
  473. lightning_sdk/utils/__init__.py +0 -0
  474. lightning_sdk/utils/dynamic.py +61 -0
  475. lightning_sdk/utils/enum.py +116 -0
  476. lightning_sdk/{utils.py → utils/resolve.py} +41 -4
  477. lightning_sdk-0.1.46.dist-info/LICENSE +21 -0
  478. {lightning_sdk-0.1.3.dist-info → lightning_sdk-0.1.46.dist-info}/METADATA +30 -4
  479. {lightning_sdk-0.1.3.dist-info → lightning_sdk-0.1.46.dist-info}/RECORD +482 -204
  480. {lightning_sdk-0.1.3.dist-info → lightning_sdk-0.1.46.dist-info}/WHEEL +1 -1
  481. lightning_sdk/lightning_cloud/openapi/models/app_id_releases_body.py +0 -541
  482. lightning_sdk/lightning_cloud/openapi/models/id_endpoint_body.py +0 -409
  483. lightning_sdk/lightning_cloud/openapi/models/id_get_body1.py +0 -333
  484. lightning_sdk/lightning_cloud/openapi/models/project_id_appsv2_body.py +0 -201
  485. lightning_sdk/lightning_cloud/openapi/models/v1_aws_cluster_driver_spec.py +0 -1039
  486. lightning_sdk/lightning_cloud/openapi/models/v1_aws_cluster_secondary_region_spec.py +0 -253
  487. lightning_sdk/lightning_cloud/openapi/models/v1_azure_cluster_driver_spec.py +0 -227
  488. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_driver.py +0 -175
  489. lightning_sdk/lightning_cloud/openapi/models/v1_container_resources.py +0 -201
  490. lightning_sdk/lightning_cloud/openapi/models/v1_eks_custer_driver_status.py +0 -387
  491. lightning_sdk/lightning_cloud/openapi/models/v1_instance_type.py +0 -305
  492. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_cluster_driver.py +0 -359
  493. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_cluster_status.py +0 -279
  494. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_release.py +0 -697
  495. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_v2.py +0 -331
  496. lightning_sdk/services/uploader.py +0 -123
  497. {lightning_sdk-0.1.3.dist-info → lightning_sdk-0.1.46.dist-info}/entry_points.txt +0 -0
  498. {lightning_sdk-0.1.3.dist-info → lightning_sdk-0.1.46.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,435 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1AffiliateLink(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'code': 'str',
45
+ 'created_at': 'datetime',
46
+ 'expires_at': 'datetime',
47
+ 'id': 'str',
48
+ 'max_uses': 'int',
49
+ 'name': 'str',
50
+ 'referred_incentives': 'dict(str, str)',
51
+ 'skip_waitlist': 'bool',
52
+ 'type': 'str',
53
+ 'updated_at': 'datetime',
54
+ 'url': 'str',
55
+ 'usage_count': 'int',
56
+ 'user_id': 'str'
57
+ }
58
+
59
+ attribute_map = {
60
+ 'code': 'code',
61
+ 'created_at': 'createdAt',
62
+ 'expires_at': 'expiresAt',
63
+ 'id': 'id',
64
+ 'max_uses': 'maxUses',
65
+ 'name': 'name',
66
+ 'referred_incentives': 'referredIncentives',
67
+ 'skip_waitlist': 'skipWaitlist',
68
+ 'type': 'type',
69
+ 'updated_at': 'updatedAt',
70
+ 'url': 'url',
71
+ 'usage_count': 'usageCount',
72
+ 'user_id': 'userId'
73
+ }
74
+
75
+ def __init__(self, code: 'str' =None, created_at: 'datetime' =None, expires_at: 'datetime' =None, id: 'str' =None, max_uses: 'int' =None, name: 'str' =None, referred_incentives: 'dict(str, str)' =None, skip_waitlist: 'bool' =None, type: 'str' =None, updated_at: 'datetime' =None, url: 'str' =None, usage_count: 'int' =None, user_id: 'str' =None): # noqa: E501
76
+ """V1AffiliateLink - a model defined in Swagger""" # noqa: E501
77
+ self._code = None
78
+ self._created_at = None
79
+ self._expires_at = None
80
+ self._id = None
81
+ self._max_uses = None
82
+ self._name = None
83
+ self._referred_incentives = None
84
+ self._skip_waitlist = None
85
+ self._type = None
86
+ self._updated_at = None
87
+ self._url = None
88
+ self._usage_count = None
89
+ self._user_id = None
90
+ self.discriminator = None
91
+ if code is not None:
92
+ self.code = code
93
+ if created_at is not None:
94
+ self.created_at = created_at
95
+ if expires_at is not None:
96
+ self.expires_at = expires_at
97
+ if id is not None:
98
+ self.id = id
99
+ if max_uses is not None:
100
+ self.max_uses = max_uses
101
+ if name is not None:
102
+ self.name = name
103
+ if referred_incentives is not None:
104
+ self.referred_incentives = referred_incentives
105
+ if skip_waitlist is not None:
106
+ self.skip_waitlist = skip_waitlist
107
+ if type is not None:
108
+ self.type = type
109
+ if updated_at is not None:
110
+ self.updated_at = updated_at
111
+ if url is not None:
112
+ self.url = url
113
+ if usage_count is not None:
114
+ self.usage_count = usage_count
115
+ if user_id is not None:
116
+ self.user_id = user_id
117
+
118
+ @property
119
+ def code(self) -> 'str':
120
+ """Gets the code of this V1AffiliateLink. # noqa: E501
121
+
122
+
123
+ :return: The code of this V1AffiliateLink. # noqa: E501
124
+ :rtype: str
125
+ """
126
+ return self._code
127
+
128
+ @code.setter
129
+ def code(self, code: 'str'):
130
+ """Sets the code of this V1AffiliateLink.
131
+
132
+
133
+ :param code: The code of this V1AffiliateLink. # noqa: E501
134
+ :type: str
135
+ """
136
+
137
+ self._code = code
138
+
139
+ @property
140
+ def created_at(self) -> 'datetime':
141
+ """Gets the created_at of this V1AffiliateLink. # noqa: E501
142
+
143
+
144
+ :return: The created_at of this V1AffiliateLink. # noqa: E501
145
+ :rtype: datetime
146
+ """
147
+ return self._created_at
148
+
149
+ @created_at.setter
150
+ def created_at(self, created_at: 'datetime'):
151
+ """Sets the created_at of this V1AffiliateLink.
152
+
153
+
154
+ :param created_at: The created_at of this V1AffiliateLink. # noqa: E501
155
+ :type: datetime
156
+ """
157
+
158
+ self._created_at = created_at
159
+
160
+ @property
161
+ def expires_at(self) -> 'datetime':
162
+ """Gets the expires_at of this V1AffiliateLink. # noqa: E501
163
+
164
+
165
+ :return: The expires_at of this V1AffiliateLink. # noqa: E501
166
+ :rtype: datetime
167
+ """
168
+ return self._expires_at
169
+
170
+ @expires_at.setter
171
+ def expires_at(self, expires_at: 'datetime'):
172
+ """Sets the expires_at of this V1AffiliateLink.
173
+
174
+
175
+ :param expires_at: The expires_at of this V1AffiliateLink. # noqa: E501
176
+ :type: datetime
177
+ """
178
+
179
+ self._expires_at = expires_at
180
+
181
+ @property
182
+ def id(self) -> 'str':
183
+ """Gets the id of this V1AffiliateLink. # noqa: E501
184
+
185
+
186
+ :return: The id of this V1AffiliateLink. # noqa: E501
187
+ :rtype: str
188
+ """
189
+ return self._id
190
+
191
+ @id.setter
192
+ def id(self, id: 'str'):
193
+ """Sets the id of this V1AffiliateLink.
194
+
195
+
196
+ :param id: The id of this V1AffiliateLink. # noqa: E501
197
+ :type: str
198
+ """
199
+
200
+ self._id = id
201
+
202
+ @property
203
+ def max_uses(self) -> 'int':
204
+ """Gets the max_uses of this V1AffiliateLink. # noqa: E501
205
+
206
+
207
+ :return: The max_uses of this V1AffiliateLink. # noqa: E501
208
+ :rtype: int
209
+ """
210
+ return self._max_uses
211
+
212
+ @max_uses.setter
213
+ def max_uses(self, max_uses: 'int'):
214
+ """Sets the max_uses of this V1AffiliateLink.
215
+
216
+
217
+ :param max_uses: The max_uses of this V1AffiliateLink. # noqa: E501
218
+ :type: int
219
+ """
220
+
221
+ self._max_uses = max_uses
222
+
223
+ @property
224
+ def name(self) -> 'str':
225
+ """Gets the name of this V1AffiliateLink. # noqa: E501
226
+
227
+
228
+ :return: The name of this V1AffiliateLink. # noqa: E501
229
+ :rtype: str
230
+ """
231
+ return self._name
232
+
233
+ @name.setter
234
+ def name(self, name: 'str'):
235
+ """Sets the name of this V1AffiliateLink.
236
+
237
+
238
+ :param name: The name of this V1AffiliateLink. # noqa: E501
239
+ :type: str
240
+ """
241
+
242
+ self._name = name
243
+
244
+ @property
245
+ def referred_incentives(self) -> 'dict(str, str)':
246
+ """Gets the referred_incentives of this V1AffiliateLink. # noqa: E501
247
+
248
+
249
+ :return: The referred_incentives of this V1AffiliateLink. # noqa: E501
250
+ :rtype: dict(str, str)
251
+ """
252
+ return self._referred_incentives
253
+
254
+ @referred_incentives.setter
255
+ def referred_incentives(self, referred_incentives: 'dict(str, str)'):
256
+ """Sets the referred_incentives of this V1AffiliateLink.
257
+
258
+
259
+ :param referred_incentives: The referred_incentives of this V1AffiliateLink. # noqa: E501
260
+ :type: dict(str, str)
261
+ """
262
+
263
+ self._referred_incentives = referred_incentives
264
+
265
+ @property
266
+ def skip_waitlist(self) -> 'bool':
267
+ """Gets the skip_waitlist of this V1AffiliateLink. # noqa: E501
268
+
269
+
270
+ :return: The skip_waitlist of this V1AffiliateLink. # noqa: E501
271
+ :rtype: bool
272
+ """
273
+ return self._skip_waitlist
274
+
275
+ @skip_waitlist.setter
276
+ def skip_waitlist(self, skip_waitlist: 'bool'):
277
+ """Sets the skip_waitlist of this V1AffiliateLink.
278
+
279
+
280
+ :param skip_waitlist: The skip_waitlist of this V1AffiliateLink. # noqa: E501
281
+ :type: bool
282
+ """
283
+
284
+ self._skip_waitlist = skip_waitlist
285
+
286
+ @property
287
+ def type(self) -> 'str':
288
+ """Gets the type of this V1AffiliateLink. # noqa: E501
289
+
290
+
291
+ :return: The type of this V1AffiliateLink. # noqa: E501
292
+ :rtype: str
293
+ """
294
+ return self._type
295
+
296
+ @type.setter
297
+ def type(self, type: 'str'):
298
+ """Sets the type of this V1AffiliateLink.
299
+
300
+
301
+ :param type: The type of this V1AffiliateLink. # noqa: E501
302
+ :type: str
303
+ """
304
+
305
+ self._type = type
306
+
307
+ @property
308
+ def updated_at(self) -> 'datetime':
309
+ """Gets the updated_at of this V1AffiliateLink. # noqa: E501
310
+
311
+
312
+ :return: The updated_at of this V1AffiliateLink. # noqa: E501
313
+ :rtype: datetime
314
+ """
315
+ return self._updated_at
316
+
317
+ @updated_at.setter
318
+ def updated_at(self, updated_at: 'datetime'):
319
+ """Sets the updated_at of this V1AffiliateLink.
320
+
321
+
322
+ :param updated_at: The updated_at of this V1AffiliateLink. # noqa: E501
323
+ :type: datetime
324
+ """
325
+
326
+ self._updated_at = updated_at
327
+
328
+ @property
329
+ def url(self) -> 'str':
330
+ """Gets the url of this V1AffiliateLink. # noqa: E501
331
+
332
+
333
+ :return: The url of this V1AffiliateLink. # noqa: E501
334
+ :rtype: str
335
+ """
336
+ return self._url
337
+
338
+ @url.setter
339
+ def url(self, url: 'str'):
340
+ """Sets the url of this V1AffiliateLink.
341
+
342
+
343
+ :param url: The url of this V1AffiliateLink. # noqa: E501
344
+ :type: str
345
+ """
346
+
347
+ self._url = url
348
+
349
+ @property
350
+ def usage_count(self) -> 'int':
351
+ """Gets the usage_count of this V1AffiliateLink. # noqa: E501
352
+
353
+
354
+ :return: The usage_count of this V1AffiliateLink. # noqa: E501
355
+ :rtype: int
356
+ """
357
+ return self._usage_count
358
+
359
+ @usage_count.setter
360
+ def usage_count(self, usage_count: 'int'):
361
+ """Sets the usage_count of this V1AffiliateLink.
362
+
363
+
364
+ :param usage_count: The usage_count of this V1AffiliateLink. # noqa: E501
365
+ :type: int
366
+ """
367
+
368
+ self._usage_count = usage_count
369
+
370
+ @property
371
+ def user_id(self) -> 'str':
372
+ """Gets the user_id of this V1AffiliateLink. # noqa: E501
373
+
374
+
375
+ :return: The user_id of this V1AffiliateLink. # noqa: E501
376
+ :rtype: str
377
+ """
378
+ return self._user_id
379
+
380
+ @user_id.setter
381
+ def user_id(self, user_id: 'str'):
382
+ """Sets the user_id of this V1AffiliateLink.
383
+
384
+
385
+ :param user_id: The user_id of this V1AffiliateLink. # noqa: E501
386
+ :type: str
387
+ """
388
+
389
+ self._user_id = user_id
390
+
391
+ def to_dict(self) -> dict:
392
+ """Returns the model properties as a dict"""
393
+ result = {}
394
+
395
+ for attr, _ in six.iteritems(self.swagger_types):
396
+ value = getattr(self, attr)
397
+ if isinstance(value, list):
398
+ result[attr] = list(map(
399
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
400
+ value
401
+ ))
402
+ elif hasattr(value, "to_dict"):
403
+ result[attr] = value.to_dict()
404
+ elif isinstance(value, dict):
405
+ result[attr] = dict(map(
406
+ lambda item: (item[0], item[1].to_dict())
407
+ if hasattr(item[1], "to_dict") else item,
408
+ value.items()
409
+ ))
410
+ else:
411
+ result[attr] = value
412
+ if issubclass(V1AffiliateLink, dict):
413
+ for key, value in self.items():
414
+ result[key] = value
415
+
416
+ return result
417
+
418
+ def to_str(self) -> str:
419
+ """Returns the string representation of the model"""
420
+ return pprint.pformat(self.to_dict())
421
+
422
+ def __repr__(self) -> str:
423
+ """For `print` and `pprint`"""
424
+ return self.to_str()
425
+
426
+ def __eq__(self, other: 'V1AffiliateLink') -> bool:
427
+ """Returns true if both objects are equal"""
428
+ if not isinstance(other, V1AffiliateLink):
429
+ return False
430
+
431
+ return self.__dict__ == other.__dict__
432
+
433
+ def __ne__(self, other: 'V1AffiliateLink') -> bool:
434
+ """Returns true if both objects are not equal"""
435
+ return not self == other
@@ -41,6 +41,7 @@ class V1AgentJob(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'cache_id': 'str',
44
45
  'cloudspace_id': 'str',
45
46
  'cluster_id': 'str',
46
47
  'command': 'str',
@@ -48,10 +49,14 @@ class V1AgentJob(object):
48
49
  'conda_env_path': 'str',
49
50
  'created_at': 'datetime',
50
51
  'id': 'str',
52
+ 'linked_project_id': 'str',
53
+ 'linked_user_id': 'str',
51
54
  'message': 'str',
52
55
  'name': 'str',
56
+ 'num_gpus': 'str',
53
57
  'project_id': 'str',
54
58
  'run_id': 'str',
59
+ 'service_id': 'str',
55
60
  'slurm_v1_status': 'V1SlurmV1JobStatus',
56
61
  'state': 'str',
57
62
  'updated_at': 'datetime',
@@ -62,6 +67,7 @@ class V1AgentJob(object):
62
67
  }
63
68
 
64
69
  attribute_map = {
70
+ 'cache_id': 'cacheId',
65
71
  'cloudspace_id': 'cloudspaceId',
66
72
  'cluster_id': 'clusterId',
67
73
  'command': 'command',
@@ -69,10 +75,14 @@ class V1AgentJob(object):
69
75
  'conda_env_path': 'condaEnvPath',
70
76
  'created_at': 'createdAt',
71
77
  'id': 'id',
78
+ 'linked_project_id': 'linkedProjectId',
79
+ 'linked_user_id': 'linkedUserId',
72
80
  'message': 'message',
73
81
  'name': 'name',
82
+ 'num_gpus': 'numGpus',
74
83
  'project_id': 'projectId',
75
84
  'run_id': 'runId',
85
+ 'service_id': 'serviceId',
76
86
  'slurm_v1_status': 'slurmV1Status',
77
87
  'state': 'state',
78
88
  'updated_at': 'updatedAt',
@@ -82,8 +92,9 @@ class V1AgentJob(object):
82
92
  'work_dir': 'workDir'
83
93
  }
84
94
 
85
- def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, conda_checksum: 'str' =None, conda_env_path: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, run_id: 'str' =None, slurm_v1_status: 'V1SlurmV1JobStatus' =None, state: 'str' =None, updated_at: 'datetime' =None, upload_eta_seconds: 'str' =None, upload_in_progress: 'bool' =None, upload_percentage: 'str' =None, work_dir: 'str' =None): # noqa: E501
95
+ def __init__(self, cache_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, conda_checksum: 'str' =None, conda_env_path: 'str' =None, created_at: 'datetime' =None, id: 'str' =None, linked_project_id: 'str' =None, linked_user_id: 'str' =None, message: 'str' =None, name: 'str' =None, num_gpus: 'str' =None, project_id: 'str' =None, run_id: 'str' =None, service_id: 'str' =None, slurm_v1_status: 'V1SlurmV1JobStatus' =None, state: 'str' =None, updated_at: 'datetime' =None, upload_eta_seconds: 'str' =None, upload_in_progress: 'bool' =None, upload_percentage: 'str' =None, work_dir: 'str' =None): # noqa: E501
86
96
  """V1AgentJob - a model defined in Swagger""" # noqa: E501
97
+ self._cache_id = None
87
98
  self._cloudspace_id = None
88
99
  self._cluster_id = None
89
100
  self._command = None
@@ -91,10 +102,14 @@ class V1AgentJob(object):
91
102
  self._conda_env_path = None
92
103
  self._created_at = None
93
104
  self._id = None
105
+ self._linked_project_id = None
106
+ self._linked_user_id = None
94
107
  self._message = None
95
108
  self._name = None
109
+ self._num_gpus = None
96
110
  self._project_id = None
97
111
  self._run_id = None
112
+ self._service_id = None
98
113
  self._slurm_v1_status = None
99
114
  self._state = None
100
115
  self._updated_at = None
@@ -103,6 +118,8 @@ class V1AgentJob(object):
103
118
  self._upload_percentage = None
104
119
  self._work_dir = None
105
120
  self.discriminator = None
121
+ if cache_id is not None:
122
+ self.cache_id = cache_id
106
123
  if cloudspace_id is not None:
107
124
  self.cloudspace_id = cloudspace_id
108
125
  if cluster_id is not None:
@@ -117,14 +134,22 @@ class V1AgentJob(object):
117
134
  self.created_at = created_at
118
135
  if id is not None:
119
136
  self.id = id
137
+ if linked_project_id is not None:
138
+ self.linked_project_id = linked_project_id
139
+ if linked_user_id is not None:
140
+ self.linked_user_id = linked_user_id
120
141
  if message is not None:
121
142
  self.message = message
122
143
  if name is not None:
123
144
  self.name = name
145
+ if num_gpus is not None:
146
+ self.num_gpus = num_gpus
124
147
  if project_id is not None:
125
148
  self.project_id = project_id
126
149
  if run_id is not None:
127
150
  self.run_id = run_id
151
+ if service_id is not None:
152
+ self.service_id = service_id
128
153
  if slurm_v1_status is not None:
129
154
  self.slurm_v1_status = slurm_v1_status
130
155
  if state is not None:
@@ -140,6 +165,27 @@ class V1AgentJob(object):
140
165
  if work_dir is not None:
141
166
  self.work_dir = work_dir
142
167
 
168
+ @property
169
+ def cache_id(self) -> 'str':
170
+ """Gets the cache_id of this V1AgentJob. # noqa: E501
171
+
172
+
173
+ :return: The cache_id of this V1AgentJob. # noqa: E501
174
+ :rtype: str
175
+ """
176
+ return self._cache_id
177
+
178
+ @cache_id.setter
179
+ def cache_id(self, cache_id: 'str'):
180
+ """Sets the cache_id of this V1AgentJob.
181
+
182
+
183
+ :param cache_id: The cache_id of this V1AgentJob. # noqa: E501
184
+ :type: str
185
+ """
186
+
187
+ self._cache_id = cache_id
188
+
143
189
  @property
144
190
  def cloudspace_id(self) -> 'str':
145
191
  """Gets the cloudspace_id of this V1AgentJob. # noqa: E501
@@ -287,6 +333,48 @@ class V1AgentJob(object):
287
333
 
288
334
  self._id = id
289
335
 
336
+ @property
337
+ def linked_project_id(self) -> 'str':
338
+ """Gets the linked_project_id of this V1AgentJob. # noqa: E501
339
+
340
+
341
+ :return: The linked_project_id of this V1AgentJob. # noqa: E501
342
+ :rtype: str
343
+ """
344
+ return self._linked_project_id
345
+
346
+ @linked_project_id.setter
347
+ def linked_project_id(self, linked_project_id: 'str'):
348
+ """Sets the linked_project_id of this V1AgentJob.
349
+
350
+
351
+ :param linked_project_id: The linked_project_id of this V1AgentJob. # noqa: E501
352
+ :type: str
353
+ """
354
+
355
+ self._linked_project_id = linked_project_id
356
+
357
+ @property
358
+ def linked_user_id(self) -> 'str':
359
+ """Gets the linked_user_id of this V1AgentJob. # noqa: E501
360
+
361
+
362
+ :return: The linked_user_id of this V1AgentJob. # noqa: E501
363
+ :rtype: str
364
+ """
365
+ return self._linked_user_id
366
+
367
+ @linked_user_id.setter
368
+ def linked_user_id(self, linked_user_id: 'str'):
369
+ """Sets the linked_user_id of this V1AgentJob.
370
+
371
+
372
+ :param linked_user_id: The linked_user_id of this V1AgentJob. # noqa: E501
373
+ :type: str
374
+ """
375
+
376
+ self._linked_user_id = linked_user_id
377
+
290
378
  @property
291
379
  def message(self) -> 'str':
292
380
  """Gets the message of this V1AgentJob. # noqa: E501
@@ -329,6 +417,27 @@ class V1AgentJob(object):
329
417
 
330
418
  self._name = name
331
419
 
420
+ @property
421
+ def num_gpus(self) -> 'str':
422
+ """Gets the num_gpus of this V1AgentJob. # noqa: E501
423
+
424
+
425
+ :return: The num_gpus of this V1AgentJob. # noqa: E501
426
+ :rtype: str
427
+ """
428
+ return self._num_gpus
429
+
430
+ @num_gpus.setter
431
+ def num_gpus(self, num_gpus: 'str'):
432
+ """Sets the num_gpus of this V1AgentJob.
433
+
434
+
435
+ :param num_gpus: The num_gpus of this V1AgentJob. # noqa: E501
436
+ :type: str
437
+ """
438
+
439
+ self._num_gpus = num_gpus
440
+
332
441
  @property
333
442
  def project_id(self) -> 'str':
334
443
  """Gets the project_id of this V1AgentJob. # noqa: E501
@@ -371,6 +480,27 @@ class V1AgentJob(object):
371
480
 
372
481
  self._run_id = run_id
373
482
 
483
+ @property
484
+ def service_id(self) -> 'str':
485
+ """Gets the service_id of this V1AgentJob. # noqa: E501
486
+
487
+
488
+ :return: The service_id of this V1AgentJob. # noqa: E501
489
+ :rtype: str
490
+ """
491
+ return self._service_id
492
+
493
+ @service_id.setter
494
+ def service_id(self, service_id: 'str'):
495
+ """Sets the service_id of this V1AgentJob.
496
+
497
+
498
+ :param service_id: The service_id of this V1AgentJob. # noqa: E501
499
+ :type: str
500
+ """
501
+
502
+ self._service_id = service_id
503
+
374
504
  @property
375
505
  def slurm_v1_status(self) -> 'V1SlurmV1JobStatus':
376
506
  """Gets the slurm_v1_status of this V1AgentJob. # noqa: E501