oh-pi 0.1.21 → 0.1.23

package/README.fr.md

@@ -0,0 +1,196 @@
+<div align="center">
+
+<img src="./logo.svg" width="180" alt="oh-pi logo"/>
+
+# 🐜 oh-pi
+
+**Une seule commande pour booster [pi-coding-agent](https://github.com/badlogic/pi-mono).**
+
+Comme oh-my-zsh pour pi — mais avec une colonie de fourmis autonome.
+
+[![npm](https://img.shields.io/npm/v/oh-pi)](https://www.npmjs.com/package/oh-pi)
+[![license](https://img.shields.io/npm/l/oh-pi)](./LICENSE)
+[![node](https://img.shields.io/node/v/oh-pi)](https://nodejs.org)
+
+[English](./README.md) | [中文](./README.zh.md) | [Français](./README.fr.md)
+
+```bash
+npx oh-pi
+```
+
+</div>
+
+---
+
+## Pourquoi
+
+pi-coding-agent est puissant dès l'installation. Mais configurer manuellement les fournisseurs, thèmes, extensions, compétences et modèles de prompts est fastidieux. oh-pi vous offre une TUI moderne qui fait tout en moins d'une minute — et intègre un **essaim de fourmis** qui transforme pi en système multi-agents.
+
+## Démarrage rapide
+
+```bash
+npx oh-pi    # tout configurer
+pi           # commencer à coder
+```
+
+C'est tout. oh-pi détecte votre environnement, vous guide dans la configuration et génère `~/.pi/agent/` pour vous.
+
+Vous avez déjà une config ? oh-pi la détecte et propose une **sauvegarde avant écrasement**.
+
+## Ce que vous obtenez
+
+```
+~/.pi/agent/
+├── auth.json            Clés API (permissions 0600)
+├── settings.json        Modèle, thème, niveau de réflexion
+├── keybindings.json     Raccourcis Vim/Emacs (optionnel)
+├── AGENTS.md            Directives IA par rôle
+├── extensions/          4 extensions
+│   ├── safe-guard       Confirmation des commandes dangereuses + protection des chemins
+│   ├── git-guard        Checkpoints auto stash + alerte dépôt modifié
+│   ├── auto-session     Nommage de session depuis le premier message
+│   └── ant-colony/      🐜 Essaim multi-agents autonome
+├── prompts/             10 modèles (/review /fix /commit /test ...)
+├── skills/              4 compétences (debug, git, setup, colony)
+└── themes/              6 thèmes personnalisés
+```
+
+## Modes de configuration
+
+| Mode | Étapes | Pour |
+|------|--------|------|
+| 🚀 **Rapide** | 3 | Choisir fournisseur → entrer la clé → terminé |
+| 📦 **Préréglage** | 2 | Choisir un profil de rôle → entrer la clé |
+| 🎛️ **Personnalisé** | 6 | Tout choisir soi-même |
+
+### Préréglages
+
+| | Thème | Réflexion | Inclut |
+|---|-------|-----------|--------|
+| 🟢 Débutant | oh-pi Dark | moyen | Sécurité + bases git |
+| 🔵 Développeur Pro | Catppuccin | élevé | Chaîne d'outils complète |
+| 🟣 Chercheur en sécurité | Cyberpunk | élevé | Audit + pentesting |
+| 🟠 Data & IA | Tokyo Night | moyen | MLOps + pipelines |
+| 🔴 Minimal | Default | désactivé | Noyau uniquement |
+| ⚫ Pleine puissance | oh-pi Dark | élevé | Tout + colonie de fourmis |
+
+### Fournisseurs
+
+Anthropic · OpenAI · Google Gemini · Groq · OpenRouter · xAI · Mistral · [FOXNIO](https://www.foxnio.com) (fournisseur Claude d'intérêt public recommandé)
+
+Détection automatique des clés API depuis les variables d'environnement.
+
+## 🐜 Colonie de fourmis
+
+La fonctionnalité phare. Un essaim multi-agents inspiré de l'écologie réelle des fourmis.
+
+```
+Vous : "Refactorer l'auth des sessions vers JWT"
+
+oh-pi :
+  🔍 Fourmis éclaireuses explorent le code (haiku — rapide, économique)
+  📋 Pool de tâches généré à partir des découvertes
+  ⚒️  Fourmis ouvrières exécutent en parallèle (sonnet — capable)
+  🛡️ Fourmis soldats révisent tous les changements (sonnet — minutieux)
+  ✅ Terminé — rapport de synthèse avec métriques
+```
+
+### Pourquoi des fourmis ?
+
+Les vraies colonies de fourmis résolvent des problèmes complexes sans contrôle central. Chaque fourmi suit des règles simples, communique par **pistes de phéromones**, et la colonie s'auto-organise. oh-pi reproduit directement ce modèle :
+
+| Vraies fourmis | oh-pi |
+|----------------|-------|
+| L'éclaireuse trouve la nourriture | L'éclaireuse scanne le code, identifie les cibles |
+| Piste de phéromones | `.ant-colony/pheromone.jsonl` — découvertes partagées |
+| L'ouvrière transporte la nourriture | L'ouvrière exécute la tâche sur les fichiers assignés |
+| Le soldat garde le nid | Le soldat révise les changements, demande des corrections |
+| Plus de nourriture → plus de fourmis | Plus de tâches → concurrence plus élevée (auto-adaptée) |
+| Les phéromones s'évaporent | Demi-vie de 10 min — les infos obsolètes s'estompent |
+
+### Déclenchement automatique
+
+Le LLM décide quand déployer la colonie. Vous n'avez pas à y penser :
+
+- **≥3 fichiers** à modifier → colonie
+- **Flux de travail parallèles** possibles → colonie
+- **Un seul fichier** à modifier → exécution directe (pas de surcharge colonie)
+
+Ou déclencher manuellement :
+
+```
+/colony migrer tout le projet de CJS vers ESM
+```
+
+### Concurrence adaptative
+
+La colonie trouve automatiquement le parallélisme optimal pour votre machine :
+
+```
+Démarrage à froid  →  1-2 fourmis (conservateur)
+Exploration        →  +1 par vague, surveillance du débit
+Débit en baisse    →  verrouiller l'optimal, stabiliser
+CPU > 85%          →  réduire immédiatement
+429 rate limit     →  diviser la concurrence par 2 + backoff exponentiel (15s→30s→60s)
+Tâches terminées   →  réduire au minimum
+```
+
+### Sécurité des fichiers
+
+Une fourmi par fichier. Toujours. Les tâches en conflit sont automatiquement bloquées et reprennent à la libération des verrous.
+
+## Thèmes
+
+| | |
+|---|---|
+| 🌙 **oh-pi Dark** | Cyan + violet, contraste élevé |
+| 🌙 **Cyberpunk** | Magenta néon + cyan électrique |
+| 🌙 **Nord** | Palette bleu arctique |
+| 🌙 **Catppuccin Mocha** | Pastel sur fond sombre |
+| 🌙 **Tokyo Night** | Crépuscule bleu + violet |
+| 🌙 **Gruvbox Dark** | Tons rétro chaleureux |
+
+## Modèles de prompts
+
+```
+/review    Revue de code : bugs, sécurité, performance
+/fix       Corriger les erreurs avec un minimum de changements
+/explain   Expliquer le code, du simple au détaillé
+/refactor  Refactorer en préservant le comportement
+/test      Générer des tests
+/commit    Message Conventional Commit
+/pr        Description de Pull Request
+/security  Audit de sécurité OWASP
+/optimize  Optimisation des performances
+/document  Générer la documentation
+```
+
+## Modèles AGENTS.md
+
+| Modèle | Focus |
+|--------|-------|
+| Développeur généraliste | Directives de codage universelles |
+| Développeur Full-Stack | Frontend + backend + BDD |
+| Chercheur en sécurité | Pentesting & audit |
+| Ingénieur Data & IA | MLOps & pipelines |
+| 🐜 Opérateur de colonie | Orchestration multi-agents |
+
+## Aussi un package Pi
+
+Passez le configurateur, installez directement les ressources :
+
+```bash
+pi install npm:oh-pi
+```
+
+Ajoute tous les thèmes, prompts, compétences et extensions à votre configuration pi existante.
+
+## Prérequis
+
+- Node.js ≥ 20
+- Au moins une clé API LLM
+- pi-coding-agent (installé automatiquement si absent)
+
+## Licence
+
+MIT

package/README.md

@@ -12,6 +12,8 @@ Like oh-my-zsh for pi — but with an autonomous ant colony.
 [![license](https://img.shields.io/npm/l/oh-pi)](./LICENSE)
 [![node](https://img.shields.io/node/v/oh-pi)](https://nodejs.org)
 
+[English](./README.md) | [中文](./README.zh.md) | [Français](./README.fr.md)
+
 ```bash
 npx oh-pi
 ```
@@ -74,7 +76,7 @@ Already have a config? oh-pi detects it and offers **backup before overwriting**
 
 ### Providers
 
-Anthropic · OpenAI · Google Gemini · Groq · OpenRouter · xAI · Mistral
+Anthropic · OpenAI · Google Gemini · Groq · OpenRouter · xAI · Mistral · [FOXNIO](https://www.foxnio.com) (recommended public-benefit Claude provider)
 
 Auto-detects API keys from environment variables.
 

package/README.zh.md

@@ -0,0 +1,196 @@
+<div align="center">
+
+<img src="./logo.svg" width="180" alt="oh-pi logo"/>
+
+# 🐜 oh-pi
+
+**一条命令，增强 [pi-coding-agent](https://github.com/badlogic/pi-mono)。**
+
+就像 oh-my-zsh 之于 pi —— 但带有自主蚁群系统。
+
+[![npm](https://img.shields.io/npm/v/oh-pi)](https://www.npmjs.com/package/oh-pi)
+[![license](https://img.shields.io/npm/l/oh-pi)](./LICENSE)
+[![node](https://img.shields.io/node/v/oh-pi)](https://nodejs.org)
+
+[English](./README.md) | [中文](./README.zh.md) | [Français](./README.fr.md)
+
+```bash
+npx oh-pi
+```
+
+</div>
+
+---
+
+## 为什么选择 oh-pi
+
+pi-coding-agent 开箱即用已经很强大，但手动配置供应商、主题、扩展、技能和提示词模板非常繁琐。oh-pi 提供现代化 TUI，一分钟内搞定一切 —— 还附带**蚁群集群**，将 pi 变成多智能体系统。
+
+## 快速开始
+
+```bash
+npx oh-pi    # 配置一切
+pi           # 开始编码
+```
+
+就这么简单。oh-pi 检测你的环境，引导你完成设置，并为你生成 `~/.pi/agent/` 配置。
+
+已有配置？oh-pi 会检测到并提供**覆盖前备份**。
+
+## 你将获得
+
+```
+~/.pi/agent/
+├── auth.json            API 密钥（0600 权限）
+├── settings.json        模型、主题、思考级别
+├── keybindings.json     Vim/Emacs 快捷键（可选）
+├── AGENTS.md            角色专属 AI 指南
+├── extensions/          4 个扩展
+│   ├── safe-guard       危险命令确认 + 路径保护
+│   ├── git-guard        自动 stash 检查点 + 脏仓库警告
+│   ├── auto-session     根据首条消息自动命名会话
+│   └── ant-colony/      🐜 自主多智能体蚁群
+├── prompts/             10 个模板（/review /fix /commit /test ...）
+├── skills/              4 个技能（debug、git、setup、colony）
+└── themes/              6 个自定义主题
+```
+
+## 设置模式
+
+| 模式 | 步骤 | 适用场景 |
+|------|------|----------|
+| 🚀 **快速** | 3 | 选供应商 → 输入密钥 → 完成 |
+| 📦 **预设** | 2 | 选择角色配置 → 输入密钥 |
+| 🎛️ **自定义** | 6 | 自己选择所有选项 |
+
+### 预设方案
+
+| | 主题 | 思考级别 | 包含 |
+|---|------|----------|------|
+| 🟢 入门 | oh-pi Dark | 中等 | 安全 + git 基础 |
+| 🔵 专业开发者 | Catppuccin | 高 | 完整工具链 |
+| 🟣 安全研究员 | Cyberpunk | 高 | 审计 + 渗透测试 |
+| 🟠 数据 & AI | Tokyo Night | 中等 | MLOps + 流水线 |
+| 🔴 极简 | Default | 关闭 | 仅核心功能 |
+| ⚫ 全功率 | oh-pi Dark | 高 | 所有功能 + 蚁群 |
+
+### 供应商
+
+Anthropic · OpenAI · Google Gemini · Groq · OpenRouter · xAI · Mistral · [FOXNIO](https://www.foxnio.com)（推荐公益 Claude 供应商）
+
+自动从环境变量检测 API 密钥。
+
+## 🐜 蚁群系统
+
+核心特性。模拟真实蚂蚁生态的多智能体集群。
+
+```
+你："把认证从 session 重构为 JWT"
+
+oh-pi:
+  🔍 侦察蚁探索代码库（haiku — 快速、低成本）
+  📋 根据发现生成任务池
+  ⚒️  工蚁并行执行任务（sonnet — 能力强）
+  🛡️ 兵蚁审查所有变更（sonnet — 细致）
+  ✅ 完成 — 汇总报告及指标
+```
+
+### 为什么是蚂蚁？
+
+真实蚁群无需中央控制即可解决复杂问题。每只蚂蚁遵循简单规则，通过**信息素轨迹**通信，蚁群自组织运作。oh-pi 直接映射了这一模型：
+
+| 真实蚂蚁 | oh-pi |
+|----------|-------|
+| 侦察蚁发现食物 | 侦察蚁扫描代码库，识别目标 |
+| 信息素轨迹 | `.ant-colony/pheromone.jsonl` — 共享发现 |
+| 工蚁搬运食物 | 工蚁在分配的文件上执行任务 |
+| 兵蚁守卫巢穴 | 兵蚁审查变更，请求修复 |
+| 食物越多 → 蚂蚁越多 | 任务越多 → 并发越高（自动适配） |
+| 信息素蒸发 | 10 分钟半衰期 — 过时信息自动淡化 |
+
+### 自动触发
+
+LLM 自行决定何时部署蚁群，你无需操心：
+
+- **≥3 个文件**需要修改 → 蚁群
+- 可**并行工作流** → 蚁群
+- **单文件**修改 → 直接执行（无蚁群开销）
+
+或手动触发：
+
+```
+/colony 将整个项目从 CJS 迁移到 ESM
+```
+
+### 自适应并发
+
+蚁群自动为你的机器找到最优并行度：
+
+```
+冷启动       →  1-2 只蚂蚁（保守）
+探索阶段     →  每波 +1，监控吞吐量
+吞吐量下降   →  锁定最优值，稳定运行
+CPU > 85%   →  立即减少
+429 限流     →  并发减半 + 指数退避（15s→30s→60s）
+任务完成     →  缩减至最小值
+```
+
+### 文件安全
+
+一个文件一只蚂蚁，始终如此。冲突任务自动阻塞，锁释放后恢复执行。
+
+## 主题
+
+| | |
+|---|---|
+| 🌙 **oh-pi Dark** | 青色 + 紫色，高对比度 |
+| 🌙 **Cyberpunk** | 霓虹洋红 + 电光青 |
+| 🌙 **Nord** | 北极蓝色调 |
+| 🌙 **Catppuccin Mocha** | 暗底柔和色 |
+| 🌙 **Tokyo Night** | 蓝紫暮光 |
+| 🌙 **Gruvbox Dark** | 暖色复古风 |
+
+## 提示词模板
+
+```
+/review    代码审查：bug、安全、性能
+/fix       最小改动修复错误
+/explain   代码解释，由浅入深
+/refactor  保持行为的重构
+/test      生成测试
+/commit    Conventional Commit 提交信息
+/pr        Pull Request 描述
+/security  OWASP 安全审计
+/optimize  性能优化
+/document  生成文档
+```
+
+## AGENTS.md 模板
+
+| 模板 | 侧重 |
+|------|------|
+| 通用开发者 | 通用编码指南 |
+| 全栈开发者 | 前端 + 后端 + 数据库 |
+| 安全研究员 | 渗透测试 & 审计 |
+| 数据 & AI 工程师 | MLOps & 流水线 |
+| 🐜 蚁群操作员 | 多智能体编排 |
+
+## 也是 Pi 包
+
+跳过配置器，直接安装资源：
+
+```bash
+pi install npm:oh-pi
+```
+
+将所有主题、提示词、技能和扩展添加到你现有的 pi 配置中。
+
+## 环境要求
+
+- Node.js ≥ 20
+- 至少一个 LLM API 密钥
+- pi-coding-agent（缺失时自动安装）
+
+## 许可证
+
+MIT

package/dist/index.js

@@ -63,7 +63,7 @@ async function customFlow(env) {
         p.cancel(t("cancelled"));
         process.exit(0);
     }
-    let compactThreshold = 0.75;
+    let compactThreshold = 0.80;
     if (wantAdvanced) {
         const threshold = await p.text({
             message: t("advanced.compactThreshold"),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oh-pi",
-  "version": "0.1.21",
+  "version": "0.1.23",
   "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

@@ -37,7 +37,7 @@ function formatTokens(n: number): string {
 function statusIcon(status: string): string {
   const icons: Record<string, string> = {
     scouting: "🔍", working: "⚒️", reviewing: "🛡️",
-    done: "✅", failed: "❌",
+    done: "✅", failed: "❌", budget_exceeded: "💰",
   };
   return icons[status] || "🐜";
 }
@@ -79,11 +79,20 @@ For simple single-file tasks, work directly without the colony.`,
     parameters: Type.Object({
       goal: Type.String({ description: "What the colony should accomplish" }),
       maxAnts: Type.Optional(Type.Number({ description: "Max concurrent ants (default: auto-adapt)", minimum: 1, maximum: 8 })),
+      maxCost: Type.Optional(Type.Number({ description: "Max cost budget in USD (default: unlimited)", minimum: 0.01 })),
     }),
 
     async execute(_toolCallId, params, signal, onUpdate, ctx) {
       const details: ColonyDetails = { state: null, phase: "initializing", log: [] };
 
+      // Resolve models: use current session model for worker/soldier, find a cheap model for scout
+      const currentModel = ctx.model?.id;
+      const modelOverrides: Record<string, string> = {};
+      if (currentModel) {
+        modelOverrides.worker = currentModel;
+        modelOverrides.soldier = currentModel;
+      }
+
       const emit = () => {
         const summary = details.state
           ? `${statusIcon(details.state.status)} Colony: ${details.phase}`
@@ -126,6 +135,8 @@ For simple single-file tasks, work directly without the colony.`,
           cwd: ctx.cwd,
           goal: params.goal,
           maxAnts: params.maxAnts,
+          maxCost: params.maxCost,
+          modelOverrides,
           signal: signal ?? undefined,
           callbacks,
         });
@@ -140,6 +151,7 @@ For simple single-file tasks, work directly without the colony.`,
           `**Goal:** ${state.goal}`,
           `**Status:** ${statusIcon(state.status)} ${state.status}`,
           `**Duration:** ${elapsed}`,
+          ...(state.maxCost != null ? [`**Budget:** ${formatCost(m.totalCost)} / ${formatCost(state.maxCost)}`] : []),
           ``,
           `### Metrics`,
           `- Tasks: ${m.tasksDone}/${m.tasksTotal} done, ${m.tasksFailed} failed`,
@@ -165,7 +177,7 @@ For simple single-file tasks, work directly without the colony.`,
         return {
           content: [{ type: "text", text: report }],
           details: { ...details },
-          isError: state.status === "failed",
+          isError: state.status === "failed" || state.status === "budget_exceeded",
         };
       } catch (e) {
         return {
@@ -184,6 +196,7 @@ For simple single-file tasks, work directly without the colony.`,
       const goal = args.goal?.length > 60 ? args.goal.slice(0, 57) + "..." : args.goal;
       text += "\n  " + theme.fg("dim", goal || "...");
       if (args.maxAnts) text += theme.fg("muted", ` (max: ${args.maxAnts})`);
+      if (args.maxCost) text += theme.fg("warning", ` (budget: $${args.maxCost})`);
       return new Text(text, 0, 0);
     },
 

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

@@ -14,7 +14,7 @@
 
 import type {
   ColonyState, Task, Ant, AntCaste, ColonyMetrics,
-  ConcurrencyConfig, TaskPriority,
+  ConcurrencyConfig, TaskPriority, ModelOverrides,
 } from "./types.js";
 import { DEFAULT_ANT_CONFIGS } from "./types.js";
 import { Nest } from "./nest.js";
@@ -33,6 +33,8 @@ export interface QueenOptions {
   cwd: string;
   goal: string;
   maxAnts?: number;
+  maxCost?: number;
+  modelOverrides?: ModelOverrides;
   signal?: AbortSignal;
   callbacks: QueenCallbacks;
 }
@@ -138,8 +140,11 @@ async function runAntWave(
   caste: AntCaste,
   signal: AbortSignal | undefined,
   callbacks: QueenCallbacks,
-): Promise<void> {
-  const config = DEFAULT_ANT_CONFIGS[caste];
+  modelOverrides?: ModelOverrides,
+  maxCost?: number,
+): Promise<"ok" | "budget_exceeded"> {
+  const config = { ...DEFAULT_ANT_CONFIGS[caste] };
+  if (modelOverrides?.[caste]) config.model = modelOverrides[caste];
 
   let backoffMs = 0; // 429 退避时间
 
@@ -191,6 +196,15 @@ async function runAntWave(
 
   // 调度循环：持续派蚂蚁直到没有待处理任务
   while (!signal?.aborted) {
+    // Budget check
+    if (maxCost != null) {
+      const currentCost = nest.getState().ants.reduce((s, a) => s + a.usage.cost, 0);
+      if (currentCost >= maxCost) {
+        callbacks.onPhase("working", `Budget exceeded ($${currentCost.toFixed(3)} >= $${maxCost.toFixed(2)}). Stopping.`);
+        return "budget_exceeded";
+      }
+    }
+
     const state = nest.getState();
     const pending = state.tasks.filter(t => t.status === "pending" && t.caste === caste);
     if (pending.length === 0) break;
@@ -254,6 +268,7 @@ async function runAntWave(
       backoffMs = 0;
     }
   }
+  return "ok";
 }
 
 /**
@@ -276,6 +291,8 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
       antsSpawned: 0, totalCost: 0, totalTokens: 0,
       startTime: Date.now(), throughputHistory: [],
     },
+    maxCost: opts.maxCost ?? null,
+    modelOverrides: opts.modelOverrides ?? {},
     createdAt: Date.now(),
     finishedAt: null,
   };
@@ -286,11 +303,22 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
 
   nest.init(initialState);
   const { signal, callbacks } = opts;
+  const mo = opts.modelOverrides;
+  const mc = opts.maxCost;
+
+  const budgetStop = (phase: string) => {
+    nest.updateState({ status: "budget_exceeded", finishedAt: Date.now() });
+    callbacks.onPhase("budget_exceeded" as any, phase);
+    const s = nest.getState();
+    callbacks.onComplete(s);
+    return s;
+  };
 
   try {
     // ═══ Phase 1: 侦察 ═══
     callbacks.onPhase("scouting", "Dispatching scout ants to explore codebase...");
-    await runAntWave(nest, opts.cwd, "scout", signal, callbacks);
+    if (await runAntWave(nest, opts.cwd, "scout", signal, callbacks, mo, mc) === "budget_exceeded")
+      return budgetStop("Budget exceeded during scouting");
 
     // 检查侦察结果，如果没有产生工蚁任务，用侦察结果让女王自己拆
     const postScout = nest.getAllTasks();
@@ -306,7 +334,8 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
     // ═══ Phase 2: 工作 ═══
     nest.updateState({ status: "working" });
     callbacks.onPhase("working", `${workerTasks.length} tasks discovered. Dispatching worker ants...`);
-    await runAntWave(nest, opts.cwd, "worker", signal, callbacks);
+    if (await runAntWave(nest, opts.cwd, "worker", signal, callbacks, mo, mc) === "budget_exceeded")
+      return budgetStop("Budget exceeded during working");
 
     // 处理工蚁产生的子任务（可能有多轮）
     let rounds = 0;
@@ -317,7 +346,8 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
       if (remaining.length === 0) break;
       rounds++;
       callbacks.onPhase("working", `Round ${rounds + 1}: ${remaining.length} sub-tasks from workers...`);
-      await runAntWave(nest, opts.cwd, "worker", signal, callbacks);
+      if (await runAntWave(nest, opts.cwd, "worker", signal, callbacks, mo, mc) === "budget_exceeded")
+        return budgetStop("Budget exceeded during sub-tasks");
     }
 
     // ═══ Phase 3: 审查 ═══
@@ -327,7 +357,8 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
       callbacks.onPhase("reviewing", "Dispatching soldier ants to review changes...");
       const reviewTask = makeReviewTask(completedWorkerTasks);
       nest.writeTask(reviewTask);
-      await runAntWave(nest, opts.cwd, "soldier", signal, callbacks);
+      if (await runAntWave(nest, opts.cwd, "soldier", signal, callbacks, mo, mc) === "budget_exceeded")
+        return budgetStop("Budget exceeded during review");
 
       // 兵蚁产生的修复任务
       const fixTasks = nest.getAllTasks().filter(t =>
@@ -336,7 +367,8 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
       if (fixTasks.length > 0) {
         nest.updateState({ status: "working" });
         callbacks.onPhase("working", `${fixTasks.length} fix tasks from review. Dispatching workers...`);
-        await runAntWave(nest, opts.cwd, "worker", signal, callbacks);
+        if (await runAntWave(nest, opts.cwd, "worker", signal, callbacks, mo, mc) === "budget_exceeded")
+          return budgetStop("Budget exceeded during fixes");
       }
     }
 

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

@@ -19,6 +19,9 @@ export const DEFAULT_ANT_CONFIGS: Record<AntCaste, Omit<AntConfig, "systemPrompt
   soldier: { caste: "soldier", model: "claude-sonnet-4-5",  tools: ["read", "bash", "grep", "find", "ls"], maxTurns: 8 },
 };
 
+/** Per-caste model overrides from user config */
+export type ModelOverrides = Partial<Record<AntCaste, string>>;
+
 // ═══ 任务 (Food Source) ═══
 export type TaskStatus = "pending" | "claimed" | "active" | "done" | "failed" | "blocked";
 export type TaskPriority = 1 | 2 | 3 | 4 | 5; // 1=highest
@@ -79,12 +82,14 @@ export interface Ant {
 export interface ColonyState {
   id: string;
   goal: string;
-  status: "scouting" | "working" | "reviewing" | "done" | "failed";
+  status: "scouting" | "working" | "reviewing" | "done" | "failed" | "budget_exceeded";
   tasks: Task[];
   ants: Ant[];
   pheromones: Pheromone[];
   concurrency: ConcurrencyConfig;
   metrics: ColonyMetrics;
+  maxCost: number | null;       // cost budget in USD, null = unlimited
+  modelOverrides: ModelOverrides;
   createdAt: number;
   finishedAt: number | null;
 }

package/pi-package/extensions/custom-footer.ts

@@ -75,13 +75,15 @@ export default function (pi: ExtensionAPI) {
 
 					const thinking = pi.getThinkingLevel();
 					const modelId = ctx.model?.id || "no-model";
-					const right = theme.fg("dim", `${modelId} ${thinking}`);
+					const modelStr = theme.fg("accent", `◆ ${modelId}`);
 
 					const sep = theme.fg("dim", " | ");
-					const leftParts = [tokenStats, elapsed, cwdStr];
+					const leftParts = [modelStr, tokenStats, elapsed, cwdStr];
 					if (branchStr) leftParts.push(branchStr);
 					const left = leftParts.join(sep);
 
+					const right = theme.fg("dim", thinking);
+
 					const leftW = visibleWidth(left);
 					const rightW = visibleWidth(right);
 					const gap = width - leftW - rightW;