oh-pi 0.1.70 → 0.1.72

package/pi-package/extensions/ant-colony/parser.test.ts

@@ -0,0 +1,110 @@
+import { describe, it, expect, vi } from "vitest";
+
+vi.mock("@mariozechner/pi-coding-agent", () => ({
+  AuthStorage: class {},
+  createAgentSession: vi.fn(),
+  createReadTool: vi.fn(), createBashTool: vi.fn(), createEditTool: vi.fn(),
+  createWriteTool: vi.fn(), createGrepTool: vi.fn(), createFindTool: vi.fn(),
+  createLsTool: vi.fn(), ModelRegistry: class {}, SessionManager: { inMemory: vi.fn() },
+  SettingsManager: { inMemory: vi.fn() }, createExtensionRuntime: vi.fn(),
+}));
+vi.mock("@mariozechner/pi-ai", () => ({ getModel: vi.fn() }));
+vi.mock("./spawner.js", async () => {
+  const actual = await vi.importActual<any>("./spawner.js");
+  return { ...actual, makePheromoneId: () => "p-test" };
+});
+
+import { parseSubTasks, extractPheromones } from "./parser.js";
+
+describe("parseSubTasks", () => {
+  it("parses markdown TASK blocks", () => {
+    const output = `## Recommended Tasks
+### TASK: Fix login
+- description: Fix the login bug
+- files: src/auth.ts
+- caste: worker
+- priority: 2`;
+    const tasks = parseSubTasks(output);
+    expect(tasks).toHaveLength(1);
+    expect(tasks[0].title).toBe("Fix login");
+    expect(tasks[0].description).toBe("Fix the login bug");
+    expect(tasks[0].files).toEqual(["src/auth.ts"]);
+    expect(tasks[0].caste).toBe("worker");
+    expect(tasks[0].priority).toBe(2);
+  });
+
+  it("parses JSON block", () => {
+    const output = '```json\n[{"title":"Task A","description":"Do A","files":["a.ts"],"caste":"scout","priority":1}]\n```';
+    const tasks = parseSubTasks(output);
+    expect(tasks).toHaveLength(1);
+    expect(tasks[0].title).toBe("Task A");
+    expect(tasks[0].caste).toBe("scout");
+  });
+
+  it("defaults caste to worker for invalid", () => {
+    const tasks = parseSubTasks('```json\n[{"title":"X","caste":"invalid"}]\n```');
+    expect(tasks[0].caste).toBe("worker");
+  });
+
+  it("defaults priority to 3", () => {
+    const tasks = parseSubTasks('```json\n[{"title":"X"}]\n```');
+    expect(tasks[0].priority).toBe(3);
+  });
+
+  it("returns empty for no tasks", () => {
+    expect(parseSubTasks("no tasks here")).toEqual([]);
+  });
+
+  it("parses multiple markdown tasks", () => {
+    const output = `### TASK: A
+- description: Do A
+- files: a.ts
+- caste: worker
+- priority: 1
+
+### TASK: B
+- description: Do B
+- files: b.ts
+- caste: soldier
+- priority: 2`;
+    const tasks = parseSubTasks(output);
+    expect(tasks).toHaveLength(2);
+  });
+
+  it("parses context field", () => {
+    const output = `### TASK: Fix it
+- description: Fix bug
+- files: x.ts
+- caste: worker
+- priority: 3
+- context: some relevant code`;
+    const tasks = parseSubTasks(output);
+    expect(tasks[0].context).toBeTruthy();
+  });
+});
+
+describe("extractPheromones", () => {
+  it("extracts discovery section", () => {
+    const p = extractPheromones("ant-1", "scout", "t-1", "## Discoveries\n- Found auth\n\n## Other\nstuff", ["a.ts"]);
+    expect(p.some(x => x.type === "discovery")).toBe(true);
+  });
+
+  it("extracts warning section", () => {
+    const p = extractPheromones("ant-1", "scout", "t-1", "## Warnings\n- Conflict\n", []);
+    expect(p.some(x => x.type === "warning")).toBe(true);
+  });
+
+  it("adds repellent on failure", () => {
+    const p = extractPheromones("ant-1", "worker", "t-1", "output", ["a.ts"], true);
+    expect(p.some(x => x.type === "repellent")).toBe(true);
+  });
+
+  it("returns empty for no matching sections", () => {
+    expect(extractPheromones("ant-1", "worker", "t-1", "nothing", [])).toEqual([]);
+  });
+
+  it("extracts Files Changed as completion", () => {
+    const p = extractPheromones("ant-1", "worker", "t-1", "## Files Changed\n- src/foo.ts\n", ["src/foo.ts"]);
+    expect(p.some(x => x.type === "completion")).toBe(true);
+  });
+});

