changes

Author: abidlabs

.github/workflows/pr-wheel-preview-build.yml

@@ -0,0 +1,73 @@
+name: PR wheel preview build
+
+on:
+  pull_request:
+    branches: [main]
+
+concurrency:
+  group: pr-wheel-build-${{ github.event.pull_request.number }}-${{ github.head_ref || github.run_id }}
+  cancel-in-progress: true
+
+permissions: {}
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      pull-requests: read
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v4
+        with:
+          python-version: "3.10"
+
+      - uses: actions/setup-node@v4
+        with:
+          node-version: "20"
+          cache: npm
+          cache-dependency-path: trackio/frontend/package-lock.json
+
+      - name: Build frontend
+        run: |
+          cd trackio/frontend
+          npm ci
+          npm run build
+
+      - name: Install build
+        run: python -m pip install build
+
+      - name: Build wheel
+        env:
+          SKIP_FRONTEND_BUILD: "1"
+        run: python -m build -w
+
+      - name: Write metadata
+        run: |
+          python3 << 'PY'
+          import json
+          from pathlib import Path
+
+          wheels = sorted(Path("dist").glob("trackio-*.whl"))
+          if len(wheels) != 1:
+              raise SystemExit(f"expected one trackio wheel, got {[p.name for p in wheels]}")
+          meta = {
+              "pr_number": ${{ github.event.pull_request.number }},
+              "sha": "${{ github.event.pull_request.head.sha }}",
+              "wheel_basename": wheels[0].name,
+          }
+          Path("output.json").write_text(json.dumps(meta))
+          PY
+
+      - name: Upload wheel
+        uses: actions/upload-artifact@v4
+        with:
+          name: trackio-wheel
+          path: dist/*.whl
+
+      - name: Upload metadata
+        uses: actions/upload-artifact@v4
+        with:
+          name: pr-wheel-preview-metadata
+          path: output.json

.github/workflows/pr-wheel-preview-deploy.yml

@@ -0,0 +1,75 @@
+name: PR wheel preview deploy
+
+on:
+  workflow_run:
+    workflows: ["PR wheel preview build"]
+    types:
+      - completed
+
+concurrency:
+  group: pr-wheel-deploy-${{ github.event.workflow_run.head_sha }}
+  cancel-in-progress: true
+
+jobs:
+  deploy:
+    if: github.event.workflow_run.conclusion == 'success' && github.event.workflow_run.event == 'pull_request'
+    runs-on: ubuntu-latest
+    permissions:
+      actions: read
+    outputs:
+      pr_number: ${{ steps.meta.outputs.pr_number }}
+      sha: ${{ steps.meta.outputs.sha }}
+      wheel_basename: ${{ steps.meta.outputs.wheel_basename }}
+    steps:
+      - name: Download metadata
+        uses: actions/download-artifact@v4
+        with:
+          name: pr-wheel-preview-metadata
+          path: metadata
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          repository: ${{ github.repository }}
+          run-id: ${{ github.event.workflow_run.id }}
+
+      - name: Download wheel
+        uses: actions/download-artifact@v4
+        with:
+          name: trackio-wheel
+          path: wheel
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          repository: ${{ github.repository }}
+          run-id: ${{ github.event.workflow_run.id }}
+
+      - name: Parse metadata
+        id: meta
+        run: |
+          echo "pr_number=$(jq -r .pr_number metadata/output.json)" >> $GITHUB_OUTPUT
+          echo "sha=$(jq -r .sha metadata/output.json)" >> $GITHUB_OUTPUT
+          echo "wheel_basename=$(jq -r .wheel_basename metadata/output.json)" >> $GITHUB_OUTPUT
+
+      - name: Install hf CLI
+        run: curl -LsSf https://hf.co/cli/install.sh | bash
+
+      - name: Upload wheel to Hugging Face bucket
+        env:
+          HF_TOKEN: ${{ secrets.HF_PYPI_PREVIEWS_TOKEN }}
+        run: |
+          echo "$HOME/.local/bin" >> $GITHUB_PATH
+          wheel_file=$(find wheel -maxdepth 1 -type f -name "*.whl" -print -quit)
+          hf buckets cp "$wheel_file" "hf://buckets/trackio/trackio-wheels/${{ steps.meta.outputs.sha }}/"
+
+  comment:
+    needs: deploy
+    if: needs.deploy.result == 'success'
+    uses: ./.github/workflows/comment-queue.yml
+    secrets:
+      gh_token: ${{ secrets.COMMENT_BOT_TOKEN }}
+    with:
+      pr_number: ${{ needs.deploy.outputs.pr_number }}
+      tag: trackio-pr-wheel
+      message: trackio-wheel~success~null
+      additional_text: |
+        **Install Trackio from this PR (includes built frontend)**
+
+        ```bash
+        pip install "https://huggingface.co/buckets/trackio/trackio-wheels/resolve/${{ needs.deploy.outputs.sha }}/${{ needs.deploy.outputs.wheel_basename }}"
+        ```