lightning-sdk 0.1.2__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 (521) 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 +174 -83
  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 -47
  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 +314 -43
  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 +3115 -895
  42. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +908 -141
  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 +1658 -159
  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 -590
  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 +304 -41
  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/agents_id_body.py +643 -0
  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_body.py +175 -0
  74. lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versionpublications_body1.py +175 -0
  75. lightning_sdk/lightning_cloud/openapi/models/cloud_space_id_versions_body.py +409 -0
  76. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_runs_body.py +27 -1
  77. lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py +84 -6
  78. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityblock_body.py +253 -0
  79. lightning_sdk/lightning_cloud/openapi/models/cluster_id_capacityreservations_body.py +55 -3
  80. lightning_sdk/lightning_cloud/openapi/models/cluster_id_proxies_body.py +123 -0
  81. lightning_sdk/lightning_cloud/openapi/models/clusters_id_body.py +27 -1
  82. lightning_sdk/lightning_cloud/openapi/models/create.py +157 -1
  83. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +383 -0
  84. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +565 -0
  85. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +513 -0
  86. lightning_sdk/lightning_cloud/openapi/models/endpoints_id_body.py +43 -17
  87. lightning_sdk/lightning_cloud/openapi/models/experiment_name_variant_name_body.py +123 -0
  88. lightning_sdk/lightning_cloud/openapi/models/externalv1_cloud_space_instance_status.py +107 -1
  89. lightning_sdk/lightning_cloud/openapi/models/externalv1_lightningapp_instance.py +27 -1
  90. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +27 -1
  91. lightning_sdk/lightning_cloud/openapi/models/fileendpoints_id_body.py +131 -1
  92. lightning_sdk/lightning_cloud/openapi/models/id_codeconfig_body.py +29 -55
  93. lightning_sdk/lightning_cloud/openapi/models/id_complete_body.py +175 -0
  94. lightning_sdk/lightning_cloud/openapi/models/id_engage_body.py +3 -29
  95. lightning_sdk/lightning_cloud/openapi/models/id_engage_body1.py +149 -0
  96. lightning_sdk/lightning_cloud/openapi/models/id_execute_body.py +3 -55
  97. lightning_sdk/lightning_cloud/openapi/models/id_execute_body1.py +175 -0
  98. lightning_sdk/lightning_cloud/openapi/models/id_fork_body.py +31 -31
  99. lightning_sdk/lightning_cloud/openapi/models/id_fork_body1.py +175 -0
  100. lightning_sdk/lightning_cloud/openapi/models/id_get_body.py +133 -3
  101. lightning_sdk/lightning_cloud/openapi/models/id_index_body.py +67 -15
  102. lightning_sdk/lightning_cloud/openapi/models/id_index_body2.py +123 -0
  103. lightning_sdk/lightning_cloud/openapi/models/id_index_body3.py +175 -0
  104. lightning_sdk/lightning_cloud/openapi/models/id_reportlogsactivity_body.py +123 -0
  105. lightning_sdk/lightning_cloud/openapi/models/id_start_body.py +29 -3
  106. lightning_sdk/lightning_cloud/openapi/models/id_storage_body.py +175 -0
  107. lightning_sdk/lightning_cloud/openapi/models/id_uploads_body1.py +175 -0
  108. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body.py +123 -0
  109. lightning_sdk/lightning_cloud/openapi/models/jobs_id_body1.py +27 -1
  110. lightning_sdk/lightning_cloud/openapi/models/jobs_id_body2.py +17 -43
  111. lightning_sdk/lightning_cloud/openapi/models/jobs_id_body3.py +149 -0
  112. lightning_sdk/lightning_cloud/openapi/models/litloggermetrics_id_body.py +175 -0
  113. lightning_sdk/lightning_cloud/openapi/models/litpages_id_body.py +27 -1
  114. lightning_sdk/lightning_cloud/openapi/models/loggermetrics_id_body.py +123 -0
  115. lightning_sdk/lightning_cloud/openapi/models/metrics_stream_id_loggerartifacts_body.py +123 -0
  116. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +383 -0
  117. lightning_sdk/lightning_cloud/openapi/models/{v1_cloud_space_id_list.py → metricsstream_delete_body.py} +10 -10
  118. lightning_sdk/lightning_cloud/openapi/models/metricsstream_id_body.py +175 -0
  119. lightning_sdk/lightning_cloud/openapi/models/{v1_upload_model_response.py → model_id_versions_body.py} +25 -51
  120. lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py +123 -0
  121. lightning_sdk/lightning_cloud/openapi/models/models_model_id_body.py +149 -0
  122. lightning_sdk/lightning_cloud/openapi/models/multimachinejobs_id_body.py +123 -0
  123. lightning_sdk/lightning_cloud/openapi/models/org_id_memberships_body.py +27 -1
  124. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +183 -1
  125. lightning_sdk/lightning_cloud/openapi/models/profiler_captures_body.py +279 -0
  126. lightning_sdk/lightning_cloud/openapi/models/profiler_enabled_body.py +149 -0
  127. lightning_sdk/lightning_cloud/openapi/models/project_id_agentmanagedendpoints_body.py +149 -0
  128. lightning_sdk/lightning_cloud/openapi/models/{project_id_assistants_body.py → project_id_agents_body.py} +181 -51
  129. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +71 -19
  130. lightning_sdk/lightning_cloud/openapi/models/project_id_fileendpoints_body.py +79 -1
  131. lightning_sdk/lightning_cloud/openapi/models/project_id_jobs_body.py +175 -0
  132. lightning_sdk/lightning_cloud/openapi/models/project_id_litregistry_body.py +123 -0
  133. lightning_sdk/lightning_cloud/openapi/models/project_id_memberships_body.py +53 -1
  134. lightning_sdk/lightning_cloud/openapi/models/{v1_upload_model_request.py → project_id_models_body.py} +70 -70
  135. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +227 -0
  136. lightning_sdk/lightning_cloud/openapi/models/project_id_secrets_body.py +27 -1
  137. lightning_sdk/lightning_cloud/openapi/models/project_id_serviceexecution_body.py +175 -0
  138. lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py +123 -0
  139. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +29 -3
  140. lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py +175 -0
  141. lightning_sdk/lightning_cloud/openapi/models/secrets_id_body1.py +123 -0
  142. lightning_sdk/lightning_cloud/openapi/models/servers_server_id_body.py +123 -0
  143. lightning_sdk/lightning_cloud/openapi/models/service_artifact_artifact_kind.py +104 -0
  144. lightning_sdk/lightning_cloud/openapi/models/serviceexecution_id_body.py +331 -0
  145. lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py +79 -1
  146. lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py +305 -0
  147. lightning_sdk/lightning_cloud/openapi/models/{v1_download_model_response.py → snowflake_query_body.py} +51 -51
  148. lightning_sdk/lightning_cloud/openapi/models/storage_complete_body.py +27 -1
  149. lightning_sdk/lightning_cloud/openapi/models/update.py +105 -1
  150. lightning_sdk/lightning_cloud/openapi/models/upload_id_complete_body.py +149 -0
  151. lightning_sdk/lightning_cloud/openapi/models/upload_id_parts_body.py +149 -0
  152. lightning_sdk/lightning_cloud/openapi/models/user_id_affiliatelinks_body.py +149 -0
  153. lightning_sdk/lightning_cloud/openapi/models/v1_accelerator_quota_info.py +201 -0
  154. lightning_sdk/lightning_cloud/openapi/models/v1_ack_user_storage_violation_response.py +97 -0
  155. lightning_sdk/lightning_cloud/openapi/models/v1_add_job_timing_response.py +97 -0
  156. lightning_sdk/lightning_cloud/openapi/models/v1_affiliate_link.py +435 -0
  157. lightning_sdk/lightning_cloud/openapi/models/v1_agent_job.py +131 -1
  158. lightning_sdk/lightning_cloud/openapi/models/v1_api_pricing_spec.py +149 -0
  159. lightning_sdk/lightning_cloud/openapi/models/v1_app_type.py +104 -0
  160. lightning_sdk/lightning_cloud/openapi/models/v1_append_logger_metrics_response.py +97 -0
  161. lightning_sdk/lightning_cloud/openapi/models/v1_approve_auto_join_domain_response.py +123 -0
  162. lightning_sdk/lightning_cloud/openapi/models/v1_assign_variant_response.py +97 -0
  163. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +157 -1
  164. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_knowledge_item_status.py +253 -0
  165. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_knowledge_status.py +123 -0
  166. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +104 -0
  167. lightning_sdk/lightning_cloud/openapi/models/v1_auto_join_domain_validation.py +175 -0
  168. lightning_sdk/lightning_cloud/openapi/models/v1_auto_join_org_response.py +149 -0
  169. lightning_sdk/lightning_cloud/openapi/models/v1_autoscaling_spec.py +305 -0
  170. lightning_sdk/lightning_cloud/openapi/models/v1_autoscaling_target_metric.py +149 -0
  171. lightning_sdk/lightning_cloud/openapi/models/v1_aws_direct_v1.py +133 -3
  172. lightning_sdk/lightning_cloud/openapi/models/v1_batch_update_lightningwork_response.py +97 -0
  173. lightning_sdk/lightning_cloud/openapi/models/v1_body.py +123 -0
  174. lightning_sdk/lightning_cloud/openapi/models/v1_cancellation_metadata.py +149 -0
  175. lightning_sdk/lightning_cloud/openapi/models/v1_capacity_block_offering.py +383 -0
  176. lightning_sdk/lightning_cloud/openapi/models/v1_check_snowflake_connection_response.py +123 -0
  177. lightning_sdk/lightning_cloud/openapi/models/v1_checkbox.py +201 -0
  178. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +162 -6
  179. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_code_version.py +383 -0
  180. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_code_version_status.py +105 -0
  181. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_instance_config.py +1 -53
  182. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_instance_startup_status.py +79 -1
  183. lightning_sdk/lightning_cloud/openapi/models/{v1_cluster_log_service.py → v1_cloud_space_session.py} +49 -49
  184. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version.py +669 -0
  185. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version_publication.py +201 -0
  186. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +445 -3
  187. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +55 -3
  188. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_names.py +123 -0
  189. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_proxy.py +201 -0
  190. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_resource_tag.py +149 -0
  191. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +357 -0
  192. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +157 -107
  193. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_state.py +1 -0
  194. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +17 -43
  195. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py +175 -0
  196. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_type.py +0 -1
  197. lightning_sdk/lightning_cloud/openapi/models/v1_command_argument.py +131 -1
  198. lightning_sdk/lightning_cloud/openapi/models/v1_complete_model_upload_response.py +97 -0
  199. lightning_sdk/lightning_cloud/openapi/models/v1_complete_multi_part_upload_response.py +97 -0
  200. lightning_sdk/lightning_cloud/openapi/models/v1_complete_upload_service_execution_artifact_response.py +97 -0
  201. lightning_sdk/lightning_cloud/openapi/models/v1_completed_part.py +149 -0
  202. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +15 -15
  203. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +26 -26
  204. lightning_sdk/lightning_cloud/openapi/models/v1_count_metrics_streams_response.py +123 -0
  205. lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py +53 -1
  206. lightning_sdk/lightning_cloud/openapi/models/v1_create_cluster_request.py +27 -1
  207. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_template_request.py +539 -0
  208. lightning_sdk/lightning_cloud/openapi/models/v1_create_multi_part_upload_response.py +123 -0
  209. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +159 -3
  210. lightning_sdk/lightning_cloud/openapi/models/v1_create_project_request.py +27 -1
  211. lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_request.py +201 -0
  212. lightning_sdk/lightning_cloud/openapi/models/v1_create_shared_metrics_stream_response.py +123 -0
  213. lightning_sdk/lightning_cloud/openapi/models/v1_create_snowflake_connection_response.py +123 -0
  214. lightning_sdk/lightning_cloud/openapi/models/v1_create_user_secret_request.py +149 -0
  215. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +365 -1
  216. lightning_sdk/lightning_cloud/openapi/models/{v1_cluster_performance_profile.py → v1_data_connection_state.py} +11 -9
  217. lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py +201 -0
  218. lightning_sdk/lightning_cloud/openapi/models/v1_delete_affiliate_link_response.py +97 -0
  219. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cloud_space_session_response.py +97 -0
  220. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cloud_space_version_publication_response.py +97 -0
  221. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cloud_space_version_response.py +97 -0
  222. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_proxy_response.py +97 -0
  223. lightning_sdk/lightning_cloud/openapi/models/v1_delete_deployment_release_response.py +97 -0
  224. lightning_sdk/lightning_cloud/openapi/models/{v1_azure_cluster_driver_status.py → v1_delete_deployment_response.py} +6 -6
  225. lightning_sdk/lightning_cloud/openapi/models/v1_delete_index_response.py +97 -0
  226. lightning_sdk/lightning_cloud/openapi/models/{v1_byom_cluster_driver.py → v1_delete_job_response.py} +6 -6
  227. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_lightningapp_v2_response.py → v1_delete_logger_artifact_response.py} +6 -6
  228. lightning_sdk/lightning_cloud/openapi/models/v1_delete_managed_endpoint_response.py +97 -0
  229. lightning_sdk/lightning_cloud/openapi/models/v1_delete_metrics_stream_response.py +97 -0
  230. lightning_sdk/lightning_cloud/openapi/models/v1_delete_multi_machine_job_response.py +97 -0
  231. lightning_sdk/lightning_cloud/openapi/models/v1_delete_profiler_capture_response.py +97 -0
  232. lightning_sdk/lightning_cloud/openapi/models/v1_delete_service_execution_response.py +97 -0
  233. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_lightningapp_release_response.py → v1_delete_shared_metrics_stream_response.py} +6 -6
  234. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +617 -0
  235. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +227 -0
  236. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event.py +357 -0
  237. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event_type.py +104 -0
  238. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py +149 -0
  239. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_performance.py +305 -0
  240. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_release.py +331 -0
  241. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_spec.py +201 -0
  242. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py +105 -0
  243. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +279 -0
  244. lightning_sdk/lightning_cloud/openapi/models/{appsv2_id_body.py → v1_deployment_strategy.py} +39 -39
  245. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template.py +721 -0
  246. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_engagement_response.py +97 -0
  247. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_gallery_response.py +591 -0
  248. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter.py +435 -0
  249. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_placement.py +106 -0
  250. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_parameter_type.py +106 -0
  251. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template_summary.py +591 -0
  252. lightning_sdk/lightning_cloud/openapi/models/{instance_type_availability.py → v1_deployment_template_type.py} +10 -10
  253. lightning_sdk/lightning_cloud/openapi/models/{v1_get_lightningapp_source_code_download_url_response.py → v1_download_job_logs_response.py} +10 -10
  254. lightning_sdk/lightning_cloud/openapi/models/v1_download_service_execution_artifact_response.py +175 -0
  255. lightning_sdk/lightning_cloud/openapi/models/v1_ebs.py +279 -0
  256. lightning_sdk/lightning_cloud/openapi/models/v1_efs_config.py +201 -0
  257. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint.py +53 -27
  258. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint_auth.py +27 -1
  259. lightning_sdk/lightning_cloud/openapi/models/v1_endpoint_type.py +104 -0
  260. lightning_sdk/lightning_cloud/openapi/models/v1_execute_cloud_space_command_response.py +29 -3
  261. lightning_sdk/lightning_cloud/openapi/models/v1_execute_in_cloud_space_session_response.py +97 -0
  262. lightning_sdk/lightning_cloud/openapi/models/v1_execute_snowflake_query_response.py +149 -0
  263. lightning_sdk/lightning_cloud/openapi/models/v1_experiment.py +409 -0
  264. lightning_sdk/lightning_cloud/openapi/models/v1_export_snowflake_query_response.py +123 -0
  265. lightning_sdk/lightning_cloud/openapi/models/v1_external_search_user.py +279 -0
  266. lightning_sdk/lightning_cloud/openapi/models/v1_file_endpoint.py +131 -1
  267. lightning_sdk/lightning_cloud/openapi/models/{v1_instance_spec.py → v1_filesystem_app.py} +85 -59
  268. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_cloud_space.py +149 -0
  269. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_dataset.py +123 -0
  270. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_job.py +175 -0
  271. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_slurm_job.py +149 -0
  272. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_snowflake_connection.py +123 -0
  273. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_work.py +149 -0
  274. lightning_sdk/lightning_cloud/openapi/models/v1_find_capacity_block_offering_response.py +123 -0
  275. lightning_sdk/lightning_cloud/openapi/models/v1_gcp_data_connection.py +27 -1
  276. lightning_sdk/lightning_cloud/openapi/models/v1_gcs_folder_data_connection.py +123 -0
  277. lightning_sdk/lightning_cloud/openapi/models/v1_get_affiliate_link_response.py +123 -0
  278. lightning_sdk/lightning_cloud/openapi/models/{v1_list_cluster_instance_types_response.py → v1_get_deployment_routing_telemetry_aggregated_response.py} +23 -23
  279. lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_response.py +123 -0
  280. lightning_sdk/lightning_cloud/openapi/models/v1_get_folder_index_response.py +27 -1
  281. lightning_sdk/lightning_cloud/openapi/models/{v1_cluster_driver_status.py → v1_get_job_stats_response.py} +39 -39
  282. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_system_metrics_response.py +123 -0
  283. lightning_sdk/lightning_cloud/openapi/models/v1_get_lightningapp_instance_open_ports_response.py +123 -0
  284. lightning_sdk/lightning_cloud/openapi/models/v1_get_logger_metrics_response.py +123 -0
  285. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_upload_urls_response.py +123 -0
  286. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_url_response.py +175 -0
  287. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +279 -0
  288. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py +149 -0
  289. lightning_sdk/lightning_cloud/openapi/models/{v1_lightningwork_cluster_driver.py → v1_get_project_artifact_response.py} +33 -14
  290. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +27 -1
  291. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_storage_metadata_response.py +383 -0
  292. lightning_sdk/lightning_cloud/openapi/models/v1_get_service_execution_status_response.py +175 -0
  293. lightning_sdk/lightning_cloud/openapi/models/v1_get_snowflake_query_response.py +149 -0
  294. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_balance_response.py +27 -1
  295. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +105 -1
  296. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py +279 -0
  297. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py +201 -0
  298. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +163 -3
  299. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1_status.py +123 -0
  300. lightning_sdk/lightning_cloud/openapi/models/v1_header.py +175 -0
  301. lightning_sdk/lightning_cloud/openapi/models/{v1_external_kubeconfig.py → v1_health_check_exec.py} +21 -21
  302. lightning_sdk/lightning_cloud/openapi/models/v1_health_check_http_get.py +149 -0
  303. lightning_sdk/lightning_cloud/openapi/models/v1_ids_logger_metrics.py +123 -0
  304. lightning_sdk/lightning_cloud/openapi/models/v1_input.py +175 -0
  305. lightning_sdk/lightning_cloud/openapi/models/v1_interrupt_server_response.py +97 -0
  306. lightning_sdk/lightning_cloud/openapi/models/v1_job.py +697 -0
  307. lightning_sdk/lightning_cloud/openapi/models/v1_job_health_check_config.py +253 -0
  308. lightning_sdk/lightning_cloud/openapi/models/v1_job_log_entry.py +175 -0
  309. lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_page.py +227 -0
  310. lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_response.py +149 -0
  311. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +669 -0
  312. lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py +201 -0
  313. lightning_sdk/lightning_cloud/openapi/models/v1_joinable_organization.py +331 -0
  314. lightning_sdk/lightning_cloud/openapi/models/v1_knowledge_configuration.py +279 -0
  315. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +125 -0
  316. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +27 -1
  317. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_instance_spec.py +105 -27
  318. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_instance_status.py +53 -1
  319. lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_spec.py +27 -27
  320. lightning_sdk/lightning_cloud/openapi/models/v1_lightningwork_status.py +81 -3
  321. lightning_sdk/lightning_cloud/openapi/models/v1_list_affiliate_links_response.py +123 -0
  322. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_python_versions_response.py +123 -0
  323. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_sessions_response.py +123 -0
  324. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_tags_response.py +6 -6
  325. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_version_publications_response.py +123 -0
  326. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloud_space_versions_response.py +123 -0
  327. lightning_sdk/lightning_cloud/openapi/models/v1_list_cluster_availabilities_response.py +123 -0
  328. lightning_sdk/lightning_cloud/openapi/models/v1_list_cluster_proxies_response.py +123 -0
  329. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_events_response.py +123 -0
  330. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_releases_response.py +123 -0
  331. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_tags_response.py +123 -0
  332. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_templates_response.py +175 -0
  333. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployments_response.py +123 -0
  334. lightning_sdk/lightning_cloud/openapi/models/v1_list_experiments_response.py +149 -0
  335. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_apps_response.py +123 -0
  336. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_cloud_spaces_response.py +123 -0
  337. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_datasets_response.py +123 -0
  338. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_jobs_response.py +123 -0
  339. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_slurm_jobs_response.py +123 -0
  340. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_snowflake_response.py +123 -0
  341. lightning_sdk/lightning_cloud/openapi/models/{v1_list_lightningapps_v2_response.py → v1_list_jobs_response.py} +31 -31
  342. lightning_sdk/lightning_cloud/openapi/models/v1_list_joinable_organizations_response.py +123 -0
  343. lightning_sdk/lightning_cloud/openapi/models/v1_list_logger_artifact_response.py +123 -0
  344. lightning_sdk/lightning_cloud/openapi/models/v1_list_metrics_streams_response.py +123 -0
  345. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_job_events_response.py +123 -0
  346. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_jobs_response.py +123 -0
  347. lightning_sdk/lightning_cloud/openapi/models/v1_list_new_features_for_user_response.py +123 -0
  348. lightning_sdk/lightning_cloud/openapi/models/v1_list_profiler_captures_response.py +123 -0
  349. lightning_sdk/lightning_cloud/openapi/models/v1_list_project_locked_resources_response.py +123 -0
  350. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_cloud_spaces_response.py +3 -29
  351. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_deployment_templates_response.py +175 -0
  352. lightning_sdk/lightning_cloud/openapi/models/v1_list_service_execution_lightningapp_instances_response.py +175 -0
  353. lightning_sdk/lightning_cloud/openapi/models/v1_list_service_executions_response.py +123 -0
  354. lightning_sdk/lightning_cloud/openapi/models/v1_lit_page.py +27 -1
  355. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +227 -0
  356. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +279 -0
  357. lightning_sdk/lightning_cloud/openapi/models/v1_locked_resource.py +227 -0
  358. lightning_sdk/lightning_cloud/openapi/models/v1_logger_artifact.py +227 -0
  359. lightning_sdk/lightning_cloud/openapi/models/v1_machines_selector.py +149 -0
  360. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +81 -3
  361. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +53 -1
  362. lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py +175 -19
  363. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +305 -0
  364. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py +175 -0
  365. lightning_sdk/lightning_cloud/openapi/models/v1_membership.py +79 -1
  366. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +138 -8
  367. lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py +6 -6
  368. lightning_sdk/lightning_cloud/openapi/models/v1_message_content_type.py +103 -0
  369. lightning_sdk/lightning_cloud/openapi/models/v1_metadata.py +131 -1
  370. lightning_sdk/lightning_cloud/openapi/models/v1_metric_value.py +175 -0
  371. lightning_sdk/lightning_cloud/openapi/models/v1_metrics.py +175 -0
  372. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +799 -0
  373. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_tags.py +201 -0
  374. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_tracker.py +383 -0
  375. lightning_sdk/lightning_cloud/openapi/models/v1_model.py +95 -17
  376. lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py +175 -0
  377. lightning_sdk/lightning_cloud/openapi/models/v1_model_version_archive.py +131 -1
  378. lightning_sdk/lightning_cloud/openapi/models/v1_mount_target.py +201 -0
  379. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +487 -0
  380. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py +331 -0
  381. lightning_sdk/lightning_cloud/openapi/models/{get_cluster_health_response_health_status.py → v1_multi_machine_job_event_type.py} +9 -9
  382. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance.py +149 -0
  383. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance_strategy.py +105 -0
  384. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py +108 -0
  385. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +305 -0
  386. lightning_sdk/lightning_cloud/openapi/models/v1_named_get_logger_metrics.py +123 -0
  387. lightning_sdk/lightning_cloud/openapi/models/v1_new_feature.py +383 -0
  388. lightning_sdk/lightning_cloud/openapi/models/v1_onboarding_event_request.py +175 -0
  389. lightning_sdk/lightning_cloud/openapi/models/v1_onboarding_event_response.py +97 -0
  390. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +209 -1
  391. lightning_sdk/lightning_cloud/openapi/models/v1_parameterization_spec.py +227 -0
  392. lightning_sdk/lightning_cloud/openapi/models/v1_path_telemetry.py +123 -0
  393. lightning_sdk/lightning_cloud/openapi/models/v1_phase_type.py +104 -0
  394. lightning_sdk/lightning_cloud/openapi/models/v1_profiler_capture.py +357 -0
  395. lightning_sdk/lightning_cloud/openapi/models/v1_profiler_enabled_response.py +123 -0
  396. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +131 -1
  397. lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py +263 -3
  398. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +29 -3
  399. lightning_sdk/lightning_cloud/openapi/models/v1_project_storage.py +279 -0
  400. lightning_sdk/lightning_cloud/openapi/models/v1_published_cloud_space_response.py +188 -6
  401. lightning_sdk/lightning_cloud/openapi/models/v1_purchase_capacity_block_response.py +175 -0
  402. lightning_sdk/lightning_cloud/openapi/models/v1_query_param.py +175 -0
  403. lightning_sdk/lightning_cloud/openapi/models/v1_query_result.py +123 -0
  404. lightning_sdk/lightning_cloud/openapi/models/v1_query_result_row.py +123 -0
  405. lightning_sdk/lightning_cloud/openapi/models/v1_refresh_path_response.py +97 -0
  406. lightning_sdk/lightning_cloud/openapi/models/v1_regional_load_balancer.py +149 -0
  407. lightning_sdk/lightning_cloud/openapi/models/v1_report_logs_activity_response.py +97 -0
  408. lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_request.py +175 -0
  409. lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_response.py +97 -0
  410. lightning_sdk/lightning_cloud/openapi/models/{v1_cloud_space_tag.py → v1_resource_tag.py} +18 -18
  411. lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py +123 -0
  412. lightning_sdk/lightning_cloud/openapi/models/v1_resources.py +37 -11
  413. lightning_sdk/lightning_cloud/openapi/models/v1_restore_deployment_release_response.py +97 -0
  414. lightning_sdk/lightning_cloud/openapi/models/v1_rolling_update_strategy.py +149 -0
  415. lightning_sdk/lightning_cloud/openapi/models/v1_rule_condition.py +29 -3
  416. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +11 -0
  417. lightning_sdk/lightning_cloud/openapi/models/v1_s3_folder_data_connection.py +123 -0
  418. lightning_sdk/lightning_cloud/openapi/models/v1_search_job_logs_response.py +149 -0
  419. lightning_sdk/lightning_cloud/openapi/models/v1_search_users_response.py +6 -6
  420. lightning_sdk/lightning_cloud/openapi/models/v1_secret.py +107 -3
  421. lightning_sdk/lightning_cloud/openapi/models/{v1_lightningapp_cluster_driver.py → v1_secret_type.py} +10 -9
  422. lightning_sdk/lightning_cloud/openapi/models/v1_select.py +149 -0
  423. lightning_sdk/lightning_cloud/openapi/models/v1_server_check_in_response.py +97 -0
  424. lightning_sdk/lightning_cloud/openapi/models/{v1_update_cluster_response.py → v1_service_artifact.py} +54 -54
  425. lightning_sdk/lightning_cloud/openapi/models/v1_service_execution.py +383 -0
  426. lightning_sdk/lightning_cloud/openapi/models/v1_should_start_syncing_response.py +123 -0
  427. lightning_sdk/lightning_cloud/openapi/models/v1_signed_url.py +149 -0
  428. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py +131 -1
  429. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_node.py +53 -1
  430. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1.py +27 -1
  431. lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py +253 -0
  432. lightning_sdk/lightning_cloud/openapi/models/v1_status_code_telemetry.py +123 -0
  433. lightning_sdk/lightning_cloud/openapi/models/v1_storage_asset.py +253 -0
  434. lightning_sdk/lightning_cloud/openapi/models/v1_storage_asset_type.py +105 -0
  435. lightning_sdk/lightning_cloud/openapi/models/v1_studio_job_app.py +1 -0
  436. lightning_sdk/lightning_cloud/openapi/models/v1_system_info.py +617 -0
  437. lightning_sdk/lightning_cloud/openapi/models/v1_telemetry.py +331 -0
  438. lightning_sdk/lightning_cloud/openapi/models/v1_timestamp_code_telemetry.py +123 -0
  439. lightning_sdk/lightning_cloud/openapi/models/v1_transaction.py +27 -1
  440. lightning_sdk/lightning_cloud/openapi/models/v1_transfer_user_balance_request.py +27 -1
  441. lightning_sdk/lightning_cloud/openapi/models/v1_update_billing_subscription_request.py +53 -1
  442. lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_accelerators_request.py +149 -0
  443. lightning_sdk/lightning_cloud/openapi/models/v1_update_index_response.py +97 -0
  444. lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py +97 -0
  445. lightning_sdk/lightning_cloud/openapi/models/v1_update_model_visibility_response.py +97 -0
  446. lightning_sdk/lightning_cloud/openapi/models/v1_update_shared_metrics_stream_response.py +97 -0
  447. lightning_sdk/lightning_cloud/openapi/models/v1_update_snowflake_query_response.py +97 -0
  448. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +53 -1
  449. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_viewed_new_features_request.py +123 -0
  450. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_viewed_new_features_response.py +97 -0
  451. lightning_sdk/lightning_cloud/openapi/models/v1_upload_service_execution_artifact_parts_response.py +123 -0
  452. lightning_sdk/lightning_cloud/openapi/models/v1_upload_service_execution_artifact_response.py +149 -0
  453. lightning_sdk/lightning_cloud/openapi/models/v1_upstream_cloud_space.py +97 -19
  454. lightning_sdk/lightning_cloud/openapi/models/v1_upstream_job.py +227 -0
  455. lightning_sdk/lightning_cloud/openapi/models/v1_usage.py +27 -1
  456. lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py +107 -3
  457. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +1030 -302
  458. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +45 -45
  459. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py +29 -29
  460. lightning_sdk/lightning_cloud/openapi/models/v1_validate_assistant_status_response.py +149 -0
  461. lightning_sdk/lightning_cloud/openapi/models/v1_validate_auto_join_domain_response.py +97 -0
  462. lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py +107 -3
  463. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +149 -0
  464. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py +97 -0
  465. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_request.py +175 -0
  466. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py +123 -0
  467. lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_model_response.py +123 -0
  468. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +175 -0
  469. lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py +125 -0
  470. lightning_sdk/lightning_cloud/openapi/models/validate.py +97 -19
  471. lightning_sdk/lightning_cloud/openapi/models/validateautojoindomain_domain_body.py +123 -0
  472. lightning_sdk/lightning_cloud/openapi/models/version_uploads_body.py +123 -0
  473. lightning_sdk/lightning_cloud/openapi/models/versions_id_body.py +357 -0
  474. lightning_sdk/lightning_cloud/rest_client.py +49 -23
  475. lightning_sdk/lightning_cloud/source_code/logs_socket_api.py +1 -1
  476. lightning_sdk/lightning_cloud/source_code/tar.py +1 -3
  477. lightning_sdk/lightning_cloud/utils/data_connection.py +145 -7
  478. lightning_sdk/machine.py +17 -4
  479. lightning_sdk/mmt/__init__.py +4 -0
  480. lightning_sdk/mmt/base.py +288 -0
  481. lightning_sdk/mmt/mmt.py +282 -0
  482. lightning_sdk/mmt/v1.py +185 -0
  483. lightning_sdk/mmt/v2.py +193 -0
  484. lightning_sdk/models.py +153 -0
  485. lightning_sdk/organization.py +1 -1
  486. lightning_sdk/plugin.py +207 -41
  487. lightning_sdk/services/__init__.py +5 -0
  488. lightning_sdk/services/file_endpoint.py +223 -0
  489. lightning_sdk/services/finetune/__init__.py +35 -0
  490. lightning_sdk/services/utilities.py +111 -0
  491. lightning_sdk/status.py +2 -1
  492. lightning_sdk/studio.py +90 -17
  493. lightning_sdk/teamspace.py +189 -11
  494. lightning_sdk/user.py +1 -1
  495. lightning_sdk/utils/__init__.py +0 -0
  496. lightning_sdk/utils/dynamic.py +61 -0
  497. lightning_sdk/utils/enum.py +116 -0
  498. lightning_sdk/{utils.py → utils/resolve.py} +41 -4
  499. lightning_sdk-0.1.46.dist-info/LICENSE +21 -0
  500. {lightning_sdk-0.1.2.dist-info → lightning_sdk-0.1.46.dist-info}/METADATA +30 -4
  501. lightning_sdk-0.1.46.dist-info/RECORD +858 -0
  502. {lightning_sdk-0.1.2.dist-info → lightning_sdk-0.1.46.dist-info}/WHEEL +1 -1
  503. lightning_sdk/lightning_cloud/openapi/models/app_id_releases_body.py +0 -541
  504. lightning_sdk/lightning_cloud/openapi/models/assistants_id_body.py +0 -487
  505. lightning_sdk/lightning_cloud/openapi/models/id_endpoint_body.py +0 -409
  506. lightning_sdk/lightning_cloud/openapi/models/id_get_body1.py +0 -333
  507. lightning_sdk/lightning_cloud/openapi/models/project_id_appsv2_body.py +0 -201
  508. lightning_sdk/lightning_cloud/openapi/models/v1_aws_cluster_driver_spec.py +0 -1039
  509. lightning_sdk/lightning_cloud/openapi/models/v1_aws_cluster_secondary_region_spec.py +0 -253
  510. lightning_sdk/lightning_cloud/openapi/models/v1_azure_cluster_driver_spec.py +0 -227
  511. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_driver.py +0 -175
  512. lightning_sdk/lightning_cloud/openapi/models/v1_container_resources.py +0 -201
  513. lightning_sdk/lightning_cloud/openapi/models/v1_eks_custer_driver_status.py +0 -387
  514. lightning_sdk/lightning_cloud/openapi/models/v1_instance_type.py +0 -305
  515. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_cluster_driver.py +0 -359
  516. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_cluster_status.py +0 -279
  517. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_release.py +0 -697
  518. lightning_sdk/lightning_cloud/openapi/models/v1_lightningapp_v2.py +0 -331
  519. lightning_sdk-0.1.2.dist-info/RECORD +0 -550
  520. {lightning_sdk-0.1.2.dist-info → lightning_sdk-0.1.46.dist-info}/entry_points.txt +0 -0
  521. {lightning_sdk-0.1.2.dist-info → lightning_sdk-0.1.46.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,686 @@
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
+ from __future__ import absolute_import
21
+
22
+ import re # noqa: F401
23
+ from typing import TYPE_CHECKING, Any
24
+
25
+ # python 2 and python 3 compatibility library
26
+ import six
27
+
28
+ from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
29
+
30
+ if TYPE_CHECKING:
31
+ from datetime import datetime
32
+ from lightning_sdk.lightning_cloud.openapi.models import *
33
+
34
+ class SnowflakeServiceApi(object):
35
+ """NOTE: This class is auto generated by the swagger code generator program.
36
+
37
+ Do not edit the class manually.
38
+ Ref: https://github.com/swagger-api/swagger-codegen
39
+ """
40
+
41
+ def __init__(self, api_client=None):
42
+ if api_client is None:
43
+ api_client = ApiClient()
44
+ self.api_client = api_client
45
+
46
+ def snowflake_service_check_snowflake_connection(self, project_id: 'str', **kwargs) -> 'V1CheckSnowflakeConnectionResponse': # noqa: E501
47
+ """snowflake_service_check_snowflake_connection # noqa: E501
48
+
49
+ This method makes a synchronous HTTP request by default. To make an
50
+ asynchronous HTTP request, please pass async_req=True
51
+ >>> thread = api.snowflake_service_check_snowflake_connection(project_id, async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :param str project_id: (required)
56
+ :param str username:
57
+ :param str password:
58
+ :param str account:
59
+ :param str private_key_file:
60
+ :param str private_key_password:
61
+ :return: V1CheckSnowflakeConnectionResponse
62
+ If the method is called asynchronously,
63
+ returns the request thread.
64
+ """
65
+ kwargs['_return_http_data_only'] = True
66
+ if kwargs.get('async_req'):
67
+ return self.snowflake_service_check_snowflake_connection_with_http_info(project_id, **kwargs) # noqa: E501
68
+ else:
69
+ (data) = self.snowflake_service_check_snowflake_connection_with_http_info(project_id, **kwargs) # noqa: E501
70
+ return data
71
+
72
+ def snowflake_service_check_snowflake_connection_with_http_info(self, project_id: 'str', **kwargs) -> 'V1CheckSnowflakeConnectionResponse': # noqa: E501
73
+ """snowflake_service_check_snowflake_connection # noqa: E501
74
+
75
+ This method makes a synchronous HTTP request by default. To make an
76
+ asynchronous HTTP request, please pass async_req=True
77
+ >>> thread = api.snowflake_service_check_snowflake_connection_with_http_info(project_id, async_req=True)
78
+ >>> result = thread.get()
79
+
80
+ :param async_req bool
81
+ :param str project_id: (required)
82
+ :param str username:
83
+ :param str password:
84
+ :param str account:
85
+ :param str private_key_file:
86
+ :param str private_key_password:
87
+ :return: V1CheckSnowflakeConnectionResponse
88
+ If the method is called asynchronously,
89
+ returns the request thread.
90
+ """
91
+
92
+ all_params = ['project_id', 'username', 'password', 'account', 'private_key_file', 'private_key_password'] # noqa: E501
93
+ all_params.append('async_req')
94
+ all_params.append('_return_http_data_only')
95
+ all_params.append('_preload_content')
96
+ all_params.append('_request_timeout')
97
+
98
+ params = locals()
99
+ for key, val in six.iteritems(params['kwargs']):
100
+ if key not in all_params:
101
+ raise TypeError(
102
+ "Got an unexpected keyword argument '%s'"
103
+ " to method snowflake_service_check_snowflake_connection" % key
104
+ )
105
+ params[key] = val
106
+ del params['kwargs']
107
+ # verify the required parameter 'project_id' is set
108
+ if ('project_id' not in params or
109
+ params['project_id'] is None):
110
+ raise ValueError("Missing the required parameter `project_id` when calling `snowflake_service_check_snowflake_connection`") # noqa: E501
111
+
112
+ collection_formats = {}
113
+
114
+ path_params = {}
115
+ if 'project_id' in params:
116
+ path_params['projectId'] = params['project_id'] # noqa: E501
117
+
118
+ query_params = []
119
+ if 'username' in params:
120
+ query_params.append(('username', params['username'])) # noqa: E501
121
+ if 'password' in params:
122
+ query_params.append(('password', params['password'])) # noqa: E501
123
+ if 'account' in params:
124
+ query_params.append(('account', params['account'])) # noqa: E501
125
+ if 'private_key_file' in params:
126
+ query_params.append(('privateKeyFile', params['private_key_file'])) # noqa: E501
127
+ if 'private_key_password' in params:
128
+ query_params.append(('privateKeyPassword', params['private_key_password'])) # noqa: E501
129
+
130
+ header_params = {}
131
+
132
+ form_params = []
133
+ local_var_files = {}
134
+
135
+ body_params = None
136
+ # HTTP header `Accept`
137
+ header_params['Accept'] = self.api_client.select_header_accept(
138
+ ['application/json']) # noqa: E501
139
+
140
+ # Authentication setting
141
+ auth_settings = [] # noqa: E501
142
+
143
+ return self.api_client.call_api(
144
+ '/v1/projects/{projectId}/snowflake', 'GET',
145
+ path_params,
146
+ query_params,
147
+ header_params,
148
+ body=body_params,
149
+ post_params=form_params,
150
+ files=local_var_files,
151
+ response_type='V1CheckSnowflakeConnectionResponse', # noqa: E501
152
+ auth_settings=auth_settings,
153
+ async_req=params.get('async_req'),
154
+ _return_http_data_only=params.get('_return_http_data_only'),
155
+ _preload_content=params.get('_preload_content', True),
156
+ _request_timeout=params.get('_request_timeout'),
157
+ collection_formats=collection_formats)
158
+
159
+ def snowflake_service_create_snowflake_connection(self, body: 'ProjectIdSnowflakeBody', project_id: 'str', **kwargs) -> 'V1CreateSnowflakeConnectionResponse': # noqa: E501
160
+ """snowflake_service_create_snowflake_connection # noqa: E501
161
+
162
+ This method makes a synchronous HTTP request by default. To make an
163
+ asynchronous HTTP request, please pass async_req=True
164
+ >>> thread = api.snowflake_service_create_snowflake_connection(body, project_id, async_req=True)
165
+ >>> result = thread.get()
166
+
167
+ :param async_req bool
168
+ :param ProjectIdSnowflakeBody body: (required)
169
+ :param str project_id: (required)
170
+ :return: V1CreateSnowflakeConnectionResponse
171
+ If the method is called asynchronously,
172
+ returns the request thread.
173
+ """
174
+ kwargs['_return_http_data_only'] = True
175
+ if kwargs.get('async_req'):
176
+ return self.snowflake_service_create_snowflake_connection_with_http_info(body, project_id, **kwargs) # noqa: E501
177
+ else:
178
+ (data) = self.snowflake_service_create_snowflake_connection_with_http_info(body, project_id, **kwargs) # noqa: E501
179
+ return data
180
+
181
+ def snowflake_service_create_snowflake_connection_with_http_info(self, body: 'ProjectIdSnowflakeBody', project_id: 'str', **kwargs) -> 'V1CreateSnowflakeConnectionResponse': # noqa: E501
182
+ """snowflake_service_create_snowflake_connection # noqa: E501
183
+
184
+ This method makes a synchronous HTTP request by default. To make an
185
+ asynchronous HTTP request, please pass async_req=True
186
+ >>> thread = api.snowflake_service_create_snowflake_connection_with_http_info(body, project_id, async_req=True)
187
+ >>> result = thread.get()
188
+
189
+ :param async_req bool
190
+ :param ProjectIdSnowflakeBody body: (required)
191
+ :param str project_id: (required)
192
+ :return: V1CreateSnowflakeConnectionResponse
193
+ If the method is called asynchronously,
194
+ returns the request thread.
195
+ """
196
+
197
+ all_params = ['body', 'project_id'] # noqa: E501
198
+ all_params.append('async_req')
199
+ all_params.append('_return_http_data_only')
200
+ all_params.append('_preload_content')
201
+ all_params.append('_request_timeout')
202
+
203
+ params = locals()
204
+ for key, val in six.iteritems(params['kwargs']):
205
+ if key not in all_params:
206
+ raise TypeError(
207
+ "Got an unexpected keyword argument '%s'"
208
+ " to method snowflake_service_create_snowflake_connection" % key
209
+ )
210
+ params[key] = val
211
+ del params['kwargs']
212
+ # verify the required parameter 'body' is set
213
+ if ('body' not in params or
214
+ params['body'] is None):
215
+ raise ValueError("Missing the required parameter `body` when calling `snowflake_service_create_snowflake_connection`") # noqa: E501
216
+ # verify the required parameter 'project_id' is set
217
+ if ('project_id' not in params or
218
+ params['project_id'] is None):
219
+ raise ValueError("Missing the required parameter `project_id` when calling `snowflake_service_create_snowflake_connection`") # noqa: E501
220
+
221
+ collection_formats = {}
222
+
223
+ path_params = {}
224
+ if 'project_id' in params:
225
+ path_params['projectId'] = params['project_id'] # noqa: E501
226
+
227
+ query_params = []
228
+
229
+ header_params = {}
230
+
231
+ form_params = []
232
+ local_var_files = {}
233
+
234
+ body_params = None
235
+ if 'body' in params:
236
+ body_params = params['body']
237
+ # HTTP header `Accept`
238
+ header_params['Accept'] = self.api_client.select_header_accept(
239
+ ['application/json']) # noqa: E501
240
+
241
+ # HTTP header `Content-Type`
242
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
243
+ ['application/json']) # noqa: E501
244
+
245
+ # Authentication setting
246
+ auth_settings = [] # noqa: E501
247
+
248
+ return self.api_client.call_api(
249
+ '/v1/projects/{projectId}/snowflake', 'POST',
250
+ path_params,
251
+ query_params,
252
+ header_params,
253
+ body=body_params,
254
+ post_params=form_params,
255
+ files=local_var_files,
256
+ response_type='V1CreateSnowflakeConnectionResponse', # noqa: E501
257
+ auth_settings=auth_settings,
258
+ async_req=params.get('async_req'),
259
+ _return_http_data_only=params.get('_return_http_data_only'),
260
+ _preload_content=params.get('_preload_content', True),
261
+ _request_timeout=params.get('_request_timeout'),
262
+ collection_formats=collection_formats)
263
+
264
+ def snowflake_service_execute_snowflake_query(self, body: 'SnowflakeQueryBody', project_id: 'str', **kwargs) -> 'V1ExecuteSnowflakeQueryResponse': # noqa: E501
265
+ """snowflake_service_execute_snowflake_query # noqa: E501
266
+
267
+ This method makes a synchronous HTTP request by default. To make an
268
+ asynchronous HTTP request, please pass async_req=True
269
+ >>> thread = api.snowflake_service_execute_snowflake_query(body, project_id, async_req=True)
270
+ >>> result = thread.get()
271
+
272
+ :param async_req bool
273
+ :param SnowflakeQueryBody body: (required)
274
+ :param str project_id: (required)
275
+ :return: V1ExecuteSnowflakeQueryResponse
276
+ If the method is called asynchronously,
277
+ returns the request thread.
278
+ """
279
+ kwargs['_return_http_data_only'] = True
280
+ if kwargs.get('async_req'):
281
+ return self.snowflake_service_execute_snowflake_query_with_http_info(body, project_id, **kwargs) # noqa: E501
282
+ else:
283
+ (data) = self.snowflake_service_execute_snowflake_query_with_http_info(body, project_id, **kwargs) # noqa: E501
284
+ return data
285
+
286
+ def snowflake_service_execute_snowflake_query_with_http_info(self, body: 'SnowflakeQueryBody', project_id: 'str', **kwargs) -> 'V1ExecuteSnowflakeQueryResponse': # noqa: E501
287
+ """snowflake_service_execute_snowflake_query # noqa: E501
288
+
289
+ This method makes a synchronous HTTP request by default. To make an
290
+ asynchronous HTTP request, please pass async_req=True
291
+ >>> thread = api.snowflake_service_execute_snowflake_query_with_http_info(body, project_id, async_req=True)
292
+ >>> result = thread.get()
293
+
294
+ :param async_req bool
295
+ :param SnowflakeQueryBody body: (required)
296
+ :param str project_id: (required)
297
+ :return: V1ExecuteSnowflakeQueryResponse
298
+ If the method is called asynchronously,
299
+ returns the request thread.
300
+ """
301
+
302
+ all_params = ['body', 'project_id'] # noqa: E501
303
+ all_params.append('async_req')
304
+ all_params.append('_return_http_data_only')
305
+ all_params.append('_preload_content')
306
+ all_params.append('_request_timeout')
307
+
308
+ params = locals()
309
+ for key, val in six.iteritems(params['kwargs']):
310
+ if key not in all_params:
311
+ raise TypeError(
312
+ "Got an unexpected keyword argument '%s'"
313
+ " to method snowflake_service_execute_snowflake_query" % key
314
+ )
315
+ params[key] = val
316
+ del params['kwargs']
317
+ # verify the required parameter 'body' is set
318
+ if ('body' not in params or
319
+ params['body'] is None):
320
+ raise ValueError("Missing the required parameter `body` when calling `snowflake_service_execute_snowflake_query`") # noqa: E501
321
+ # verify the required parameter 'project_id' is set
322
+ if ('project_id' not in params or
323
+ params['project_id'] is None):
324
+ raise ValueError("Missing the required parameter `project_id` when calling `snowflake_service_execute_snowflake_query`") # noqa: E501
325
+
326
+ collection_formats = {}
327
+
328
+ path_params = {}
329
+ if 'project_id' in params:
330
+ path_params['projectId'] = params['project_id'] # noqa: E501
331
+
332
+ query_params = []
333
+
334
+ header_params = {}
335
+
336
+ form_params = []
337
+ local_var_files = {}
338
+
339
+ body_params = None
340
+ if 'body' in params:
341
+ body_params = params['body']
342
+ # HTTP header `Accept`
343
+ header_params['Accept'] = self.api_client.select_header_accept(
344
+ ['application/json']) # noqa: E501
345
+
346
+ # HTTP header `Content-Type`
347
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
348
+ ['application/json']) # noqa: E501
349
+
350
+ # Authentication setting
351
+ auth_settings = [] # noqa: E501
352
+
353
+ return self.api_client.call_api(
354
+ '/v1/projects/{projectId}/snowflake/query', 'POST',
355
+ path_params,
356
+ query_params,
357
+ header_params,
358
+ body=body_params,
359
+ post_params=form_params,
360
+ files=local_var_files,
361
+ response_type='V1ExecuteSnowflakeQueryResponse', # noqa: E501
362
+ auth_settings=auth_settings,
363
+ async_req=params.get('async_req'),
364
+ _return_http_data_only=params.get('_return_http_data_only'),
365
+ _preload_content=params.get('_preload_content', True),
366
+ _request_timeout=params.get('_request_timeout'),
367
+ collection_formats=collection_formats)
368
+
369
+ def snowflake_service_export_snowflake_query(self, body: 'SnowflakeExportBody', project_id: 'str', **kwargs) -> 'V1ExportSnowflakeQueryResponse': # noqa: E501
370
+ """snowflake_service_export_snowflake_query # noqa: E501
371
+
372
+ This method makes a synchronous HTTP request by default. To make an
373
+ asynchronous HTTP request, please pass async_req=True
374
+ >>> thread = api.snowflake_service_export_snowflake_query(body, project_id, async_req=True)
375
+ >>> result = thread.get()
376
+
377
+ :param async_req bool
378
+ :param SnowflakeExportBody body: (required)
379
+ :param str project_id: (required)
380
+ :return: V1ExportSnowflakeQueryResponse
381
+ If the method is called asynchronously,
382
+ returns the request thread.
383
+ """
384
+ kwargs['_return_http_data_only'] = True
385
+ if kwargs.get('async_req'):
386
+ return self.snowflake_service_export_snowflake_query_with_http_info(body, project_id, **kwargs) # noqa: E501
387
+ else:
388
+ (data) = self.snowflake_service_export_snowflake_query_with_http_info(body, project_id, **kwargs) # noqa: E501
389
+ return data
390
+
391
+ def snowflake_service_export_snowflake_query_with_http_info(self, body: 'SnowflakeExportBody', project_id: 'str', **kwargs) -> 'V1ExportSnowflakeQueryResponse': # noqa: E501
392
+ """snowflake_service_export_snowflake_query # noqa: E501
393
+
394
+ This method makes a synchronous HTTP request by default. To make an
395
+ asynchronous HTTP request, please pass async_req=True
396
+ >>> thread = api.snowflake_service_export_snowflake_query_with_http_info(body, project_id, async_req=True)
397
+ >>> result = thread.get()
398
+
399
+ :param async_req bool
400
+ :param SnowflakeExportBody body: (required)
401
+ :param str project_id: (required)
402
+ :return: V1ExportSnowflakeQueryResponse
403
+ If the method is called asynchronously,
404
+ returns the request thread.
405
+ """
406
+
407
+ all_params = ['body', 'project_id'] # noqa: E501
408
+ all_params.append('async_req')
409
+ all_params.append('_return_http_data_only')
410
+ all_params.append('_preload_content')
411
+ all_params.append('_request_timeout')
412
+
413
+ params = locals()
414
+ for key, val in six.iteritems(params['kwargs']):
415
+ if key not in all_params:
416
+ raise TypeError(
417
+ "Got an unexpected keyword argument '%s'"
418
+ " to method snowflake_service_export_snowflake_query" % key
419
+ )
420
+ params[key] = val
421
+ del params['kwargs']
422
+ # verify the required parameter 'body' is set
423
+ if ('body' not in params or
424
+ params['body'] is None):
425
+ raise ValueError("Missing the required parameter `body` when calling `snowflake_service_export_snowflake_query`") # noqa: E501
426
+ # verify the required parameter 'project_id' is set
427
+ if ('project_id' not in params or
428
+ params['project_id'] is None):
429
+ raise ValueError("Missing the required parameter `project_id` when calling `snowflake_service_export_snowflake_query`") # noqa: E501
430
+
431
+ collection_formats = {}
432
+
433
+ path_params = {}
434
+ if 'project_id' in params:
435
+ path_params['projectId'] = params['project_id'] # noqa: E501
436
+
437
+ query_params = []
438
+
439
+ header_params = {}
440
+
441
+ form_params = []
442
+ local_var_files = {}
443
+
444
+ body_params = None
445
+ if 'body' in params:
446
+ body_params = params['body']
447
+ # HTTP header `Accept`
448
+ header_params['Accept'] = self.api_client.select_header_accept(
449
+ ['application/json']) # noqa: E501
450
+
451
+ # HTTP header `Content-Type`
452
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
453
+ ['application/json']) # noqa: E501
454
+
455
+ # Authentication setting
456
+ auth_settings = [] # noqa: E501
457
+
458
+ return self.api_client.call_api(
459
+ '/v1/projects/{projectId}/snowflake/export', 'POST',
460
+ path_params,
461
+ query_params,
462
+ header_params,
463
+ body=body_params,
464
+ post_params=form_params,
465
+ files=local_var_files,
466
+ response_type='V1ExportSnowflakeQueryResponse', # noqa: E501
467
+ auth_settings=auth_settings,
468
+ async_req=params.get('async_req'),
469
+ _return_http_data_only=params.get('_return_http_data_only'),
470
+ _preload_content=params.get('_preload_content', True),
471
+ _request_timeout=params.get('_request_timeout'),
472
+ collection_formats=collection_formats)
473
+
474
+ def snowflake_service_get_snowflake_query(self, project_id: 'str', query_id: 'str', **kwargs) -> 'V1GetSnowflakeQueryResponse': # noqa: E501
475
+ """snowflake_service_get_snowflake_query # noqa: E501
476
+
477
+ This method makes a synchronous HTTP request by default. To make an
478
+ asynchronous HTTP request, please pass async_req=True
479
+ >>> thread = api.snowflake_service_get_snowflake_query(project_id, query_id, async_req=True)
480
+ >>> result = thread.get()
481
+
482
+ :param async_req bool
483
+ :param str project_id: (required)
484
+ :param str query_id: (required)
485
+ :return: V1GetSnowflakeQueryResponse
486
+ If the method is called asynchronously,
487
+ returns the request thread.
488
+ """
489
+ kwargs['_return_http_data_only'] = True
490
+ if kwargs.get('async_req'):
491
+ return self.snowflake_service_get_snowflake_query_with_http_info(project_id, query_id, **kwargs) # noqa: E501
492
+ else:
493
+ (data) = self.snowflake_service_get_snowflake_query_with_http_info(project_id, query_id, **kwargs) # noqa: E501
494
+ return data
495
+
496
+ def snowflake_service_get_snowflake_query_with_http_info(self, project_id: 'str', query_id: 'str', **kwargs) -> 'V1GetSnowflakeQueryResponse': # noqa: E501
497
+ """snowflake_service_get_snowflake_query # noqa: E501
498
+
499
+ This method makes a synchronous HTTP request by default. To make an
500
+ asynchronous HTTP request, please pass async_req=True
501
+ >>> thread = api.snowflake_service_get_snowflake_query_with_http_info(project_id, query_id, async_req=True)
502
+ >>> result = thread.get()
503
+
504
+ :param async_req bool
505
+ :param str project_id: (required)
506
+ :param str query_id: (required)
507
+ :return: V1GetSnowflakeQueryResponse
508
+ If the method is called asynchronously,
509
+ returns the request thread.
510
+ """
511
+
512
+ all_params = ['project_id', 'query_id'] # noqa: E501
513
+ all_params.append('async_req')
514
+ all_params.append('_return_http_data_only')
515
+ all_params.append('_preload_content')
516
+ all_params.append('_request_timeout')
517
+
518
+ params = locals()
519
+ for key, val in six.iteritems(params['kwargs']):
520
+ if key not in all_params:
521
+ raise TypeError(
522
+ "Got an unexpected keyword argument '%s'"
523
+ " to method snowflake_service_get_snowflake_query" % key
524
+ )
525
+ params[key] = val
526
+ del params['kwargs']
527
+ # verify the required parameter 'project_id' is set
528
+ if ('project_id' not in params or
529
+ params['project_id'] is None):
530
+ raise ValueError("Missing the required parameter `project_id` when calling `snowflake_service_get_snowflake_query`") # noqa: E501
531
+ # verify the required parameter 'query_id' is set
532
+ if ('query_id' not in params or
533
+ params['query_id'] is None):
534
+ raise ValueError("Missing the required parameter `query_id` when calling `snowflake_service_get_snowflake_query`") # noqa: E501
535
+
536
+ collection_formats = {}
537
+
538
+ path_params = {}
539
+ if 'project_id' in params:
540
+ path_params['projectId'] = params['project_id'] # noqa: E501
541
+ if 'query_id' in params:
542
+ path_params['queryId'] = params['query_id'] # noqa: E501
543
+
544
+ query_params = []
545
+
546
+ header_params = {}
547
+
548
+ form_params = []
549
+ local_var_files = {}
550
+
551
+ body_params = None
552
+ # HTTP header `Accept`
553
+ header_params['Accept'] = self.api_client.select_header_accept(
554
+ ['application/json']) # noqa: E501
555
+
556
+ # Authentication setting
557
+ auth_settings = [] # noqa: E501
558
+
559
+ return self.api_client.call_api(
560
+ '/v1/projects/{projectId}/snowflake/query/{queryId}', 'GET',
561
+ path_params,
562
+ query_params,
563
+ header_params,
564
+ body=body_params,
565
+ post_params=form_params,
566
+ files=local_var_files,
567
+ response_type='V1GetSnowflakeQueryResponse', # noqa: E501
568
+ auth_settings=auth_settings,
569
+ async_req=params.get('async_req'),
570
+ _return_http_data_only=params.get('_return_http_data_only'),
571
+ _preload_content=params.get('_preload_content', True),
572
+ _request_timeout=params.get('_request_timeout'),
573
+ collection_formats=collection_formats)
574
+
575
+ def snowflake_service_update_snowflake_query(self, body: 'QueryQueryIdBody', project_id: 'str', query_id: 'str', **kwargs) -> 'V1UpdateSnowflakeQueryResponse': # noqa: E501
576
+ """snowflake_service_update_snowflake_query # noqa: E501
577
+
578
+ This method makes a synchronous HTTP request by default. To make an
579
+ asynchronous HTTP request, please pass async_req=True
580
+ >>> thread = api.snowflake_service_update_snowflake_query(body, project_id, query_id, async_req=True)
581
+ >>> result = thread.get()
582
+
583
+ :param async_req bool
584
+ :param QueryQueryIdBody body: (required)
585
+ :param str project_id: (required)
586
+ :param str query_id: (required)
587
+ :return: V1UpdateSnowflakeQueryResponse
588
+ If the method is called asynchronously,
589
+ returns the request thread.
590
+ """
591
+ kwargs['_return_http_data_only'] = True
592
+ if kwargs.get('async_req'):
593
+ return self.snowflake_service_update_snowflake_query_with_http_info(body, project_id, query_id, **kwargs) # noqa: E501
594
+ else:
595
+ (data) = self.snowflake_service_update_snowflake_query_with_http_info(body, project_id, query_id, **kwargs) # noqa: E501
596
+ return data
597
+
598
+ def snowflake_service_update_snowflake_query_with_http_info(self, body: 'QueryQueryIdBody', project_id: 'str', query_id: 'str', **kwargs) -> 'V1UpdateSnowflakeQueryResponse': # noqa: E501
599
+ """snowflake_service_update_snowflake_query # noqa: E501
600
+
601
+ This method makes a synchronous HTTP request by default. To make an
602
+ asynchronous HTTP request, please pass async_req=True
603
+ >>> thread = api.snowflake_service_update_snowflake_query_with_http_info(body, project_id, query_id, async_req=True)
604
+ >>> result = thread.get()
605
+
606
+ :param async_req bool
607
+ :param QueryQueryIdBody body: (required)
608
+ :param str project_id: (required)
609
+ :param str query_id: (required)
610
+ :return: V1UpdateSnowflakeQueryResponse
611
+ If the method is called asynchronously,
612
+ returns the request thread.
613
+ """
614
+
615
+ all_params = ['body', 'project_id', 'query_id'] # noqa: E501
616
+ all_params.append('async_req')
617
+ all_params.append('_return_http_data_only')
618
+ all_params.append('_preload_content')
619
+ all_params.append('_request_timeout')
620
+
621
+ params = locals()
622
+ for key, val in six.iteritems(params['kwargs']):
623
+ if key not in all_params:
624
+ raise TypeError(
625
+ "Got an unexpected keyword argument '%s'"
626
+ " to method snowflake_service_update_snowflake_query" % key
627
+ )
628
+ params[key] = val
629
+ del params['kwargs']
630
+ # verify the required parameter 'body' is set
631
+ if ('body' not in params or
632
+ params['body'] is None):
633
+ raise ValueError("Missing the required parameter `body` when calling `snowflake_service_update_snowflake_query`") # noqa: E501
634
+ # verify the required parameter 'project_id' is set
635
+ if ('project_id' not in params or
636
+ params['project_id'] is None):
637
+ raise ValueError("Missing the required parameter `project_id` when calling `snowflake_service_update_snowflake_query`") # noqa: E501
638
+ # verify the required parameter 'query_id' is set
639
+ if ('query_id' not in params or
640
+ params['query_id'] is None):
641
+ raise ValueError("Missing the required parameter `query_id` when calling `snowflake_service_update_snowflake_query`") # noqa: E501
642
+
643
+ collection_formats = {}
644
+
645
+ path_params = {}
646
+ if 'project_id' in params:
647
+ path_params['projectId'] = params['project_id'] # noqa: E501
648
+ if 'query_id' in params:
649
+ path_params['queryId'] = params['query_id'] # noqa: E501
650
+
651
+ query_params = []
652
+
653
+ header_params = {}
654
+
655
+ form_params = []
656
+ local_var_files = {}
657
+
658
+ body_params = None
659
+ if 'body' in params:
660
+ body_params = params['body']
661
+ # HTTP header `Accept`
662
+ header_params['Accept'] = self.api_client.select_header_accept(
663
+ ['application/json']) # noqa: E501
664
+
665
+ # HTTP header `Content-Type`
666
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
667
+ ['application/json']) # noqa: E501
668
+
669
+ # Authentication setting
670
+ auth_settings = [] # noqa: E501
671
+
672
+ return self.api_client.call_api(
673
+ '/v1/projects/{projectId}/snowflake/query/{queryId}', 'PUT',
674
+ path_params,
675
+ query_params,
676
+ header_params,
677
+ body=body_params,
678
+ post_params=form_params,
679
+ files=local_var_files,
680
+ response_type='V1UpdateSnowflakeQueryResponse', # noqa: E501
681
+ auth_settings=auth_settings,
682
+ async_req=params.get('async_req'),
683
+ _return_http_data_only=params.get('_return_http_data_only'),
684
+ _preload_content=params.get('_preload_content', True),
685
+ _request_timeout=params.get('_request_timeout'),
686
+ collection_formats=collection_formats)