Rebase the fork on upstream 0.22.2+ for Nebius Dynamic Spot pricing (keep fork features)

.github/FUNDING.yml

@@ -0,0 +1,3 @@
+# These are supported funding model platforms
+
+github: [dstackai]

.github/workflows/build-artifacts.yml

@@ -64,7 +64,7 @@ jobs:
     strategy:
       matrix:
         os: [macos-latest, ubuntu-latest, windows-latest]
-        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
+        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
     steps:
       - uses: actions/checkout@v4
       - name: Set up Python ${{ matrix.python-version }}
@@ -74,7 +74,7 @@ jobs:
       - name: Install dependencies
         run: uv sync --all-extras
       - name: Run pyright
-        uses: jakebailey/pyright-action@v2
+        uses: jakebailey/pyright-action@v3
         with:
           pylance-version: latest-release
       - name: Download frontend build
@@ -100,10 +100,7 @@ jobs:
     defaults:
       run:
         working-directory: runner
-    runs-on: ${{ matrix.os }}
-    strategy:
-      matrix:
-        os: [ubuntu-latest, macos-latest]
+    runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
       - name: Set up Go
@@ -126,11 +123,6 @@ jobs:
           if [[ "${{ inputs.go-integration-tests }}" == "true" ]]; then
             SHORT=""
           fi
-          # Skip failing integration tests on macOS for release builds.
-          # TODO: https://github.com/dstackai/dstack/issues/3005
-          if [[ "${{ inputs.staging }}" == "false" && "${{ startsWith(matrix.os, 'macos') }}" == "true" ]]; then
-            SHORT="-short"
-          fi
           go version
           go fmt $(go list ./... | grep -v /vendor/)
           go vet $(go list ./... | grep -v /vendor/)
@@ -171,33 +163,6 @@ jobs:
             runner/dstack-shim-${{ matrix.goos }}-${{ matrix.goarch }}
           retention-days: 1
 
-  gateway-build:
-    runs-on: ubuntu-latest
-    steps:
-      - uses: actions/checkout@v4
-      - name: Set up uv
-        uses: astral-sh/setup-uv@v5
-        with:
-          python-version: 3.11
-      - name: Build package
-        working-directory: gateway
-        run: |
-          echo "__version__ = \"${{ inputs.version }}\"" > src/dstack/gateway/version.py
-          # TODO: depend on a specific dstackai/dstack commit for staging builds?
-          if [[ "${{ inputs.staging }}" == "false" ]]; then
-            sed \
-              -i.old \
-              "s|@ https://github.com/dstackai/dstack/archive/refs/heads/master.tar.gz|== ${{ inputs.version }}|" \
-              pyproject.toml
-            diff pyproject.toml pyproject.toml.old > /dev/null && echo "Could not set version" && exit 1
-          fi
-          uv build
-      - uses: actions/upload-artifact@v4
-        with:
-          name: dstack-gateway
-          path: gateway/dist/dstack_gateway-${{ inputs.version }}-py3-none-any.whl
-          retention-days: 1
-
   python-build:
     needs: [code-lint, frontend-build]
     runs-on: ubuntu-latest
@@ -224,12 +189,14 @@ jobs:
             VERSION=${{ inputs.version }}
             IS_RELEASE=True
           fi
-          BASE_IMAGE=$(cat src/dstack/version.py | grep "base_image = ")
-          BASE_IMAGE_UBUNTU_VERSION=$(cat src/dstack/version.py | grep "base_image_ubuntu_version = ")
+          DOCKER_BASE_IMAGE=$(cat src/dstack/version.py | grep "docker_base_image = ")
+          DOCKER_BASE_IMAGE_UBUNTU_VERSION=$(cat src/dstack/version.py | grep "docker_base_image_ubuntu_version = ")
+          VM_BASE_IMAGE=$(cat src/dstack/version.py | grep "vm_base_image = ")
           echo "__version__ = \"$VERSION\"" > src/dstack/version.py
           echo "__is_release__ = $IS_RELEASE" >> src/dstack/version.py
-          echo $BASE_IMAGE >> src/dstack/version.py
-          echo $BASE_IMAGE_UBUNTU_VERSION >> src/dstack/version.py
+          echo $DOCKER_BASE_IMAGE >> src/dstack/version.py
+          echo $DOCKER_BASE_IMAGE_UBUNTU_VERSION >> src/dstack/version.py
+          echo $VM_BASE_IMAGE >> src/dstack/version.py
           cp README.md src
           uv build
       - uses: actions/upload-artifact@v4
