oh-pi 0.1.60 → 0.1.62

package/package.json

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

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

@@ -169,16 +169,11 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget"> {
   const retriedTasks = new Set<string>(); // 防止重复重试
 
   const runOne = async (): Promise<"done" | "empty" | "rate_limited" | "budget"> => {
-    // Budget 刹车：剩余预算不够一只蚂蚁的预估成本就不出发
+    // Budget 刹车：预算用完就不出发（drone 免费，不检查）
     const state = nest.getState();
     if (state.maxCost != null && caste !== "drone") {
       const spent = state.ants.reduce((s, a) => s + a.usage.cost, 0);
-      const remaining = state.maxCost - spent;
-      const doneAnts = state.ants.filter(a => a.status === "done" && a.usage.cost > 0);
-      const avgCost = doneAnts.length > 0
-        ? doneAnts.reduce((s, a) => s + a.usage.cost, 0) / doneAnts.length
-        : 0.05;
-      if (remaining < avgCost * 1.5) return "budget";
+      if (spent >= state.maxCost) return "budget";
     }
 
     const task = nest.nextPendingTask(caste);
@@ -468,6 +463,37 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
       }
     }
 
+    // ═══ 持续探索：Worker 完成后检查是否有新发现，有则再派 Scout ═══
+    const discoveries = nest.getAllPheromones().filter(p => p.type === "discovery");
+    const allDone = nest.getAllTasks().filter(t => t.status === "done");
+    if (discoveries.length > allDone.length && nest.getState().maxCost != null) {
+      const spent = nest.getState().ants.reduce((s, a) => s + a.usage.cost, 0);
+      if (spent < (nest.getState().maxCost ?? Infinity)) {
+        callbacks.onPhase?.("scouting", "Re-exploring based on new discoveries...");
+        emitSignal("scouting", "Re-exploring...");
+        await runAntWave({ ...waveBase, caste: "scout" });
+
+        const newTasks = nest.getAllTasks().filter(t =>
+          (t.caste === "worker" || t.caste === "drone") && t.status === "pending"
+        );
+        if (newTasks.length > 0) {
+          const drones = newTasks.filter(t => t.caste === "drone");
+          if (drones.length > 0) await runAntWave({ ...waveBase, caste: "drone" });
+
+          callbacks.onPhase?.("working", `${newTasks.length} new tasks from re-exploration`);
+          emitSignal("working", `${newTasks.length} new tasks`);
+          const result = await runAntWave({ ...waveBase, caste: "worker" });
+          if (result === "budget") {
+            nest.updateState({ status: "budget_exceeded", finishedAt: Date.now() });
+            emitSignal("budget_exceeded", "Budget exhausted");
+            const budgetState = nest.getState();
+            callbacks.onComplete?.(budgetState);
+            return budgetState;
+          }
+        }
+      }
+    }
+
     // ═══ Auto-check: run tsc before soldier review ═══
     let tscPassed = true;
     try {