oh-pi 0.1.39 → 0.1.41

package/README.fr.md

@@ -123,6 +123,29 @@ Arrêt forcé        →  À maxTurns+1 : SIGTERM → SIGKILL si nécessaire
 
 Éclaireuse : 8 tours · Ouvrière : 15 tours · Soldat : 8 tours
 
+### Sélection des modèles
+
+La colonie détecte automatiquement les modèles disponibles et laisse le LLM choisir le meilleur par rôle :
+
+| Rôle | Stratégie | Exemple |
+|------|-----------|---------|
+| Éclaireuse | Rapide & économique — lecture seule | `claude-haiku-4-5`, `gpt-4o-mini` |
+| Ouvrière | Capable — modifie le code | `claude-sonnet-4-0`, `gpt-4o` |
+| Soldat | Même que ouvrière ou légèrement moins cher | `claude-sonnet-4-0` |
+
+Remplacement manuel si nécessaire :
+
+```
+ant_colony({
+  goal: "Migrer vers ESM",
+  scoutModel: "claude-haiku-4-5",
+  workerModel: "claude-sonnet-4-0",
+  soldierModel: "claude-sonnet-4-0"
+})
+```
+
+Omettez les trois pour utiliser le modèle de session actuel pour chaque fourmi.
+
 ### Rapport de coûts
 
 La colonie suit le coût par fourmi et le total, rapporté dans le résumé final. **Le coût n'interrompt jamais l'exécution** — les limites de tours et le contrôle de concurrence gèrent les ressources.

package/README.md

@@ -123,6 +123,29 @@ Hard kill       →  At maxTurns+1: SIGTERM → SIGKILL if needed
 
 Scout: 8 turns · Worker: 15 turns · Soldier: 8 turns
 
+### Model Selection
+
+The colony auto-detects available models and lets the LLM pick the best fit per role:
+
+| Role | Strategy | Example |
+|------|----------|---------|
+| Scout | Fast & cheap — only reads, no edits | `claude-haiku-4-5`, `gpt-4o-mini` |
+| Worker | Capable — makes code changes | `claude-sonnet-4-0`, `gpt-4o` |
+| Soldier | Same as worker or slightly cheaper | `claude-sonnet-4-0` |
+
+Override manually if needed:
+
+```
+ant_colony({
+  goal: "Migrate to ESM",
+  scoutModel: "claude-haiku-4-5",
+  workerModel: "claude-sonnet-4-0",
+  soldierModel: "claude-sonnet-4-0"
+})
+```
+
+Omit all three to use the current session model for every ant.
+
 ### Cost Reporting
 
 The colony tracks cost per ant and total spend, reported in the final summary. **Cost never interrupts execution** — turn limits and concurrency control handle resource management.

package/README.zh.md

@@ -123,6 +123,29 @@ oh-pi:
 
 侦察蚁：8 轮 · 工蚁：15 轮 · 兵蚁：8 轮
 
+### 模型选择
+
+蚁群自动检测可用模型，让 LLM 按角色选择最优模型：
+
+| 角色 | 策略 | 示例 |
+|------|------|------|
+| 侦察蚁 | 快速便宜 — 只读不改 | `claude-haiku-4-5`, `gpt-4o-mini` |
+| 工蚁 | 能力强 — 执行代码修改 | `claude-sonnet-4-0`, `gpt-4o` |
+| 兵蚁 | 与工蚁相同或稍便宜 | `claude-sonnet-4-0` |
+
+手动覆盖：
+
+```
+ant_colony({
+  goal: "迁移到 ESM",
+  scoutModel: "claude-haiku-4-5",
+  workerModel: "claude-sonnet-4-0",
+  soldierModel: "claude-sonnet-4-0"
+})
+```
+
+不指定则默认使用当前会话模型。
+
 ### 成本报告
 
 蚁群追踪每只蚂蚁和总花费，在最终报告中汇总。**成本不会中断执行** —— 轮次限制和并发控制负责资源管理。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oh-pi",
-  "version": "0.1.39",
+  "version": "0.1.41",
   "description": "One-click setup for pi-coding-agent. Like oh-my-zsh for pi.",
   "type": "module",
   "bin": {

package/pi-package/extensions/ant-colony/index.ts

@@ -79,7 +79,19 @@ function phasePipeline(status: string, theme: any): string {
 export default function antColonyExtension(pi: ExtensionAPI) {
 
   // ═══ Auto-trigger: 注入蚁群意识，LLM 自动判断何时启动 ═══
-  pi.on("before_agent_start", async () => {
+  pi.on("before_agent_start", async (ctx) => {
+    // 获取可用模型列表
+    let modelList = "";
+    try {
+      const { execSync } = await import("node:child_process");
+      const output = execSync("pi --list-models 2>/dev/null", { encoding: "utf-8", timeout: 5000 });
+      const models = output.trim().split("\n").slice(1) // skip header
+        .map(l => l.trim().split(/\s+/))
+        .filter(p => p.length >= 2)
+        .map(p => `${p[0]}/${p[1]}`);
+      if (models.length > 0) modelList = models.join(", ");
+    } catch { /* ignore */ }
+
     return {
       message: {
         customType: "ant-colony-context",
@@ -90,7 +102,17 @@ You have the ant_colony tool. Use it automatically when:
 - Task is complex enough that serial execution is inefficient
 
 Do NOT ask for confirmation. Call ant_colony directly with a clear goal.
-For simple single-file tasks, work directly without the colony.`,
+For simple single-file tasks, work directly without the colony.
+${modelList ? `
+[COLONY MODEL SELECTION]
+Available models: ${modelList}
+
+Strategy for choosing per-caste models:
+- scoutModel: Use a fast/cheap model (e.g. haiku, flash, gpt-4o-mini). Scouts only read, no edits.
+- workerModel: Use a capable model (e.g. sonnet, opus, gpt-4o). Workers make code changes.
+- soldierModel: Use same as worker or slightly cheaper. Soldiers review but don't edit.
+- If unsure, omit all three — defaults to current session model.
+- Prefer latest model versions for best quality.` : ""}`,
         display: false,
       },
     };