@@ -251,8 +218,8 @@ jobs:
       - name: Generate json schema
         run: |
           mkdir /tmp/json-schemas
-          uv run python -c "from dstack._internal.core.models.configurations import DstackConfiguration; print(DstackConfiguration.schema_json())" > /tmp/json-schemas/configuration.json
-          uv run python -c "from dstack._internal.core.models.profiles import ProfilesConfig; print(ProfilesConfig.schema_json())" > /tmp/json-schemas/profiles.json
+          uv run python -c "from dstack._internal.core.models.configurations import DstackConfiguration; import json; print(json.dumps(DstackConfiguration.model_json_schema()))" > /tmp/json-schemas/configuration.json
+          uv run python -c "from dstack._internal.core.models.profiles import ProfilesConfig; import json; print(json.dumps(ProfilesConfig.model_json_schema()))" > /tmp/json-schemas/profiles.json
       - uses: actions/upload-artifact@v4
         with:
           name: json-schemas

.github/workflows/build-docs.yml

@@ -1,4 +1,8 @@
-name: Build Docs
+name: Build Site
+
+# Reusable build of the full site: the React landing (website/) overlaid onto the
+# MkDocs docs + blog build, via scripts/docs/build_site.sh. Produces the `site` artifact
+# consumed by docs.yaml (deploy) and build.yml (PR build check).
 
 on:
   workflow_call:
@@ -11,15 +15,21 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           python-version: 3.11
+      - uses: actions/setup-node@v4
+        with:
+          node-version: 20
+          cache: npm
+          cache-dependency-path: website/package-lock.json
       - name: Install dstack
         run: |
           uv sync --extra server
       - name: Build
         run: |
           sudo apt-get update && sudo apt-get install -y libcairo2-dev libfreetype6-dev libffi-dev libjpeg-dev libpng-dev libz-dev
-          uv run mkdocs build -s
+          ./scripts/docs/build_site.sh
       - uses: actions/upload-artifact@v4
         with:
           name: site
           path: site
           retention-days: 1
+          include-hidden-files: true

.github/workflows/build.yml

@@ -5,7 +5,7 @@ on:
     branches:
       - "master"
     paths-ignore:
-      - "docs/**"
+      - "mkdocs/**"
       - "mkdocs.yml"
   pull_request:
     branches:

.github/workflows/docker-tt-smi.yml

@@ -11,6 +11,11 @@ on:
         description: "TT SMI version"
         required: true
         default: "3.0.25"
+      tag_latest:
+        description: "Update 'latest'"
+        type: boolean
+        required: true
+        default: false
 
 jobs:
   build-tt-smi:
@@ -38,8 +43,12 @@ jobs:
             --build-arg IMAGE_NAME=${IMAGE_NAME} \
             --build-arg TT_SMI_VERSION=${{ inputs.tt_smi_version }} \
             --build-arg BUILD_DATE=$(date --utc --iso-8601=seconds)Z \
-            --tag ${IMAGE_NAME}:latest
-          VERSION=$(docker inspect --format '{{ index .Config.Labels "org.opencontainers.image.version" }}' ${IMAGE_NAME})
-          docker tag ${IMAGE_NAME}:latest ${IMAGE_NAME}:${VERSION}
+            --tag ${IMAGE_NAME}:${{ inputs.tt_smi_version }}
+          VERSION=$(docker inspect --format '{{ index .Config.Labels "org.opencontainers.image.version" }}' ${IMAGE_NAME}:${{ inputs.tt_smi_version }})
+          docker tag ${IMAGE_NAME}:${{ inputs.tt_smi_version }} ${IMAGE_NAME}:${VERSION}
           docker push ${IMAGE_NAME}:${VERSION}
-          docker push ${IMAGE_NAME}:latest
+      - name: Tag and push latest
+        if: ${{ inputs.tag_latest }}
+        run: |
+          docker tag ${{ inputs.image_name }}:${{ inputs.tt_smi_version }} ${{ inputs.image_name }}:latest
+          docker push ${{ inputs.image_name }}:latest