oh-pi 0.1.53 → 0.1.54

package/package.json

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

@@ -72,7 +72,7 @@ function makeInitialScoutTask(goal: string): Task {
 
 function childTaskFromParsed(
   parentId: string,
-  parsed: { title: string; description: string; files: string[]; caste: AntCaste; priority: TaskPriority },
+  parsed: { title: string; description: string; files: string[]; caste: AntCaste; priority: TaskPriority; context?: string },
 ): Task {
   return {
     id: makeTaskId(),
@@ -83,6 +83,7 @@ function childTaskFromParsed(
     status: "pending",
     priority: parsed.priority,
     files: parsed.files,
+    context: parsed.context || undefined,
     claimedBy: null,
     result: null,
     error: null,
@@ -398,9 +399,18 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
       await runAntWave({ ...waveBase, caste: "worker" });
     }
 
+    // ═══ Auto-check: run tsc before soldier review ═══
+    let tscPassed = true;
+    try {
+      const { execSync } = await import("node:child_process");
+      execSync("npx tsc --noEmit", { cwd: opts.cwd, timeout: 30000, stdio: "pipe" });
+    } catch {
+      tscPassed = false;
+    }
+
     // ═══ Phase 3: 审查 ═══
     const completedWorkerTasks = nest.getAllTasks().filter(t => t.caste === "worker" && t.status === "done");
-    if (completedWorkerTasks.length > 0) {
+    if (completedWorkerTasks.length > 0 && (!tscPassed || completedWorkerTasks.length > 3)) {
       nest.updateState({ status: "reviewing" });
       callbacks.onPhase("reviewing", "Dispatching soldier ants to review changes...");
       const reviewTask = makeReviewTask(completedWorkerTasks);

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

@@ -57,6 +57,7 @@ export interface ParsedSubTask {
   files: string[];
   caste: AntCaste;
   priority: 1 | 2 | 3 | 4 | 5;
+  context?: string;
 }
 
 const CASTE_PROMPTS: Record<AntCaste, string> = {
@@ -67,6 +68,7 @@ Behavior:
 - Identify files, functions, dependencies related to the goal
 - IMPORTANT: After EACH tool call, summarize what you found so far. Do NOT wait until the end.
 - Report findings as structured intelligence for Worker Ants
+- For each recommended task, include the KEY code snippets (with file:line) the worker will need — this saves workers from re-reading files
 
 Output format (MUST follow exactly):
 ## Discoveries
@@ -79,6 +81,7 @@ For each task the colony should do next:
 - files: <comma-separated file paths>
 - caste: worker
 - priority: <1-5, 1=highest>
+- context: <relevant code snippets that the worker will need, with file:line references>
 
 ## Warnings
 Any risks, blockers, or conflicts detected.`,
@@ -142,6 +145,9 @@ function buildPrompt(task: Task, pheromoneContext: string, castePrompt: string,
   if (task.files.length > 0) {
     prompt += `**Files scope:** ${task.files.join(", ")}\n`;
   }
+  if (task.context) {
+    prompt += `\n## Pre-loaded Context (from scout)\n${task.context}\n`;
+  }
   if (/[\u4e00-\u9fff]/.test(task.description)) {
     prompt += '\nIMPORTANT: Follow the language requirements specified in the task description. If the task says to write in Chinese, write in Chinese.\n';
   }
@@ -152,13 +158,18 @@ function buildPrompt(task: Task, pheromoneContext: string, castePrompt: string,
 function parseSubTasks(output: string): ParsedSubTask[] {
   const tasks: ParsedSubTask[] = [];
   const regex = /### TASK:\s*(.+)\n(?:- description:\s*(.+)\n)?(?:- files:\s*(.+)\n)?(?:- caste:\s*(\w+)\n)?(?:- priority:\s*(\d))?/g;
+  const taskBlocks = output.split(/(?=### TASK:)/);
   for (const m of output.matchAll(regex)) {
+    const block = taskBlocks.find(b => b.includes(`### TASK: ${m[1]?.trim()}`)) || "";
+    const ctxMatch = block.match(/- context:\s*([\s\S]*?)(?=### TASK:|## |\n\n|$)/);
+    const context = ctxMatch?.[1]?.trim() || undefined;
     tasks.push({
       title: m[1]?.trim() || "Untitled",
       description: m[2]?.trim() || m[1]?.trim() || "",
       files: (m[3]?.trim() || "").split(",").map(f => f.trim()).filter(Boolean),
       caste: (m[4]?.trim() as AntCaste) || "worker",
       priority: (parseInt(m[5] || "3") as 1 | 2 | 3 | 4 | 5) || 3,
+      context,
     });
   }
   return tasks;

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

@@ -35,6 +35,7 @@ export interface Task {
   status: TaskStatus;
   priority: TaskPriority;
   files: string[];           // 锁定的文件
+  context?: string;          // Scout 预加载的代码片段
   claimedBy: string | null;  // ant id
   result: string | null;
   error: string | null;