oh-pi 0.1.51 → 0.1.52

package/package.json

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

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

@@ -16,7 +16,7 @@ You command an autonomous ant colony. Complex tasks are delegated to the swarm,
 
 ## Workflow
 1. Assess task scope
-2. If colony-worthy → use `/colony <goal>` command with clear goal
+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
 

package/pi-package/extensions/ant-colony/README.md

@@ -36,16 +36,15 @@
 
 ## 使用方式
 
-### 作为 Tool（LLM 自动调用）
+### 使用方式
 
-LLM 会在判断任务复杂度足够时自动调用 `ant_colony` tool。
+LLM 在判断任务复杂度足够时会自动调用 `ant_colony` tool，无需手动触发。
 
 ### 命令
 
 ```
-/colony <目标描述>          启动蚁群
-/colony-status              查看最近蚁群状态
-Ctrl+Alt+A                  快捷启动
+/colony-stop                中止运行中的蚁群
+Ctrl+Shift+A                展开蚁群详情面板
 ```
 
 ### 示例
@@ -114,4 +113,4 @@ npx oh-pi  # 选择 Full Power 预设
 | `concurrency.ts` | 115 | 自适应并发：系统采样，探索/稳态双阶段调节 |
 | `spawner.ts` | 316 | 蚂蚁孵化：进程管理，prompt 构建，输出解析 |
 | `queen.ts` | 331 | 女王调度：生命周期，任务波次，多轮迭代 |
-| `index.ts` | 324 | 扩展入口：tool/command/shortcut 注册，TUI 渲染 |
+| `index.ts` | 324 | 扩展入口：tool/shortcut 注册，TUI 渲染 |

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

@@ -467,37 +467,94 @@ export default function antColonyExtension(pi: ExtensionAPI) {
     },
   });
 