package/pi-package/extensions/ant-colony/prompts.test.ts

@@ -0,0 +1,57 @@
+import { describe, it, expect } from "vitest";
+import { CASTE_PROMPTS, buildPrompt } from "./prompts.js";
+import type { Task } from "./types.js";
+
+const mkTask = (overrides: Partial<Task> = {}): Task => ({
+  id: "t-1", parentId: null, title: "Test task", description: "Do something",
+  caste: "worker", status: "pending", priority: 3, files: [], claimedBy: null,
+  result: null, error: null, spawnedTasks: [], createdAt: 0, startedAt: null, finishedAt: null,
+  ...overrides,
+});
+
+describe("CASTE_PROMPTS", () => {
+  it("has all castes", () => {
+    for (const c of ["scout", "worker", "soldier"] as const) {
+      expect(typeof CASTE_PROMPTS[c]).toBe("string");
+      expect(CASTE_PROMPTS[c].length).toBeGreaterThan(0);
+    }
+  });
+});
+
+describe("buildPrompt", () => {
+  it("includes task title and description", () => {
+    const r = buildPrompt(mkTask(), "", "System");
+    expect(r).toContain("Test task");
+    expect(r).toContain("Do something");
+  });
+
+  it("includes pheromone context", () => {
+    const r = buildPrompt(mkTask(), "Found auth at src/auth.ts", "System");
+    expect(r).toContain("Found auth at src/auth.ts");
+  });
+
+  it("includes files scope", () => {
+    const r = buildPrompt(mkTask({ files: ["a.ts", "b.ts"] }), "", "System");
+    expect(r).toContain("a.ts, b.ts");
+  });
+
+  it("includes turn limit when provided", () => {
+    const r = buildPrompt(mkTask(), "", "System", 10);
+    expect(r).toContain("10");
+    expect(r).toContain("Turn Limit");
+  });
+
+  it("omits turn limit when not provided", () => {
+    expect(buildPrompt(mkTask(), "", "System")).not.toContain("Turn Limit");
+  });
+
+  it("includes pre-loaded context", () => {
+    const r = buildPrompt(mkTask({ context: "code snippet" }), "", "System");
+    expect(r).toContain("code snippet");
+  });
+
+  it("adds Chinese hint for Chinese descriptions", () => {
+    const r = buildPrompt(mkTask({ description: "修复登录问题" }), "", "System");
+    expect(r).toContain("Chinese");
+  });
+});

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

@@ -76,7 +76,7 @@ Output format (MUST follow exactly):
 PASS or FAIL with summary.`,
 };
 
-export function buildPrompt(task: Task, pheromoneContext: string, castePrompt: string, maxTurns?: number): string {
+export function buildPrompt(task: Task, pheromoneContext: string, castePrompt: string, maxTurns?: number, tandem?: { parentResult?: string; priorError?: string }): string {
   let prompt = castePrompt + "\n\n";
   if (maxTurns) {
     prompt += `## ⚠️ Turn Limit\nYou have a MAXIMUM of ${maxTurns} turns. Plan accordingly — reserve your LAST turn to output the structured result format above. Do NOT waste turns on unnecessary exploration.\n\n`;
