dstack 0.0.9__py3-none-any.whl → 0.20.7__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.
- dstack/_internal/cli/commands/__init__.py +80 -0
- dstack/_internal/cli/commands/apply.py +100 -0
- dstack/_internal/cli/commands/attach.py +161 -0
- dstack/_internal/cli/commands/completion.py +22 -0
- dstack/_internal/cli/commands/delete.py +44 -0
- dstack/_internal/cli/commands/event.py +168 -0
- dstack/_internal/cli/commands/fleet.py +161 -0
- dstack/_internal/cli/commands/gateway.py +159 -0
- dstack/_internal/cli/commands/init.py +64 -0
- dstack/_internal/cli/commands/login.py +352 -0
- dstack/_internal/cli/commands/logs.py +62 -0
- dstack/_internal/cli/commands/metrics.py +153 -0
- dstack/_internal/cli/commands/offer.py +146 -0
- dstack/_internal/cli/commands/project.py +259 -0
- dstack/_internal/cli/commands/ps.py +81 -0
- dstack/_internal/cli/commands/run.py +69 -0
- dstack/_internal/cli/commands/secrets.py +92 -0
- dstack/_internal/cli/commands/server.py +96 -0
- dstack/_internal/cli/commands/stop.py +26 -0
- dstack/_internal/cli/commands/volume.py +117 -0
- dstack/_internal/cli/main.py +101 -0
- dstack/_internal/cli/models/gateways.py +16 -0
- dstack/_internal/cli/models/offers.py +47 -0
- dstack/_internal/cli/models/runs.py +16 -0
- dstack/_internal/cli/services/args.py +31 -0
- dstack/_internal/cli/services/completion.py +91 -0
- dstack/_internal/cli/services/configurators/__init__.py +86 -0
- dstack/_internal/cli/services/configurators/base.py +103 -0
- dstack/_internal/cli/services/configurators/fleet.py +475 -0
- dstack/_internal/cli/services/configurators/gateway.py +231 -0
- dstack/_internal/cli/services/configurators/run.py +882 -0
- dstack/_internal/cli/services/configurators/volume.py +222 -0
- dstack/_internal/cli/services/events.py +68 -0
- dstack/_internal/cli/services/profile.py +182 -0
- dstack/_internal/cli/services/repos.py +71 -0
- dstack/_internal/cli/services/resources.py +54 -0
- dstack/_internal/cli/utils/common.py +159 -0
- dstack/_internal/cli/utils/fleet.py +106 -0
- dstack/_internal/cli/utils/gateway.py +56 -0
- dstack/_internal/cli/utils/gpu.py +178 -0
- dstack/_internal/cli/utils/rich.py +156 -0
- dstack/_internal/cli/utils/run.py +517 -0
- dstack/_internal/cli/utils/secrets.py +25 -0
- dstack/_internal/cli/utils/updates.py +98 -0
- dstack/_internal/cli/utils/volume.py +58 -0
- dstack/_internal/compat.py +3 -0
- dstack/_internal/core/backends/amddevcloud/__init__.py +1 -0
- dstack/_internal/core/backends/amddevcloud/backend.py +16 -0
- dstack/_internal/core/backends/amddevcloud/compute.py +5 -0
- dstack/_internal/core/backends/amddevcloud/configurator.py +29 -0
- dstack/_internal/core/backends/aws/auth.py +30 -0
- dstack/_internal/core/backends/aws/backend.py +31 -0
- dstack/_internal/core/backends/aws/compute.py +1153 -0
- dstack/_internal/core/backends/aws/configurator.py +191 -0
- dstack/_internal/core/backends/aws/models.py +135 -0
- dstack/_internal/core/backends/aws/resources.py +700 -0
- dstack/_internal/core/backends/azure/auth.py +39 -0
- dstack/_internal/core/backends/azure/backend.py +21 -0
- dstack/_internal/core/backends/azure/compute.py +676 -0
- dstack/_internal/core/backends/azure/configurator.py +472 -0
- dstack/_internal/core/backends/azure/models.py +98 -0
- dstack/_internal/core/backends/azure/resources.py +116 -0
- dstack/_internal/core/backends/azure/utils.py +42 -0
- dstack/_internal/core/backends/base/backend.py +18 -0
- dstack/_internal/core/backends/base/compute.py +1101 -0
- dstack/_internal/core/backends/base/configurator.py +117 -0
- dstack/_internal/core/backends/base/models.py +24 -0
- dstack/_internal/core/backends/base/offers.py +232 -0
- dstack/_internal/core/backends/cloudrift/api_client.py +220 -0
- dstack/_internal/core/backends/cloudrift/backend.py +16 -0
- dstack/_internal/core/backends/cloudrift/compute.py +138 -0
- dstack/_internal/core/backends/cloudrift/configurator.py +72 -0
- dstack/_internal/core/backends/cloudrift/models.py +40 -0
- dstack/_internal/core/backends/configurators.py +181 -0
- dstack/_internal/core/backends/cudo/__init__.py +0 -0
- dstack/_internal/core/backends/cudo/api_client.py +111 -0
- dstack/_internal/core/backends/cudo/backend.py +16 -0
- dstack/_internal/core/backends/cudo/compute.py +174 -0
- dstack/_internal/core/backends/cudo/configurator.py +63 -0
- dstack/_internal/core/backends/cudo/models.py +37 -0
- dstack/_internal/core/backends/datacrunch/__init__.py +1 -0
- dstack/_internal/core/backends/datacrunch/backend.py +18 -0
- dstack/_internal/core/backends/datacrunch/compute.py +8 -0
- dstack/_internal/core/backends/datacrunch/configurator.py +17 -0
- dstack/_internal/core/backends/digitalocean/__init__.py +1 -0
- dstack/_internal/core/backends/digitalocean/backend.py +16 -0
- dstack/_internal/core/backends/digitalocean/compute.py +5 -0
- dstack/_internal/core/backends/digitalocean/configurator.py +31 -0
- dstack/_internal/core/backends/digitalocean_base/__init__.py +1 -0
- dstack/_internal/core/backends/digitalocean_base/api_client.py +104 -0
- dstack/_internal/core/backends/digitalocean_base/backend.py +5 -0
- dstack/_internal/core/backends/digitalocean_base/compute.py +174 -0
- dstack/_internal/core/backends/digitalocean_base/configurator.py +57 -0
- dstack/_internal/core/backends/digitalocean_base/models.py +43 -0
- dstack/_internal/core/backends/dstack/__init__.py +0 -0
- dstack/_internal/core/backends/dstack/models.py +26 -0
- dstack/_internal/core/backends/features.py +74 -0
- dstack/_internal/core/backends/gcp/__init__.py +0 -0
- dstack/_internal/core/backends/gcp/auth.py +57 -0
- dstack/_internal/core/backends/gcp/backend.py +17 -0
- dstack/_internal/core/backends/gcp/compute.py +1257 -0
- dstack/_internal/core/backends/gcp/configurator.py +206 -0
- dstack/_internal/core/backends/gcp/features/__init__.py +0 -0
- dstack/_internal/core/backends/gcp/features/tcpx.py +65 -0
- dstack/_internal/core/backends/gcp/models.py +160 -0
- dstack/_internal/core/backends/gcp/resources.py +585 -0
- dstack/_internal/core/backends/hotaisle/__init__.py +1 -0
- dstack/_internal/core/backends/hotaisle/api_client.py +101 -0
- dstack/_internal/core/backends/hotaisle/backend.py +16 -0
- dstack/_internal/core/backends/hotaisle/compute.py +188 -0
- dstack/_internal/core/backends/hotaisle/configurator.py +66 -0
- dstack/_internal/core/backends/hotaisle/models.py +45 -0
- dstack/_internal/core/backends/kubernetes/__init__.py +0 -0
- dstack/_internal/core/backends/kubernetes/backend.py +16 -0
- dstack/_internal/core/backends/kubernetes/compute.py +1077 -0
- dstack/_internal/core/backends/kubernetes/configurator.py +61 -0
- dstack/_internal/core/backends/kubernetes/models.py +71 -0
- dstack/_internal/core/backends/kubernetes/utils.py +81 -0
- dstack/_internal/core/backends/lambdalabs/__init__.py +0 -0
- dstack/_internal/core/backends/lambdalabs/api_client.py +87 -0
- dstack/_internal/core/backends/lambdalabs/backend.py +17 -0
- dstack/_internal/core/backends/lambdalabs/compute.py +233 -0
- dstack/_internal/core/backends/lambdalabs/configurator.py +65 -0
- dstack/_internal/core/backends/lambdalabs/models.py +37 -0
- dstack/_internal/core/backends/local/__init__.py +0 -0
- dstack/_internal/core/backends/local/backend.py +14 -0
- dstack/_internal/core/backends/local/compute.py +130 -0
- dstack/_internal/core/backends/models.py +158 -0
- dstack/_internal/core/backends/nebius/__init__.py +0 -0
- dstack/_internal/core/backends/nebius/backend.py +16 -0
- dstack/_internal/core/backends/nebius/compute.py +401 -0
- dstack/_internal/core/backends/nebius/configurator.py +98 -0
- dstack/_internal/core/backends/nebius/models.py +185 -0
- dstack/_internal/core/backends/nebius/resources.py +433 -0
- dstack/_internal/core/backends/oci/__init__.py +0 -0
- dstack/_internal/core/backends/oci/auth.py +21 -0
- dstack/_internal/core/backends/oci/backend.py +16 -0
- dstack/_internal/core/backends/oci/compute.py +209 -0
- dstack/_internal/core/backends/oci/configurator.py +156 -0
- dstack/_internal/core/backends/oci/exceptions.py +15 -0
- dstack/_internal/core/backends/oci/models.py +87 -0
- dstack/_internal/core/backends/oci/region.py +86 -0
- dstack/_internal/core/backends/oci/resources.py +836 -0
- dstack/_internal/core/backends/runpod/__init__.py +0 -0
- dstack/_internal/core/backends/runpod/api_client.py +627 -0
- dstack/_internal/core/backends/runpod/backend.py +16 -0
- dstack/_internal/core/backends/runpod/compute.py +444 -0
- dstack/_internal/core/backends/runpod/configurator.py +63 -0
- dstack/_internal/core/backends/runpod/models.py +54 -0
- dstack/_internal/core/backends/template/__init__.py +0 -0
- dstack/_internal/core/backends/template/backend.py.jinja +16 -0
- dstack/_internal/core/backends/template/compute.py.jinja +95 -0
- dstack/_internal/core/backends/template/configurator.py.jinja +69 -0
- dstack/_internal/core/backends/template/models.py.jinja +62 -0
- dstack/_internal/core/backends/tensordock/models.py +40 -0
- dstack/_internal/core/backends/vastai/__init__.py +0 -0
- dstack/_internal/core/backends/vastai/api_client.py +143 -0
- dstack/_internal/core/backends/vastai/backend.py +16 -0
- dstack/_internal/core/backends/vastai/compute.py +141 -0
- dstack/_internal/core/backends/vastai/configurator.py +69 -0
- dstack/_internal/core/backends/vastai/models.py +37 -0
- dstack/_internal/core/backends/verda/__init__.py +0 -0
- dstack/_internal/core/backends/verda/backend.py +16 -0
- dstack/_internal/core/backends/verda/compute.py +266 -0
- dstack/_internal/core/backends/verda/configurator.py +73 -0
- dstack/_internal/core/backends/verda/models.py +38 -0
- dstack/_internal/core/backends/vultr/__init__.py +0 -0
- dstack/_internal/core/backends/vultr/api_client.py +116 -0
- dstack/_internal/core/backends/vultr/backend.py +16 -0
- dstack/_internal/core/backends/vultr/compute.py +167 -0
- dstack/_internal/core/backends/vultr/configurator.py +71 -0
- dstack/_internal/core/backends/vultr/models.py +34 -0
- dstack/_internal/core/compatibility/__init__.py +0 -0
- dstack/_internal/core/compatibility/events.py +13 -0
- dstack/_internal/core/compatibility/fleets.py +58 -0
- dstack/_internal/core/compatibility/gateways.py +39 -0
- dstack/_internal/core/compatibility/gpus.py +13 -0
- dstack/_internal/core/compatibility/logs.py +14 -0
- dstack/_internal/core/compatibility/runs.py +86 -0
- dstack/_internal/core/compatibility/volumes.py +37 -0
- dstack/_internal/core/consts.py +8 -0
- dstack/_internal/core/errors.py +160 -0
- dstack/_internal/core/models/__init__.py +0 -0
- dstack/_internal/core/models/auth.py +28 -0
- dstack/_internal/core/models/backends/__init__.py +0 -0
- dstack/_internal/core/models/backends/base.py +48 -0
- dstack/_internal/core/models/common.py +143 -0
- dstack/_internal/core/models/compute_groups.py +39 -0
- dstack/_internal/core/models/config.py +28 -0
- dstack/_internal/core/models/configurations.py +1123 -0
- dstack/_internal/core/models/envs.py +149 -0
- dstack/_internal/core/models/events.py +98 -0
- dstack/_internal/core/models/files.py +67 -0
- dstack/_internal/core/models/fleets.py +437 -0
- dstack/_internal/core/models/gateways.py +146 -0
- dstack/_internal/core/models/gpus.py +45 -0
- dstack/_internal/core/models/health.py +28 -0
- dstack/_internal/core/models/instances.py +346 -0
- dstack/_internal/core/models/logs.py +27 -0
- dstack/_internal/core/models/metrics.py +14 -0
- dstack/_internal/core/models/placement.py +27 -0
- dstack/_internal/core/models/profiles.py +431 -0
- dstack/_internal/core/models/projects.py +46 -0
- dstack/_internal/core/models/repos/__init__.py +34 -0
- dstack/_internal/core/models/repos/base.py +36 -0
- dstack/_internal/core/models/repos/local.py +96 -0
- dstack/_internal/core/models/repos/remote.py +341 -0
- dstack/_internal/core/models/repos/virtual.py +85 -0
- dstack/_internal/core/models/resources.py +424 -0
- dstack/_internal/core/models/routers.py +24 -0
- dstack/_internal/core/models/runs.py +618 -0
- dstack/_internal/core/models/secrets.py +16 -0
- dstack/_internal/core/models/server.py +7 -0
- dstack/_internal/core/models/services.py +76 -0
- dstack/_internal/core/models/unix.py +53 -0
- dstack/_internal/core/models/users.py +60 -0
- dstack/_internal/core/models/volumes.py +221 -0
- dstack/_internal/core/services/__init__.py +16 -0
- dstack/_internal/core/services/api_client.py +15 -0
- dstack/_internal/core/services/configs/__init__.py +116 -0
- dstack/_internal/core/services/diff.py +71 -0
- dstack/_internal/core/services/logs.py +58 -0
- dstack/_internal/core/services/profiles.py +46 -0
- dstack/_internal/core/services/repos.py +236 -0
- dstack/_internal/core/services/ssh/__init__.py +27 -0
- dstack/_internal/core/services/ssh/attach.py +241 -0
- dstack/_internal/core/services/ssh/client.py +113 -0
- dstack/_internal/core/services/ssh/key_manager.py +53 -0
- dstack/_internal/core/services/ssh/ports.py +89 -0
- dstack/_internal/core/services/ssh/tunnel.py +337 -0
- dstack/_internal/proxy/__init__.py +8 -0
- dstack/_internal/proxy/gateway/__init__.py +0 -0
- dstack/_internal/proxy/gateway/app.py +89 -0
- dstack/_internal/proxy/gateway/auth.py +26 -0
- dstack/_internal/proxy/gateway/const.py +7 -0
- dstack/_internal/proxy/gateway/deps.py +73 -0
- dstack/_internal/proxy/gateway/main.py +17 -0
- dstack/_internal/proxy/gateway/models.py +23 -0
- dstack/_internal/proxy/gateway/repo/__init__.py +0 -0
- dstack/_internal/proxy/gateway/repo/repo.py +121 -0
- dstack/_internal/proxy/gateway/repo/state_v1.py +164 -0
- dstack/_internal/proxy/gateway/resources/nginx/00-log-format.conf +11 -0
- dstack/_internal/proxy/gateway/resources/nginx/entrypoint.jinja2 +27 -0
- dstack/_internal/proxy/gateway/resources/nginx/router_workers.jinja2 +23 -0
- dstack/_internal/proxy/gateway/resources/nginx/service.jinja2 +105 -0
- dstack/_internal/proxy/gateway/routers/__init__.py +0 -0
- dstack/_internal/proxy/gateway/routers/auth.py +10 -0
- dstack/_internal/proxy/gateway/routers/config.py +28 -0
- dstack/_internal/proxy/gateway/routers/registry.py +124 -0
- dstack/_internal/proxy/gateway/routers/stats.py +18 -0
- dstack/_internal/proxy/gateway/schemas/__init__.py +0 -0
- dstack/_internal/proxy/gateway/schemas/common.py +5 -0
- dstack/_internal/proxy/gateway/schemas/config.py +9 -0
- dstack/_internal/proxy/gateway/schemas/registry.py +63 -0
- dstack/_internal/proxy/gateway/schemas/stats.py +15 -0
- dstack/_internal/proxy/gateway/services/__init__.py +0 -0
- dstack/_internal/proxy/gateway/services/model_routers/__init__.py +18 -0
- dstack/_internal/proxy/gateway/services/model_routers/base.py +91 -0
- dstack/_internal/proxy/gateway/services/model_routers/sglang.py +269 -0
- dstack/_internal/proxy/gateway/services/nginx.py +455 -0
- dstack/_internal/proxy/gateway/services/registry.py +426 -0
- dstack/_internal/proxy/gateway/services/server_client.py +95 -0
- dstack/_internal/proxy/gateway/services/stats.py +170 -0
- dstack/_internal/proxy/gateway/testing/__init__.py +0 -0
- dstack/_internal/proxy/gateway/testing/common.py +13 -0
- dstack/_internal/proxy/lib/__init__.py +0 -0
- dstack/_internal/proxy/lib/auth.py +7 -0
- dstack/_internal/proxy/lib/deps.py +106 -0
- dstack/_internal/proxy/lib/errors.py +14 -0
- dstack/_internal/proxy/lib/models.py +112 -0
- dstack/_internal/proxy/lib/repo.py +27 -0
- dstack/_internal/proxy/lib/routers/__init__.py +0 -0
- dstack/_internal/proxy/lib/routers/model_proxy.py +102 -0
- dstack/_internal/proxy/lib/schemas/__init__.py +0 -0
- dstack/_internal/proxy/lib/schemas/model_proxy.py +77 -0
- dstack/_internal/proxy/lib/services/__init__.py +0 -0
- dstack/_internal/proxy/lib/services/model_proxy/__init__.py +0 -0
- dstack/_internal/proxy/lib/services/model_proxy/clients/__init__.py +0 -0
- dstack/_internal/proxy/lib/services/model_proxy/clients/base.py +18 -0
- dstack/_internal/proxy/lib/services/model_proxy/clients/openai.py +67 -0
- dstack/_internal/proxy/lib/services/model_proxy/clients/tgi.py +208 -0
- dstack/_internal/proxy/lib/services/model_proxy/model_proxy.py +23 -0
- dstack/_internal/proxy/lib/services/service_connection.py +160 -0
- dstack/_internal/proxy/lib/testing/__init__.py +0 -0
- dstack/_internal/proxy/lib/testing/auth.py +11 -0
- dstack/_internal/proxy/lib/testing/common.py +51 -0
- dstack/_internal/server/__init__.py +0 -0
- dstack/_internal/server/alembic.ini +100 -0
- dstack/_internal/server/app.py +432 -0
- dstack/_internal/server/background/__init__.py +142 -0
- dstack/_internal/server/background/tasks/__init__.py +0 -0
- dstack/_internal/server/background/tasks/common.py +24 -0
- dstack/_internal/server/background/tasks/process_compute_groups.py +167 -0
- dstack/_internal/server/background/tasks/process_events.py +17 -0
- dstack/_internal/server/background/tasks/process_fleets.py +289 -0
- dstack/_internal/server/background/tasks/process_gateways.py +188 -0
- dstack/_internal/server/background/tasks/process_idle_volumes.py +145 -0
- dstack/_internal/server/background/tasks/process_instances.py +1186 -0
- dstack/_internal/server/background/tasks/process_metrics.py +172 -0
- dstack/_internal/server/background/tasks/process_placement_groups.py +104 -0
- dstack/_internal/server/background/tasks/process_probes.py +164 -0
- dstack/_internal/server/background/tasks/process_prometheus_metrics.py +150 -0
- dstack/_internal/server/background/tasks/process_running_jobs.py +1238 -0
- dstack/_internal/server/background/tasks/process_runs.py +842 -0
- dstack/_internal/server/background/tasks/process_submitted_jobs.py +1106 -0
- dstack/_internal/server/background/tasks/process_terminating_jobs.py +108 -0
- dstack/_internal/server/background/tasks/process_volumes.py +129 -0
- dstack/_internal/server/compatibility/__init__.py +0 -0
- dstack/_internal/server/compatibility/common.py +20 -0
- dstack/_internal/server/compatibility/gpus.py +22 -0
- dstack/_internal/server/db.py +127 -0
- dstack/_internal/server/deps.py +19 -0
- dstack/_internal/server/main.py +4 -0
- dstack/_internal/server/migrations/__init__.py +0 -0
- dstack/_internal/server/migrations/env.py +112 -0
- dstack/_internal/server/migrations/script.py.mako +28 -0
- dstack/_internal/server/migrations/versions/006512f572b4_add_projects_original_name.py +38 -0
- dstack/_internal/server/migrations/versions/065588ec72b8_add_vultr_to_backendtype_enum.py +81 -0
- dstack/_internal/server/migrations/versions/06e977bc61c7_add_usermodel_deleted_and_original_name.py +45 -0
- dstack/_internal/server/migrations/versions/0e33559e16ed_update_instancestatus.py +64 -0
- dstack/_internal/server/migrations/versions/112753bc17dd_remove_nullable_fields.py +50 -0
- dstack/_internal/server/migrations/versions/1338b788b612_reverse_job_instance_relationship.py +71 -0
- dstack/_internal/server/migrations/versions/14f2cb002fc2_add_jobmodel_removed_flag.py +44 -0
- dstack/_internal/server/migrations/versions/1a48dfe44a40_rework_termination_handling.py +42 -0
- dstack/_internal/server/migrations/versions/1aa9638ad963_added_email_index.py +31 -0
- dstack/_internal/server/migrations/versions/1e3fb39ef74b_add_remote_connection_details.py +26 -0
- dstack/_internal/server/migrations/versions/1e76fb0dde87_add_jobmodel_inactivity_secs.py +32 -0
- dstack/_internal/server/migrations/versions/20166748b60c_add_jobmodel_disconnected_at.py +100 -0
- dstack/_internal/server/migrations/versions/22d74df9897e_add_events_and_event_targets.py +99 -0
- dstack/_internal/server/migrations/versions/23e01c56279a_make_blob_nullable.py +32 -0
- dstack/_internal/server/migrations/versions/2498ab323443_add_fleetmodel_consolidation_attempt_.py +44 -0
- dstack/_internal/server/migrations/versions/252d3743b641_.py +40 -0
- dstack/_internal/server/migrations/versions/25479f540245_add_probes.py +43 -0
- dstack/_internal/server/migrations/versions/27d3e55759fa_add_pools.py +152 -0
- dstack/_internal/server/migrations/versions/29826f417010_remove_instancemodel_retry_policy.py +34 -0
- dstack/_internal/server/migrations/versions/29c08c6a8cb3_.py +36 -0
- dstack/_internal/server/migrations/versions/35e90e1b0d3e_add_rolling_deployment_fields.py +42 -0
- dstack/_internal/server/migrations/versions/35f732ee4cf5_add_projectmodel_is_public.py +39 -0
- dstack/_internal/server/migrations/versions/3cf77fb8bcf1_store_repo_clone_url.py +85 -0
- dstack/_internal/server/migrations/versions/3d7f6c2ec000_add_jobmodel_registered.py +28 -0
- dstack/_internal/server/migrations/versions/3dbdce90d0e0_fix_code_uq_constraint.py +33 -0
- dstack/_internal/server/migrations/versions/48ad3ecbaea2_do_not_delete_projects_and_runs.py +46 -0
- dstack/_internal/server/migrations/versions/4ae1a5b0e7f1_add_run_list_index.py +34 -0
- dstack/_internal/server/migrations/versions/4b4319398164_introduce_runs_processing.py +144 -0
- dstack/_internal/server/migrations/versions/50dd7ea98639_index_status_columns.py +55 -0
- dstack/_internal/server/migrations/versions/51d45659d574_add_instancemodel_blocks_fields.py +43 -0
- dstack/_internal/server/migrations/versions/54a77e19c64c_add_manager_project_role.py +67 -0
- dstack/_internal/server/migrations/versions/555138b1f77f_change_instancemodel_for_asynchronous_.py +61 -0
- dstack/_internal/server/migrations/versions/58aa5162dcc3_add_gatewaymodel_configuration.py +32 -0
- dstack/_internal/server/migrations/versions/5ad8debc8fe6_fixes_for_psql.py +329 -0
- dstack/_internal/server/migrations/versions/5ec538b70e71_replace_instansestatus.py +31 -0
- dstack/_internal/server/migrations/versions/5f1707c525d2_add_filearchivemodel.py +39 -0
- dstack/_internal/server/migrations/versions/5fd659afca82_add_ix_instances_fleet_id.py +31 -0
- dstack/_internal/server/migrations/versions/60e444118b6d_add_jobprometheusmetrics.py +40 -0
- dstack/_internal/server/migrations/versions/63c3f19cb184_add_jobterminationreason_inactivity_.py +83 -0
- dstack/_internal/server/migrations/versions/644b8a114187_add_secretmodel.py +49 -0
- dstack/_internal/server/migrations/versions/686fb8341ea5_add_user_emails.py +32 -0
- dstack/_internal/server/migrations/versions/6c1a9d6530ee_add_jobmodel_exit_status.py +26 -0
- dstack/_internal/server/migrations/versions/706e0acc3a7d_add_runmodel_desired_replica_counts.py +26 -0
- dstack/_internal/server/migrations/versions/710e5b3fac8f_add_encryption.py +54 -0
- dstack/_internal/server/migrations/versions/728b1488b1b4_add_instance_health.py +50 -0
- dstack/_internal/server/migrations/versions/74a1f55209bd_store_enums_as_strings.py +484 -0
- dstack/_internal/server/migrations/versions/7b24b1c8eba7_add_instancemodel_last_processed_at.py +68 -0
- dstack/_internal/server/migrations/versions/7ba3b59d7ca6_add_runmodel_resubmission_attempt.py +35 -0
- dstack/_internal/server/migrations/versions/7bc2586e8b9e_make_instancemodel_pool_id_optional.py +36 -0
- dstack/_internal/server/migrations/versions/7d1ec2b920ac_add_computegroupmodel.py +91 -0
- dstack/_internal/server/migrations/versions/803c7e9ed85d_add_jobmodel_job_runtime_data.py +32 -0
- dstack/_internal/server/migrations/versions/82b32a135ea2_.py +58 -0
- dstack/_internal/server/migrations/versions/866ec1d67184_replace_retrypolicy_limit_with_.py +93 -0
- dstack/_internal/server/migrations/versions/903c91e24634_add_instances_termination_reason_message.py +34 -0
- dstack/_internal/server/migrations/versions/91a12fff6c76_add_repocredsmodel.py +43 -0
- dstack/_internal/server/migrations/versions/91ac5e543037_extend_repos_creds_column.py +36 -0
- dstack/_internal/server/migrations/versions/98cd9c8b5927_add_volumemodel.py +73 -0
- dstack/_internal/server/migrations/versions/98d1b92988bc_add_jobterminationreason_terminated_due_.py +140 -0
- dstack/_internal/server/migrations/versions/99b4c8c954ea_add_termination_reason_message.py +71 -0
- dstack/_internal/server/migrations/versions/9eea6af28e10_added_fail_reason_for_instancemodel.py +36 -0
- dstack/_internal/server/migrations/versions/__init__.py +0 -0
- dstack/_internal/server/migrations/versions/a060e2440936_.py +206 -0
- dstack/_internal/server/migrations/versions/a751ef183f27_move_attachment_data_to_volumes_.py +34 -0
- dstack/_internal/server/migrations/versions/a7b46c073fa1_add_placementgroupmodel.py +58 -0
- dstack/_internal/server/migrations/versions/afbc600ff2b2_add_created_at_to_usermodel_and_.py +102 -0
- dstack/_internal/server/migrations/versions/b4d6ad60db08_add_instancemodel_unreachable.py +37 -0
- dstack/_internal/server/migrations/versions/b88d55c2a07d_replace_instancestatus_ready.py +21 -0
- dstack/_internal/server/migrations/versions/bc8ca4a505c6_store_backendtype_as_string.py +171 -0
- dstack/_internal/server/migrations/versions/bca2fdf130bf_add_runmodel_priority.py +34 -0
- dstack/_internal/server/migrations/versions/bfba43f6def2_.py +32 -0
- dstack/_internal/server/migrations/versions/c00090eaef21_support_fleets.py +108 -0
- dstack/_internal/server/migrations/versions/c154eece89da_add_fields_for_async_gateway_creation.py +74 -0
- dstack/_internal/server/migrations/versions/c20626d03cfb_add_jobmetricspoint.py +43 -0
- dstack/_internal/server/migrations/versions/c48df7985d57_add_instance_termination_retries.py +38 -0
- dstack/_internal/server/migrations/versions/c83d45f9a971_replace_string_with_text.py +150 -0
- dstack/_internal/server/migrations/versions/d0bb68e48b9f_add_project_owners_and_quotas.py +106 -0
- dstack/_internal/server/migrations/versions/d3e8af4786fa_gateway_compute_flag_deleted.py +34 -0
- dstack/_internal/server/migrations/versions/d4d9dc26cf58_add_ix_jobs_run_id.py +31 -0
- dstack/_internal/server/migrations/versions/d5863798bf41_add_volumemodel_last_job_processed_at.py +40 -0
- dstack/_internal/server/migrations/versions/d6b11105f659_add_usermodel_active.py +36 -0
- dstack/_internal/server/migrations/versions/da574e93fee0_add_jobmodel_volumes_detached_at.py +40 -0
- dstack/_internal/server/migrations/versions/dfffd6a1165c_add_fields_for_gateways_behind_alb.py +36 -0
- dstack/_internal/server/migrations/versions/e2d08cd1b8d9_add_jobmodel_fleet.py +41 -0
- dstack/_internal/server/migrations/versions/e3b7db07727f_add_gatewaycomputemodel_app_updated_at.py +61 -0
- dstack/_internal/server/migrations/versions/e6391ca6a264_separate_gateways_from_compute.py +72 -0
- dstack/_internal/server/migrations/versions/ea60480f82bb_add_membermodel_member_num.py +32 -0
- dstack/_internal/server/migrations/versions/ec02a26a256c_add_runmodel_next_triggered_at.py +38 -0
- dstack/_internal/server/migrations/versions/ed0ca30e13bb_migrate_instancestatus_provisioning.py +29 -0
- dstack/_internal/server/migrations/versions/fe72c4de8376_add_gateways.py +81 -0
- dstack/_internal/server/migrations/versions/ff1d94f65b08_user_ssh_key.py +34 -0
- dstack/_internal/server/migrations/versions/ffa99edd1988_add_jobterminationreason_max_duration_.py +81 -0
- dstack/_internal/server/models.py +930 -0
- dstack/_internal/server/routers/__init__.py +0 -0
- dstack/_internal/server/routers/auth.py +34 -0
- dstack/_internal/server/routers/backends.py +142 -0
- dstack/_internal/server/routers/events.py +60 -0
- dstack/_internal/server/routers/files.py +68 -0
- dstack/_internal/server/routers/fleets.py +202 -0
- dstack/_internal/server/routers/gateways.py +109 -0
- dstack/_internal/server/routers/gpus.py +32 -0
- dstack/_internal/server/routers/instances.py +77 -0
- dstack/_internal/server/routers/logs.py +34 -0
- dstack/_internal/server/routers/metrics.py +82 -0
- dstack/_internal/server/routers/projects.py +205 -0
- dstack/_internal/server/routers/prometheus.py +35 -0
- dstack/_internal/server/routers/repos.py +118 -0
- dstack/_internal/server/routers/runs.py +216 -0
- dstack/_internal/server/routers/secrets.py +86 -0
- dstack/_internal/server/routers/server.py +19 -0
- dstack/_internal/server/routers/users.py +158 -0
- dstack/_internal/server/routers/volumes.py +122 -0
- dstack/_internal/server/schemas/__init__.py +0 -0
- dstack/_internal/server/schemas/auth.py +83 -0
- dstack/_internal/server/schemas/backends.py +16 -0
- dstack/_internal/server/schemas/common.py +9 -0
- dstack/_internal/server/schemas/events.py +211 -0
- dstack/_internal/server/schemas/files.py +5 -0
- dstack/_internal/server/schemas/fleets.py +49 -0
- dstack/_internal/server/schemas/gateways.py +31 -0
- dstack/_internal/server/schemas/gpus.py +26 -0
- dstack/_internal/server/schemas/health/__init__.py +0 -0
- dstack/_internal/server/schemas/health/dcgm.py +56 -0
- dstack/_internal/server/schemas/instances.py +47 -0
- dstack/_internal/server/schemas/logs.py +17 -0
- dstack/_internal/server/schemas/projects.py +81 -0
- dstack/_internal/server/schemas/repos.py +24 -0
- dstack/_internal/server/schemas/runner.py +269 -0
- dstack/_internal/server/schemas/runs.py +66 -0
- dstack/_internal/server/schemas/secrets.py +16 -0
- dstack/_internal/server/schemas/users.py +72 -0
- dstack/_internal/server/schemas/volumes.py +29 -0
- dstack/_internal/server/security/__init__.py +0 -0
- dstack/_internal/server/security/permissions.py +251 -0
- dstack/_internal/server/services/__init__.py +0 -0
- dstack/_internal/server/services/auth.py +77 -0
- dstack/_internal/server/services/backends/__init__.py +404 -0
- dstack/_internal/server/services/backends/handlers.py +105 -0
- dstack/_internal/server/services/compute_groups.py +22 -0
- dstack/_internal/server/services/config.py +279 -0
- dstack/_internal/server/services/docker.py +162 -0
- dstack/_internal/server/services/encryption/__init__.py +102 -0
- dstack/_internal/server/services/encryption/keys/__init__.py +0 -0
- dstack/_internal/server/services/encryption/keys/aes.py +68 -0
- dstack/_internal/server/services/encryption/keys/base.py +19 -0
- dstack/_internal/server/services/encryption/keys/identity.py +28 -0
- dstack/_internal/server/services/events.py +477 -0
- dstack/_internal/server/services/files.py +91 -0
- dstack/_internal/server/services/fleets.py +1224 -0
- dstack/_internal/server/services/gateways/__init__.py +686 -0
- dstack/_internal/server/services/gateways/client.py +209 -0
- dstack/_internal/server/services/gateways/connection.py +139 -0
- dstack/_internal/server/services/gateways/pool.py +58 -0
- dstack/_internal/server/services/gpus.py +387 -0
- dstack/_internal/server/services/instances.py +731 -0
- dstack/_internal/server/services/jobs/__init__.py +840 -0
- dstack/_internal/server/services/jobs/configurators/__init__.py +0 -0
- dstack/_internal/server/services/jobs/configurators/base.py +469 -0
- dstack/_internal/server/services/jobs/configurators/dev.py +69 -0
- dstack/_internal/server/services/jobs/configurators/extensions/__init__.py +0 -0
- dstack/_internal/server/services/jobs/configurators/extensions/base.py +15 -0
- dstack/_internal/server/services/jobs/configurators/extensions/cursor.py +42 -0
- dstack/_internal/server/services/jobs/configurators/extensions/vscode.py +42 -0
- dstack/_internal/server/services/jobs/configurators/extensions/windsurf.py +43 -0
- dstack/_internal/server/services/jobs/configurators/service.py +28 -0
- dstack/_internal/server/services/jobs/configurators/task.py +39 -0
- dstack/_internal/server/services/locking.py +187 -0
- dstack/_internal/server/services/logging.py +29 -0
- dstack/_internal/server/services/logs/__init__.py +122 -0
- dstack/_internal/server/services/logs/aws.py +373 -0
- dstack/_internal/server/services/logs/base.py +47 -0
- dstack/_internal/server/services/logs/filelog.py +261 -0
- dstack/_internal/server/services/logs/fluentbit.py +329 -0
- dstack/_internal/server/services/logs/gcp.py +181 -0
- dstack/_internal/server/services/metrics.py +172 -0
- dstack/_internal/server/services/offers.py +249 -0
- dstack/_internal/server/services/permissions.py +37 -0
- dstack/_internal/server/services/placement.py +234 -0
- dstack/_internal/server/services/plugins.py +109 -0
- dstack/_internal/server/services/probes.py +10 -0
- dstack/_internal/server/services/projects.py +835 -0
- dstack/_internal/server/services/prometheus/__init__.py +0 -0
- dstack/_internal/server/services/prometheus/client_metrics.py +55 -0
- dstack/_internal/server/services/prometheus/custom_metrics.py +327 -0
- dstack/_internal/server/services/proxy/__init__.py +3 -0
- dstack/_internal/server/services/proxy/auth.py +12 -0
- dstack/_internal/server/services/proxy/deps.py +18 -0
- dstack/_internal/server/services/proxy/repo.py +189 -0
- dstack/_internal/server/services/proxy/routers/__init__.py +0 -0
- dstack/_internal/server/services/proxy/routers/service_proxy.py +49 -0
- dstack/_internal/server/services/proxy/services/__init__.py +0 -0
- dstack/_internal/server/services/proxy/services/service_proxy.py +135 -0
- dstack/_internal/server/services/repos.py +362 -0
- dstack/_internal/server/services/requirements/__init__.py +0 -0
- dstack/_internal/server/services/requirements/combine.py +260 -0
- dstack/_internal/server/services/resources.py +21 -0
- dstack/_internal/server/services/runner/__init__.py +0 -0
- dstack/_internal/server/services/runner/client.py +646 -0
- dstack/_internal/server/services/runner/ssh.py +128 -0
- dstack/_internal/server/services/runs/__init__.py +1026 -0
- dstack/_internal/server/services/runs/plan.py +703 -0
- dstack/_internal/server/services/runs/replicas.py +317 -0
- dstack/_internal/server/services/runs/spec.py +191 -0
- dstack/_internal/server/services/secrets.py +245 -0
- dstack/_internal/server/services/services/__init__.py +345 -0
- dstack/_internal/server/services/services/autoscalers.py +140 -0
- dstack/_internal/server/services/services/options.py +53 -0
- dstack/_internal/server/services/ssh.py +67 -0
- dstack/_internal/server/services/storage/__init__.py +37 -0
- dstack/_internal/server/services/storage/base.py +48 -0
- dstack/_internal/server/services/storage/gcs.py +66 -0
- dstack/_internal/server/services/storage/s3.py +69 -0
- dstack/_internal/server/services/users.py +461 -0
- dstack/_internal/server/services/volumes.py +496 -0
- dstack/_internal/server/settings.py +161 -0
- dstack/_internal/server/statics/00a6e1fb461ed2929fb9.png +0 -0
- dstack/_internal/server/statics/0cae4d9f0a36034984a7.png +0 -0
- dstack/_internal/server/statics/391de232cc0e30cae513.png +0 -0
- dstack/_internal/server/statics/4e0eead8c1a73689ef9d.svg +1 -0
- dstack/_internal/server/statics/544afa2f63428c2235b0.png +0 -0
- dstack/_internal/server/statics/54a4f50f74c6b9381530.svg +7 -0
- dstack/_internal/server/statics/68dd1360a7d2611e0132.svg +4 -0
- dstack/_internal/server/statics/69544b4c81973b54a66f.png +0 -0
- dstack/_internal/server/statics/77a8b02b17af19e39266.png +0 -0
- dstack/_internal/server/statics/83a93a8871c219104367.svg +9 -0
- dstack/_internal/server/statics/8f28bb8e9999e5e6a48b.svg +4 -0
- dstack/_internal/server/statics/9124086961ab8c366bc4.svg +9 -0
- dstack/_internal/server/statics/9a9ebaeb54b025dbac0a.svg +5 -0
- dstack/_internal/server/statics/a3428392dc534f3b15c4.svg +7 -0
- dstack/_internal/server/statics/ae22625574d69361f72c.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-144x144.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-192x192.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-256x256.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-36x36.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-384x384.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-48x48.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-512x512.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-72x72.png +0 -0
- dstack/_internal/server/statics/assets/android-chrome-96x96.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-1024x1024.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-114x114.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-120x120.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-144x144.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-152x152.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-167x167.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-180x180.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-57x57.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-60x60.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-72x72.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-76x76.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon-precomposed.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-icon.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1125x2436.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1136x640.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1170x2532.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1179x2556.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1242x2208.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1242x2688.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1284x2778.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1290x2796.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1334x750.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1488x2266.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1536x2048.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1620x2160.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1640x2160.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1668x2224.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1668x2388.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-1792x828.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2048x1536.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2048x2732.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2160x1620.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2160x1640.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2208x1242.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2224x1668.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2266x1488.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2388x1668.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2436x1125.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2532x1170.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2556x1179.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2688x1242.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2732x2048.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2778x1284.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-2796x1290.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-640x1136.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-750x1334.png +0 -0
- dstack/_internal/server/statics/assets/apple-touch-startup-image-828x1792.png +0 -0
- dstack/_internal/server/statics/assets/browserconfig.xml +12 -0
- dstack/_internal/server/statics/assets/favicon-16x16.png +0 -0
- dstack/_internal/server/statics/assets/favicon-32x32.png +0 -0
- dstack/_internal/server/statics/assets/favicon-48x48.png +0 -0
- dstack/_internal/server/statics/assets/favicon.ico +0 -0
- dstack/{dashboard/statics/assets/manifest.json → _internal/server/statics/assets/manifest.webmanifest} +18 -9
- dstack/_internal/server/statics/assets/mstile-144x144.png +0 -0
- dstack/_internal/server/statics/assets/mstile-150x150.png +0 -0
- dstack/_internal/server/statics/assets/mstile-310x150.png +0 -0
- dstack/_internal/server/statics/assets/mstile-310x310.png +0 -0
- dstack/_internal/server/statics/assets/mstile-70x70.png +0 -0
- dstack/_internal/server/statics/assets/yandex-browser-50x50.png +0 -0
- dstack/_internal/server/statics/b7ae68f44193474fc578.png +0 -0
- dstack/_internal/server/statics/d2f008c75b2b5b191f3f.png +0 -0
- dstack/_internal/server/statics/d44c33e1b92e05c379fd.png +0 -0
- dstack/_internal/server/statics/dd43ff0552815179d7ab.png +0 -0
- dstack/_internal/server/statics/dd4e7166c0b9aac197d7.png +0 -0
- dstack/_internal/server/statics/e30b27916930d43d2271.png +0 -0
- dstack/_internal/server/statics/e467d7d60aae81ab198b.svg +6 -0
- dstack/_internal/server/statics/eb9b344b73818fe2b71a.png +0 -0
- dstack/_internal/server/statics/f517dd626eb964120de0.png +0 -0
- dstack/_internal/server/statics/f958aecddee5d8e3222c.png +0 -0
- dstack/_internal/server/statics/index.html +3 -0
- dstack/_internal/server/statics/logo-notext.svg +116 -0
- dstack/_internal/server/statics/main-2e6967bad9f29395eea6.css +3 -0
- dstack/_internal/server/statics/main-7dc0f6d20b8b41659acc.js +155547 -0
- dstack/_internal/server/statics/main-7dc0f6d20b8b41659acc.js.map +1 -0
- dstack/{dashboard → _internal/server}/statics/manifest.json +2 -2
- dstack/_internal/server/statics/static/media/entraID.d65d1f3e9486a8e56d24fc07b3230885.svg +9 -0
- dstack/_internal/server/statics/static/media/google.b194b06fafd0a52aeb566922160ea514.svg +1 -0
- dstack/{dashboard/statics/static/media/logo.f9d7170678f68f796e270698633770ec.svg → _internal/server/statics/static/media/logo.f602feeb138844eda97c8cb641461448.svg} +8 -6
- dstack/_internal/server/statics/static/media/okta.12f178e6873a1100965f2a4dbd18fcec.svg +2 -0
- dstack/_internal/server/statics/static/media/theme.3994c817bb7dda191c1c9640dee0bf42.svg +3 -0
- dstack/_internal/server/testing/__init__.py +0 -0
- dstack/_internal/server/testing/common.py +1220 -0
- dstack/_internal/server/testing/conf.py +53 -0
- dstack/_internal/server/testing/matchers.py +31 -0
- dstack/_internal/server/utils/__init__.py +0 -0
- dstack/_internal/server/utils/common.py +55 -0
- dstack/_internal/server/utils/logging.py +51 -0
- dstack/_internal/server/utils/provisioning.py +368 -0
- dstack/_internal/server/utils/routers.py +166 -0
- dstack/_internal/server/utils/sentry_utils.py +24 -0
- dstack/_internal/settings.py +49 -0
- dstack/_internal/utils/__init__.py +0 -0
- dstack/_internal/utils/common.py +318 -0
- dstack/_internal/utils/cron.py +5 -0
- dstack/_internal/utils/crypto.py +40 -0
- dstack/_internal/utils/env.py +88 -0
- dstack/_internal/utils/event_loop.py +30 -0
- dstack/_internal/utils/files.py +69 -0
- dstack/_internal/utils/gpu.py +59 -0
- dstack/_internal/utils/hash.py +31 -0
- dstack/_internal/utils/interpolator.py +91 -0
- dstack/_internal/utils/json_schema.py +11 -0
- dstack/_internal/utils/json_utils.py +54 -0
- dstack/_internal/utils/logging.py +5 -0
- dstack/_internal/utils/nested_list.py +47 -0
- dstack/_internal/utils/network.py +50 -0
- dstack/_internal/utils/path.py +57 -0
- dstack/_internal/utils/random_names.py +258 -0
- dstack/_internal/utils/ssh.py +346 -0
- dstack/_internal/utils/tags.py +42 -0
- dstack/_internal/utils/typing.py +14 -0
- dstack/_internal/utils/version.py +22 -0
- dstack/api/__init__.py +46 -0
- dstack/api/_public/__init__.py +96 -0
- dstack/api/_public/backends.py +42 -0
- dstack/api/_public/common.py +5 -0
- dstack/api/_public/repos.py +202 -0
- dstack/api/_public/runs.py +714 -0
- dstack/api/server/__init__.py +206 -0
- dstack/api/server/_auth.py +30 -0
- dstack/api/server/_backends.py +38 -0
- dstack/api/server/_events.py +64 -0
- dstack/api/server/_files.py +18 -0
- dstack/api/server/_fleets.py +82 -0
- dstack/api/server/_gateways.py +54 -0
- dstack/api/server/_gpus.py +27 -0
- dstack/api/server/_group.py +22 -0
- dstack/api/server/_logs.py +15 -0
- dstack/api/server/_metrics.py +23 -0
- dstack/api/server/_projects.py +124 -0
- dstack/api/server/_repos.py +64 -0
- dstack/api/server/_runs.py +102 -0
- dstack/api/server/_secrets.py +36 -0
- dstack/api/server/_users.py +82 -0
- dstack/api/server/_volumes.py +39 -0
- dstack/api/server/utils.py +34 -0
- dstack/api/utils.py +105 -0
- dstack/core/__init__.py +0 -0
- dstack/plugins/__init__.py +8 -0
- dstack/plugins/_base.py +72 -0
- dstack/plugins/_models.py +8 -0
- dstack/plugins/_utils.py +19 -0
- dstack/plugins/builtin/__init__.py +0 -0
- dstack/plugins/builtin/rest_plugin/__init__.py +18 -0
- dstack/plugins/builtin/rest_plugin/_models.py +48 -0
- dstack/plugins/builtin/rest_plugin/_plugin.py +147 -0
- dstack/version.py +3 -1
- dstack-0.20.7.dist-info/METADATA +519 -0
- dstack-0.20.7.dist-info/RECORD +720 -0
- {dstack-0.0.9.dist-info → dstack-0.20.7.dist-info}/WHEEL +1 -2
- dstack-0.20.7.dist-info/entry_points.txt +2 -0
- dstack-0.20.7.dist-info/licenses/LICENSE.md +353 -0
- dstack/aws/__init__.py +0 -180
- dstack/aws/artifacts.py +0 -111
- dstack/aws/config.py +0 -40
- dstack/aws/jobs.py +0 -245
- dstack/aws/logs.py +0 -186
- dstack/aws/repos.py +0 -137
- dstack/aws/run_names.py +0 -17
- dstack/aws/runners.py +0 -693
- dstack/aws/runs.py +0 -79
- dstack/aws/secrets.py +0 -99
- dstack/aws/tags.py +0 -138
- dstack/backend.py +0 -299
- dstack/cli/app.py +0 -41
- dstack/cli/artifacts.py +0 -87
- dstack/cli/common.py +0 -57
- dstack/cli/config.py +0 -194
- dstack/cli/dashboard.py +0 -26
- dstack/cli/delete.py +0 -49
- dstack/cli/init.py +0 -33
- dstack/cli/logs.py +0 -87
- dstack/cli/main.py +0 -81
- dstack/cli/restart.py +0 -43
- dstack/cli/run.py +0 -223
- dstack/cli/schema.py +0 -46
- dstack/cli/secrets.py +0 -97
- dstack/cli/status.py +0 -140
- dstack/cli/stop.py +0 -53
- dstack/cli/tags.py +0 -100
- dstack/config.py +0 -80
- dstack/dashboard/artifacts.py +0 -26
- dstack/dashboard/logs.py +0 -73
- dstack/dashboard/main.py +0 -45
- dstack/dashboard/repos.py +0 -41
- dstack/dashboard/runs.py +0 -140
- dstack/dashboard/secrets.py +0 -53
- dstack/dashboard/statics/4d6a4e032505c1efd23c.png +0 -0
- dstack/dashboard/statics/7e018c3e5566d7c349a8.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-144x144.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-192x192.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-256x256.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-36x36.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-384x384.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-48x48.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-512x512.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-72x72.png +0 -0
- dstack/dashboard/statics/assets/android-chrome-96x96.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-1024x1024.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-114x114.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-120x120.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-144x144.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-152x152.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-167x167.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-180x180.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-57x57.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-60x60.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-72x72.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-76x76.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon-precomposed.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-icon.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1125x2436.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1136x640.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1242x2208.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1242x2688.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1334x750.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1536x2048.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1620x2160.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1668x2224.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1668x2388.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-1792x828.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2048x1536.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2048x2732.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2160x1620.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2208x1242.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2224x1668.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2388x1668.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2436x1125.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2688x1242.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-2732x2048.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-640x1136.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-750x1334.png +0 -0
- dstack/dashboard/statics/assets/apple-touch-startup-image-828x1792.png +0 -0
- dstack/dashboard/statics/assets/browserconfig.xml +0 -15
- dstack/dashboard/statics/assets/coast-228x228.png +0 -0
- dstack/dashboard/statics/assets/favicon-16x16.png +0 -0
- dstack/dashboard/statics/assets/favicon-32x32.png +0 -0
- dstack/dashboard/statics/assets/favicon-48x48.png +0 -0
- dstack/dashboard/statics/assets/favicon.ico +0 -0
- dstack/dashboard/statics/assets/firefox_app_128x128.png +0 -0
- dstack/dashboard/statics/assets/firefox_app_512x512.png +0 -0
- dstack/dashboard/statics/assets/firefox_app_60x60.png +0 -0
- dstack/dashboard/statics/assets/manifest.webapp +0 -14
- dstack/dashboard/statics/assets/mstile-144x144.png +0 -0
- dstack/dashboard/statics/assets/mstile-150x150.png +0 -0
- dstack/dashboard/statics/assets/mstile-310x150.png +0 -0
- dstack/dashboard/statics/assets/mstile-310x310.png +0 -0
- dstack/dashboard/statics/assets/mstile-70x70.png +0 -0
- dstack/dashboard/statics/assets/yandex-browser-50x50.png +0 -0
- dstack/dashboard/statics/d0f71e48806e25d72553.png +0 -0
- dstack/dashboard/statics/index.html +0 -7
- dstack/dashboard/statics/main-1d87e34eb0454da8ebb4.js +0 -3
- dstack/dashboard/statics/main-1d87e34eb0454da8ebb4.js.LICENSE.txt +0 -102
- dstack/dashboard/statics/main-1d87e34eb0454da8ebb4.js.map +0 -1
- dstack/dashboard/statics/main.css +0 -5058
- dstack/dashboard/statics/splash_thumbnail.png +0 -0
- dstack/dashboard/statics/static/media/check.3f68ffc787a15c0476793a6d18ecb71a.svg +0 -3
- dstack/dashboard/statics/static/media/chevron-down.bfd8f22c4a5db4d443e76bca3b02f334.svg +0 -3
- dstack/dashboard/statics/static/media/chevron-up.bade0c5d82d741cead615813264140c9.svg +0 -3
- dstack/dashboard/statics/static/media/clock.583b744f29b9d143718a55e7c35fe38e.svg +0 -3
- dstack/dashboard/statics/static/media/close.a8bb9e47361b03a3b5084dad676ba1da.svg +0 -3
- dstack/dashboard/statics/static/media/content-copy.73f5f2a175094757758e315243a4111e.svg +0 -3
- dstack/dashboard/statics/static/media/delete-outline.6a8abf4e4f9cb777781967efd56efe9b.svg +0 -3
- dstack/dashboard/statics/static/media/dots-vertical.82fc618192e0c7dc4d615ff93269246a.svg +0 -3
- dstack/dashboard/statics/static/media/earth.1ad57c7f59f4be5c8bb2fa00439c3149.svg +0 -3
- dstack/dashboard/statics/static/media/email.320bc3af24a5f1bb41ebd85f66a5dd70.svg +0 -3
- dstack/dashboard/statics/static/media/external-link.99b88e699c15afb820a1779d9a2261ed.svg +0 -3
- dstack/dashboard/statics/static/media/eye-off-outline.5b4afb7ad624a44dd307518ff93d1faa.svg +0 -3
- dstack/dashboard/statics/static/media/eye-outline.ca41708feaaed1edb15c5fff021fbafe.svg +0 -3
- dstack/dashboard/statics/static/media/file-download-outline.3634b41923ba79b297ff294ef898661c.svg +0 -3
- dstack/dashboard/statics/static/media/folder-outline.33378387af61821dd1207e4b2d061a07.svg +0 -3
- dstack/dashboard/statics/static/media/github-circle.1bb85d171c31a3c2eebad07319377171.svg +0 -3
- dstack/dashboard/statics/static/media/infinity.915f92939afc0a37f94adba211ceb172.svg +0 -3
- dstack/dashboard/statics/static/media/layers.b4b02cea267a617d7aa44c2719250c89.svg +0 -3
- dstack/dashboard/statics/static/media/linkedin.1c52fae553eee54397f0e63a79455a5e.svg +0 -3
- dstack/dashboard/statics/static/media/loading.e466be7b2c1f0ac9e7e51ca929d0e37d.svg +0 -3
- dstack/dashboard/statics/static/media/lock.4a4c7768d0fa60c716609ddc483470ef.svg +0 -3
- dstack/dashboard/statics/static/media/magnify.0c803314d039d21f3cb1504ccd1437a4.svg +0 -3
- dstack/dashboard/statics/static/media/mark.3f68ffc787a15c0476793a6d18ecb71a.svg +0 -3
- dstack/dashboard/statics/static/media/menu-close.3ee84714181017c6ff837830297c8437.svg +0 -3
- dstack/dashboard/statics/static/media/menu.922f81e0972fbcbb5adcd8def20c86a3.svg +0 -3
- dstack/dashboard/statics/static/media/pencil.f706a3b9dcbff4959a91bf72e1e6324f.svg +0 -3
- dstack/dashboard/statics/static/media/refresh.a80edb948e98b322cd73b67814a57a48.svg +0 -3
- dstack/dashboard/statics/static/media/shape-plus.63b093c7f4b44c3def774f30fcfbceca.svg +0 -3
- dstack/dashboard/statics/static/media/slack.ec2fca99c6b944950ac65404ddd26880.svg +0 -4
- dstack/dashboard/statics/static/media/small-logo.b9cc8d09f646a553e65fa336dafd8b10.svg +0 -116
- dstack/dashboard/statics/static/media/source-branch.b8d22cfc42a7bed81f0fc08130818e85.svg +0 -3
- dstack/dashboard/statics/static/media/source-commit.be2bb53c081b9b6836adffccc0b8d3e6.svg +0 -3
- dstack/dashboard/statics/static/media/stop.11488ff1437ad929476be8924a3b7075.svg +0 -3
- dstack/dashboard/statics/static/media/tag-minus.15680a815b0b8d027e973c84832c05e6.svg +0 -3
- dstack/dashboard/statics/static/media/tag-outline.19b0bf86a8afd7d6d9c716e9a91d94ca.svg +0 -3
- dstack/dashboard/statics/static/media/twitter.4af18861c84a2f3044c7546b55d5739c.svg +0 -3
- dstack/dashboard/tags.py +0 -119
- dstack/jobs.py +0 -255
- dstack/providers/__init__.py +0 -316
- dstack/providers/_python/main.py +0 -88
- dstack/providers/_tensorboard/main.py +0 -93
- dstack/providers/_torchrun/main.py +0 -121
- dstack/providers/bash/main.py +0 -90
- dstack/providers/code/main.py +0 -95
- dstack/providers/docker/main.py +0 -79
- dstack/providers/lab/main.py +0 -95
- dstack/providers/notebook/main.py +0 -90
- dstack/random_name.py +0 -29
- dstack/repo.py +0 -135
- dstack/runners.py +0 -35
- dstack/util.py +0 -15
- dstack-0.0.9.dist-info/METADATA +0 -176
- dstack-0.0.9.dist-info/RECORD +0 -179
- dstack-0.0.9.dist-info/entry_points.txt +0 -3
- dstack-0.0.9.dist-info/top_level.txt +0 -2
- tests/test_config.py +0 -70
- /dstack/{cli → _internal}/__init__.py +0 -0
- /dstack/{dashboard → _internal/cli}/__init__.py +0 -0
- /dstack/{providers/_python → _internal/cli/models}/__init__.py +0 -0
- /dstack/{providers/_tensorboard → _internal/cli/services}/__init__.py +0 -0
- /dstack/{providers/_torchrun → _internal/cli/utils}/__init__.py +0 -0
- /dstack/{providers/bash → _internal/core}/__init__.py +0 -0
- /dstack/{providers/code → _internal/core/backends}/__init__.py +0 -0
- /dstack/{providers/docker → _internal/core/backends/aws}/__init__.py +0 -0
- /dstack/{providers/lab → _internal/core/backends/azure}/__init__.py +0 -0
- /dstack/{providers/notebook → _internal/core/backends/base}/__init__.py +0 -0
- {tests → dstack/_internal/core/backends/cloudrift}/__init__.py +0 -0
- /dstack/{dashboard → _internal/server}/statics/assets/yandex-browser-manifest.json +0 -0
- /dstack/{dashboard → _internal/server}/statics/robots.txt +0 -0
|
@@ -0,0 +1,517 @@
|
|
|
1
|
+
import shutil
|
|
2
|
+
from typing import Any, Dict, List, Optional, Union
|
|
3
|
+
|
|
4
|
+
from rich.markup import escape
|
|
5
|
+
from rich.table import Table
|
|
6
|
+
|
|
7
|
+
from dstack._internal.cli.models.offers import OfferCommandOutput, OfferRequirements
|
|
8
|
+
from dstack._internal.cli.models.runs import PsCommandOutput
|
|
9
|
+
from dstack._internal.cli.utils.common import (
|
|
10
|
+
NO_FLEETS_WARNING,
|
|
11
|
+
NO_OFFERS_WARNING,
|
|
12
|
+
add_row_from_dict,
|
|
13
|
+
console,
|
|
14
|
+
format_instance_availability,
|
|
15
|
+
)
|
|
16
|
+
from dstack._internal.core.models.backends.base import BackendType
|
|
17
|
+
from dstack._internal.core.models.configurations import DevEnvironmentConfiguration
|
|
18
|
+
from dstack._internal.core.models.instances import (
|
|
19
|
+
InstanceOfferWithAvailability,
|
|
20
|
+
InstanceType,
|
|
21
|
+
)
|
|
22
|
+
from dstack._internal.core.models.profiles import (
|
|
23
|
+
DEFAULT_RUN_TERMINATION_IDLE_TIME,
|
|
24
|
+
SpotPolicy,
|
|
25
|
+
TerminationPolicy,
|
|
26
|
+
)
|
|
27
|
+
from dstack._internal.core.models.runs import (
|
|
28
|
+
Job,
|
|
29
|
+
JobStatus,
|
|
30
|
+
JobSubmission,
|
|
31
|
+
Probe,
|
|
32
|
+
ProbeSpec,
|
|
33
|
+
RunPlan,
|
|
34
|
+
RunStatus,
|
|
35
|
+
get_policy_map,
|
|
36
|
+
)
|
|
37
|
+
from dstack._internal.core.models.runs import (
|
|
38
|
+
Run as CoreRun,
|
|
39
|
+
)
|
|
40
|
+
from dstack._internal.core.services.profiles import get_termination
|
|
41
|
+
from dstack._internal.utils.common import (
|
|
42
|
+
DateFormatter,
|
|
43
|
+
batched,
|
|
44
|
+
format_duration_multiunit,
|
|
45
|
+
format_pretty_duration,
|
|
46
|
+
pretty_date,
|
|
47
|
+
)
|
|
48
|
+
from dstack.api import Run
|
|
49
|
+
|
|
50
|
+
|
|
51
|
+
def print_offers_json(run_plan: RunPlan, run_spec):
|
|
52
|
+
"""Print offers information in JSON format."""
|
|
53
|
+
job_plan = run_plan.job_plans[0]
|
|
54
|
+
|
|
55
|
+
requirements = OfferRequirements(
|
|
56
|
+
resources=job_plan.job_spec.requirements.resources,
|
|
57
|
+
max_price=job_plan.job_spec.requirements.max_price,
|
|
58
|
+
spot=get_policy_map(run_spec.configuration.spot_policy, default=SpotPolicy.AUTO),
|
|
59
|
+
reservation=run_plan.run_spec.configuration.reservation,
|
|
60
|
+
)
|
|
61
|
+
|
|
62
|
+
output = OfferCommandOutput(
|
|
63
|
+
project=run_plan.project_name,
|
|
64
|
+
user=run_plan.user,
|
|
65
|
+
requirements=requirements,
|
|
66
|
+
offers=job_plan.offers,
|
|
67
|
+
total_offers=job_plan.total_offers,
|
|
68
|
+
)
|
|
69
|
+
|
|
70
|
+
print(output.json())
|
|
71
|
+
|
|
72
|
+
|
|
73
|
+
def print_runs_json(project: str, runs: List[Run]) -> None:
|
|
74
|
+
"""Print runs information in JSON format."""
|
|
75
|
+
output = PsCommandOutput(
|
|
76
|
+
project=project,
|
|
77
|
+
runs=[r._run for r in runs],
|
|
78
|
+
)
|
|
79
|
+
print(output.json())
|
|
80
|
+
|
|
81
|
+
|
|
82
|
+
def print_run_plan(
|
|
83
|
+
run_plan: RunPlan,
|
|
84
|
+
max_offers: Optional[int] = None,
|
|
85
|
+
include_run_properties: bool = True,
|
|
86
|
+
no_fleets: bool = False,
|
|
87
|
+
):
|
|
88
|
+
run_spec = run_plan.get_effective_run_spec()
|
|
89
|
+
job_plan = run_plan.job_plans[0]
|
|
90
|
+
|
|
91
|
+
props = Table(box=None, show_header=False)
|
|
92
|
+
props.add_column(no_wrap=True) # key
|
|
93
|
+
props.add_column() # value
|
|
94
|
+
|
|
95
|
+
req = job_plan.job_spec.requirements
|
|
96
|
+
pretty_req = req.pretty_format(resources_only=True)
|
|
97
|
+
max_price = f"${req.max_price:3f}".rstrip("0").rstrip(".") if req.max_price else "-"
|
|
98
|
+
max_duration = (
|
|
99
|
+
format_pretty_duration(job_plan.job_spec.max_duration)
|
|
100
|
+
if job_plan.job_spec.max_duration
|
|
101
|
+
else "-"
|
|
102
|
+
)
|
|
103
|
+
if include_run_properties:
|
|
104
|
+
inactivity_duration = None
|
|
105
|
+
if isinstance(run_spec.configuration, DevEnvironmentConfiguration):
|
|
106
|
+
inactivity_duration = "-"
|
|
107
|
+
if isinstance(run_spec.configuration.inactivity_duration, int):
|
|
108
|
+
inactivity_duration = format_pretty_duration(
|
|
109
|
+
run_spec.configuration.inactivity_duration
|
|
110
|
+
)
|
|
111
|
+
if job_plan.job_spec.retry is None:
|
|
112
|
+
retry = "-"
|
|
113
|
+
else:
|
|
114
|
+
retry = escape(job_plan.job_spec.retry.pretty_format())
|
|
115
|
+
|
|
116
|
+
profile = run_spec.merged_profile
|
|
117
|
+
creation_policy = profile.creation_policy
|
|
118
|
+
# FIXME: This assumes the default idle_duration is the same for client and server.
|
|
119
|
+
# If the server changes idle_duration, old clients will see incorrect value.
|
|
120
|
+
termination_policy, termination_idle_time = get_termination(
|
|
121
|
+
profile, DEFAULT_RUN_TERMINATION_IDLE_TIME
|
|
122
|
+
)
|
|
123
|
+
if termination_policy == TerminationPolicy.DONT_DESTROY:
|
|
124
|
+
idle_duration = "-"
|
|
125
|
+
else:
|
|
126
|
+
idle_duration = format_pretty_duration(termination_idle_time)
|
|
127
|
+
|
|
128
|
+
if req.spot is None:
|
|
129
|
+
spot_policy = "auto"
|
|
130
|
+
elif req.spot:
|
|
131
|
+
spot_policy = "spot"
|
|
132
|
+
else:
|
|
133
|
+
spot_policy = "on-demand"
|
|
134
|
+
|
|
135
|
+
def th(s: str) -> str:
|
|
136
|
+
return f"[bold]{s}[/bold]"
|
|
137
|
+
|
|
138
|
+
props.add_row(th("Project"), run_plan.project_name)
|
|
139
|
+
props.add_row(th("User"), run_plan.user)
|
|
140
|
+
if include_run_properties:
|
|
141
|
+
props.add_row(th("Configuration"), run_spec.configuration_path)
|
|
142
|
+
configuration_type = run_spec.configuration.type
|
|
143
|
+
if run_spec.configuration.type == "task":
|
|
144
|
+
configuration_type += f" (nodes={run_spec.configuration.nodes})"
|
|
145
|
+
props.add_row(th("Type"), configuration_type)
|
|
146
|
+
props.add_row(th("Resources"), pretty_req)
|
|
147
|
+
props.add_row(th("Spot policy"), spot_policy)
|
|
148
|
+
props.add_row(th("Max price"), max_price)
|
|
149
|
+
if include_run_properties:
|
|
150
|
+
props.add_row(th("Retry policy"), retry)
|
|
151
|
+
props.add_row(th("Creation policy"), creation_policy)
|
|
152
|
+
props.add_row(th("Idle duration"), idle_duration)
|
|
153
|
+
props.add_row(th("Max duration"), max_duration)
|
|
154
|
+
if inactivity_duration is not None: # None means n/a
|
|
155
|
+
props.add_row(th("Inactivity duration"), inactivity_duration)
|
|
156
|
+
props.add_row(th("Reservation"), run_spec.configuration.reservation or "-")
|
|
157
|
+
|
|
158
|
+
offers = Table(box=None, expand=shutil.get_terminal_size(fallback=(120, 40)).columns <= 110)
|
|
159
|
+
offers.add_column("#")
|
|
160
|
+
offers.add_column("BACKEND", style="grey58", ratio=2)
|
|
161
|
+
offers.add_column("RESOURCES", ratio=4)
|
|
162
|
+
offers.add_column("INSTANCE TYPE", style="grey58", no_wrap=True, ratio=2)
|
|
163
|
+
offers.add_column("PRICE", style="grey58", ratio=1)
|
|
164
|
+
offers.add_column()
|
|
165
|
+
|
|
166
|
+
job_plan.offers = job_plan.offers[:max_offers] if max_offers else job_plan.offers
|
|
167
|
+
|
|
168
|
+
for i, offer in enumerate(job_plan.offers, start=1):
|
|
169
|
+
r = offer.instance.resources
|
|
170
|
+
|
|
171
|
+
instance = offer.instance.name
|
|
172
|
+
if offer.total_blocks > 1:
|
|
173
|
+
instance += f" ({offer.blocks}/{offer.total_blocks})"
|
|
174
|
+
offers.add_row(
|
|
175
|
+
f"{i}",
|
|
176
|
+
offer.backend.replace("remote", "ssh") + " (" + offer.region + ")",
|
|
177
|
+
r.pretty_format(include_spot=True),
|
|
178
|
+
instance,
|
|
179
|
+
f"${offer.price:.4f}".rstrip("0").rstrip("."),
|
|
180
|
+
format_instance_availability(offer.availability),
|
|
181
|
+
style=None if i == 1 or not include_run_properties else "secondary",
|
|
182
|
+
)
|
|
183
|
+
if job_plan.total_offers > len(job_plan.offers):
|
|
184
|
+
offers.add_row("", "...", style="secondary")
|
|
185
|
+
|
|
186
|
+
console.print(props)
|
|
187
|
+
console.print()
|
|
188
|
+
if len(job_plan.offers) > 0:
|
|
189
|
+
console.print(offers)
|
|
190
|
+
if job_plan.total_offers > len(job_plan.offers):
|
|
191
|
+
console.print(
|
|
192
|
+
f"[secondary] Shown {len(job_plan.offers)} of {job_plan.total_offers} offers, "
|
|
193
|
+
f"${job_plan.max_price:3f}".rstrip("0").rstrip(".")
|
|
194
|
+
+ "max[/]"
|
|
195
|
+
)
|
|
196
|
+
console.print()
|
|
197
|
+
else:
|
|
198
|
+
console.print(NO_FLEETS_WARNING if no_fleets else NO_OFFERS_WARNING)
|
|
199
|
+
|
|
200
|
+
|
|
201
|
+
def _format_run_status(run) -> str:
|
|
202
|
+
status_text = (
|
|
203
|
+
run.latest_job_submission.status_message
|
|
204
|
+
if run.status.is_finished() and run.latest_job_submission
|
|
205
|
+
else run.status_message
|
|
206
|
+
)
|
|
207
|
+
# Inline of _get_run_status_style
|
|
208
|
+
color_map = {
|
|
209
|
+
RunStatus.PENDING: "white",
|
|
210
|
+
RunStatus.SUBMITTED: "grey",
|
|
211
|
+
RunStatus.PROVISIONING: "deep_sky_blue1",
|
|
212
|
+
RunStatus.RUNNING: "sea_green3",
|
|
213
|
+
RunStatus.TERMINATING: "deep_sky_blue1",
|
|
214
|
+
RunStatus.TERMINATED: "grey",
|
|
215
|
+
RunStatus.FAILED: "indian_red1",
|
|
216
|
+
RunStatus.DONE: "grey",
|
|
217
|
+
}
|
|
218
|
+
if status_text in ("no offers", "interrupted"):
|
|
219
|
+
color = "gold1"
|
|
220
|
+
elif status_text == "no fleets":
|
|
221
|
+
color = "indian_red1"
|
|
222
|
+
elif status_text == "pulling":
|
|
223
|
+
color = "sea_green3"
|
|
224
|
+
else:
|
|
225
|
+
color = color_map.get(run.status, "white")
|
|
226
|
+
status_style = f"bold {color}" if not run.status.is_finished() else color
|
|
227
|
+
return f"[{status_style}]{status_text}[/]"
|
|
228
|
+
|
|
229
|
+
|
|
230
|
+
def _format_job_submission_status(job_submission: JobSubmission, verbose: bool) -> str:
|
|
231
|
+
status_message = job_submission.status_message
|
|
232
|
+
job_status = job_submission.status
|
|
233
|
+
if status_message in ("no offers", "interrupted"):
|
|
234
|
+
color = "gold1"
|
|
235
|
+
elif status_message == "no fleets":
|
|
236
|
+
color = "indian_red1"
|
|
237
|
+
elif status_message == "stopped":
|
|
238
|
+
color = "grey"
|
|
239
|
+
else:
|
|
240
|
+
color_map = {
|
|
241
|
+
JobStatus.SUBMITTED: "grey",
|
|
242
|
+
JobStatus.PROVISIONING: "deep_sky_blue1",
|
|
243
|
+
JobStatus.PULLING: "sea_green3",
|
|
244
|
+
JobStatus.RUNNING: "sea_green3",
|
|
245
|
+
JobStatus.TERMINATING: "deep_sky_blue1",
|
|
246
|
+
JobStatus.TERMINATED: "grey",
|
|
247
|
+
JobStatus.ABORTED: "gold1",
|
|
248
|
+
JobStatus.FAILED: "indian_red1",
|
|
249
|
+
JobStatus.DONE: "grey",
|
|
250
|
+
}
|
|
251
|
+
color = color_map.get(job_status, "white")
|
|
252
|
+
status_style = f"bold {color}" if not job_status.is_finished() else color
|
|
253
|
+
formatted_status_message = f"[{status_style}]{status_message}[/]"
|
|
254
|
+
if verbose and job_submission.inactivity_secs:
|
|
255
|
+
inactive_for = format_duration_multiunit(job_submission.inactivity_secs)
|
|
256
|
+
formatted_status_message += f" (inactive for {inactive_for})"
|
|
257
|
+
return formatted_status_message
|
|
258
|
+
|
|
259
|
+
|
|
260
|
+
def _get_show_deployment_replica_job(run: CoreRun, verbose: bool) -> tuple[bool, bool, bool]:
|
|
261
|
+
show_deployment_num = (
|
|
262
|
+
verbose and run.run_spec.configuration.type == "service"
|
|
263
|
+
) or run.is_deployment_in_progress()
|
|
264
|
+
|
|
265
|
+
replica_nums = {job.job_spec.replica_num for job in run.jobs}
|
|
266
|
+
show_replica = len(replica_nums) > 1
|
|
267
|
+
|
|
268
|
+
jobs_by_replica: Dict[int, List[Any]] = {}
|
|
269
|
+
for job in run.jobs:
|
|
270
|
+
replica_num = job.job_spec.replica_num
|
|
271
|
+
if replica_num not in jobs_by_replica:
|
|
272
|
+
jobs_by_replica[replica_num] = []
|
|
273
|
+
jobs_by_replica[replica_num].append(job)
|
|
274
|
+
|
|
275
|
+
show_job = any(
|
|
276
|
+
len({j.job_spec.job_num for j in jobs}) > 1 for jobs in jobs_by_replica.values()
|
|
277
|
+
)
|
|
278
|
+
|
|
279
|
+
return show_deployment_num, show_replica, show_job
|
|
280
|
+
|
|
281
|
+
|
|
282
|
+
def _format_job_name(
|
|
283
|
+
job: Job,
|
|
284
|
+
latest_job_submission: JobSubmission,
|
|
285
|
+
show_deployment_num: bool,
|
|
286
|
+
show_replica: bool,
|
|
287
|
+
show_job: bool,
|
|
288
|
+
group_index: Optional[int] = None,
|
|
289
|
+
last_shown_group_index: Optional[int] = None,
|
|
290
|
+
) -> str:
|
|
291
|
+
name_parts = []
|
|
292
|
+
prefix = ""
|
|
293
|
+
if show_replica:
|
|
294
|
+
# Show group information if replica groups are used
|
|
295
|
+
if group_index is not None:
|
|
296
|
+
# Show group=X replica=Y when group changes, or just replica=Y when same group
|
|
297
|
+
if group_index != last_shown_group_index:
|
|
298
|
+
# First job in group: use 3 spaces indent
|
|
299
|
+
prefix = " "
|
|
300
|
+
name_parts.append(f"group={group_index} replica={job.job_spec.replica_num}")
|
|
301
|
+
else:
|
|
302
|
+
# Subsequent job in same group: align "replica=" with first job's "replica="
|
|
303
|
+
# Calculate padding: width of " group={last_shown_group_index} "
|
|
304
|
+
padding_width = 3 + len(f"group={last_shown_group_index}") + 1
|
|
305
|
+
prefix = " " * padding_width
|
|
306
|
+
name_parts.append(f"replica={job.job_spec.replica_num}")
|
|
307
|
+
else:
|
|
308
|
+
# Legacy behavior: no replica groups
|
|
309
|
+
prefix = " "
|
|
310
|
+
name_parts.append(f"replica={job.job_spec.replica_num}")
|
|
311
|
+
else:
|
|
312
|
+
prefix = " "
|
|
313
|
+
|
|
314
|
+
if show_job:
|
|
315
|
+
name_parts.append(f"job={job.job_spec.job_num}")
|
|
316
|
+
name_suffix = (
|
|
317
|
+
f" deployment={latest_job_submission.deployment_num}" if show_deployment_num else ""
|
|
318
|
+
)
|
|
319
|
+
name_value = prefix + (" ".join(name_parts) if name_parts else "")
|
|
320
|
+
name_value += name_suffix
|
|
321
|
+
return name_value
|
|
322
|
+
|
|
323
|
+
|
|
324
|
+
def _format_price(price: float, is_spot: bool) -> str:
|
|
325
|
+
price_str = f"${price:.4f}".rstrip("0").rstrip(".")
|
|
326
|
+
if is_spot:
|
|
327
|
+
price_str += " (spot)"
|
|
328
|
+
return price_str
|
|
329
|
+
|
|
330
|
+
|
|
331
|
+
def _format_backend(backend_type: BackendType, region: str) -> str:
|
|
332
|
+
backend_str = backend_type.value
|
|
333
|
+
if backend_type == BackendType.REMOTE:
|
|
334
|
+
backend_str = "ssh"
|
|
335
|
+
return f"{backend_str} ({region})"
|
|
336
|
+
|
|
337
|
+
|
|
338
|
+
def _format_instance_type(
|
|
339
|
+
instance_type: InstanceType,
|
|
340
|
+
shared_offer: Optional[InstanceOfferWithAvailability],
|
|
341
|
+
reservation: Optional[str],
|
|
342
|
+
) -> str:
|
|
343
|
+
instance_type_str = instance_type.name
|
|
344
|
+
if shared_offer is not None:
|
|
345
|
+
instance_type_str += f" ({shared_offer.blocks}/{shared_offer.total_blocks})"
|
|
346
|
+
if reservation is not None:
|
|
347
|
+
instance_type_str += f" ({reservation})"
|
|
348
|
+
return instance_type_str
|
|
349
|
+
|
|
350
|
+
|
|
351
|
+
def _format_run_name(run: CoreRun, show_deployment_num: bool) -> str:
|
|
352
|
+
parts: List[str] = [run.run_spec.run_name]
|
|
353
|
+
if show_deployment_num:
|
|
354
|
+
parts.append(f" [secondary]deployment={run.deployment_num}[/]")
|
|
355
|
+
return "".join(parts)
|
|
356
|
+
|
|
357
|
+
|
|
358
|
+
def get_runs_table(
|
|
359
|
+
runs: List[Run], verbose: bool = False, format_date: DateFormatter = pretty_date
|
|
360
|
+
) -> Table:
|
|
361
|
+
table = Table(box=None, expand=shutil.get_terminal_size(fallback=(120, 40)).columns <= 110)
|
|
362
|
+
table.add_column("NAME", style="bold", no_wrap=True, ratio=2)
|
|
363
|
+
table.add_column("BACKEND", style="grey58", ratio=2)
|
|
364
|
+
if verbose:
|
|
365
|
+
table.add_column("RESOURCES", style="grey58", ratio=3)
|
|
366
|
+
table.add_column("INSTANCE TYPE", style="grey58", no_wrap=True, ratio=1)
|
|
367
|
+
else:
|
|
368
|
+
table.add_column("GPU", ratio=2)
|
|
369
|
+
table.add_column("PRICE", style="grey58", ratio=1)
|
|
370
|
+
table.add_column("STATUS", no_wrap=True, ratio=1)
|
|
371
|
+
if verbose or any(
|
|
372
|
+
run._run.is_deployment_in_progress()
|
|
373
|
+
and any(job.job_submissions[-1].probes for job in run._run.jobs)
|
|
374
|
+
for run in runs
|
|
375
|
+
):
|
|
376
|
+
table.add_column("PROBES", ratio=1)
|
|
377
|
+
table.add_column("SUBMITTED", style="grey58", no_wrap=True, ratio=1)
|
|
378
|
+
if verbose:
|
|
379
|
+
table.add_column("ERROR", no_wrap=True, ratio=2)
|
|
380
|
+
|
|
381
|
+
for run in runs:
|
|
382
|
+
run = run._run # TODO(egor-s): make public attribute
|
|
383
|
+
show_deployment_num, show_replica, show_job = _get_show_deployment_replica_job(
|
|
384
|
+
run, verbose
|
|
385
|
+
)
|
|
386
|
+
merge_job_rows = len(run.jobs) == 1 and not show_deployment_num
|
|
387
|
+
|
|
388
|
+
group_name_to_index: Dict[str, int] = {}
|
|
389
|
+
if run.run_spec.configuration.type == "service" and hasattr(
|
|
390
|
+
run.run_spec.configuration, "replica_groups"
|
|
391
|
+
):
|
|
392
|
+
replica_groups = run.run_spec.configuration.replica_groups
|
|
393
|
+
if replica_groups:
|
|
394
|
+
for idx, group in enumerate(replica_groups):
|
|
395
|
+
assert group.name is not None, "Group name is always set"
|
|
396
|
+
group_name = group.name
|
|
397
|
+
group_name_to_index[group_name] = idx
|
|
398
|
+
|
|
399
|
+
run_row: Dict[Union[str, int], Any] = {
|
|
400
|
+
"NAME": _format_run_name(run, show_deployment_num),
|
|
401
|
+
"SUBMITTED": format_date(run.submitted_at),
|
|
402
|
+
"STATUS": _format_run_status(run),
|
|
403
|
+
"RESOURCES": "-",
|
|
404
|
+
"GPU": "-",
|
|
405
|
+
"PRICE": "-",
|
|
406
|
+
}
|
|
407
|
+
if run.error:
|
|
408
|
+
run_row["ERROR"] = run.error
|
|
409
|
+
if not merge_job_rows:
|
|
410
|
+
add_row_from_dict(table, run_row)
|
|
411
|
+
|
|
412
|
+
# Sort jobs by group index first, then by replica_num within each group
|
|
413
|
+
def get_job_sort_key(job: Job) -> tuple:
|
|
414
|
+
group_index = None
|
|
415
|
+
if group_name_to_index:
|
|
416
|
+
group_index = group_name_to_index.get(job.job_spec.replica_group)
|
|
417
|
+
# Use a large number for jobs without groups to put them at the end
|
|
418
|
+
return (group_index if group_index is not None else 999999, job.job_spec.replica_num)
|
|
419
|
+
|
|
420
|
+
sorted_jobs = sorted(run.jobs, key=get_job_sort_key)
|
|
421
|
+
|
|
422
|
+
last_shown_group_index: Optional[int] = None
|
|
423
|
+
for job in sorted_jobs:
|
|
424
|
+
latest_job_submission = job.job_submissions[-1]
|
|
425
|
+
status_formatted = _format_job_submission_status(latest_job_submission, verbose)
|
|
426
|
+
|
|
427
|
+
# Get group index for this job
|
|
428
|
+
group_index: Optional[int] = None
|
|
429
|
+
if group_name_to_index:
|
|
430
|
+
group_index = group_name_to_index.get(job.job_spec.replica_group)
|
|
431
|
+
|
|
432
|
+
job_row: Dict[Union[str, int], Any] = {
|
|
433
|
+
"NAME": _format_job_name(
|
|
434
|
+
job,
|
|
435
|
+
latest_job_submission,
|
|
436
|
+
show_deployment_num,
|
|
437
|
+
show_replica,
|
|
438
|
+
show_job,
|
|
439
|
+
group_index=group_index,
|
|
440
|
+
last_shown_group_index=last_shown_group_index,
|
|
441
|
+
),
|
|
442
|
+
"STATUS": status_formatted,
|
|
443
|
+
"PROBES": _format_job_probes(
|
|
444
|
+
job.job_spec.probes, latest_job_submission.probes, latest_job_submission.status
|
|
445
|
+
),
|
|
446
|
+
"SUBMITTED": format_date(latest_job_submission.submitted_at),
|
|
447
|
+
"ERROR": latest_job_submission.error,
|
|
448
|
+
"RESOURCES": "-",
|
|
449
|
+
"GPU": "-",
|
|
450
|
+
"PRICE": "-",
|
|
451
|
+
}
|
|
452
|
+
# Update last shown group index for next iteration
|
|
453
|
+
if group_index is not None:
|
|
454
|
+
last_shown_group_index = group_index
|
|
455
|
+
jpd = latest_job_submission.job_provisioning_data
|
|
456
|
+
if jpd is not None:
|
|
457
|
+
shared_offer: Optional[InstanceOfferWithAvailability] = None
|
|
458
|
+
instance_type = jpd.instance_type
|
|
459
|
+
price = jpd.price
|
|
460
|
+
jrd = latest_job_submission.job_runtime_data
|
|
461
|
+
if jrd is not None and jrd.offer is not None and jrd.offer.total_blocks > 1:
|
|
462
|
+
# We only use offer data from jrd if the job is/was running on a shared
|
|
463
|
+
# instance (the instance blocks feature). In that case, jpd contains the full
|
|
464
|
+
# instance offer data, while jrd contains the shared offer (a fraction of
|
|
465
|
+
# the full offer). Although jrd always contains the offer, we don't use it in
|
|
466
|
+
# other cases, as, unlike jpd offer data, jrd offer is not updated after
|
|
467
|
+
# Compute.update_provisioning_data() call, but some backends, namely
|
|
468
|
+
# Kubernetes, may update offer data via that method.
|
|
469
|
+
# As long as we don't have a backend which both supports the blocks feature
|
|
470
|
+
# and may update offer data in update_provisioning_data(), this logic is fine.
|
|
471
|
+
shared_offer = jrd.offer
|
|
472
|
+
instance_type = shared_offer.instance
|
|
473
|
+
price = shared_offer.price
|
|
474
|
+
resources = instance_type.resources
|
|
475
|
+
job_row.update(
|
|
476
|
+
{
|
|
477
|
+
"BACKEND": _format_backend(jpd.backend, jpd.region),
|
|
478
|
+
"RESOURCES": resources.pretty_format(include_spot=False),
|
|
479
|
+
"GPU": resources.pretty_format(gpu_only=True, include_spot=False),
|
|
480
|
+
"INSTANCE TYPE": _format_instance_type(
|
|
481
|
+
instance_type, shared_offer, jpd.reservation
|
|
482
|
+
),
|
|
483
|
+
"PRICE": _format_price(price, resources.spot),
|
|
484
|
+
}
|
|
485
|
+
)
|
|
486
|
+
if merge_job_rows:
|
|
487
|
+
_status = job_row["STATUS"]
|
|
488
|
+
_resources = job_row["RESOURCES"]
|
|
489
|
+
_gpu = job_row["GPU"]
|
|
490
|
+
_price = job_row["PRICE"]
|
|
491
|
+
job_row.update(run_row)
|
|
492
|
+
job_row["RESOURCES"] = _resources
|
|
493
|
+
job_row["GPU"] = _gpu
|
|
494
|
+
job_row["PRICE"] = _price
|
|
495
|
+
job_row["STATUS"] = _status
|
|
496
|
+
add_row_from_dict(table, job_row, style="secondary" if len(run.jobs) != 1 else None)
|
|
497
|
+
|
|
498
|
+
return table
|
|
499
|
+
|
|
500
|
+
|
|
501
|
+
def _format_job_probes(
|
|
502
|
+
probe_specs: list[ProbeSpec], probes: list[Probe], job_status: JobStatus
|
|
503
|
+
) -> str:
|
|
504
|
+
if not probes or job_status != JobStatus.RUNNING:
|
|
505
|
+
return ""
|
|
506
|
+
statuses = []
|
|
507
|
+
for probe_spec, probe in zip(probe_specs, probes):
|
|
508
|
+
# NOTE: the symbols are documented in concepts/services.md, keep in sync.
|
|
509
|
+
if probe.success_streak >= probe_spec.ready_after:
|
|
510
|
+
status = "[code]✓[/]"
|
|
511
|
+
elif probe.success_streak > 0:
|
|
512
|
+
status = "[warning]~[/]"
|
|
513
|
+
else:
|
|
514
|
+
status = "[error]×[/]"
|
|
515
|
+
statuses.append(status)
|
|
516
|
+
# split into whitespace-delimited batches to allow column wrapping
|
|
517
|
+
return " ".join("".join(batch) for batch in batched(statuses, 5))
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
from typing import List
|
|
2
|
+
|
|
3
|
+
from rich.table import Table
|
|
4
|
+
|
|
5
|
+
from dstack._internal.cli.utils.common import add_row_from_dict, console
|
|
6
|
+
from dstack._internal.core.models.secrets import Secret
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
def print_secrets_table(secrets: List[Secret]) -> None:
|
|
10
|
+
console.print(get_secrets_table(secrets))
|
|
11
|
+
console.print()
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
def get_secrets_table(secrets: List[Secret]) -> Table:
|
|
15
|
+
table = Table(box=None)
|
|
16
|
+
table.add_column("NAME", no_wrap=True)
|
|
17
|
+
table.add_column("VALUE")
|
|
18
|
+
|
|
19
|
+
for secret in secrets:
|
|
20
|
+
row = {
|
|
21
|
+
"NAME": secret.name,
|
|
22
|
+
"VALUE": secret.value or "*" * 6,
|
|
23
|
+
}
|
|
24
|
+
add_row_from_dict(table, row)
|
|
25
|
+
return table
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
import uuid
|
|
2
|
+
from datetime import datetime
|
|
3
|
+
from pathlib import Path
|
|
4
|
+
from typing import Optional
|
|
5
|
+
|
|
6
|
+
import requests
|
|
7
|
+
from packaging import version as pkg_version
|
|
8
|
+
|
|
9
|
+
from dstack import version
|
|
10
|
+
from dstack._internal.cli.utils.common import console
|
|
11
|
+
from dstack._internal.utils.logging import get_logger
|
|
12
|
+
|
|
13
|
+
logger = get_logger(__name__)
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
def anonymous_installation_id() -> str:
|
|
17
|
+
ai_id_path = Path.home() / ".cache" / ".dstack" / "anonymous_installation_id"
|
|
18
|
+
if not ai_id_path.exists():
|
|
19
|
+
ai_id_path.parent.mkdir(parents=True, exist_ok=True)
|
|
20
|
+
with open(ai_id_path, "w") as f:
|
|
21
|
+
ai_id = uuid.uuid4().hex
|
|
22
|
+
f.write(ai_id)
|
|
23
|
+
else:
|
|
24
|
+
with open(ai_id_path, "r") as f:
|
|
25
|
+
ai_id = f.readline().strip()
|
|
26
|
+
return ai_id
|
|
27
|
+
|
|
28
|
+
|
|
29
|
+
def get_latest_version() -> Optional[str]:
|
|
30
|
+
latest_version_url = f"https://get.dstack.ai/{_get_channel()}/latest-version"
|
|
31
|
+
try:
|
|
32
|
+
response = requests.get(
|
|
33
|
+
latest_version_url,
|
|
34
|
+
params={
|
|
35
|
+
"anonymous_installation_id": anonymous_installation_id(),
|
|
36
|
+
"version": version.__version__,
|
|
37
|
+
},
|
|
38
|
+
timeout=5,
|
|
39
|
+
)
|
|
40
|
+
if response.status_code == 200:
|
|
41
|
+
return response.text.strip()
|
|
42
|
+
else:
|
|
43
|
+
return None
|
|
44
|
+
except Exception:
|
|
45
|
+
return None
|
|
46
|
+
|
|
47
|
+
|
|
48
|
+
def _is_last_check_time_outdated() -> bool:
|
|
49
|
+
path = _get_last_check_path()
|
|
50
|
+
if not path.exists():
|
|
51
|
+
return True
|
|
52
|
+
current_datetime = datetime.now()
|
|
53
|
+
modified_datetime = datetime.fromtimestamp(path.stat().st_mtime)
|
|
54
|
+
return (
|
|
55
|
+
modified_datetime.timetuple().tm_yday != current_datetime.timetuple().tm_yday
|
|
56
|
+
or modified_datetime.year != current_datetime.year
|
|
57
|
+
)
|
|
58
|
+
|
|
59
|
+
|
|
60
|
+
def is_update_available(current_version: str, latest_version: str) -> bool:
|
|
61
|
+
"""
|
|
62
|
+
Return True if latest_version is newer than current_version.
|
|
63
|
+
Pre-releases are only considered if the current version is also a pre-release.
|
|
64
|
+
"""
|
|
65
|
+
_current_version = pkg_version.parse(str(current_version))
|
|
66
|
+
_latest_version = pkg_version.parse(str(latest_version))
|
|
67
|
+
return _current_version < _latest_version and (
|
|
68
|
+
not _latest_version.is_prerelease or _current_version.is_prerelease
|
|
69
|
+
)
|
|
70
|
+
|
|
71
|
+
|
|
72
|
+
def _check_version():
|
|
73
|
+
latest_version = get_latest_version()
|
|
74
|
+
if latest_version is not None:
|
|
75
|
+
if is_update_available(version.__version__, latest_version):
|
|
76
|
+
console.print(f"A new version of dstack is available: [code]{latest_version}[/]\n")
|
|
77
|
+
|
|
78
|
+
|
|
79
|
+
def _update_last_check_time():
|
|
80
|
+
path = _get_last_check_path()
|
|
81
|
+
path.parent.mkdir(parents=True, exist_ok=True)
|
|
82
|
+
path.touch(exist_ok=True)
|
|
83
|
+
|
|
84
|
+
|
|
85
|
+
def _get_channel() -> str:
|
|
86
|
+
return "cli" if version.__is_release__ else "stgn-cli"
|
|
87
|
+
|
|
88
|
+
|
|
89
|
+
def _get_last_check_path() -> Path:
|
|
90
|
+
return Path.home() / ".cache" / ".dstack" / _get_channel() / "last_check"
|
|
91
|
+
|
|
92
|
+
|
|
93
|
+
def check_for_updates():
|
|
94
|
+
if version.__is_release__:
|
|
95
|
+
if _is_last_check_time_outdated():
|
|
96
|
+
logger.debug("Checking for updates...")
|
|
97
|
+
_check_version()
|
|
98
|
+
_update_last_check_time()
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
from typing import List
|
|
2
|
+
|
|
3
|
+
from rich.table import Table
|
|
4
|
+
|
|
5
|
+
from dstack._internal.cli.utils.common import add_row_from_dict, console
|
|
6
|
+
from dstack._internal.core.models.volumes import Volume
|
|
7
|
+
from dstack._internal.utils.common import DateFormatter, pretty_date
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
def print_volumes_table(volumes: List[Volume], verbose: bool = False):
|
|
11
|
+
table = get_volumes_table(volumes, verbose=verbose)
|
|
12
|
+
console.print(table)
|
|
13
|
+
console.print()
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
def get_volumes_table(
|
|
17
|
+
volumes: List[Volume], verbose: bool = False, format_date: DateFormatter = pretty_date
|
|
18
|
+
) -> Table:
|
|
19
|
+
table = Table(box=None)
|
|
20
|
+
table.add_column("NAME", no_wrap=True)
|
|
21
|
+
table.add_column("BACKEND")
|
|
22
|
+
if verbose:
|
|
23
|
+
table.add_column("REGION")
|
|
24
|
+
table.add_column("STATUS")
|
|
25
|
+
if verbose:
|
|
26
|
+
table.add_column("ATTACHED")
|
|
27
|
+
table.add_column("CREATED")
|
|
28
|
+
if verbose:
|
|
29
|
+
table.add_column("ERROR")
|
|
30
|
+
|
|
31
|
+
for volume in volumes:
|
|
32
|
+
backend = f"{volume.configuration.backend.value} ({volume.configuration.region})"
|
|
33
|
+
region = volume.configuration.region
|
|
34
|
+
if verbose:
|
|
35
|
+
backend = volume.configuration.backend.value
|
|
36
|
+
if (
|
|
37
|
+
verbose
|
|
38
|
+
and volume.provisioning_data is not None
|
|
39
|
+
and volume.provisioning_data.availability_zone is not None
|
|
40
|
+
):
|
|
41
|
+
region += f" ({volume.provisioning_data.availability_zone})"
|
|
42
|
+
attached = "-"
|
|
43
|
+
if volume.attachments is not None:
|
|
44
|
+
attached = ", ".join(
|
|
45
|
+
{va.instance.fleet_name for va in volume.attachments if va.instance.fleet_name}
|
|
46
|
+
)
|
|
47
|
+
attached = attached or "-"
|
|
48
|
+
row = {
|
|
49
|
+
"NAME": volume.name,
|
|
50
|
+
"BACKEND": backend,
|
|
51
|
+
"REGION": region,
|
|
52
|
+
"STATUS": volume.status,
|
|
53
|
+
"ATTACHED": attached,
|
|
54
|
+
"CREATED": format_date(volume.created_at),
|
|
55
|
+
"ERROR": volume.status_message,
|
|
56
|
+
}
|
|
57
|
+
add_row_from_dict(table, row)
|
|
58
|
+
return table
|