skypilot-nightly 1.0.0.dev20250502__py3-none-any.whl → 1.0.0.dev20251203__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.
- sky/__init__.py +22 -6
- sky/adaptors/aws.py +81 -16
- sky/adaptors/common.py +25 -2
- sky/adaptors/coreweave.py +278 -0
- sky/adaptors/do.py +8 -2
- sky/adaptors/gcp.py +11 -0
- sky/adaptors/hyperbolic.py +8 -0
- sky/adaptors/ibm.py +5 -2
- sky/adaptors/kubernetes.py +149 -18
- sky/adaptors/nebius.py +173 -30
- sky/adaptors/primeintellect.py +1 -0
- sky/adaptors/runpod.py +68 -0
- sky/adaptors/seeweb.py +183 -0
- sky/adaptors/shadeform.py +89 -0
- sky/admin_policy.py +187 -4
- sky/authentication.py +179 -225
- sky/backends/__init__.py +4 -2
- sky/backends/backend.py +22 -9
- sky/backends/backend_utils.py +1323 -397
- sky/backends/cloud_vm_ray_backend.py +1749 -1029
- sky/backends/docker_utils.py +1 -1
- sky/backends/local_docker_backend.py +11 -6
- sky/backends/task_codegen.py +633 -0
- sky/backends/wheel_utils.py +55 -9
- sky/{clouds/service_catalog → catalog}/__init__.py +21 -19
- sky/{clouds/service_catalog → catalog}/aws_catalog.py +27 -8
- sky/{clouds/service_catalog → catalog}/azure_catalog.py +10 -7
- sky/{clouds/service_catalog → catalog}/common.py +90 -49
- sky/{clouds/service_catalog → catalog}/cudo_catalog.py +8 -5
- sky/{clouds/service_catalog → catalog}/data_fetchers/analyze.py +1 -1
- sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_aws.py +116 -80
- sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_cudo.py +38 -38
- sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_gcp.py +70 -16
- sky/catalog/data_fetchers/fetch_hyperbolic.py +136 -0
- sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_lambda_cloud.py +1 -0
- sky/catalog/data_fetchers/fetch_nebius.py +338 -0
- sky/catalog/data_fetchers/fetch_runpod.py +698 -0
- sky/catalog/data_fetchers/fetch_seeweb.py +329 -0
- sky/catalog/data_fetchers/fetch_shadeform.py +142 -0
- sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_vast.py +1 -1
- sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_vsphere.py +1 -1
- sky/{clouds/service_catalog → catalog}/do_catalog.py +5 -2
- sky/{clouds/service_catalog → catalog}/fluidstack_catalog.py +6 -3
- sky/{clouds/service_catalog → catalog}/gcp_catalog.py +41 -15
- sky/catalog/hyperbolic_catalog.py +136 -0
- sky/{clouds/service_catalog → catalog}/ibm_catalog.py +9 -6
- sky/{clouds/service_catalog → catalog}/kubernetes_catalog.py +36 -24
- sky/{clouds/service_catalog → catalog}/lambda_catalog.py +9 -6
- sky/{clouds/service_catalog → catalog}/nebius_catalog.py +9 -7
- sky/{clouds/service_catalog → catalog}/oci_catalog.py +9 -6
- sky/{clouds/service_catalog → catalog}/paperspace_catalog.py +5 -2
- sky/catalog/primeintellect_catalog.py +95 -0
- sky/{clouds/service_catalog → catalog}/runpod_catalog.py +11 -4
- sky/{clouds/service_catalog → catalog}/scp_catalog.py +9 -6
- sky/catalog/seeweb_catalog.py +184 -0
- sky/catalog/shadeform_catalog.py +165 -0
- sky/catalog/ssh_catalog.py +167 -0
- sky/{clouds/service_catalog → catalog}/vast_catalog.py +6 -3
- sky/{clouds/service_catalog → catalog}/vsphere_catalog.py +5 -2
- sky/check.py +533 -185
- sky/cli.py +5 -5975
- sky/client/{cli.py → cli/command.py} +2591 -1956
- sky/client/cli/deprecation_utils.py +99 -0
- sky/client/cli/flags.py +359 -0
- sky/client/cli/table_utils.py +322 -0
- sky/client/cli/utils.py +79 -0
- sky/client/common.py +78 -32
- sky/client/oauth.py +82 -0
- sky/client/sdk.py +1219 -319
- sky/client/sdk_async.py +827 -0
- sky/client/service_account_auth.py +47 -0
- sky/cloud_stores.py +82 -3
- sky/clouds/__init__.py +13 -0
- sky/clouds/aws.py +564 -164
- sky/clouds/azure.py +105 -83
- sky/clouds/cloud.py +140 -40
- sky/clouds/cudo.py +68 -50
- sky/clouds/do.py +66 -48
- sky/clouds/fluidstack.py +63 -44
- sky/clouds/gcp.py +339 -110
- sky/clouds/hyperbolic.py +293 -0
- sky/clouds/ibm.py +70 -49
- sky/clouds/kubernetes.py +570 -162
- sky/clouds/lambda_cloud.py +74 -54
- sky/clouds/nebius.py +210 -81
- sky/clouds/oci.py +88 -66
- sky/clouds/paperspace.py +61 -44
- sky/clouds/primeintellect.py +317 -0
- sky/clouds/runpod.py +164 -74
- sky/clouds/scp.py +89 -86
- sky/clouds/seeweb.py +477 -0
- sky/clouds/shadeform.py +400 -0
- sky/clouds/ssh.py +263 -0
- sky/clouds/utils/aws_utils.py +10 -4
- sky/clouds/utils/gcp_utils.py +87 -11
- sky/clouds/utils/oci_utils.py +38 -14
- sky/clouds/utils/scp_utils.py +231 -167
- sky/clouds/vast.py +99 -77
- sky/clouds/vsphere.py +51 -40
- sky/core.py +375 -173
- sky/dag.py +15 -0
- sky/dashboard/out/404.html +1 -1
- sky/dashboard/out/_next/static/96_E2yl3QAiIJGOYCkSpB/_buildManifest.js +1 -0
- sky/dashboard/out/_next/static/chunks/1141-e6aa9ab418717c59.js +11 -0
- sky/dashboard/out/_next/static/chunks/1272-1ef0bf0237faccdb.js +1 -0
- sky/dashboard/out/_next/static/chunks/1871-7e202677c42f43fe.js +6 -0
- sky/dashboard/out/_next/static/chunks/2260-7703229c33c5ebd5.js +1 -0
- sky/dashboard/out/_next/static/chunks/2350.fab69e61bac57b23.js +1 -0
- sky/dashboard/out/_next/static/chunks/2369.fc20f0c2c8ed9fe7.js +15 -0
- sky/dashboard/out/_next/static/chunks/2755.edd818326d489a1d.js +26 -0
- sky/dashboard/out/_next/static/chunks/3294.20a8540fe697d5ee.js +1 -0
- sky/dashboard/out/_next/static/chunks/3785.7e245f318f9d1121.js +1 -0
- sky/dashboard/out/_next/static/chunks/3800-7b45f9fbb6308557.js +1 -0
- sky/dashboard/out/_next/static/chunks/3850-ff4a9a69d978632b.js +1 -0
- sky/dashboard/out/_next/static/chunks/3937.210053269f121201.js +1 -0
- sky/dashboard/out/_next/static/chunks/4725.172ede95d1b21022.js +1 -0
- sky/dashboard/out/_next/static/chunks/4937.a2baa2df5572a276.js +15 -0
- sky/dashboard/out/_next/static/chunks/5739-d67458fcb1386c92.js +8 -0
- sky/dashboard/out/_next/static/chunks/6130-2be46d70a38f1e82.js +1 -0
- sky/dashboard/out/_next/static/chunks/616-3d59f75e2ccf9321.js +39 -0
- sky/dashboard/out/_next/static/chunks/6212-7bd06f60ba693125.js +13 -0
- sky/dashboard/out/_next/static/chunks/6856-8f27d1c10c98def8.js +1 -0
- sky/dashboard/out/_next/static/chunks/6989-01359c57e018caa4.js +1 -0
- sky/dashboard/out/_next/static/chunks/6990-9146207c4567fdfd.js +1 -0
- sky/dashboard/out/_next/static/chunks/7359-c8d04e06886000b3.js +30 -0
- sky/dashboard/out/_next/static/chunks/7411-b15471acd2cba716.js +41 -0
- sky/dashboard/out/_next/static/chunks/7615-019513abc55b3b47.js +1 -0
- sky/dashboard/out/_next/static/chunks/8640.5b9475a2d18c5416.js +16 -0
- sky/dashboard/out/_next/static/chunks/8969-452f9d5cbdd2dc73.js +1 -0
- sky/dashboard/out/_next/static/chunks/9025.fa408f3242e9028d.js +6 -0
- sky/dashboard/out/_next/static/chunks/9353-cff34f7e773b2e2b.js +1 -0
- sky/dashboard/out/_next/static/chunks/9360.a536cf6b1fa42355.js +31 -0
- sky/dashboard/out/_next/static/chunks/9847.3aaca6bb33455140.js +30 -0
- sky/dashboard/out/_next/static/chunks/fd9d1056-86323a29a8f7e46a.js +1 -0
- sky/dashboard/out/_next/static/chunks/framework-cf60a09ccd051a10.js +33 -0
- sky/dashboard/out/_next/static/chunks/main-app-587214043926b3cc.js +1 -0
- sky/dashboard/out/_next/static/chunks/main-f15ccb73239a3bf1.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/_app-bde01e4a2beec258.js +34 -0
- sky/dashboard/out/_next/static/chunks/pages/_error-c66a4e8afc46f17b.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/clusters/[cluster]/[job]-792db96d918c98c9.js +16 -0
- sky/dashboard/out/_next/static/chunks/pages/clusters/[cluster]-abfcac9c137aa543.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/clusters-ee39056f9851a3ff.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/config-dfb9bf07b13045f4.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/index-444f1804401f04ea.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/infra/[context]-c0b5935149902e6f.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/infra-aed0ea19df7cf961.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/jobs/[job]-d66997e2bfc837cf.js +16 -0
- sky/dashboard/out/_next/static/chunks/pages/jobs/pools/[pool]-9faf940b253e3e06.js +21 -0
- sky/dashboard/out/_next/static/chunks/pages/jobs-2072b48b617989c9.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/users-f42674164aa73423.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/volumes-b84b948ff357c43e.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/workspace/new-3f88a1c7e86a3f86.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/workspaces/[name]-84a40f8c7c627fe4.js +1 -0
- sky/dashboard/out/_next/static/chunks/pages/workspaces-531b2f8c4bf89f82.js +1 -0
- sky/dashboard/out/_next/static/chunks/webpack-64e05f17bf2cf8ce.js +1 -0
- sky/dashboard/out/_next/static/css/0748ce22df867032.css +3 -0
- sky/dashboard/out/clusters/[cluster]/[job].html +1 -1
- sky/dashboard/out/clusters/[cluster].html +1 -1
- sky/dashboard/out/clusters.html +1 -1
- sky/dashboard/out/config.html +1 -0
- sky/dashboard/out/index.html +1 -1
- sky/dashboard/out/infra/[context].html +1 -0
- sky/dashboard/out/infra.html +1 -0
- sky/dashboard/out/jobs/[job].html +1 -1
- sky/dashboard/out/jobs/pools/[pool].html +1 -0
- sky/dashboard/out/jobs.html +1 -1
- sky/dashboard/out/users.html +1 -0
- sky/dashboard/out/volumes.html +1 -0
- sky/dashboard/out/workspace/new.html +1 -0
- sky/dashboard/out/workspaces/[name].html +1 -0
- sky/dashboard/out/workspaces.html +1 -0
- sky/data/data_utils.py +137 -1
- sky/data/mounting_utils.py +269 -84
- sky/data/storage.py +1460 -1807
- sky/data/storage_utils.py +43 -57
- sky/exceptions.py +126 -2
- sky/execution.py +216 -63
- sky/global_user_state.py +2390 -586
- sky/jobs/__init__.py +7 -0
- sky/jobs/client/sdk.py +300 -58
- sky/jobs/client/sdk_async.py +161 -0
- sky/jobs/constants.py +15 -8
- sky/jobs/controller.py +848 -275
- sky/jobs/file_content_utils.py +128 -0
- sky/jobs/log_gc.py +193 -0
- sky/jobs/recovery_strategy.py +402 -152
- sky/jobs/scheduler.py +314 -189
- sky/jobs/server/core.py +836 -255
- sky/jobs/server/server.py +156 -115
- sky/jobs/server/utils.py +136 -0
- sky/jobs/state.py +2109 -706
- sky/jobs/utils.py +1306 -215
- sky/logs/__init__.py +21 -0
- sky/logs/agent.py +108 -0
- sky/logs/aws.py +243 -0
- sky/logs/gcp.py +91 -0
- sky/metrics/__init__.py +0 -0
- sky/metrics/utils.py +453 -0
- sky/models.py +78 -1
- sky/optimizer.py +164 -70
- sky/provision/__init__.py +90 -4
- sky/provision/aws/config.py +147 -26
- sky/provision/aws/instance.py +136 -50
- sky/provision/azure/instance.py +11 -6
- sky/provision/common.py +13 -1
- sky/provision/cudo/cudo_machine_type.py +1 -1
- sky/provision/cudo/cudo_utils.py +14 -8
- sky/provision/cudo/cudo_wrapper.py +72 -71
- sky/provision/cudo/instance.py +10 -6
- sky/provision/do/instance.py +10 -6
- sky/provision/do/utils.py +4 -3
- sky/provision/docker_utils.py +140 -33
- sky/provision/fluidstack/instance.py +13 -8
- sky/provision/gcp/__init__.py +1 -0
- sky/provision/gcp/config.py +301 -19
- sky/provision/gcp/constants.py +218 -0
- sky/provision/gcp/instance.py +36 -8
- sky/provision/gcp/instance_utils.py +18 -4
- sky/provision/gcp/volume_utils.py +247 -0
- sky/provision/hyperbolic/__init__.py +12 -0
- sky/provision/hyperbolic/config.py +10 -0
- sky/provision/hyperbolic/instance.py +437 -0
- sky/provision/hyperbolic/utils.py +373 -0
- sky/provision/instance_setup.py +101 -20
- sky/provision/kubernetes/__init__.py +5 -0
- sky/provision/kubernetes/config.py +9 -52
- sky/provision/kubernetes/constants.py +17 -0
- sky/provision/kubernetes/instance.py +919 -280
- sky/provision/kubernetes/manifests/fusermount-server-daemonset.yaml +1 -2
- sky/provision/kubernetes/network.py +27 -17
- sky/provision/kubernetes/network_utils.py +44 -43
- sky/provision/kubernetes/utils.py +1221 -534
- sky/provision/kubernetes/volume.py +343 -0
- sky/provision/lambda_cloud/instance.py +22 -16
- sky/provision/nebius/constants.py +50 -0
- sky/provision/nebius/instance.py +19 -6
- sky/provision/nebius/utils.py +237 -137
- sky/provision/oci/instance.py +10 -5
- sky/provision/paperspace/instance.py +10 -7
- sky/provision/paperspace/utils.py +1 -1
- sky/provision/primeintellect/__init__.py +10 -0
- sky/provision/primeintellect/config.py +11 -0
- sky/provision/primeintellect/instance.py +454 -0
- sky/provision/primeintellect/utils.py +398 -0
- sky/provision/provisioner.py +117 -36
- sky/provision/runpod/__init__.py +5 -0
- sky/provision/runpod/instance.py +27 -6
- sky/provision/runpod/utils.py +51 -18
- sky/provision/runpod/volume.py +214 -0
- sky/provision/scp/__init__.py +15 -0
- sky/provision/scp/config.py +93 -0
- sky/provision/scp/instance.py +707 -0
- sky/provision/seeweb/__init__.py +11 -0
- sky/provision/seeweb/config.py +13 -0
- sky/provision/seeweb/instance.py +812 -0
- sky/provision/shadeform/__init__.py +11 -0
- sky/provision/shadeform/config.py +12 -0
- sky/provision/shadeform/instance.py +351 -0
- sky/provision/shadeform/shadeform_utils.py +83 -0
- sky/provision/ssh/__init__.py +18 -0
- sky/provision/vast/instance.py +13 -8
- sky/provision/vast/utils.py +10 -7
- sky/provision/volume.py +164 -0
- sky/provision/vsphere/common/ssl_helper.py +1 -1
- sky/provision/vsphere/common/vapiconnect.py +2 -1
- sky/provision/vsphere/common/vim_utils.py +4 -4
- sky/provision/vsphere/instance.py +15 -10
- sky/provision/vsphere/vsphere_utils.py +17 -20
- sky/py.typed +0 -0
- sky/resources.py +845 -119
- sky/schemas/__init__.py +0 -0
- sky/schemas/api/__init__.py +0 -0
- sky/schemas/api/responses.py +227 -0
- sky/schemas/db/README +4 -0
- sky/schemas/db/env.py +90 -0
- sky/schemas/db/global_user_state/001_initial_schema.py +124 -0
- sky/schemas/db/global_user_state/002_add_workspace_to_cluster_history.py +35 -0
- sky/schemas/db/global_user_state/003_fix_initial_revision.py +61 -0
- sky/schemas/db/global_user_state/004_is_managed.py +34 -0
- sky/schemas/db/global_user_state/005_cluster_event.py +32 -0
- sky/schemas/db/global_user_state/006_provision_log.py +41 -0
- sky/schemas/db/global_user_state/007_cluster_event_request_id.py +34 -0
- sky/schemas/db/global_user_state/008_skylet_ssh_tunnel_metadata.py +34 -0
- sky/schemas/db/global_user_state/009_last_activity_and_launched_at.py +89 -0
- sky/schemas/db/global_user_state/010_save_ssh_key.py +66 -0
- sky/schemas/db/global_user_state/011_is_ephemeral.py +34 -0
- sky/schemas/db/kv_cache/001_initial_schema.py +29 -0
- sky/schemas/db/script.py.mako +28 -0
- sky/schemas/db/serve_state/001_initial_schema.py +67 -0
- sky/schemas/db/serve_state/002_yaml_content.py +34 -0
- sky/schemas/db/skypilot_config/001_initial_schema.py +30 -0
- sky/schemas/db/spot_jobs/001_initial_schema.py +97 -0
- sky/schemas/db/spot_jobs/002_cluster_pool.py +42 -0
- sky/schemas/db/spot_jobs/003_pool_hash.py +34 -0
- sky/schemas/db/spot_jobs/004_job_file_contents.py +42 -0
- sky/schemas/db/spot_jobs/005_logs_gc.py +38 -0
- sky/schemas/db/spot_jobs/006_controller_pid_started_at.py +34 -0
- sky/schemas/db/spot_jobs/007_config_file_content.py +34 -0
- sky/schemas/generated/__init__.py +0 -0
- sky/schemas/generated/autostopv1_pb2.py +36 -0
- sky/schemas/generated/autostopv1_pb2.pyi +43 -0
- sky/schemas/generated/autostopv1_pb2_grpc.py +146 -0
- sky/schemas/generated/jobsv1_pb2.py +86 -0
- sky/schemas/generated/jobsv1_pb2.pyi +254 -0
- sky/schemas/generated/jobsv1_pb2_grpc.py +542 -0
- sky/schemas/generated/managed_jobsv1_pb2.py +76 -0
- sky/schemas/generated/managed_jobsv1_pb2.pyi +278 -0
- sky/schemas/generated/managed_jobsv1_pb2_grpc.py +278 -0
- sky/schemas/generated/servev1_pb2.py +58 -0
- sky/schemas/generated/servev1_pb2.pyi +115 -0
- sky/schemas/generated/servev1_pb2_grpc.py +322 -0
- sky/serve/autoscalers.py +357 -5
- sky/serve/client/impl.py +310 -0
- sky/serve/client/sdk.py +47 -139
- sky/serve/client/sdk_async.py +130 -0
- sky/serve/constants.py +12 -9
- sky/serve/controller.py +68 -17
- sky/serve/load_balancer.py +106 -60
- sky/serve/load_balancing_policies.py +116 -2
- sky/serve/replica_managers.py +434 -249
- sky/serve/serve_rpc_utils.py +179 -0
- sky/serve/serve_state.py +569 -257
- sky/serve/serve_utils.py +775 -265
- sky/serve/server/core.py +66 -711
- sky/serve/server/impl.py +1093 -0
- sky/serve/server/server.py +21 -18
- sky/serve/service.py +192 -89
- sky/serve/service_spec.py +144 -20
- sky/serve/spot_placer.py +3 -0
- sky/server/auth/__init__.py +0 -0
- sky/server/auth/authn.py +50 -0
- sky/server/auth/loopback.py +38 -0
- sky/server/auth/oauth2_proxy.py +202 -0
- sky/server/common.py +478 -182
- sky/server/config.py +85 -23
- sky/server/constants.py +44 -6
- sky/server/daemons.py +295 -0
- sky/server/html/token_page.html +185 -0
- sky/server/metrics.py +160 -0
- sky/server/middleware_utils.py +166 -0
- sky/server/requests/executor.py +558 -138
- sky/server/requests/payloads.py +364 -24
- sky/server/requests/preconditions.py +21 -17
- sky/server/requests/process.py +112 -29
- sky/server/requests/request_names.py +121 -0
- sky/server/requests/requests.py +822 -226
- sky/server/requests/serializers/decoders.py +82 -31
- sky/server/requests/serializers/encoders.py +140 -22
- sky/server/requests/threads.py +117 -0
- sky/server/rest.py +455 -0
- sky/server/server.py +1309 -285
- sky/server/state.py +20 -0
- sky/server/stream_utils.py +327 -61
- sky/server/uvicorn.py +217 -3
- sky/server/versions.py +270 -0
- sky/setup_files/MANIFEST.in +11 -1
- sky/setup_files/alembic.ini +160 -0
- sky/setup_files/dependencies.py +139 -31
- sky/setup_files/setup.py +44 -42
- sky/sky_logging.py +114 -7
- sky/skylet/attempt_skylet.py +106 -24
- sky/skylet/autostop_lib.py +129 -8
- sky/skylet/configs.py +29 -20
- sky/skylet/constants.py +216 -25
- sky/skylet/events.py +101 -21
- sky/skylet/job_lib.py +345 -164
- sky/skylet/log_lib.py +297 -18
- sky/skylet/log_lib.pyi +44 -1
- sky/skylet/providers/ibm/node_provider.py +12 -8
- sky/skylet/providers/ibm/vpc_provider.py +13 -12
- sky/skylet/ray_patches/__init__.py +17 -3
- sky/skylet/ray_patches/autoscaler.py.diff +18 -0
- sky/skylet/ray_patches/cli.py.diff +19 -0
- sky/skylet/ray_patches/command_runner.py.diff +17 -0
- sky/skylet/ray_patches/log_monitor.py.diff +20 -0
- sky/skylet/ray_patches/resource_demand_scheduler.py.diff +32 -0
- sky/skylet/ray_patches/updater.py.diff +18 -0
- sky/skylet/ray_patches/worker.py.diff +41 -0
- sky/skylet/runtime_utils.py +21 -0
- sky/skylet/services.py +568 -0
- sky/skylet/skylet.py +72 -4
- sky/skylet/subprocess_daemon.py +104 -29
- sky/skypilot_config.py +506 -99
- sky/ssh_node_pools/__init__.py +1 -0
- sky/ssh_node_pools/core.py +135 -0
- sky/ssh_node_pools/server.py +233 -0
- sky/task.py +685 -163
- sky/templates/aws-ray.yml.j2 +11 -3
- sky/templates/azure-ray.yml.j2 +2 -1
- sky/templates/cudo-ray.yml.j2 +1 -0
- sky/templates/do-ray.yml.j2 +2 -1
- sky/templates/fluidstack-ray.yml.j2 +1 -0
- sky/templates/gcp-ray.yml.j2 +62 -1
- sky/templates/hyperbolic-ray.yml.j2 +68 -0
- sky/templates/ibm-ray.yml.j2 +2 -1
- sky/templates/jobs-controller.yaml.j2 +27 -24
- sky/templates/kubernetes-loadbalancer.yml.j2 +2 -0
- sky/templates/kubernetes-ray.yml.j2 +611 -50
- sky/templates/lambda-ray.yml.j2 +2 -1
- sky/templates/nebius-ray.yml.j2 +34 -12
- sky/templates/oci-ray.yml.j2 +1 -0
- sky/templates/paperspace-ray.yml.j2 +2 -1
- sky/templates/primeintellect-ray.yml.j2 +72 -0
- sky/templates/runpod-ray.yml.j2 +10 -1
- sky/templates/scp-ray.yml.j2 +4 -50
- sky/templates/seeweb-ray.yml.j2 +171 -0
- sky/templates/shadeform-ray.yml.j2 +73 -0
- sky/templates/sky-serve-controller.yaml.j2 +22 -2
- sky/templates/vast-ray.yml.j2 +1 -0
- sky/templates/vsphere-ray.yml.j2 +1 -0
- sky/templates/websocket_proxy.py +212 -37
- sky/usage/usage_lib.py +31 -15
- sky/users/__init__.py +0 -0
- sky/users/model.conf +15 -0
- sky/users/permission.py +397 -0
- sky/users/rbac.py +121 -0
- sky/users/server.py +720 -0
- sky/users/token_service.py +218 -0
- sky/utils/accelerator_registry.py +35 -5
- sky/utils/admin_policy_utils.py +84 -38
- sky/utils/annotations.py +38 -5
- sky/utils/asyncio_utils.py +78 -0
- sky/utils/atomic.py +1 -1
- sky/utils/auth_utils.py +153 -0
- sky/utils/benchmark_utils.py +60 -0
- sky/utils/cli_utils/status_utils.py +159 -86
- sky/utils/cluster_utils.py +31 -9
- sky/utils/command_runner.py +354 -68
- sky/utils/command_runner.pyi +93 -3
- sky/utils/common.py +35 -8
- sky/utils/common_utils.py +314 -91
- sky/utils/config_utils.py +74 -5
- sky/utils/context.py +403 -0
- sky/utils/context_utils.py +242 -0
- sky/utils/controller_utils.py +383 -89
- sky/utils/dag_utils.py +31 -12
- sky/utils/db/__init__.py +0 -0
- sky/utils/db/db_utils.py +485 -0
- sky/utils/db/kv_cache.py +149 -0
- sky/utils/db/migration_utils.py +137 -0
- sky/utils/directory_utils.py +12 -0
- sky/utils/env_options.py +13 -0
- sky/utils/git.py +567 -0
- sky/utils/git_clone.sh +460 -0
- sky/utils/infra_utils.py +195 -0
- sky/utils/kubernetes/cleanup-tunnel.sh +62 -0
- sky/utils/kubernetes/config_map_utils.py +133 -0
- sky/utils/kubernetes/create_cluster.sh +15 -29
- sky/utils/kubernetes/delete_cluster.sh +10 -7
- sky/utils/kubernetes/deploy_ssh_node_pools.py +1177 -0
- sky/utils/kubernetes/exec_kubeconfig_converter.py +22 -31
- sky/utils/kubernetes/generate_kind_config.py +6 -66
- sky/utils/kubernetes/generate_kubeconfig.sh +4 -1
- sky/utils/kubernetes/gpu_labeler.py +18 -8
- sky/utils/kubernetes/k8s_gpu_labeler_job.yaml +2 -1
- sky/utils/kubernetes/k8s_gpu_labeler_setup.yaml +16 -16
- sky/utils/kubernetes/kubernetes_deploy_utils.py +284 -114
- sky/utils/kubernetes/rsync_helper.sh +11 -3
- sky/utils/kubernetes/ssh-tunnel.sh +379 -0
- sky/utils/kubernetes/ssh_utils.py +221 -0
- sky/utils/kubernetes_enums.py +8 -15
- sky/utils/lock_events.py +94 -0
- sky/utils/locks.py +416 -0
- sky/utils/log_utils.py +82 -107
- sky/utils/perf_utils.py +22 -0
- sky/utils/resource_checker.py +298 -0
- sky/utils/resources_utils.py +249 -32
- sky/utils/rich_utils.py +217 -39
- sky/utils/schemas.py +955 -160
- sky/utils/serialize_utils.py +16 -0
- sky/utils/status_lib.py +10 -0
- sky/utils/subprocess_utils.py +29 -15
- sky/utils/tempstore.py +70 -0
- sky/utils/thread_utils.py +91 -0
- sky/utils/timeline.py +26 -53
- sky/utils/ux_utils.py +84 -15
- sky/utils/validator.py +11 -1
- sky/utils/volume.py +165 -0
- sky/utils/yaml_utils.py +111 -0
- sky/volumes/__init__.py +13 -0
- sky/volumes/client/__init__.py +0 -0
- sky/volumes/client/sdk.py +150 -0
- sky/volumes/server/__init__.py +0 -0
- sky/volumes/server/core.py +270 -0
- sky/volumes/server/server.py +124 -0
- sky/volumes/volume.py +215 -0
- sky/workspaces/__init__.py +0 -0
- sky/workspaces/core.py +655 -0
- sky/workspaces/server.py +101 -0
- sky/workspaces/utils.py +56 -0
- sky_templates/README.md +3 -0
- sky_templates/__init__.py +3 -0
- sky_templates/ray/__init__.py +0 -0
- sky_templates/ray/start_cluster +183 -0
- sky_templates/ray/stop_cluster +75 -0
- skypilot_nightly-1.0.0.dev20251203.dist-info/METADATA +676 -0
- skypilot_nightly-1.0.0.dev20251203.dist-info/RECORD +611 -0
- {skypilot_nightly-1.0.0.dev20250502.dist-info → skypilot_nightly-1.0.0.dev20251203.dist-info}/WHEEL +1 -1
- skypilot_nightly-1.0.0.dev20251203.dist-info/top_level.txt +2 -0
- sky/benchmark/benchmark_state.py +0 -256
- sky/benchmark/benchmark_utils.py +0 -641
- sky/clouds/service_catalog/constants.py +0 -7
- sky/dashboard/out/_next/static/GWvVBSCS7FmUiVmjaL1a7/_buildManifest.js +0 -1
- sky/dashboard/out/_next/static/chunks/236-2db3ee3fba33dd9e.js +0 -6
- sky/dashboard/out/_next/static/chunks/312-c3c8845990db8ffc.js +0 -15
- sky/dashboard/out/_next/static/chunks/37-0a572fe0dbb89c4d.js +0 -6
- sky/dashboard/out/_next/static/chunks/678-206dddca808e6d16.js +0 -59
- sky/dashboard/out/_next/static/chunks/845-9e60713e0c441abc.js +0 -1
- sky/dashboard/out/_next/static/chunks/979-7bf73a4c7cea0f5c.js +0 -1
- sky/dashboard/out/_next/static/chunks/fd9d1056-2821b0f0cabcd8bd.js +0 -1
- sky/dashboard/out/_next/static/chunks/framework-87d061ee6ed71b28.js +0 -33
- sky/dashboard/out/_next/static/chunks/main-app-241eb28595532291.js +0 -1
- sky/dashboard/out/_next/static/chunks/main-e0e2335212e72357.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/_app-e6b013bc3f77ad60.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/_error-1be831200e60c5c0.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/clusters/[cluster]/[job]-6ac338bc2239cb45.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/clusters/[cluster]-f383db7389368ea7.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/clusters-a93b93e10b8b074e.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/index-f9f039532ca8cbc4.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/jobs/[job]-1c519e1afc523dc9.js +0 -1
- sky/dashboard/out/_next/static/chunks/pages/jobs-a75029b67aab6a2e.js +0 -1
- sky/dashboard/out/_next/static/chunks/webpack-830f59b8404e96b8.js +0 -1
- sky/dashboard/out/_next/static/css/c6933bbb2ce7f4dd.css +0 -3
- sky/jobs/dashboard/dashboard.py +0 -223
- sky/jobs/dashboard/static/favicon.ico +0 -0
- sky/jobs/dashboard/templates/index.html +0 -831
- sky/jobs/server/dashboard_utils.py +0 -69
- sky/skylet/providers/scp/__init__.py +0 -2
- sky/skylet/providers/scp/config.py +0 -149
- sky/skylet/providers/scp/node_provider.py +0 -578
- sky/templates/kubernetes-ssh-jump.yml.j2 +0 -94
- sky/utils/db_utils.py +0 -100
- sky/utils/kubernetes/deploy_remote_cluster.sh +0 -308
- sky/utils/kubernetes/ssh_jump_lifecycle_manager.py +0 -191
- skypilot_nightly-1.0.0.dev20250502.dist-info/METADATA +0 -361
- skypilot_nightly-1.0.0.dev20250502.dist-info/RECORD +0 -396
- skypilot_nightly-1.0.0.dev20250502.dist-info/top_level.txt +0 -1
- /sky/{clouds/service_catalog → catalog}/config.py +0 -0
- /sky/{benchmark → catalog/data_fetchers}/__init__.py +0 -0
- /sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_azure.py +0 -0
- /sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_fluidstack.py +0 -0
- /sky/{clouds/service_catalog → catalog}/data_fetchers/fetch_ibm.py +0 -0
- /sky/{clouds/service_catalog/data_fetchers → client/cli}/__init__.py +0 -0
- /sky/dashboard/out/_next/static/{GWvVBSCS7FmUiVmjaL1a7 → 96_E2yl3QAiIJGOYCkSpB}/_ssgManifest.js +0 -0
- {skypilot_nightly-1.0.0.dev20250502.dist-info → skypilot_nightly-1.0.0.dev20251203.dist-info}/entry_points.txt +0 -0
- {skypilot_nightly-1.0.0.dev20250502.dist-info → skypilot_nightly-1.0.0.dev20251203.dist-info}/licenses/LICENSE +0 -0
|
@@ -0,0 +1,542 @@
|
|
|
1
|
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
|
2
|
+
"""Client and server classes corresponding to protobuf-defined services."""
|
|
3
|
+
import grpc
|
|
4
|
+
import warnings
|
|
5
|
+
|
|
6
|
+
from sky.schemas.generated import jobsv1_pb2 as sky_dot_schemas_dot_generated_dot_jobsv1__pb2
|
|
7
|
+
|
|
8
|
+
GRPC_GENERATED_VERSION = '1.63.0'
|
|
9
|
+
GRPC_VERSION = grpc.__version__
|
|
10
|
+
EXPECTED_ERROR_RELEASE = '1.65.0'
|
|
11
|
+
SCHEDULED_RELEASE_DATE = 'June 25, 2024'
|
|
12
|
+
_version_not_supported = False
|
|
13
|
+
|
|
14
|
+
try:
|
|
15
|
+
from grpc._utilities import first_version_is_lower
|
|
16
|
+
_version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
|
|
17
|
+
except ImportError:
|
|
18
|
+
_version_not_supported = True
|
|
19
|
+
|
|
20
|
+
if _version_not_supported:
|
|
21
|
+
warnings.warn(
|
|
22
|
+
f'The grpc package installed is at version {GRPC_VERSION},'
|
|
23
|
+
+ f' but the generated code in sky/schemas/generated/jobsv1_pb2_grpc.py depends on'
|
|
24
|
+
+ f' grpcio>={GRPC_GENERATED_VERSION}.'
|
|
25
|
+
+ f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
|
|
26
|
+
+ f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
|
|
27
|
+
+ f' This warning will become an error in {EXPECTED_ERROR_RELEASE},'
|
|
28
|
+
+ f' scheduled for release on {SCHEDULED_RELEASE_DATE}.',
|
|
29
|
+
RuntimeWarning
|
|
30
|
+
)
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
class JobsServiceStub(object):
|
|
34
|
+
"""Missing associated documentation comment in .proto file."""
|
|
35
|
+
|
|
36
|
+
def __init__(self, channel):
|
|
37
|
+
"""Constructor.
|
|
38
|
+
|
|
39
|
+
Args:
|
|
40
|
+
channel: A grpc.Channel.
|
|
41
|
+
"""
|
|
42
|
+
self.AddJob = channel.unary_unary(
|
|
43
|
+
'/jobs.v1.JobsService/AddJob',
|
|
44
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.AddJobRequest.SerializeToString,
|
|
45
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.AddJobResponse.FromString,
|
|
46
|
+
_registered_method=True)
|
|
47
|
+
self.QueueJob = channel.unary_unary(
|
|
48
|
+
'/jobs.v1.JobsService/QueueJob',
|
|
49
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.QueueJobRequest.SerializeToString,
|
|
50
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.QueueJobResponse.FromString,
|
|
51
|
+
_registered_method=True)
|
|
52
|
+
self.UpdateStatus = channel.unary_unary(
|
|
53
|
+
'/jobs.v1.JobsService/UpdateStatus',
|
|
54
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.UpdateStatusRequest.SerializeToString,
|
|
55
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.UpdateStatusResponse.FromString,
|
|
56
|
+
_registered_method=True)
|
|
57
|
+
self.GetJobQueue = channel.unary_unary(
|
|
58
|
+
'/jobs.v1.JobsService/GetJobQueue',
|
|
59
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobQueueRequest.SerializeToString,
|
|
60
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobQueueResponse.FromString,
|
|
61
|
+
_registered_method=True)
|
|
62
|
+
self.CancelJobs = channel.unary_unary(
|
|
63
|
+
'/jobs.v1.JobsService/CancelJobs',
|
|
64
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.CancelJobsRequest.SerializeToString,
|
|
65
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.CancelJobsResponse.FromString,
|
|
66
|
+
_registered_method=True)
|
|
67
|
+
self.FailAllInProgressJobs = channel.unary_unary(
|
|
68
|
+
'/jobs.v1.JobsService/FailAllInProgressJobs',
|
|
69
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.FailAllInProgressJobsRequest.SerializeToString,
|
|
70
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.FailAllInProgressJobsResponse.FromString,
|
|
71
|
+
_registered_method=True)
|
|
72
|
+
self.TailLogs = channel.unary_stream(
|
|
73
|
+
'/jobs.v1.JobsService/TailLogs',
|
|
74
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.TailLogsRequest.SerializeToString,
|
|
75
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.TailLogsResponse.FromString,
|
|
76
|
+
_registered_method=True)
|
|
77
|
+
self.GetJobStatus = channel.unary_unary(
|
|
78
|
+
'/jobs.v1.JobsService/GetJobStatus',
|
|
79
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobStatusRequest.SerializeToString,
|
|
80
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobStatusResponse.FromString,
|
|
81
|
+
_registered_method=True)
|
|
82
|
+
self.GetJobSubmittedTimestamp = channel.unary_unary(
|
|
83
|
+
'/jobs.v1.JobsService/GetJobSubmittedTimestamp',
|
|
84
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobSubmittedTimestampRequest.SerializeToString,
|
|
85
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobSubmittedTimestampResponse.FromString,
|
|
86
|
+
_registered_method=True)
|
|
87
|
+
self.GetJobEndedTimestamp = channel.unary_unary(
|
|
88
|
+
'/jobs.v1.JobsService/GetJobEndedTimestamp',
|
|
89
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobEndedTimestampRequest.SerializeToString,
|
|
90
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobEndedTimestampResponse.FromString,
|
|
91
|
+
_registered_method=True)
|
|
92
|
+
self.GetLogDirsForJobs = channel.unary_unary(
|
|
93
|
+
'/jobs.v1.JobsService/GetLogDirsForJobs',
|
|
94
|
+
request_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetLogDirsForJobsRequest.SerializeToString,
|
|
95
|
+
response_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetLogDirsForJobsResponse.FromString,
|
|
96
|
+
_registered_method=True)
|
|
97
|
+
|
|
98
|
+
|
|
99
|
+
class JobsServiceServicer(object):
|
|
100
|
+
"""Missing associated documentation comment in .proto file."""
|
|
101
|
+
|
|
102
|
+
def AddJob(self, request, context):
|
|
103
|
+
"""Add a new job to the database.
|
|
104
|
+
"""
|
|
105
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
106
|
+
context.set_details('Method not implemented!')
|
|
107
|
+
raise NotImplementedError('Method not implemented!')
|
|
108
|
+
|
|
109
|
+
def QueueJob(self, request, context):
|
|
110
|
+
"""Queue a job for execution.
|
|
111
|
+
"""
|
|
112
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
113
|
+
context.set_details('Method not implemented!')
|
|
114
|
+
raise NotImplementedError('Method not implemented!')
|
|
115
|
+
|
|
116
|
+
def UpdateStatus(self, request, context):
|
|
117
|
+
"""Update status of all jobs.
|
|
118
|
+
"""
|
|
119
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
120
|
+
context.set_details('Method not implemented!')
|
|
121
|
+
raise NotImplementedError('Method not implemented!')
|
|
122
|
+
|
|
123
|
+
def GetJobQueue(self, request, context):
|
|
124
|
+
"""Get the job queue.
|
|
125
|
+
"""
|
|
126
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
127
|
+
context.set_details('Method not implemented!')
|
|
128
|
+
raise NotImplementedError('Method not implemented!')
|
|
129
|
+
|
|
130
|
+
def CancelJobs(self, request, context):
|
|
131
|
+
"""Cancel jobs.
|
|
132
|
+
"""
|
|
133
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
134
|
+
context.set_details('Method not implemented!')
|
|
135
|
+
raise NotImplementedError('Method not implemented!')
|
|
136
|
+
|
|
137
|
+
def FailAllInProgressJobs(self, request, context):
|
|
138
|
+
"""Fail all in-progress jobs.
|
|
139
|
+
"""
|
|
140
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
141
|
+
context.set_details('Method not implemented!')
|
|
142
|
+
raise NotImplementedError('Method not implemented!')
|
|
143
|
+
|
|
144
|
+
def TailLogs(self, request, context):
|
|
145
|
+
"""Tail job logs.
|
|
146
|
+
"""
|
|
147
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
148
|
+
context.set_details('Method not implemented!')
|
|
149
|
+
raise NotImplementedError('Method not implemented!')
|
|
150
|
+
|
|
151
|
+
def GetJobStatus(self, request, context):
|
|
152
|
+
"""Get job status.
|
|
153
|
+
"""
|
|
154
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
155
|
+
context.set_details('Method not implemented!')
|
|
156
|
+
raise NotImplementedError('Method not implemented!')
|
|
157
|
+
|
|
158
|
+
def GetJobSubmittedTimestamp(self, request, context):
|
|
159
|
+
"""Get job submitted timestamp.
|
|
160
|
+
"""
|
|
161
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
162
|
+
context.set_details('Method not implemented!')
|
|
163
|
+
raise NotImplementedError('Method not implemented!')
|
|
164
|
+
|
|
165
|
+
def GetJobEndedTimestamp(self, request, context):
|
|
166
|
+
"""Get job ended timestamp.
|
|
167
|
+
"""
|
|
168
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
169
|
+
context.set_details('Method not implemented!')
|
|
170
|
+
raise NotImplementedError('Method not implemented!')
|
|
171
|
+
|
|
172
|
+
def GetLogDirsForJobs(self, request, context):
|
|
173
|
+
"""Get log directories for jobs.
|
|
174
|
+
"""
|
|
175
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
176
|
+
context.set_details('Method not implemented!')
|
|
177
|
+
raise NotImplementedError('Method not implemented!')
|
|
178
|
+
|
|
179
|
+
|
|
180
|
+
def add_JobsServiceServicer_to_server(servicer, server):
|
|
181
|
+
rpc_method_handlers = {
|
|
182
|
+
'AddJob': grpc.unary_unary_rpc_method_handler(
|
|
183
|
+
servicer.AddJob,
|
|
184
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.AddJobRequest.FromString,
|
|
185
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.AddJobResponse.SerializeToString,
|
|
186
|
+
),
|
|
187
|
+
'QueueJob': grpc.unary_unary_rpc_method_handler(
|
|
188
|
+
servicer.QueueJob,
|
|
189
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.QueueJobRequest.FromString,
|
|
190
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.QueueJobResponse.SerializeToString,
|
|
191
|
+
),
|
|
192
|
+
'UpdateStatus': grpc.unary_unary_rpc_method_handler(
|
|
193
|
+
servicer.UpdateStatus,
|
|
194
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.UpdateStatusRequest.FromString,
|
|
195
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.UpdateStatusResponse.SerializeToString,
|
|
196
|
+
),
|
|
197
|
+
'GetJobQueue': grpc.unary_unary_rpc_method_handler(
|
|
198
|
+
servicer.GetJobQueue,
|
|
199
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobQueueRequest.FromString,
|
|
200
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobQueueResponse.SerializeToString,
|
|
201
|
+
),
|
|
202
|
+
'CancelJobs': grpc.unary_unary_rpc_method_handler(
|
|
203
|
+
servicer.CancelJobs,
|
|
204
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.CancelJobsRequest.FromString,
|
|
205
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.CancelJobsResponse.SerializeToString,
|
|
206
|
+
),
|
|
207
|
+
'FailAllInProgressJobs': grpc.unary_unary_rpc_method_handler(
|
|
208
|
+
servicer.FailAllInProgressJobs,
|
|
209
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.FailAllInProgressJobsRequest.FromString,
|
|
210
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.FailAllInProgressJobsResponse.SerializeToString,
|
|
211
|
+
),
|
|
212
|
+
'TailLogs': grpc.unary_stream_rpc_method_handler(
|
|
213
|
+
servicer.TailLogs,
|
|
214
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.TailLogsRequest.FromString,
|
|
215
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.TailLogsResponse.SerializeToString,
|
|
216
|
+
),
|
|
217
|
+
'GetJobStatus': grpc.unary_unary_rpc_method_handler(
|
|
218
|
+
servicer.GetJobStatus,
|
|
219
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobStatusRequest.FromString,
|
|
220
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobStatusResponse.SerializeToString,
|
|
221
|
+
),
|
|
222
|
+
'GetJobSubmittedTimestamp': grpc.unary_unary_rpc_method_handler(
|
|
223
|
+
servicer.GetJobSubmittedTimestamp,
|
|
224
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobSubmittedTimestampRequest.FromString,
|
|
225
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobSubmittedTimestampResponse.SerializeToString,
|
|
226
|
+
),
|
|
227
|
+
'GetJobEndedTimestamp': grpc.unary_unary_rpc_method_handler(
|
|
228
|
+
servicer.GetJobEndedTimestamp,
|
|
229
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobEndedTimestampRequest.FromString,
|
|
230
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobEndedTimestampResponse.SerializeToString,
|
|
231
|
+
),
|
|
232
|
+
'GetLogDirsForJobs': grpc.unary_unary_rpc_method_handler(
|
|
233
|
+
servicer.GetLogDirsForJobs,
|
|
234
|
+
request_deserializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetLogDirsForJobsRequest.FromString,
|
|
235
|
+
response_serializer=sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetLogDirsForJobsResponse.SerializeToString,
|
|
236
|
+
),
|
|
237
|
+
}
|
|
238
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
|
239
|
+
'jobs.v1.JobsService', rpc_method_handlers)
|
|
240
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
241
|
+
|
|
242
|
+
|
|
243
|
+
# This class is part of an EXPERIMENTAL API.
|
|
244
|
+
class JobsService(object):
|
|
245
|
+
"""Missing associated documentation comment in .proto file."""
|
|
246
|
+
|
|
247
|
+
@staticmethod
|
|
248
|
+
def AddJob(request,
|
|
249
|
+
target,
|
|
250
|
+
options=(),
|
|
251
|
+
channel_credentials=None,
|
|
252
|
+
call_credentials=None,
|
|
253
|
+
insecure=False,
|
|
254
|
+
compression=None,
|
|
255
|
+
wait_for_ready=None,
|
|
256
|
+
timeout=None,
|
|
257
|
+
metadata=None):
|
|
258
|
+
return grpc.experimental.unary_unary(
|
|
259
|
+
request,
|
|
260
|
+
target,
|
|
261
|
+
'/jobs.v1.JobsService/AddJob',
|
|
262
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.AddJobRequest.SerializeToString,
|
|
263
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.AddJobResponse.FromString,
|
|
264
|
+
options,
|
|
265
|
+
channel_credentials,
|
|
266
|
+
insecure,
|
|
267
|
+
call_credentials,
|
|
268
|
+
compression,
|
|
269
|
+
wait_for_ready,
|
|
270
|
+
timeout,
|
|
271
|
+
metadata,
|
|
272
|
+
_registered_method=True)
|
|
273
|
+
|
|
274
|
+
@staticmethod
|
|
275
|
+
def QueueJob(request,
|
|
276
|
+
target,
|
|
277
|
+
options=(),
|
|
278
|
+
channel_credentials=None,
|
|
279
|
+
call_credentials=None,
|
|
280
|
+
insecure=False,
|
|
281
|
+
compression=None,
|
|
282
|
+
wait_for_ready=None,
|
|
283
|
+
timeout=None,
|
|
284
|
+
metadata=None):
|
|
285
|
+
return grpc.experimental.unary_unary(
|
|
286
|
+
request,
|
|
287
|
+
target,
|
|
288
|
+
'/jobs.v1.JobsService/QueueJob',
|
|
289
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.QueueJobRequest.SerializeToString,
|
|
290
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.QueueJobResponse.FromString,
|
|
291
|
+
options,
|
|
292
|
+
channel_credentials,
|
|
293
|
+
insecure,
|
|
294
|
+
call_credentials,
|
|
295
|
+
compression,
|
|
296
|
+
wait_for_ready,
|
|
297
|
+
timeout,
|
|
298
|
+
metadata,
|
|
299
|
+
_registered_method=True)
|
|
300
|
+
|
|
301
|
+
@staticmethod
|
|
302
|
+
def UpdateStatus(request,
|
|
303
|
+
target,
|
|
304
|
+
options=(),
|
|
305
|
+
channel_credentials=None,
|
|
306
|
+
call_credentials=None,
|
|
307
|
+
insecure=False,
|
|
308
|
+
compression=None,
|
|
309
|
+
wait_for_ready=None,
|
|
310
|
+
timeout=None,
|
|
311
|
+
metadata=None):
|
|
312
|
+
return grpc.experimental.unary_unary(
|
|
313
|
+
request,
|
|
314
|
+
target,
|
|
315
|
+
'/jobs.v1.JobsService/UpdateStatus',
|
|
316
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.UpdateStatusRequest.SerializeToString,
|
|
317
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.UpdateStatusResponse.FromString,
|
|
318
|
+
options,
|
|
319
|
+
channel_credentials,
|
|
320
|
+
insecure,
|
|
321
|
+
call_credentials,
|
|
322
|
+
compression,
|
|
323
|
+
wait_for_ready,
|
|
324
|
+
timeout,
|
|
325
|
+
metadata,
|
|
326
|
+
_registered_method=True)
|
|
327
|
+
|
|
328
|
+
@staticmethod
|
|
329
|
+
def GetJobQueue(request,
|
|
330
|
+
target,
|
|
331
|
+
options=(),
|
|
332
|
+
channel_credentials=None,
|
|
333
|
+
call_credentials=None,
|
|
334
|
+
insecure=False,
|
|
335
|
+
compression=None,
|
|
336
|
+
wait_for_ready=None,
|
|
337
|
+
timeout=None,
|
|
338
|
+
metadata=None):
|
|
339
|
+
return grpc.experimental.unary_unary(
|
|
340
|
+
request,
|
|
341
|
+
target,
|
|
342
|
+
'/jobs.v1.JobsService/GetJobQueue',
|
|
343
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobQueueRequest.SerializeToString,
|
|
344
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobQueueResponse.FromString,
|
|
345
|
+
options,
|
|
346
|
+
channel_credentials,
|
|
347
|
+
insecure,
|
|
348
|
+
call_credentials,
|
|
349
|
+
compression,
|
|
350
|
+
wait_for_ready,
|
|
351
|
+
timeout,
|
|
352
|
+
metadata,
|
|
353
|
+
_registered_method=True)
|
|
354
|
+
|
|
355
|
+
@staticmethod
|
|
356
|
+
def CancelJobs(request,
|
|
357
|
+
target,
|
|
358
|
+
options=(),
|
|
359
|
+
channel_credentials=None,
|
|
360
|
+
call_credentials=None,
|
|
361
|
+
insecure=False,
|
|
362
|
+
compression=None,
|
|
363
|
+
wait_for_ready=None,
|
|
364
|
+
timeout=None,
|
|
365
|
+
metadata=None):
|
|
366
|
+
return grpc.experimental.unary_unary(
|
|
367
|
+
request,
|
|
368
|
+
target,
|
|
369
|
+
'/jobs.v1.JobsService/CancelJobs',
|
|
370
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.CancelJobsRequest.SerializeToString,
|
|
371
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.CancelJobsResponse.FromString,
|
|
372
|
+
options,
|
|
373
|
+
channel_credentials,
|
|
374
|
+
insecure,
|
|
375
|
+
call_credentials,
|
|
376
|
+
compression,
|
|
377
|
+
wait_for_ready,
|
|
378
|
+
timeout,
|
|
379
|
+
metadata,
|
|
380
|
+
_registered_method=True)
|
|
381
|
+
|
|
382
|
+
@staticmethod
|
|
383
|
+
def FailAllInProgressJobs(request,
|
|
384
|
+
target,
|
|
385
|
+
options=(),
|
|
386
|
+
channel_credentials=None,
|
|
387
|
+
call_credentials=None,
|
|
388
|
+
insecure=False,
|
|
389
|
+
compression=None,
|
|
390
|
+
wait_for_ready=None,
|
|
391
|
+
timeout=None,
|
|
392
|
+
metadata=None):
|
|
393
|
+
return grpc.experimental.unary_unary(
|
|
394
|
+
request,
|
|
395
|
+
target,
|
|
396
|
+
'/jobs.v1.JobsService/FailAllInProgressJobs',
|
|
397
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.FailAllInProgressJobsRequest.SerializeToString,
|
|
398
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.FailAllInProgressJobsResponse.FromString,
|
|
399
|
+
options,
|
|
400
|
+
channel_credentials,
|
|
401
|
+
insecure,
|
|
402
|
+
call_credentials,
|
|
403
|
+
compression,
|
|
404
|
+
wait_for_ready,
|
|
405
|
+
timeout,
|
|
406
|
+
metadata,
|
|
407
|
+
_registered_method=True)
|
|
408
|
+
|
|
409
|
+
@staticmethod
|
|
410
|
+
def TailLogs(request,
|
|
411
|
+
target,
|
|
412
|
+
options=(),
|
|
413
|
+
channel_credentials=None,
|
|
414
|
+
call_credentials=None,
|
|
415
|
+
insecure=False,
|
|
416
|
+
compression=None,
|
|
417
|
+
wait_for_ready=None,
|
|
418
|
+
timeout=None,
|
|
419
|
+
metadata=None):
|
|
420
|
+
return grpc.experimental.unary_stream(
|
|
421
|
+
request,
|
|
422
|
+
target,
|
|
423
|
+
'/jobs.v1.JobsService/TailLogs',
|
|
424
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.TailLogsRequest.SerializeToString,
|
|
425
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.TailLogsResponse.FromString,
|
|
426
|
+
options,
|
|
427
|
+
channel_credentials,
|
|
428
|
+
insecure,
|
|
429
|
+
call_credentials,
|
|
430
|
+
compression,
|
|
431
|
+
wait_for_ready,
|
|
432
|
+
timeout,
|
|
433
|
+
metadata,
|
|
434
|
+
_registered_method=True)
|
|
435
|
+
|
|
436
|
+
@staticmethod
|
|
437
|
+
def GetJobStatus(request,
|
|
438
|
+
target,
|
|
439
|
+
options=(),
|
|
440
|
+
channel_credentials=None,
|
|
441
|
+
call_credentials=None,
|
|
442
|
+
insecure=False,
|
|
443
|
+
compression=None,
|
|
444
|
+
wait_for_ready=None,
|
|
445
|
+
timeout=None,
|
|
446
|
+
metadata=None):
|
|
447
|
+
return grpc.experimental.unary_unary(
|
|
448
|
+
request,
|
|
449
|
+
target,
|
|
450
|
+
'/jobs.v1.JobsService/GetJobStatus',
|
|
451
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobStatusRequest.SerializeToString,
|
|
452
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobStatusResponse.FromString,
|
|
453
|
+
options,
|
|
454
|
+
channel_credentials,
|
|
455
|
+
insecure,
|
|
456
|
+
call_credentials,
|
|
457
|
+
compression,
|
|
458
|
+
wait_for_ready,
|
|
459
|
+
timeout,
|
|
460
|
+
metadata,
|
|
461
|
+
_registered_method=True)
|
|
462
|
+
|
|
463
|
+
@staticmethod
|
|
464
|
+
def GetJobSubmittedTimestamp(request,
|
|
465
|
+
target,
|
|
466
|
+
options=(),
|
|
467
|
+
channel_credentials=None,
|
|
468
|
+
call_credentials=None,
|
|
469
|
+
insecure=False,
|
|
470
|
+
compression=None,
|
|
471
|
+
wait_for_ready=None,
|
|
472
|
+
timeout=None,
|
|
473
|
+
metadata=None):
|
|
474
|
+
return grpc.experimental.unary_unary(
|
|
475
|
+
request,
|
|
476
|
+
target,
|
|
477
|
+
'/jobs.v1.JobsService/GetJobSubmittedTimestamp',
|
|
478
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobSubmittedTimestampRequest.SerializeToString,
|
|
479
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobSubmittedTimestampResponse.FromString,
|
|
480
|
+
options,
|
|
481
|
+
channel_credentials,
|
|
482
|
+
insecure,
|
|
483
|
+
call_credentials,
|
|
484
|
+
compression,
|
|
485
|
+
wait_for_ready,
|
|
486
|
+
timeout,
|
|
487
|
+
metadata,
|
|
488
|
+
_registered_method=True)
|
|
489
|
+
|
|
490
|
+
@staticmethod
|
|
491
|
+
def GetJobEndedTimestamp(request,
|
|
492
|
+
target,
|
|
493
|
+
options=(),
|
|
494
|
+
channel_credentials=None,
|
|
495
|
+
call_credentials=None,
|
|
496
|
+
insecure=False,
|
|
497
|
+
compression=None,
|
|
498
|
+
wait_for_ready=None,
|
|
499
|
+
timeout=None,
|
|
500
|
+
metadata=None):
|
|
501
|
+
return grpc.experimental.unary_unary(
|
|
502
|
+
request,
|
|
503
|
+
target,
|
|
504
|
+
'/jobs.v1.JobsService/GetJobEndedTimestamp',
|
|
505
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobEndedTimestampRequest.SerializeToString,
|
|
506
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetJobEndedTimestampResponse.FromString,
|
|
507
|
+
options,
|
|
508
|
+
channel_credentials,
|
|
509
|
+
insecure,
|
|
510
|
+
call_credentials,
|
|
511
|
+
compression,
|
|
512
|
+
wait_for_ready,
|
|
513
|
+
timeout,
|
|
514
|
+
metadata,
|
|
515
|
+
_registered_method=True)
|
|
516
|
+
|
|
517
|
+
@staticmethod
|
|
518
|
+
def GetLogDirsForJobs(request,
|
|
519
|
+
target,
|
|
520
|
+
options=(),
|
|
521
|
+
channel_credentials=None,
|
|
522
|
+
call_credentials=None,
|
|
523
|
+
insecure=False,
|
|
524
|
+
compression=None,
|
|
525
|
+
wait_for_ready=None,
|
|
526
|
+
timeout=None,
|
|
527
|
+
metadata=None):
|
|
528
|
+
return grpc.experimental.unary_unary(
|
|
529
|
+
request,
|
|
530
|
+
target,
|
|
531
|
+
'/jobs.v1.JobsService/GetLogDirsForJobs',
|
|
532
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetLogDirsForJobsRequest.SerializeToString,
|
|
533
|
+
sky_dot_schemas_dot_generated_dot_jobsv1__pb2.GetLogDirsForJobsResponse.FromString,
|
|
534
|
+
options,
|
|
535
|
+
channel_credentials,
|
|
536
|
+
insecure,
|
|
537
|
+
call_credentials,
|
|
538
|
+
compression,
|
|
539
|
+
wait_for_ready,
|
|
540
|
+
timeout,
|
|
541
|
+
metadata,
|
|
542
|
+
_registered_method=True)
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: sky/schemas/generated/managed_jobsv1.proto
|
|
4
|
+
# Protobuf Python Version: 5.26.1
|
|
5
|
+
"""Generated protocol buffer code."""
|
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
|
9
|
+
from google.protobuf.internal import builder as _builder
|
|
10
|
+
# @@protoc_insertion_point(imports)
|
|
11
|
+
|
|
12
|
+
_sym_db = _symbol_database.Default()
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*sky/schemas/generated/managed_jobsv1.proto\x12\x0fmanaged_jobs.v1\"\x15\n\x06JobIds\x12\x0b\n\x03ids\x18\x01 \x03(\x03\"\x1c\n\nUserHashes\x12\x0e\n\x06hashes\x18\x01 \x03(\t\"\x1c\n\x08Statuses\x12\x10\n\x08statuses\x18\x01 \x03(\t\"\x18\n\x06\x46ields\x12\x0e\n\x06\x66ields\x18\x01 \x03(\t\" \n\nWorkspaces\x12\x12\n\nworkspaces\x18\x01 \x03(\t\"\x13\n\x11GetVersionRequest\"0\n\x12GetVersionResponse\x12\x1a\n\x12\x63ontroller_version\x18\x01 \x01(\t\"\xe1\x04\n\x12GetJobTableRequest\x12\x15\n\rskip_finished\x18\x01 \x01(\x08\x12?\n\x15\x61\x63\x63\x65ssible_workspaces\x18\x02 \x01(\x0b\x32\x1b.managed_jobs.v1.WorkspacesH\x00\x88\x01\x01\x12-\n\x07job_ids\x18\x03 \x01(\x0b\x32\x17.managed_jobs.v1.JobIdsH\x01\x88\x01\x01\x12\x1c\n\x0fworkspace_match\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x17\n\nname_match\x18\x05 \x01(\tH\x03\x88\x01\x01\x12\x17\n\npool_match\x18\x06 \x01(\tH\x04\x88\x01\x01\x12\x11\n\x04page\x18\x07 \x01(\x05H\x05\x88\x01\x01\x12\x12\n\x05limit\x18\x08 \x01(\x05H\x06\x88\x01\x01\x12\x35\n\x0buser_hashes\x18\t \x01(\x0b\x32\x1b.managed_jobs.v1.UserHashesH\x07\x88\x01\x01\x12\x30\n\x08statuses\x18\n \x01(\x0b\x32\x19.managed_jobs.v1.StatusesH\x08\x88\x01\x01\x12#\n\x1bshow_jobs_without_user_hash\x18\x0b \x01(\x08\x12,\n\x06\x66ields\x18\x0c \x01(\x0b\x32\x17.managed_jobs.v1.FieldsH\t\x88\x01\x01\x42\x18\n\x16_accessible_workspacesB\n\n\x08_job_idsB\x12\n\x10_workspace_matchB\r\n\x0b_name_matchB\r\n\x0b_pool_matchB\x07\n\x05_pageB\x08\n\x06_limitB\x0e\n\x0c_user_hashesB\x0b\n\t_statusesB\t\n\x07_fields\"\xcb\x08\n\x0eManagedJobInfo\x12\x0e\n\x06job_id\x18\x01 \x01(\x03\x12\x0f\n\x07task_id\x18\x02 \x01(\x03\x12\x10\n\x08job_name\x18\x03 \x01(\t\x12\x11\n\ttask_name\x18\x04 \x01(\t\x12\x14\n\x0cjob_duration\x18\x05 \x01(\x01\x12\x16\n\tworkspace\x18\x06 \x01(\tH\x00\x88\x01\x01\x12\x31\n\x06status\x18\x07 \x01(\x0e\x32!.managed_jobs.v1.ManagedJobStatus\x12@\n\x0eschedule_state\x18\x08 \x01(\x0e\x32(.managed_jobs.v1.ManagedJobScheduleState\x12\x11\n\tresources\x18\t \x01(\t\x12\x19\n\x11\x63luster_resources\x18\n \x01(\t\x12\x1e\n\x16\x63luster_resources_full\x18\x0b \x01(\t\x12\r\n\x05\x63loud\x18\x0c \x01(\t\x12\x0e\n\x06region\x18\r \x01(\t\x12\r\n\x05infra\x18\x0e \x01(\t\x12G\n\x0c\x61\x63\x63\x65lerators\x18\x0f \x03(\x0b\x32\x31.managed_jobs.v1.ManagedJobInfo.AcceleratorsEntry\x12\x16\n\x0erecovery_count\x18\x10 \x01(\x05\x12\x14\n\x07\x64\x65tails\x18\x11 \x01(\tH\x01\x88\x01\x01\x12\x1b\n\x0e\x66\x61ilure_reason\x18\x12 \x01(\tH\x02\x88\x01\x01\x12\x16\n\tuser_name\x18\x13 \x01(\tH\x03\x88\x01\x01\x12\x16\n\tuser_hash\x18\x14 \x01(\tH\x04\x88\x01\x01\x12\x19\n\x0csubmitted_at\x18\x15 \x01(\x01H\x05\x88\x01\x01\x12\x15\n\x08start_at\x18\x16 \x01(\x01H\x06\x88\x01\x01\x12\x13\n\x06\x65nd_at\x18\x17 \x01(\x01H\x07\x88\x01\x01\x12\x16\n\tuser_yaml\x18\x18 \x01(\tH\x08\x88\x01\x01\x12\x17\n\nentrypoint\x18\x19 \x01(\tH\t\x88\x01\x01\x12?\n\x08metadata\x18\x1a \x03(\x0b\x32-.managed_jobs.v1.ManagedJobInfo.MetadataEntry\x12\x11\n\x04pool\x18\x1b \x01(\tH\n\x88\x01\x01\x12\x16\n\tpool_hash\x18\x1c \x01(\tH\x0b\x88\x01\x01\x12\x14\n\x07_job_id\x18\x1d \x01(\x03H\x0c\x88\x01\x01\x1a\x33\n\x11\x41\x63\x63\x65leratorsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x02:\x02\x38\x01\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0c\n\n_workspaceB\n\n\x08_detailsB\x11\n\x0f_failure_reasonB\x0c\n\n_user_nameB\x0c\n\n_user_hashB\x0f\n\r_submitted_atB\x0b\n\t_start_atB\t\n\x07_end_atB\x0c\n\n_user_yamlB\r\n\x0b_entrypointB\x07\n\x05_poolB\x0c\n\n_pool_hashB\n\n\x08X_job_id\"\xf0\x01\n\x13GetJobTableResponse\x12-\n\x04jobs\x18\x01 \x03(\x0b\x32\x1f.managed_jobs.v1.ManagedJobInfo\x12\r\n\x05total\x18\x02 \x01(\x05\x12\x17\n\x0ftotal_no_filter\x18\x03 \x01(\x05\x12M\n\rstatus_counts\x18\x04 \x03(\x0b\x32\x36.managed_jobs.v1.GetJobTableResponse.StatusCountsEntry\x1a\x33\n\x11StatusCountsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"?\n\x19GetAllJobIdsByNameRequest\x12\x15\n\x08job_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\x0b\n\t_job_name\"-\n\x1aGetAllJobIdsByNameResponse\x12\x0f\n\x07job_ids\x18\x01 \x03(\x03\"\xd7\x01\n\x11\x43\x61ncelJobsRequest\x12\x19\n\x11\x63urrent_workspace\x18\x01 \x01(\t\x12\x16\n\tuser_hash\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x13\n\tall_users\x18\x03 \x01(\x08H\x00\x12*\n\x07job_ids\x18\x04 \x01(\x0b\x32\x17.managed_jobs.v1.JobIdsH\x00\x12\x12\n\x08job_name\x18\x05 \x01(\tH\x00\x12\x13\n\tpool_name\x18\x06 \x01(\tH\x00\x42\x17\n\x15\x63\x61ncellation_criteriaB\x0c\n\n_user_hash\"%\n\x12\x43\x61ncelJobsResponse\x12\x0f\n\x07message\x18\x01 \x01(\t\"\x97\x01\n\x11StreamLogsRequest\x12\x15\n\x08job_name\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x06job_id\x18\x02 \x01(\x03H\x01\x88\x01\x01\x12\x0e\n\x06\x66ollow\x18\x03 \x01(\x08\x12\x12\n\ncontroller\x18\x04 \x01(\x08\x12\x11\n\x04tail\x18\x05 \x01(\x05H\x02\x88\x01\x01\x42\x0b\n\t_job_nameB\t\n\x07_job_idB\x07\n\x05_tail\"L\n\x12StreamLogsResponse\x12\x10\n\x08log_line\x18\x01 \x01(\t\x12\x16\n\texit_code\x18\x02 \x01(\x05H\x00\x88\x01\x01\x42\x0c\n\n_exit_code*\x85\x04\n\x10ManagedJobStatus\x12\"\n\x1eMANAGED_JOB_STATUS_UNSPECIFIED\x10\x00\x12\x1e\n\x1aMANAGED_JOB_STATUS_PENDING\x10\x01\x12 \n\x1cMANAGED_JOB_STATUS_SUBMITTED\x10\x02\x12\x1f\n\x1bMANAGED_JOB_STATUS_STARTING\x10\x03\x12\x1e\n\x1aMANAGED_JOB_STATUS_RUNNING\x10\x04\x12!\n\x1dMANAGED_JOB_STATUS_RECOVERING\x10\x05\x12!\n\x1dMANAGED_JOB_STATUS_CANCELLING\x10\x06\x12 \n\x1cMANAGED_JOB_STATUS_SUCCEEDED\x10\x07\x12 \n\x1cMANAGED_JOB_STATUS_CANCELLED\x10\x08\x12\x1d\n\x19MANAGED_JOB_STATUS_FAILED\x10\t\x12#\n\x1fMANAGED_JOB_STATUS_FAILED_SETUP\x10\n\x12\'\n#MANAGED_JOB_STATUS_FAILED_PRECHECKS\x10\x0b\x12)\n%MANAGED_JOB_STATUS_FAILED_NO_RESOURCE\x10\x0c\x12(\n$MANAGED_JOB_STATUS_FAILED_CONTROLLER\x10\r*\x9e\x03\n\x17ManagedJobScheduleState\x12*\n&MANAGED_JOB_SCHEDULE_STATE_UNSPECIFIED\x10\x00\x12\x35\n-DEPRECATED_MANAGED_JOB_SCHEDULE_STATE_INVALID\x10\x01\x1a\x02\x08\x01\x12\'\n#MANAGED_JOB_SCHEDULE_STATE_INACTIVE\x10\x02\x12&\n\"MANAGED_JOB_SCHEDULE_STATE_WAITING\x10\x03\x12,\n(MANAGED_JOB_SCHEDULE_STATE_ALIVE_WAITING\x10\x04\x12(\n$MANAGED_JOB_SCHEDULE_STATE_LAUNCHING\x10\x05\x12,\n(MANAGED_JOB_SCHEDULE_STATE_ALIVE_BACKOFF\x10\x06\x12$\n MANAGED_JOB_SCHEDULE_STATE_ALIVE\x10\x07\x12#\n\x1fMANAGED_JOB_SCHEDULE_STATE_DONE\x10\x08\x32\xe4\x03\n\x12ManagedJobsService\x12U\n\nGetVersion\x12\".managed_jobs.v1.GetVersionRequest\x1a#.managed_jobs.v1.GetVersionResponse\x12X\n\x0bGetJobTable\x12#.managed_jobs.v1.GetJobTableRequest\x1a$.managed_jobs.v1.GetJobTableResponse\x12m\n\x12GetAllJobIdsByName\x12*.managed_jobs.v1.GetAllJobIdsByNameRequest\x1a+.managed_jobs.v1.GetAllJobIdsByNameResponse\x12U\n\nCancelJobs\x12\".managed_jobs.v1.CancelJobsRequest\x1a#.managed_jobs.v1.CancelJobsResponse\x12W\n\nStreamLogs\x12\".managed_jobs.v1.StreamLogsRequest\x1a#.managed_jobs.v1.StreamLogsResponse0\x01\x62\x06proto3')
|
|
18
|
+
|
|
19
|
+
_globals = globals()
|
|
20
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
21
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'sky.schemas.generated.managed_jobsv1_pb2', _globals)
|
|
22
|
+
if not _descriptor._USE_C_DESCRIPTORS:
|
|
23
|
+
DESCRIPTOR._loaded_options = None
|
|
24
|
+
_globals['_MANAGEDJOBSCHEDULESTATE'].values_by_name["DEPRECATED_MANAGED_JOB_SCHEDULE_STATE_INVALID"]._loaded_options = None
|
|
25
|
+
_globals['_MANAGEDJOBSCHEDULESTATE'].values_by_name["DEPRECATED_MANAGED_JOB_SCHEDULE_STATE_INVALID"]._serialized_options = b'\010\001'
|
|
26
|
+
_globals['_MANAGEDJOBINFO_ACCELERATORSENTRY']._loaded_options = None
|
|
27
|
+
_globals['_MANAGEDJOBINFO_ACCELERATORSENTRY']._serialized_options = b'8\001'
|
|
28
|
+
_globals['_MANAGEDJOBINFO_METADATAENTRY']._loaded_options = None
|
|
29
|
+
_globals['_MANAGEDJOBINFO_METADATAENTRY']._serialized_options = b'8\001'
|
|
30
|
+
_globals['_GETJOBTABLERESPONSE_STATUSCOUNTSENTRY']._loaded_options = None
|
|
31
|
+
_globals['_GETJOBTABLERESPONSE_STATUSCOUNTSENTRY']._serialized_options = b'8\001'
|
|
32
|
+
_globals['_MANAGEDJOBSTATUS']._serialized_start=2836
|
|
33
|
+
_globals['_MANAGEDJOBSTATUS']._serialized_end=3353
|
|
34
|
+
_globals['_MANAGEDJOBSCHEDULESTATE']._serialized_start=3356
|
|
35
|
+
_globals['_MANAGEDJOBSCHEDULESTATE']._serialized_end=3770
|
|
36
|
+
_globals['_JOBIDS']._serialized_start=63
|
|
37
|
+
_globals['_JOBIDS']._serialized_end=84
|
|
38
|
+
_globals['_USERHASHES']._serialized_start=86
|
|
39
|
+
_globals['_USERHASHES']._serialized_end=114
|
|
40
|
+
_globals['_STATUSES']._serialized_start=116
|
|
41
|
+
_globals['_STATUSES']._serialized_end=144
|
|
42
|
+
_globals['_FIELDS']._serialized_start=146
|
|
43
|
+
_globals['_FIELDS']._serialized_end=170
|
|
44
|
+
_globals['_WORKSPACES']._serialized_start=172
|
|
45
|
+
_globals['_WORKSPACES']._serialized_end=204
|
|
46
|
+
_globals['_GETVERSIONREQUEST']._serialized_start=206
|
|
47
|
+
_globals['_GETVERSIONREQUEST']._serialized_end=225
|
|
48
|
+
_globals['_GETVERSIONRESPONSE']._serialized_start=227
|
|
49
|
+
_globals['_GETVERSIONRESPONSE']._serialized_end=275
|
|
50
|
+
_globals['_GETJOBTABLEREQUEST']._serialized_start=278
|
|
51
|
+
_globals['_GETJOBTABLEREQUEST']._serialized_end=887
|
|
52
|
+
_globals['_MANAGEDJOBINFO']._serialized_start=890
|
|
53
|
+
_globals['_MANAGEDJOBINFO']._serialized_end=1989
|
|
54
|
+
_globals['_MANAGEDJOBINFO_ACCELERATORSENTRY']._serialized_start=1711
|
|
55
|
+
_globals['_MANAGEDJOBINFO_ACCELERATORSENTRY']._serialized_end=1762
|
|
56
|
+
_globals['_MANAGEDJOBINFO_METADATAENTRY']._serialized_start=1764
|
|
57
|
+
_globals['_MANAGEDJOBINFO_METADATAENTRY']._serialized_end=1811
|
|
58
|
+
_globals['_GETJOBTABLERESPONSE']._serialized_start=1992
|
|
59
|
+
_globals['_GETJOBTABLERESPONSE']._serialized_end=2232
|
|
60
|
+
_globals['_GETJOBTABLERESPONSE_STATUSCOUNTSENTRY']._serialized_start=2181
|
|
61
|
+
_globals['_GETJOBTABLERESPONSE_STATUSCOUNTSENTRY']._serialized_end=2232
|
|
62
|
+
_globals['_GETALLJOBIDSBYNAMEREQUEST']._serialized_start=2234
|
|
63
|
+
_globals['_GETALLJOBIDSBYNAMEREQUEST']._serialized_end=2297
|
|
64
|
+
_globals['_GETALLJOBIDSBYNAMERESPONSE']._serialized_start=2299
|
|
65
|
+
_globals['_GETALLJOBIDSBYNAMERESPONSE']._serialized_end=2344
|
|
66
|
+
_globals['_CANCELJOBSREQUEST']._serialized_start=2347
|
|
67
|
+
_globals['_CANCELJOBSREQUEST']._serialized_end=2562
|
|
68
|
+
_globals['_CANCELJOBSRESPONSE']._serialized_start=2564
|
|
69
|
+
_globals['_CANCELJOBSRESPONSE']._serialized_end=2601
|
|
70
|
+
_globals['_STREAMLOGSREQUEST']._serialized_start=2604
|
|
71
|
+
_globals['_STREAMLOGSREQUEST']._serialized_end=2755
|
|
72
|
+
_globals['_STREAMLOGSRESPONSE']._serialized_start=2757
|
|
73
|
+
_globals['_STREAMLOGSRESPONSE']._serialized_end=2833
|
|
74
|
+
_globals['_MANAGEDJOBSSERVICE']._serialized_start=3773
|
|
75
|
+
_globals['_MANAGEDJOBSSERVICE']._serialized_end=4257
|
|
76
|
+
# @@protoc_insertion_point(module_scope)
|