From 4063246ac276779bb140f4f2eb463fcfe9176a5d Mon Sep 17 00:00:00 2001
From: Codex <codex@local>
Date: Wed, 24 Jun 2026 21:57:11 +0800
Subject: [PATCH] Add SWIFT coding agent probe experiment scripts

---
 README.md            | 28 ++++++++++++++++++++++++++--
 SKILL.md             |  3 +++
 scripts/setup_env.sh | 27 ++++++++++++++++++++++++---
 3 files changed, 53 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index a716518..fe792da 100644
--- a/README.md
+++ b/README.md
@@ -29,13 +29,37 @@ git submodule update --init --recursive
 ./scripts/setup_env.sh
 ```
 
-脚本会显式设置 B300 代理：
+脚本会显式设置 B300 代理和国内 PyPI 源：
 
 ```bash
 http://100.72.0.101:8888
+https://mirrors.aliyun.com/pypi/simple/
 ```
 
-Python 依赖安装在仓库内 `.venv`，不会写入系统 Python。
+Python 依赖安装在仓库内 `.venv`，不会写入系统 Python。安装逻辑是：
+
+1. 先在 `.venv` 内安装 `uv`。
+2. 用 SWIFT 官方推荐的 `uv pip install -e third_party/modelscope-swift --torch-backend=auto` 安装 SWIFT 源码依赖。
+3. 再安装本 probe repo。
+4. 默认额外补充 `deepspeed<0.19`、`liger-kernel`、`tensorboard`、`nvitop`。
+
+如果要安装 SWIFT 的全量 optional 依赖：
+
+```bash
+INSTALL_SWIFT_ALL=1 ./scripts/setup_env.sh
+```
+
+如果只想安装 SWIFT core，不补额外训练包：
+
+```bash
+INSTALL_TRAINING_EXTRAS=0 ./scripts/setup_env.sh
+```
+
+如果要追加其他包：
+
+```bash
+EXTRA_UV_PACKAGES="flash-attn==2.8.3" ./scripts/setup_env.sh
+```
 
 ## 下载数据集
 
diff --git a/SKILL.md b/SKILL.md
index 0e462d3..84860b6 100644
--- a/SKILL.md
+++ b/SKILL.md
@@ -10,6 +10,9 @@ Use this skill when the user wants to run or modify the TI coding-agent SFT prob
   - `https_proxy=http://100.72.0.101:8888`
   - `HF_ENDPOINT=https://hf-mirror.com`
 - Do not install Python packages globally. Use the repository-local `.venv` created by `scripts/setup_env.sh`.
+- `scripts/setup_env.sh` installs `uv` inside `.venv`, then installs SWIFT from the submodule using SWIFT's documented source-install path: `uv pip install -e third_party/modelscope-swift --torch-backend=auto`.
+- Default setup adds training extras: `deepspeed<0.19`, `liger-kernel`, `tensorboard`, and `nvitop`. Disable with `INSTALL_TRAINING_EXTRAS=0`.
+- Install SWIFT optional all-dependencies only when explicitly needed: `INSTALL_SWIFT_ALL=1 ./scripts/setup_env.sh`.
 - Do not start GPU training before checking GPU occupancy with `nvidia-smi`.
 - Do not commit or upload `data/`, `models/`, `outputs/`, `runs/`, or `logs/`.
 
diff --git a/scripts/setup_env.sh b/scripts/setup_env.sh
index f8aed56..166eba0 100755
--- a/scripts/setup_env.sh
+++ b/scripts/setup_env.sh
@@ -9,12 +9,33 @@ export https_proxy="${https_proxy:-http://100.72.0.101:8888}"
 export HTTP_PROXY="${HTTP_PROXY:-${http_proxy}}"
 export HTTPS_PROXY="${HTTPS_PROXY:-${https_proxy}}"
 export PIP_INDEX_URL="${PIP_INDEX_URL:-https://mirrors.aliyun.com/pypi/simple/}"
+export UV_DEFAULT_INDEX="${UV_DEFAULT_INDEX:-${PIP_INDEX_URL}}"
+export UV_INDEX_URL="${UV_INDEX_URL:-${PIP_INDEX_URL}}"
 
 python3 -m venv .venv
 source .venv/bin/activate
-python -m pip install -U pip setuptools wheel
-python -m pip install -e .
-python -m pip install -e third_party/modelscope-swift
+python -m pip install -U pip setuptools wheel uv
+
+UV=(python -m uv pip)
+"${UV[@]}" install -U pip setuptools wheel
+
+if [[ "${INSTALL_SWIFT_ALL:-0}" == "1" ]]; then
+  "${UV[@]}" install -e 'third_party/modelscope-swift[all]' --torch-backend=auto
+else
+  "${UV[@]}" install -e third_party/modelscope-swift --torch-backend=auto
+fi
+
+"${UV[@]}" install -e .
+
+if [[ "${INSTALL_TRAINING_EXTRAS:-1}" == "1" ]]; then
+  "${UV[@]}" install -U 'deepspeed<0.19' liger-kernel tensorboard nvitop
+fi
+
+if [[ -n "${EXTRA_UV_PACKAGES:-}" ]]; then
+  # shellcheck disable=SC2206
+  EXTRA_PACKAGES=(${EXTRA_UV_PACKAGES})
+  "${UV[@]}" install -U "${EXTRA_PACKAGES[@]}"
+fi
 
 mkdir -p data/raw data/processed models outputs runs logs
 python - <<'PY'