oh-pi 0.1.38 → 0.1.40

package/package.json

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

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

@@ -79,7 +79,19 @@ function phasePipeline(status: string, theme: any): string {
 export default function antColonyExtension(pi: ExtensionAPI) {
 
   // ═══ Auto-trigger: 注入蚁群意识，LLM 自动判断何时启动 ═══
-  pi.on("before_agent_start", async () => {
+  pi.on("before_agent_start", async (ctx) => {
+    // 获取可用模型列表
+    let modelList = "";
+    try {
+      const { execSync } = await import("node:child_process");
+      const output = execSync("pi --list-models 2>/dev/null", { encoding: "utf-8", timeout: 5000 });
+      const models = output.trim().split("\n").slice(1) // skip header
+        .map(l => l.trim().split(/\s+/))
+        .filter(p => p.length >= 2)
+        .map(p => `${p[0]}/${p[1]}`);
+      if (models.length > 0) modelList = models.join(", ");
+    } catch { /* ignore */ }
+
     return {
       message: {
         customType: "ant-colony-context",
@@ -90,7 +102,17 @@ You have the ant_colony tool. Use it automatically when:
 - Task is complex enough that serial execution is inefficient
 
 Do NOT ask for confirmation. Call ant_colony directly with a clear goal.
-For simple single-file tasks, work directly without the colony.`,
+For simple single-file tasks, work directly without the colony.
+${modelList ? `
+[COLONY MODEL SELECTION]
+Available models: ${modelList}
+
+Strategy for choosing per-caste models:
+- scoutModel: Use a fast/cheap model (e.g. haiku, flash, gpt-4o-mini). Scouts only read, no edits.
+- workerModel: Use a capable model (e.g. sonnet, opus, gpt-4o). Workers make code changes.
+- soldierModel: Use same as worker or slightly cheaper. Soldiers review but don't edit.
+- If unsure, omit all three — defaults to current session model.
+- Prefer latest model versions for best quality.` : ""}`,
         display: false,
       },
     };
@@ -110,6 +132,9 @@ For simple single-file tasks, work directly without the colony.`,
       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 execute(_toolCallId, params, signal, onUpdate, ctx) {
@@ -169,12 +194,18 @@ For simple single-file tasks, work directly without the colony.`,
           appendFileSync(gitignorePath, `${content.length && !content.endsWith("\n") ? "\n" : ""}.ant-colony/\n`);
         }
 
+        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 state = await runColony({
           cwd: ctx.cwd,
           goal: params.goal,
           maxAnts: params.maxAnts,
           maxCost: params.maxCost,
           currentModel,
+          modelOverrides,
           signal: signal ?? undefined,
           callbacks,
         });
@@ -282,11 +313,13 @@ For simple single-file tasks, work directly without the colony.`,
             const taskTitle = task?.title?.slice(0, 55) || "...";
             const dur = a.finishedAt ? formatDuration(a.finishedAt - a.startedAt) : formatDuration(Date.now() - a.startedAt);
             const turns = a.usage.turns > 0 ? `${a.usage.turns}t` : "";
+            const model = a.model ? a.model.split("/").pop()! : "";
 
             container.addChild(new Text(
               theme.fg("muted", `  ${branch} `) + statusDot + " " +
               theme.fg("accent", `@${a.id.slice(0, 20)} `) +
-              theme.fg("dim", `(${a.caste}) ${dur}${turns ? " │ " + turns : ""}`),
+              theme.fg("dim", `(${a.caste}) ${dur}${turns ? " │ " + turns : ""}`) +
+              (model ? " " + theme.fg("muted", model) : ""),
               0, 0,
             ));
             container.addChild(new Text(

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

@@ -37,6 +37,7 @@ export interface QueenOptions {
   maxAnts?: number;
   maxCost?: number;
   currentModel: string;
+  modelOverrides?: ModelOverrides;
   signal?: AbortSignal;
   callbacks: QueenCallbacks;
 }
@@ -138,6 +139,7 @@ interface WaveOptions {
   cwd: string;
   caste: AntCaste;
   currentModel: string;
+  modelOverrides?: ModelOverrides;
   signal?: AbortSignal;
   callbacks: QueenCallbacks;
 }
@@ -147,7 +149,8 @@ interface WaveOptions {
  */
 async function runAntWave(opts: WaveOptions): Promise<"ok"> {
   const { nest, cwd, caste, signal, callbacks, currentModel } = opts;
-  const config = { ...DEFAULT_ANT_CONFIGS[caste], model: currentModel };
+  const casteModel = opts.modelOverrides?.[caste] || currentModel;
+  const config = { ...DEFAULT_ANT_CONFIGS[caste], model: casteModel };
 
   let backoffMs = 0; // 429 退避时间
   let consecutiveRateLimits = 0; // 连续限流计数
@@ -160,7 +163,8 @@ async function runAntWave(opts: WaveOptions): Promise<"ok"> {
 
     const ant: Ant = {
       id: "", caste, status: "idle", taskId: task.id,
-      pid: null, usage: { input: 0, output: 0, cost: 0, turns: 0 },
+      pid: null, model: casteModel,
+      usage: { input: 0, output: 0, cost: 0, turns: 0 },
       startedAt: Date.now(), finishedAt: null,
     };
     callbacks.onAntSpawn(ant, task);
@@ -311,6 +315,7 @@ export async function runColony(opts: QueenOptions): Promise<ColonyState> {
   const waveBase: Omit<WaveOptions, "caste"> = {
     nest, cwd: opts.cwd, signal, callbacks,
     currentModel: opts.currentModel,
+    modelOverrides: opts.modelOverrides,
   };
 
   const cleanup = () => {

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

@@ -216,6 +216,7 @@ export async function spawnAnt(
     status: "working",
     taskId: task.id,
     pid: null,
+    model: antConfig.model,
     usage: { input: 0, output: 0, cost: 0, turns: 0 },
     startedAt: Date.now(),
     finishedAt: null,

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

@@ -73,6 +73,7 @@ export interface Ant {
   status: AntStatus;
   taskId: string | null;
   pid: number | null;
+  model: string;
   usage: { input: number; output: number; cost: number; turns: number };
   startedAt: number;
   finishedAt: number | null;