@@ -84,6 +84,12 @@ export function buildPrompt(task: Task, pheromoneContext: string, castePrompt: s
   if (pheromoneContext) {
     prompt += `## Colony Pheromone Trail (intelligence from other ants)\n${pheromoneContext}\n\n`;
   }
+  if (tandem?.parentResult) {
+    prompt += `## Tandem Context (from parent task)\n${tandem.parentResult.slice(0, 3000)}\n\n`;
+  }
+  if (tandem?.priorError) {
+    prompt += `## ⚠️ Prior Attempt Failed\nA previous ant failed on this task. Learn from their mistake:\n${tandem.priorError.slice(0, 1500)}\n\n`;
+  }
   prompt += `## Your Assignment\n**Task:** ${task.title}\n**Description:** ${task.description}\n`;
   if (task.files.length > 0) {
     prompt += `**Files scope:** ${task.files.join(", ")}\n`;

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

@@ -20,7 +20,7 @@ import type {
 } from "./types.js";
 import { DEFAULT_ANT_CONFIGS } from "./types.js";
 import { Nest } from "./nest.js";
-import { spawnAnt, runDrone, makeTaskId, makePheromoneId } from "./spawner.js";
+import { spawnAnt, runDrone, makeTaskId, makePheromoneId, resetAntCounter } from "./spawner.js";
 import { adapt, sampleSystem, defaultConcurrency } from "./concurrency.js";
 import { buildImportGraph, taskDependsOn, type ImportGraph } from "./deps.js";
 import type { AuthStorage, ModelRegistry } from "@mariozechner/pi-coding-agent";
@@ -98,6 +98,44 @@ function childTaskFromParsed(
   };
 }
 
+/**
+ * Bio 5: 蚁群投票 — 合并多 Scout 产生的重复任务
+ * 相同文件集合的任务合并，被多 Scout 提及的任务 priority 提升
+ */
+function quorumMergeTasks(nest: Nest): void {
+  const tasks = nest.getAllTasks().filter(t =>
+    (t.caste === "worker" || t.caste === "drone") && t.status === "pending"
+  );
+  if (tasks.length < 2) return;
+
+  // 按文件集合分组（排序后 join 作为 key）
+  const groups = new Map<string, Task[]>();
+  for (const t of tasks) {
+    const key = [...t.files].sort().join("|") || t.title;
+    const arr = groups.get(key) ?? [];
+    arr.push(t);
+    groups.set(key, arr);
+  }
+
+  for (const [, group] of groups) {
+    if (group.length < 2) continue;
+    // 保留第一个，删除重复的，合并 description
+    const keeper = group[0];
+    // Quorum 达成：被多 Scout 提及 → priority 提升
+    keeper.priority = Math.max(1, keeper.priority - 1) as 1 | 2 | 3 | 4 | 5;
+    // 合并其他任务的 context 到 keeper
+    for (let i = 1; i < group.length; i++) {
+      const dup = group[i];
+      if (dup.context && dup.context !== keeper.context) {
+        keeper.context = (keeper.context || "") + "\n\n--- Additional scout context ---\n" + dup.context;
+      }
+      // 标记重复任务为 done（已合并）
+      nest.updateTaskStatus(dup.id, "done", `Merged into ${keeper.id} (quorum)`);
+    }
+    nest.writeTask(keeper);
+  }
+}
+
 function makeReviewTask(completedTasks: Task[]): Task {
   const files = [...new Set(completedTasks.flatMap(t => t.files))];
   return {
@@ -120,8 +158,8 @@ function makeReviewTask(completedTasks: Task[]): Task {
 }
 
 function updateMetrics(nest: Nest): ColonyMetrics {
-  const tasks = nest.getAllTasks();
-  const state = nest.getState();
+  const state = nest.getStateLight();
+  const tasks = state.tasks;
   const now = Date.now();
   const elapsed = (now - state.metrics.startTime) / 60000; // minutes
 
@@ -157,25 +195,49 @@ interface WaveOptions {
   importGraph?: ImportGraph;
 }
 
+/**
+ * Bio 6: 尸体清理 — 错误模式分类
+ */
+function classifyError(errStr: string): string {
+  if (errStr.includes("TypeError") || errStr.includes("type") || errStr.includes("TS")) return "type_error";
+  if (errStr.includes("permission") || errStr.includes("401") || errStr.includes("EACCES")) return "permission";
+  if (errStr.includes("timeout") || errStr.includes("Timeout") || errStr.includes("ETIMEDOUT")) return "timeout";
+  if (errStr.includes("ENOENT") || errStr.includes("not found") || errStr.includes("Cannot find")) return "not_found";
+  if (errStr.includes("syntax") || errStr.includes("SyntaxError") || errStr.includes("Unexpected")) return "syntax";
+  if (errStr.includes("429") || errStr.includes("rate limit")) return "rate_limit";
+  return "unknown";
+}
+
 /**
  * 并发执行一批蚂蚁，自适应调节并发度
  */
 async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
   const { nest, cwd, caste, signal, callbacks, currentModel, emitSignal } = opts;
   const casteModel = opts.modelOverrides?.[caste] || currentModel;
-  const config = { ...DEFAULT_ANT_CONFIGS[caste], model: casteModel };
+  const baseConfig = { ...DEFAULT_ANT_CONFIGS[caste], model: casteModel };
 
   let backoffMs = 0; // 429 退避时间
   let consecutiveRateLimits = 0; // 连续限流计数
   const retryCount = new Map<string, number>(); // taskId → retry count
   const MAX_RETRIES = 2;
 
+  // Bio 6: 尸体清理 — 错误模式追踪
+  const errorPatterns = new Map<string, { count: number; files: Set<string>; errors: string[] }>();
+
   const runOne = async (): Promise<"done" | "empty" | "rate_limited" | "budget"> => {
     // Budget 刹车：预算用完就不出发（drone 免费，不检查）
-    const state = nest.getState();
+    const state = nest.getStateLight();
     if (state.maxCost != null && caste !== "drone") {
       const spent = state.ants.reduce((s, a) => s + a.usage.cost, 0);
       if (spent >= state.maxCost) return "budget";
+
+      // Bio 4: 巢穴温度 — 成本渐进调控
+      const temperature = spent / state.maxCost;
+      if (temperature > 0.9) {
+        // 紧急模式：只跑 priority 1 任务
+        const pending = state.tasks.filter(t => t.status === "pending" && t.caste === caste);
+        if (!pending.some(t => t.priority === 1)) return "budget";
+      }
     }
 
     const task = nest.claimNextTask(caste, "queen");
@@ -194,6 +256,14 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
       const antAbort = new AbortController();
       signal?.addEventListener("abort", () => antAbort.abort(), { once: true });
       const antSignal = antAbort.signal;
+      // Bio 7: 年龄多态 — 前期保守，后期收敛
+      const progress = state.metrics.tasksTotal > 0 ? state.metrics.tasksDone / state.metrics.tasksTotal : 0;
+      const config = { ...baseConfig };
+      if (progress < 0.3) {
+        config.maxTurns = Math.max(baseConfig.maxTurns - 3, 5); // 前期保守
+      } else if (progress > 0.7) {
+        config.maxTurns = Math.max(baseConfig.maxTurns - 5, 5); // 后期收敛，只修复收尾
+      }
       const antPromise = caste === "drone"
         ? runDrone(cwd, nest, task)
         : spawnAnt(cwd, nest, task, config, antSignal, callbacks.onAntStream, opts.authStorage, opts.modelRegistry);
@@ -211,7 +281,7 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
       }
 
       // 成本预警：超 80% 预算时发信号
-      const curState = nest.getState();
+      const curState = nest.getStateLight();
       if (curState.maxCost != null) {
         const spent = curState.ants.reduce((s, a) => s + a.usage.cost, 0);
         if (spent >= curState.maxCost * 0.8) {
@@ -220,8 +290,11 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
       }
 
       // 蚂蚁产生的子任务加入巢穴（限制繁殖上限，防止任务膨胀）
+      // Bio 7: 年龄多态 — 后期限制子任务生成
+      const m = curState.metrics;
+      const colonyProgress = m.tasksTotal > 0 ? m.tasksDone / m.tasksTotal : 0;
       const MAX_TOTAL_TASKS = 30;
-      const MAX_SUB_PER_TASK = 5;
+      const MAX_SUB_PER_TASK = colonyProgress > 0.7 ? 2 : 5; // 后期收敛
       const accepted = result.newTasks.slice(0, MAX_SUB_PER_TASK);
       for (const sub of accepted) {
         if (nest.getAllTasks().length >= MAX_TOTAL_TASKS) break;
@@ -240,13 +313,14 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
         nest.addSubTask(task.id, child);
       }
 
-      // 路径强化：成功完成释放 completion 信息素，强化相关文件路径
+      // 路径强化：成功完成释放 completion 信息素，强度与任务规模成正比（招募信号）
       if (task.files.length > 0) {
+        const recruitStrength = Math.min(1.0, 0.5 + task.files.length * 0.1 + result.newTasks.length * 0.15);
         nest.dropPheromone({
           id: makePheromoneId(), type: "completion", antId: result.ant.id,
           antCaste: caste, taskId: task.id,
           content: `Success: ${task.title}`,
-          files: task.files, strength: 1.0, createdAt: Date.now(),
+          files: task.files, strength: recruitStrength, createdAt: Date.now(),
         });
       }
 
@@ -264,16 +338,49 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
         retryCount.set(task.id, count + 1);
         nest.updateTaskStatus(task.id, "pending");
       } else {
-        // 负信息素：失败任务释放 warning，阻止后续蚂蚁走同一条路
+        // 负信息素：失败任务释放 warning，强度与任务规模成正比
         if (task.files.length > 0) {
+          const warnStrength = Math.min(1.0, 0.5 + task.files.length * 0.1);
           nest.dropPheromone({
             id: makePheromoneId(), type: "warning", antId: "queen",
             antCaste: caste, taskId: task.id,
             content: `Failed: ${task.title} — ${String(e).slice(0, 100)}`,
-            files: task.files, strength: 1.0, createdAt: Date.now(),
+            files: task.files, strength: warnStrength, createdAt: Date.now(),
           });
         }
         nest.updateTaskStatus(task.id, "failed", undefined, String(e));
+
+        // Bio 6: 尸体清理 — 错误模式追踪 + 诊断任务
+        const pattern = classifyError(errStr);
+        const entry = errorPatterns.get(pattern) ?? { count: 0, files: new Set<string>(), errors: [] };
+        entry.count++;
+        for (const f of task.files) entry.files.add(f);
+        entry.errors.push(errStr.slice(0, 200));
+        errorPatterns.set(pattern, entry);
+
+        if (entry.count >= 2 && entry.files.size > 0) {
+          const affectedFiles = [...entry.files];
+          // 释放 repellent 信息素
+          nest.dropPheromone({
+            id: makePheromoneId(), type: "repellent", antId: "queen",
+            antCaste: caste, taskId: task.id,
+            content: `Recurring ${pattern} errors (${entry.count}x): ${entry.errors[0]?.slice(0, 80)}`,
+            files: affectedFiles, strength: 1.0, createdAt: Date.now(),
+          });
+          // 生成诊断任务（仅首次触发）
+          if (entry.count === 2 && nest.getAllTasks().length < 30) {
+            const diagTask: Task = {
+              id: makeTaskId(), parentId: null,
+              title: `Diagnose recurring ${pattern} errors`,
+              description: `Multiple ants failed with ${pattern} errors on these files:\n${affectedFiles.map(f => `- ${f}`).join("\n")}\n\nErrors:\n${entry.errors.map(e => `- ${e}`).join("\n")}\n\nInvestigate root cause and generate fix tasks.`,
+              caste: "scout", status: "pending", priority: 1,
+              files: affectedFiles, claimedBy: null, result: null, error: null,
+              spawnedTasks: [], createdAt: Date.now(), startedAt: null, finishedAt: null,
+            };
+            nest.writeTask(diagTask);
+            emitSignal("working", `Diagnosing recurring ${pattern} errors...`);
+          }
+        }
       }
       return "done";
     }
@@ -282,7 +389,7 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
   // 调度循环：持续派蚂蚁直到没有待处理任务
   let lastSampleTime = 0;
   while (!signal?.aborted) {
-    const state = nest.getState();
+    const state = nest.getStateLight();
     const pending = state.tasks.filter(t => t.status === "pending" && t.caste === caste);
     if (pending.length === 0) break;
 
@@ -347,7 +454,7 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
     // 429 处理：降低并发 + 渐进退避（2s → 5s → 10s，上限 10s）+ 记录时间戳
     if (results.includes("rate_limited")) {
       consecutiveRateLimits++;
-      const cur = nest.getState().concurrency;
+      const cur = nest.getStateLight().concurrency;
       const reduced = Math.max(cur.min, cur.current - 1); // 每次只减 1，不砍半
       nest.updateState({ concurrency: { ...cur, current: reduced, lastRateLimitAt: Date.now() } });
       backoffMs = Math.min(consecutiveRateLimits * 2000, 10000);
@@ -366,6 +473,7 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
   if (!opts.goal || !opts.goal.trim()) {
     throw new Error("Colony goal is empty or undefined. Please provide a clear goal.");
   }
+  resetAntCounter();
   const colonyId = makeColonyId();
   const nest = new Nest(opts.cwd, colonyId);
 
@@ -405,7 +513,7 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
   };
 
   const emitSignal = (phase: ColonyState["status"], message: string) => {
-    const state = nest.getState();
+    const state = nest.getStateLight();
     const m = state.metrics;
     const active = state.ants.filter(a => a.status === "working").length;
     const progress = m.tasksTotal > 0 ? m.tasksDone / m.tasksTotal : 0;
@@ -421,11 +529,38 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
   };
 
   try {
-    // ═══ Phase 1: 侦察（快速单次，不再多轮接力） ═══
-    callbacks.onPhase?.("scouting", "Dispatching scout ant to explore codebase...");
-    emitSignal("scouting", "Exploring codebase...");
+    // ═══ Phase 1: 侦察（Bio 5: 蚁群投票 — 复杂目标派多 Scout） ═══
+    const scoutCount = opts.goal.length > 500 ? 3 : opts.goal.length > 200 ? 2 : 1;
+    if (scoutCount > 1) {
+      // 多 Scout 并行：为每只 Scout 创建独立任务
+      for (let i = 1; i < scoutCount; i++) {
+        const extraScout: Task = {
+          id: makeTaskId(),
+          parentId: null,
+          title: `Scout ${i + 1}: explore codebase for goal`,
+          description: `Explore the codebase from a different angle and identify files, modules, and dependencies relevant to this goal:\n\n${opts.goal}\n\nFocus on areas other scouts might miss. Be thorough.`,
+          caste: "scout",
+          status: "pending",
+          priority: 1,
+          files: [],
+          claimedBy: null,
+          result: null,
+          error: null,
+          spawnedTasks: [],
+          createdAt: Date.now(),
+          startedAt: null,
+          finishedAt: null,
+        };
+        nest.writeTask(extraScout);
+      }
+    }
+    callbacks.onPhase?.("scouting", `Dispatching ${scoutCount} scout ant(s) to explore codebase...`);
+    emitSignal("scouting", `${scoutCount} scouts exploring...`);
     await runAntWave({ ...waveBase, caste: "scout" });
 
+    // Bio 5: 合并多 Scout 产生的重复任务
+    if (scoutCount > 1) quorumMergeTasks(nest);
+
     let workerTasks = nest.getAllTasks().filter(t => (t.caste === "worker" || t.caste === "drone") && t.status === "pending");
 
     // 只在完全没有 worker 任务时才重试一次
@@ -513,11 +648,14 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
     }
 
     // ═══ 持续探索：Worker 完成后检查是否有新发现，有则再派 Scout ═══
+    // Bio 4: 巢穴温度 — 超过 50% 预算禁止新 Scout 探索
     const discoveries = nest.getAllPheromones().filter(p => p.type === "discovery");
     const allDone = nest.getAllTasks().filter(t => t.status === "done");
-    if (discoveries.length > allDone.length) {
-      const spent = nest.getState().ants.reduce((s, a) => s + a.usage.cost, 0);
-      if (spent < (nest.getState().maxCost ?? Infinity)) {
+    const preExploreSpent = nest.getStateLight().ants.reduce((s, a) => s + a.usage.cost, 0);
+    const preExploreBudget = nest.getStateLight().maxCost ?? Infinity;
+    const costTemperature = preExploreSpent / preExploreBudget;
+    if (discoveries.length > allDone.length && costTemperature < 0.5) {
+      if (preExploreSpent < preExploreBudget) {
         callbacks.onPhase?.("scouting", "Re-exploring based on new discoveries...");
         emitSignal("scouting", "Re-exploring...");
         await runAntWave({ ...waveBase, caste: "scout" });
@@ -606,7 +744,7 @@ export async function resumeColony(opts: QueenOptions): Promise<ColonyState> {
   const { signal, callbacks } = opts;
 
   const emitSignal = (phase: ColonyState["status"], message: string) => {
-    const state = nest.getState();
+    const state = nest.getStateLight();
     const m = state.metrics;
     const active = state.ants.filter(a => a.status === "working").length;
     const progress = m.tasksTotal > 0 ? m.tasksDone / m.tasksTotal : 0;

package/pi-package/extensions/ant-colony/spawner.test.ts

@@ -0,0 +1,44 @@
+import { describe, it, expect, vi } from "vitest";
+
+vi.mock("@mariozechner/pi-coding-agent", () => ({
+  AuthStorage: class {},
+  createAgentSession: vi.fn(),
+  createReadTool: vi.fn(), createBashTool: vi.fn(), createEditTool: vi.fn(),
+  createWriteTool: vi.fn(), createGrepTool: vi.fn(), createFindTool: vi.fn(),
+  createLsTool: vi.fn(), ModelRegistry: class {}, SessionManager: { inMemory: vi.fn() },
+  SettingsManager: { inMemory: vi.fn() }, createExtensionRuntime: vi.fn(),
+}));
+vi.mock("@mariozechner/pi-ai", () => ({ getModel: vi.fn() }));
+
+import { makeAntId, makePheromoneId, makeTaskId } from "./spawner.js";
+
+describe("makeAntId", () => {
+  it("includes caste name", () => {
+    expect(makeAntId("scout")).toContain("scout");
+    expect(makeAntId("worker")).toContain("worker");
+  });
+
+  it("returns unique ids", () => {
+    expect(makeAntId("worker")).not.toBe(makeAntId("worker"));
+  });
+});
+
+describe("makePheromoneId", () => {
+  it("starts with p-", () => {
+    expect(makePheromoneId()).toMatch(/^p-/);
+  });
+
+  it("returns unique ids", () => {
+    expect(makePheromoneId()).not.toBe(makePheromoneId());
+  });
+});
+
+describe("makeTaskId", () => {
+  it("starts with t-", () => {
+    expect(makeTaskId()).toMatch(/^t-/);
+  });
+
+  it("returns unique ids", () => {
+    expect(makeTaskId()).not.toBe(makeTaskId());
+  });
+});

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

@@ -32,6 +32,8 @@ import { parseSubTasks, extractPheromones, type ParsedSubTask } from "./parser.j
 
 let antCounter = 0;
 
+export function resetAntCounter(): void { antCounter = 0; }
+
 export function makeAntId(caste: AntCaste): string {
   return `${caste}-${++antCounter}-${Date.now().toString(36)}`;
 }
@@ -179,9 +181,22 @@ export async function spawnAnt(
   nest.updateAnt(ant);
   nest.updateTaskStatus(task.id, "active");
 
+  // Bio 2: 任务难度感知 — 动态 maxTurns
+  const warnings = nest.countWarnings(task.files);
+  const difficultyTurns = Math.min(25, (antConfig.maxTurns || 15) + task.files.length + warnings * 2);
+  const effectiveMaxTurns = antConfig.caste === "drone" ? 1 : difficultyTurns;
+
+  // Bio 3: 串联觅食 — 继承父任务 result 和失败前任 error
+  const tandem: { parentResult?: string; priorError?: string } = {};
+  if (task.parentId) {
+    const parent = nest.getTask(task.parentId);
+    if (parent?.result) tandem.parentResult = parent.result;
+  }
+  if (task.error) tandem.priorError = task.error;
+
   const pheromoneCtx = nest.getPheromoneContext(task.files);
   const castePrompt = CASTE_PROMPTS[antConfig.caste];
-  const systemPrompt = buildPrompt(task, pheromoneCtx, castePrompt, antConfig.maxTurns);
+  const systemPrompt = buildPrompt(task, pheromoneCtx, castePrompt, effectiveMaxTurns, tandem);
 
   const auth = authStorage ?? new AuthStorage();
   const registry = modelRegistry ?? new ModelRegistry(auth);

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

@@ -0,0 +1,36 @@
+import { describe, it, expect } from "vitest";
+import { DEFAULT_ANT_CONFIGS } from "./types.js";
+import type { AntCaste } from "./types.js";
+
+describe("DEFAULT_ANT_CONFIGS", () => {
+  const castes: AntCaste[] = ["scout", "worker", "soldier", "drone"];
+
+  it("has all castes", () => {
+    for (const c of castes) expect(DEFAULT_ANT_CONFIGS).toHaveProperty(c);
+  });
+
+  it("each config has caste/model/tools/maxTurns", () => {
+    for (const c of castes) {
+      const cfg = DEFAULT_ANT_CONFIGS[c];
+      expect(cfg.caste).toBe(c);
+      expect(typeof cfg.model).toBe("string");
+      expect(Array.isArray(cfg.tools)).toBe(true);
+      expect(cfg.maxTurns).toBeGreaterThan(0);
+    }
+  });
+
+  it("scout has no write tools", () => {
+    expect(DEFAULT_ANT_CONFIGS.scout.tools).not.toContain("edit");
+    expect(DEFAULT_ANT_CONFIGS.scout.tools).not.toContain("write");
+  });
+
+  it("worker has edit and write", () => {
+    expect(DEFAULT_ANT_CONFIGS.worker.tools).toContain("edit");
+    expect(DEFAULT_ANT_CONFIGS.worker.tools).toContain("write");
+  });
+
+  it("drone only has bash with 1 turn", () => {
+    expect(DEFAULT_ANT_CONFIGS.drone.tools).toEqual(["bash"]);
+    expect(DEFAULT_ANT_CONFIGS.drone.maxTurns).toBe(1);
+  });
+});