oh-pi 0.1.76 → 0.1.77

package/README.md

@@ -157,7 +157,7 @@ Use `/colony-stop` to abort a running colony.
 
 ### Signal Protocol
 
-The colony communicates with the main conversation via structured signals, so the model never has to guess background state:
+The colony communicates with the main conversation via structured signals, so the model never has to guess background state. Updates are passively pushed (non-blocking), so polling is optional:
 
 | Signal | Meaning |
 |--------|---------|

package/README.zh.md

@@ -163,7 +163,7 @@ pi（主进程）
 
 ### 信号协议
 
-蚁群通过结构化信号与主对话通信，让模型无需猜测后台状态：
+蚁群通过结构化信号与主对话通信，让模型无需猜测后台状态。更新采用被动推送（非阻塞），轮询仅在手动排障时需要：
 
 | 信号 | 含义 |
 |------|------|

package/dist/index.js

@@ -43,7 +43,7 @@ async function quickFlow(env) {
         providers,
         theme: "dark",
         keybindings: "default",
-        extensions: ["safe-guard", "git-guard", "auto-session-name", "custom-footer", "compact-header", "auto-update", "smart-compact"],
+        extensions: ["safe-guard", "git-guard", "auto-session-name", "custom-footer", "compact-header", "auto-update"],
         prompts: ["review", "fix", "explain", "commit", "test"],
         agents: "general-developer",
         thinking: "medium",

package/dist/registry.js

@@ -49,7 +49,6 @@ export const EXTENSIONS = [
     { name: "compact-header", label: "⚡ Compact Header — Dense startup info replacing verbose output", default: true },
     { name: "ant-colony", label: "🐜 Ant Colony — Autonomous multi-agent swarm with adaptive concurrency", default: false },
     { name: "auto-update", label: "🔄 Auto Update — Check for oh-pi updates on startup and notify", default: true },
-    { name: "smart-compact", label: "🗜️  Smart Compact — Trim large tool outputs and old messages in-flight", default: true },
     { name: "bg-process", label: "⏳ Bg Process — Auto-background long-running commands (dev servers, etc.)", default: false },
 ];
 /** 快捷键绑定方案（default / vim / emacs） */

package/package.json

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

package/pi-package/agents/colony-operator.md

@@ -17,8 +17,9 @@ You command an autonomous ant colony. Complex tasks are delegated to the swarm,
 ## Workflow
 1. Assess task scope
 2. If colony-worthy → use `ant_colony` tool with clear goal
-3. If simple → do it directly
-4. Review colony output, fix gaps manually if needed
+3. After launch, use passive mode: wait for `COLONY_SIGNAL:*` updates; do not poll `bg_colony_status` unless user explicitly asks
+4. If simple → do it directly
+5. Review colony output, fix gaps manually if needed
 
 ## Code Standards
 - Follow existing conventions

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

@@ -59,6 +59,44 @@ export default function antColonyExtension(pi: ExtensionAPI) {
   // 当前运行中的后台蚁群（同时只允许一个）
   let activeColony: BackgroundColony | null = null;
 
+  // 防止主进程主动轮询导致阻塞：仅允许显式请求的手动快照，并加冷却
+  let lastBgStatusSnapshotAt = 0;
+  const STATUS_SNAPSHOT_COOLDOWN_MS = 15_000;
+
+  const extractMessageText = (message: any): string => {
+    const c = message?.content;
+    if (typeof c === "string") return c;
+    if (Array.isArray(c)) {
+      return c.map((p: any) => {
+        if (typeof p === "string") return p;
+        if (typeof p?.text === "string") return p.text;
+        if (typeof p?.content === "string") return p.content;
+        return "";
+      }).join("\n");
+    }
+    return "";
+  };
+
+  const lastUserMessageText = (ctx: any): string => {
+    try {
+      const branch = ctx?.sessionManager?.getBranch?.() ?? [];
+      for (let i = branch.length - 1; i >= 0; i--) {
+        const e = branch[i];
+        if (e?.type === "message" && e.message?.role === "user") {
+          return extractMessageText(e.message).trim();
+        }
+      }
+    } catch {
+      // ignore
+    }
+    return "";
+  };
+
+  const isExplicitStatusRequest = (ctx: any): boolean => {
+    const text = lastUserMessageText(ctx);
+    return /(?:\/colony-status|bg_colony_status)|(?:(?:蚁群|colony).{0,20}(?:状态|进度|进展|汇报|快照|status|progress|snapshot|update|check))|(?:(?:状态|进度|进展|汇报|快照|status|progress|snapshot|update|check).{0,20}(?:蚁群|colony))/i.test(text);
+  };
+
   const calcProgress = (m?: ColonyMetrics | null) => {
     if (!m || m.tasksTotal <= 0) return 0;
     return Math.max(0, Math.min(1, m.tasksDone / m.tasksTotal));
@@ -280,6 +318,7 @@ export default function antColonyExtension(pi: ExtensionAPI) {
     colony.promise = resume ? resumeColony(colonyOpts) : runColony(colonyOpts);
 
     activeColony = colony;
+    lastBgStatusSnapshotAt = 0;
     throttledRender();
 
     // 后台等待完成，注入结果
@@ -627,7 +666,7 @@ export default function antColonyExtension(pi: ExtensionAPI) {
       launchBackgroundColony(colonyParams);
 
       return {
-        content: [{ type: "text", text: `[COLONY_SIGNAL:LAUNCHED]\n🐜 Colony launched in background.\nGoal: ${params.goal}\n\nThe colony is now running autonomously. Results will be injected when it finishes.` }],
+        content: [{ type: "text", text: `[COLONY_SIGNAL:LAUNCHED]\n🐜 Colony launched in background.\nGoal: ${params.goal}\n\nThe colony runs autonomously in passive mode. Progress is pushed via [COLONY_SIGNAL:*] follow-up messages. Do not poll bg_colony_status unless the user explicitly asks for a manual snapshot.` }],
       };
     },
 
@@ -689,9 +728,40 @@ export default function antColonyExtension(pi: ExtensionAPI) {
   pi.registerTool({
     name: "bg_colony_status",
     label: "Colony Status",
-    description: "Check the status of a running background ant colony. Use this instead of bg_status to monitor colony progress.",
+    description: "Optional manual snapshot for a running colony. Progress is pushed passively via COLONY_SIGNAL follow-up messages; call this only when the user explicitly asks.",
     parameters: Type.Object({}),
-    async execute() {
+    async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
+      if (!activeColony) {
+        return {
+          content: [{ type: "text" as const, text: "No colony is currently running." }],
+        };
+      }
+
+      const explicit = isExplicitStatusRequest(ctx);
+      if (!explicit) {
+        return {
+          content: [{
+            type: "text" as const,
+            text: "Passive mode is active. Colony progress is already pushed via [COLONY_SIGNAL:*] follow-up messages. Skipping bg_colony_status polling to avoid blocking the main process. Ask explicitly for a manual snapshot if needed.",
+          }],
+          isError: true,
+        };
+      }
+
+      const now = Date.now();
+      const delta = now - lastBgStatusSnapshotAt;
+      if (delta < STATUS_SNAPSHOT_COOLDOWN_MS) {
+        const waitSec = Math.ceil((STATUS_SNAPSHOT_COOLDOWN_MS - delta) / 1000);
+        return {
+          content: [{
+            type: "text" as const,
+            text: `Manual status snapshot is rate-limited. Please wait ${waitSec}s to avoid active polling loops.`,
+          }],
+          isError: true,
+        };
+      }
+
+      lastBgStatusSnapshotAt = now;
       return {
         content: [{ type: "text" as const, text: buildStatusText() }],
       };

package/pi-package/extensions/smart-compact.test.ts

@@ -1,64 +0,0 @@
-import { describe, it, expect } from "vitest";
-import { truncateText, compactContent } from "./smart-compact";
-
-const longMultiline = (lines: number, lineLen = 100) =>
-  Array.from({ length: lines }, (_, i) => "x".repeat(lineLen) + i).join("\n");
-
-describe("truncateText", () => {
-  it("short text returns unchanged", () => {
-    expect(truncateText("hello", 8000)).toBe("hello");
-  });
-
-  it("few lines but long chars returns unchanged", () => {
-    const text = "x".repeat(9000) + "\n" + "y".repeat(9000);
-    expect(truncateText(text, 8000)).toBe(text);
-  });
-
-  it("long text with many lines gets truncated", () => {
-    const text = longMultiline(200);
-    const result = truncateText(text, 8000);
-    expect(result).toContain("[...truncated");
-  });
-
-  it("preserves head and tail", () => {
-    const text = longMultiline(200);
-    const result = truncateText(text, 8000);
-    expect(result.startsWith(text.slice(0, 1500))).toBe(true);
-    expect(result.endsWith(text.slice(-2500))).toBe(true);
-  });
-
-  it("custom head/tail params work", () => {
-    const text = longMultiline(200);
-    const result = truncateText(text, 100, 50, 50);
-    expect(result).toContain("[...truncated");
-    expect(result.startsWith(text.slice(0, 50))).toBe(true);
-    expect(result.endsWith(text.slice(-50))).toBe(true);
-  });
-});
-
-describe("compactContent", () => {
-  it("short string returns unchanged", () => {
-    expect(compactContent("short")).toBe("short");
-  });
-
-  it("long string gets truncated", () => {
-    const text = longMultiline(200);
-    expect(compactContent(text)).toContain("[...truncated");
-  });
-
-  it("array text block gets truncated", () => {
-    const text = longMultiline(200);
-    const result = compactContent([{ type: "text", text }]);
-    expect(result[0].text).toContain("[...truncated");
-  });
-
-  it("array non-text block returned unchanged", () => {
-    const block = { type: "image", url: "x" };
-    expect(compactContent([block])).toEqual([block]);
-  });
-
-  it("non-array non-string returned as-is", () => {
-    expect(compactContent(42)).toBe(42);
-    expect(compactContent({ a: 1 })).toEqual({ a: 1 });
-  });
-});

package/pi-package/extensions/smart-compact.ts

@@ -1,89 +0,0 @@
-/**
- * 智能压缩扩展 — 在发送给 LLM 前裁剪大块内容
- *
- * 策略：
- * 1. 工具输出超过阈值 → 保留首尾，中间替换为 "[...truncated N lines]"
- * 2. 用户粘贴的大块文本 → 同上
- * 3. 越旧的消息裁剪越激进
- */
-
-const MAX_TOOL_OUTPUT_CHARS = 8000;
-const MAX_USER_BLOCK_CHARS = 12000;
-const KEEP_HEAD = 1500;
-const KEEP_TAIL = 2500;
-
-export function truncateText(text: string, max: number, head = KEEP_HEAD, tail = KEEP_TAIL): string {
-  if (text.length <= max) return text;
-  const lines = text.split("\n");
-  if (lines.length <= 10) return text; // 短文本不裁
-  const headText = text.slice(0, head);
-  const tailText = text.slice(-tail);
-  const removedLines = text.slice(head, -tail).split("\n").length;
-  return `${headText}\n\n[...truncated ${removedLines} lines...]\n\n${tailText}`;
-}
-
-export function compactContent(content: any): any {
-  if (typeof content === "string") {
-    return truncateText(content, MAX_TOOL_OUTPUT_CHARS);
-  }
-  if (!Array.isArray(content)) return content;
-  return content.map((block: any) => {
-    if (block.type === "text" && typeof block.text === "string") {
-      return { ...block, text: truncateText(block.text, MAX_TOOL_OUTPUT_CHARS) };
-    }
-    return block;
-  });
-}
-
-export default function smartCompact(pi: any) {
-  pi.on("context", async (event: any) => {
-    const messages = event.messages;
-    if (!messages || messages.length < 4) return; // 太短不处理
-
-    // 只处理非最近 3 条消息（保留最近上下文完整）
-    const cutoff = messages.length - 3;
-
-    for (let i = 0; i < cutoff; i++) {
-      const msg = messages[i];
-      if (!msg) continue;
-
-      if (msg.role === "toolResult") {
-        msg.content = compactContent(msg.content);
-      } else if (msg.role === "user") {
-        // 用户消息用更宽松的阈值
-        if (typeof msg.content === "string" && msg.content.length > MAX_USER_BLOCK_CHARS) {
-          msg.content = truncateText(msg.content, MAX_USER_BLOCK_CHARS);
-        } else if (Array.isArray(msg.content)) {
-          msg.content = msg.content.map((block: any) => {
-            if (block.type === "text" && typeof block.text === "string" && block.text.length > MAX_USER_BLOCK_CHARS) {
-              return { ...block, text: truncateText(block.text, MAX_USER_BLOCK_CHARS) };
-            }
-            return block;
-          });
-        }
-      } else if (msg.role === "assistant" && Array.isArray(msg.content)) {
-        // 裁剪 assistant 的大块工具调用参数
-        msg.content = msg.content.map((block: any) => {
-          if (block.type === "toolCall" && block.arguments) {
-            const args = JSON.stringify(block.arguments);
-            if (args.length > MAX_TOOL_OUTPUT_CHARS) {
-              try {
-                const parsed = typeof block.arguments === "string" ? JSON.parse(block.arguments) : block.arguments;
-                // 裁剪大字符串参数
-                for (const key of Object.keys(parsed)) {
-                  if (typeof parsed[key] === "string" && parsed[key].length > MAX_TOOL_OUTPUT_CHARS) {
-                    parsed[key] = truncateText(parsed[key], MAX_TOOL_OUTPUT_CHARS);
-                  }
-                }
-                return { ...block, arguments: parsed };
-              } catch { return block; }
-            }
-          }
-          return block;
-        });
-      }
-    }
-
-    return { messages };
-  });
-}