-  // ═══ Command: /colony ═══
-  pi.registerCommand("colony", {
-    description: "Launch an ant colony with a goal",
+  // ═══ Tool: ant_colony ═══
+  pi.registerTool({
+    name: "ant_colony",
+    label: "Ant Colony",
+    description: [
+      "Launch an autonomous ant colony in the BACKGROUND to accomplish a complex goal.",
+      "The colony runs asynchronously — you can continue chatting while it works.",
+      "Results are automatically injected when the colony finishes.",
+      "Scouts explore the codebase, workers execute tasks in parallel, soldiers review quality.",
+      "Use for multi-file changes, large refactors, or complex features.",
+    ].join(" "),
     parameters: Type.Object({
       goal: Type.String({ description: "What the colony should accomplish" }),
+      maxAnts: Type.Optional(Type.Number({ description: "Max concurrent ants (default: auto-adapt)", minimum: 1, maximum: 8 })),
+      maxCost: Type.Optional(Type.Number({ description: "Max cost budget in USD (default: unlimited)", minimum: 0.01 })),
+      scoutModel: Type.Optional(Type.String({ description: "Model for scout ants (default: current session model)" })),
+      workerModel: Type.Optional(Type.String({ description: "Model for worker ants (default: current session model)" })),
+      soldierModel: Type.Optional(Type.String({ description: "Model for soldier ants (default: current session model)" })),
     }),
-    async handler(args, ctx) {
+
+    async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
+      if (activeColony) {
+        return {
+          content: [{ type: "text", text: "A colony is already running in the background. Use /colony-stop to cancel it first." }],
+          isError: true,
+        };
+      }
+
       const currentModel = ctx.model ? `${ctx.model.provider}/${ctx.model.id}` : null;
       if (!currentModel) {
-        ctx.ui.notify("No model available in current session", "error");
-        return;
+        return {
+          content: [{ type: "text", text: "Colony failed: no model available in current session" }],
+          isError: true,
+        };
       }
 
+      const modelOverrides: Record<string, string> = {};
+      if (params.scoutModel) modelOverrides.scout = params.scoutModel;
+      if (params.workerModel) modelOverrides.worker = params.workerModel;
+      if (params.soldierModel) modelOverrides.soldier = params.soldierModel;
+
       const colonyParams = {
-        goal: args.goal,
+        goal: params.goal,
+        maxAnts: params.maxAnts,
+        maxCost: params.maxCost,
         currentModel,
-        modelOverrides: {},
+        modelOverrides,
         cwd: ctx.cwd,
         modelRegistry: ctx.modelRegistry ?? undefined,
       };
 
+      // 非交互模式（print mode）：同步等待蚁群完成
       if (!ctx.hasUI) {
-        const result = await runSyncColony(colonyParams);
-        if (result.isError) {
-          ctx.ui.notify("Colony failed", "error");
-        }
-        return;
+        return await runSyncColony(colonyParams, _signal);
       }
 
+      // 交互模式：后台运行
       launchBackgroundColony(colonyParams);
-      ctx.ui.notify(`🐜 Colony launched: ${args.goal.slice(0, 50)}`, "success");
+
+      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.` }],
+      };
+    },
+
+    renderCall(args, theme) {
+      const goal = args.goal?.length > 70 ? args.goal.slice(0, 67) + "..." : args.goal;
+      let text = theme.fg("toolTitle", theme.bold("🐜 ant_colony"));
+      if (args.maxAnts) text += theme.fg("muted", ` ×${args.maxAnts}`);
+      if (args.maxCost) text += theme.fg("warning", ` $${args.maxCost}`);
+      text += "\n" + theme.fg("dim", `  ${goal || "..."}`);
+      return new Text(text, 0, 0);
+    },
+
+    renderResult(result, { expanded }, theme) {
+      const text = result.content?.find((c: any) => c.type === "text")?.text || "";
+      if (result.isError) {
+        return new Text(theme.fg("error", text), 0, 0);
+      }
+      const container = new Container();
+      container.addChild(new Text(
+        theme.fg("success", "✓ ") + theme.fg("toolTitle", theme.bold("Colony launched in background")),
+        0, 0,
+      ));
+      if (activeColony) {
+        container.addChild(new Text(theme.fg("dim", `  Goal: ${activeColony.goal.slice(0, 70)}`), 0, 0));
+        container.addChild(new Text(theme.fg("muted", `  Ctrl+Shift+A for details │ /colony-stop to cancel`), 0, 0));
+      }
+      return container;
     },
   });
 

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

@@ -290,6 +290,9 @@ async function runAntWave(opts: WaveOptions): Promise<"ok"> {
  * 蚁后主循环
  */
 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.");
+  }
   const colonyId = makeColonyId();
   const nest = new Nest(opts.cwd, colonyId);
 

package/pi-package/skills/ant-colony/SKILL.md

@@ -1,59 +0,0 @@
----
-name: ant-colony
-description: Ant colony multi-agent orchestration. Use when user needs parallel multi-file changes, large refactors, or complex features. Provides colony management commands and strategies.
----
-
-# Ant Colony Skill
-
-## When to Use
-- Multi-file changes (≥3 files)
-- Parallel workstreams
-- Large refactors, migrations
-- User says: colony, swarm, parallel, multi-agent
-
-## Quick Start
-```
-/colony <goal>
-```
-
-## Colony Lifecycle
-1. **Scout** — Explore codebase, identify targets, produce task pool
-2. **Work** — Workers execute in parallel, adaptive concurrency
-3. **Review** — Soldiers audit changes, request fixes if needed
-4. **Fix** — Workers address review findings
-5. **Done** — Summary report
-
-## Strategies
-
-### File-Scoped Decomposition (default)
-Each worker owns distinct files. Zero conflict.
-```
-Worker A: [src/auth.ts, src/auth.test.ts]
-Worker B: [src/api.ts, src/api.test.ts]
-```
-
-### Module-Scoped
-Each worker owns a module directory.
-```
-Worker A: src/components/
-Worker B: src/api/
-```
-
-### Pipeline
-Sequential handoff when dependencies exist.
-```
-Scout → Worker(schema) → Worker(api) → Worker(ui) → Soldier
-```
-
-## Tuning
-- `maxAnts`: Cap concurrent ants (default: auto-adapt to CPU)
-- Colony auto-reduces concurrency on 429 rate limits
-- Blocked tasks auto-resume when file locks release
-
-## Extending Castes
-Add custom ant types by editing `ant-colony/types.ts`:
-```typescript
-// In DEFAULT_ANT_CONFIGS, add:
-mycaste: { caste: "mycaste", model: "...", tools: [...], maxTurns: 10 }
-```
-Then add matching prompt in `spawner.ts` CASTE_PROMPTS.