akg-env-setup

akg_agents 环境准备

⛔ 核心规则

1. 检查失败后只有两条路径：询问用户是否有其他可用环境、询问用户是否同意自动安装。除此之外禁止任何操作。
2. 遇到 🛑 必须调用 question 工具并等待用户回复，不得跳过或自行决定。
3. 禁止执行本文档未列出的命令。

流程总览

Step 0  解析 $HOME_DIR → 读取缓存
  ├─ 命中 ──────────────────────────────┬─ FULL_SETUP → Step 5
  │                                     └─ 否则 → 结束
  └─ 未命中 → Step 1  检查 akg_cli + LLM
                  ├─ 通过 → Step 4  采集硬件/Framework/DSL
                  │             ├─ 4a 硬件
                  │             ├─ 4b Framework
                  │             ├─ 4c DSL & 后端推断
                  │             └─ 4d 写入缓存（必须，不可跳过）
                  │                ├─ FULL_SETUP → Step 5 → (Step 6 [可选，仅需要新依赖时]) → 结束
                  │                └─ 否则 → 结束
                  └─ 失败 → Step 2  询问用户（一次收集处理方式+环境信息）
                              ├─ 已有环境 → Step 1（重试）
                              └─ 需要安装 → Step 3 → Step 1（重试）

---

Step 0: 解析 HOME_DIR & 读取环境缓存

首先执行以下命令获取 home 目录绝对路径，赋值给 $HOME_DIR：

echo $HOME

后续所有路径中的 $HOME_DIR 均使用此值。

然后读取 $HOME_DIR/.akg/check_env.md：

• 文件存在：
  • FULL_SETUP → 跳到 Step 5（每次任务都需确认参数）
  • 非 FULL_SETUP → 报告就绪，流程结束
• 文件不存在 → 进入 Step 1

---

Step 1: 执行两项检查

需要先确定 $ENV_TYPE、$CONDA_ENV/$VENV_PATH 和 $AKG_AGENTS_DIR。优先使用调用方提供的值，未提供时使用当前已激活的环境。

检查 1 — akg_cli 可用性：

conda:

conda run -n $CONDA_ENV --no-capture-output bash -c \
  "cd $AKG_AGENTS_DIR && source env.sh && which akg_cli"

venv:

bash -c "source $VENV_PATH/bin/activate && cd $AKG_AGENTS_DIR && source env.sh && which akg_cli"

检查 2 — LLM 连通性：

conda:

conda run -n $CONDA_ENV --no-capture-output bash -c \
  "cd $AKG_AGENTS_DIR && source env.sh && \
   python tools/v2/use_llm_check/test_run_llm.py"

venv:

bash -c "source $VENV_PATH/bin/activate && cd $AKG_AGENTS_DIR && source env.sh && \
  python tools/v2/use_llm_check/test_run_llm.py"

判定：

• ✅ 两项全部通过 → 进入 Step 4
• ❌ 任一失败 → 进入 Step 2，禁止自行尝试其他环境。

---

Step 2: 🛑 报告失败并询问用户