oh-pi 0.1.30 → 0.1.32

package/package.json

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

@@ -140,14 +140,13 @@ interface WaveOptions {
   currentModel: string;
   signal?: AbortSignal;
   callbacks: QueenCallbacks;
-  maxCost?: number;
 }
 
 /**
  * 并发执行一批蚂蚁，自适应调节并发度
  */
-async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget_exceeded"> {
-  const { nest, cwd, caste, signal, callbacks, maxCost, currentModel } = opts;
+async function runAntWave(opts: WaveOptions): Promise<"ok"> {
+  const { nest, cwd, caste, signal, callbacks, currentModel } = opts;
   const config = { ...DEFAULT_ANT_CONFIGS[caste], model: currentModel };
 
   let backoffMs = 0; // 429 退避时间
@@ -208,26 +207,6 @@ async function runAntWave(opts: WaveOptions): Promise<"ok" | "budget_exceeded">
   // 调度循环：持续派蚂蚁直到没有待处理任务
   let lastSampleTime = 0;
   while (!signal?.aborted) {
-    // 预算检查：派蚂蚁前预估，基于已完成蚂蚁的平均 cost
-    if (maxCost != null) {
-      const ants = nest.getState().ants;
-      const currentCost = 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 doneAnts = ants.filter(a => a.status === "done" && a.usage.cost > 0);
-      if (doneAnts.length > 0) {
-        const avgCost = doneAnts.reduce((s, a) => s + a.usage.cost, 0) / doneAnts.length;
-        const remaining = maxCost - currentCost;
-        if (remaining < avgCost * 0.5) {
-          callbacks.onPhase("working", `Budget nearly exhausted ($${currentCost.toFixed(3)}, avg/ant: $${avgCost.toFixed(3)}). 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;
@@ -331,7 +310,7 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
   const { signal, callbacks } = opts;
   const waveBase: Omit<WaveOptions, "caste"> = {
     nest, cwd: opts.cwd, signal, callbacks,
-    maxCost: opts.maxCost, currentModel: opts.currentModel,
+    currentModel: opts.currentModel,
   };
 
   const cleanup = () => {
@@ -343,15 +322,6 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
     } catch { /* ignore */ }
   };
 
-  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);
-    cleanup();
-    return s;
-  };
-
   try {
     // ═══ Phase 1: 侦察（最多重试2次） ═══
     let scoutAttempt = 0;
@@ -363,8 +333,7 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
         ? "Dispatching scout ants to explore codebase..."
         : `Scout retry ${scoutAttempt}/${MAX_SCOUT_RETRIES}...`);
 
-      if (await runAntWave({ ...waveBase, caste: "scout" }) === "budget_exceeded")
-        return budgetStop("Budget exceeded during scouting");
+      await runAntWave({ ...waveBase, caste: "scout" });
 
       workerTasks = nest.getAllTasks().filter(t => t.caste === "worker" && t.status === "pending");
       if (workerTasks.length > 0) break;
@@ -389,8 +358,7 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
     // ═══ Phase 2: 工作 ═══
     nest.updateState({ status: "working" });
     callbacks.onPhase("working", `${workerTasks.length} tasks discovered. Dispatching worker ants...`);
-    if (await runAntWave({ ...waveBase, caste: "worker" }) === "budget_exceeded")
-      return budgetStop("Budget exceeded during working");
+    await runAntWave({ ...waveBase, caste: "worker" });
 
     // 处理工蚁产生的子任务（可能有多轮）
     let rounds = 0;
@@ -401,8 +369,7 @@ 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...`);
-      if (await runAntWave({ ...waveBase, caste: "worker" }) === "budget_exceeded")
-        return budgetStop("Budget exceeded during sub-tasks");
+      await runAntWave({ ...waveBase, caste: "worker" });
     }
 
     // ═══ Phase 3: 审查 ═══
@@ -412,8 +379,7 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
       callbacks.onPhase("reviewing", "Dispatching soldier ants to review changes...");
       const reviewTask = makeReviewTask(completedWorkerTasks);
       nest.writeTask(reviewTask);
-      if (await runAntWave({ ...waveBase, caste: "soldier" }) === "budget_exceeded")
-        return budgetStop("Budget exceeded during review");
+      await runAntWave({ ...waveBase, caste: "soldier" });
 
       // 兵蚁产生的修复任务
       const fixTasks = nest.getAllTasks().filter(t =>
@@ -422,8 +388,7 @@ 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...`);
-        if (await runAntWave({ ...waveBase, caste: "worker" }) === "budget_exceeded")
-          return budgetStop("Budget exceeded during fixes");
+        await runAntWave({ ...waveBase, caste: "worker" });
       }
     }
 

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

@@ -114,8 +114,11 @@ Output format (MUST follow exactly):
 PASS or FAIL with summary.`,
 };
 
-function buildPrompt(task: Task, pheromoneContext: string, castePrompt: string): string {
+function buildPrompt(task: Task, pheromoneContext: string, castePrompt: string, maxTurns?: number): 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`;
+  }
   if (pheromoneContext) {
     prompt += `## Colony Pheromone Trail (intelligence from other ants)\n${pheromoneContext}\n\n`;
   }
@@ -223,7 +226,7 @@ export async function spawnAnt(
   // 构建 prompt
   const pheromoneCtx = nest.getPheromoneContext(task.files);
   const castePrompt = CASTE_PROMPTS[antConfig.caste];
-  const fullPrompt = buildPrompt(task, pheromoneCtx, castePrompt);
+  const fullPrompt = buildPrompt(task, pheromoneCtx, castePrompt, antConfig.maxTurns);
   const tmpFile = writePromptFile(nest.dir, antId, fullPrompt);
 
   const args = [
@@ -246,6 +249,7 @@ export async function spawnAnt(
       nest.updateAnt(ant);
 
       let buffer = "";
+      let turnCount = 0;
 
       proc.stdout.on("data", (data) => {
         buffer += data.toString();
@@ -255,6 +259,15 @@ export async function spawnAnt(
           if (!line.trim()) continue;
           try {
             const event = JSON.parse(line);
+            if (event.type === "turn_end") {
+              turnCount++;
+              if (antConfig.maxTurns && turnCount === antConfig.maxTurns) {
+                stderr += `[ant-colony] Warning: ant reached maxTurns (${antConfig.maxTurns}), 1 grace turn remaining\n`;
+              } else if (antConfig.maxTurns && turnCount > antConfig.maxTurns) {
+                proc.kill("SIGTERM");
+                setTimeout(() => { if (!proc.killed) proc.kill("SIGKILL"); }, 3000);
+              }
+            }
             if (event.type === "message_end" && event.message) {
               messages.push(event.message);
               if (event.message.role === "assistant") {

package/pi-package/skills/context7/SKILL.md

@@ -0,0 +1,34 @@
+---
+name: context7
+description: Search and query up-to-date documentation for any programming library via Context7 API. Use when you need current docs, code examples, or API references for libraries and frameworks.
+---
+
+# Context7
+
+Search for libraries and query their documentation via the Context7 API.
+
+## Search Libraries
+
+```bash
+{baseDir}/search.js "library name" "what you need help with"
+```
+
+Example:
+```bash
+{baseDir}/search.js "react" "hooks for state management"
+```
+
+Returns matching libraries with Context7-compatible IDs for use with the docs tool.
+
+## Query Documentation
+
+```bash
+{baseDir}/docs.js "/org/project" "your question"
+```
+
+Example:
+```bash
+{baseDir}/docs.js "/websites/react_dev" "useEffect cleanup"
+```
+
+Use the library ID from search results. Returns relevant documentation and code examples.

package/pi-package/skills/context7/docs.js

@@ -0,0 +1,13 @@
+#!/usr/bin/env node
+
+const [libraryId, query] = process.argv.slice(2);
+if (!libraryId || !query) { console.error("Usage: docs.js <libraryId> <query>"); process.exit(1); }
+
+const res = await fetch("https://mcp.context7.com/mcp", {
+  method: "POST",
+  headers: { "Content-Type": "application/json", "Accept": "application/json, text/event-stream" },
+  body: JSON.stringify({ jsonrpc: "2.0", id: 1, method: "tools/call", params: { name: "query-docs", arguments: { libraryId, query } } })
+});
+const data = await res.json();
+if (data.error) { console.error("Error:", data.error.message); process.exit(1); }
+console.log(data.result.content[0].text);

package/pi-package/skills/context7/search.js

@@ -0,0 +1,13 @@
+#!/usr/bin/env node
+
+const [libraryName, query = libraryName] = process.argv.slice(2);
+if (!libraryName) { console.error("Usage: search.js <libraryName> [query]"); process.exit(1); }
+
+const res = await fetch("https://mcp.context7.com/mcp", {
+  method: "POST",
+  headers: { "Content-Type": "application/json", "Accept": "application/json, text/event-stream" },
+  body: JSON.stringify({ jsonrpc: "2.0", id: 1, method: "tools/call", params: { name: "resolve-library-id", arguments: { query, libraryName } } })
+});
+const data = await res.json();
+if (data.error) { console.error("Error:", data.error.message); process.exit(1); }
+console.log(data.result.content[0].text);

package/pi-package/skills/web-fetch/SKILL.md

@@ -0,0 +1,24 @@
+---
+name: web-fetch
+description: Fetch a web page and extract readable text content. Use when user needs to retrieve or read a web page.
+---
+
+# web-fetch
+
+Fetch a web page and extract readable text content.
+
+## Usage
+
+```bash
+{baseDir}/fetch.js <url> [--raw]
+```
+
+- `<url>` — URL to fetch
+- `--raw` — Output raw HTML instead of extracted text
+
+## Examples
+
+```bash
+{baseDir}/fetch.js https://example.com
+{baseDir}/fetch.js https://example.com --raw
+```

package/pi-package/skills/web-fetch/fetch.js

@@ -0,0 +1,28 @@
+#!/usr/bin/env node
+
+process.env.NODE_TLS_REJECT_UNAUTHORIZED = '0';
+
+const args = process.argv.slice(2);
+const raw = args.includes('--raw');
+const url = args.find(a => !a.startsWith('--'));
+
+if (!url) { console.error('Usage: fetch.js <url> [--raw]'); process.exit(1); }
+
+const res = await fetch(url);
+const html = await res.text();
+
+if (raw) { console.log(html); } else {
+  const text = html
+    .replace(/<script[\s\S]*?<\/script>/gi, '')
+    .replace(/<style[\s\S]*?<\/style>/gi, '')
+    .replace(/<[^>]+>/g, ' ')
+    .replace(/&nbsp;/g, ' ')
+    .replace(/&amp;/g, '&')
+    .replace(/&lt;/g, '<')
+    .replace(/&gt;/g, '>')
+    .replace(/&#(\d+);/g, (_, n) => String.fromCharCode(n))
+    .replace(/[ \t]+/g, ' ')
+    .replace(/\n\s*\n/g, '\n')
+    .trim();
+  console.log(text);
+}

package/pi-package/skills/web-search/SKILL.md

@@ -0,0 +1,18 @@
+---
+name: web-search
+description: Web search via DuckDuckGo. Use when the user needs to look up current information online.
+---
+
+# web-search
+
+Web search via DuckDuckGo. Use when the user needs to look up current information online.
+
+## Usage
+
+```bash
+{baseDir}/search.js "query terms"
+{baseDir}/search.js -n 10 "query terms"
+```
+
+- `-n <count>` — number of results to return (default: 5)
+- Returns title, URL, and snippet for each result.

package/pi-package/skills/web-search/search.js

@@ -0,0 +1,29 @@
+#!/usr/bin/env node
+
+const args = process.argv.slice(2);
+let n = 5, query;
+
+for (let i = 0; i < args.length; i++) {
+  if (args[i] === '-n' && args[i + 1]) { n = parseInt(args[++i], 10); }
+  else { query = args[i]; }
+}
+
+if (!query) { console.error('Usage: search.js [-n count] "query"'); process.exit(1); }
+
+const res = await fetch(`https://html.duckduckgo.com/html/?q=${encodeURIComponent(query)}`, {
+  headers: { 'User-Agent': 'Mozilla/5.0' }
+});
+const html = await res.text();
+
+const results = [];
+const blockRe = /<a rel="nofollow" class="result__a" href="([^"]*)"[^>]*>([\s\S]*?)<\/a>[\s\S]*?<a class="result__snippet"[^>]*>([\s\S]*?)<\/a>/g;
+let m;
+while ((m = blockRe.exec(html)) && results.length < n) {
+  const url = decodeURIComponent(m[1].replace(/^\/\/duckduckgo\.com\/l\/\?uddg=/, '').replace(/&amp;rut=.*$/, '').replace(/&rut=.*$/, ''));
+  const title = m[2].replace(/<[^>]*>/g, '').trim();
+  const snippet = m[3].replace(/<[^>]*>/g, '').trim();
+  results.push({ title, url, snippet });
+}
+
+if (!results.length) { console.log('No results found.'); }
+else { results.forEach((r, i) => console.log(`${i + 1}. ${r.title}\n   ${r.url}\n   ${r.snippet}\n`)); }