@shog-lab/pi-toolkit 0.1.0

package/dist/extensions/mcp-bridge/mcp-client.js

@@ -0,0 +1,145 @@
+/**
+ * Minimal MCP (Model Context Protocol) stdio JSON-RPC client.
+ *
+ * Spawns an MCP server as a child process, performs the initialize handshake,
+ * and exposes tools/list + tools/call.
+ *
+ * MCP spec: https://spec.modelcontextprotocol.io/
+ */
+import { spawn } from "node:child_process";
+const PROTOCOL_VERSION = "2024-11-05";
+const HANDSHAKE_TIMEOUT_MS = 10000;
+const TOOL_CALL_TIMEOUT_MS = 60000;
+export class McpClient {
+    proc;
+    nextId = 1;
+    pending = new Map();
+    stdoutBuffer = "";
+    initialized = false;
+    closed = false;
+    initializePromise = null;
+    serverName;
+    constructor(serverName, config) {
+        this.serverName = serverName;
+        this.proc = spawn(config.command, config.args ?? [], {
+            env: { ...process.env, ...config.env },
+            stdio: ["pipe", "pipe", "pipe"],
+        });
+        this.proc.stdout?.on("data", (chunk) => this.onStdout(chunk));
+        this.proc.stderr?.on("data", (chunk) => {
+            // MCP servers commonly use stderr for logs — surface as warnings, don't fail
+            const line = chunk.toString().trim();
+            if (line)
+                console.warn(`[mcp:${this.serverName}] ${line}`);
+        });
+        this.proc.on("error", (err) => this.failAll(err));
+        this.proc.on("exit", (code, signal) => {
+            this.closed = true;
+            this.failAll(new Error(`MCP server "${this.serverName}" exited (code=${code}, signal=${signal})`));
+        });
+        this.proc.unref(); // don't keep event loop alive on our own
+    }
+    /** Run the MCP initialize handshake. Idempotent — multiple callers share one in-flight promise. */
+    initialize() {
+        if (this.initialized)
+            return Promise.resolve();
+        if (this.initializePromise)
+            return this.initializePromise;
+        this.initializePromise = (async () => {
+            await this.send("initialize", {
+                protocolVersion: PROTOCOL_VERSION,
+                capabilities: {},
+                clientInfo: { name: "pi-toolkit-mcp-bridge", version: "0.1.0" },
+            }, HANDSHAKE_TIMEOUT_MS);
+            this.notify("notifications/initialized");
+            this.initialized = true;
+        })();
+        return this.initializePromise;
+    }
+    async listTools() {
+        await this.initialize();
+        const result = await this.send("tools/list", undefined, HANDSHAKE_TIMEOUT_MS);
+        return result.tools ?? [];
+    }
+    async callTool(name, args) {
+        await this.initialize();
+        const result = await this.send("tools/call", { name, arguments: args }, TOOL_CALL_TIMEOUT_MS);
+        return result;
+    }
+    close() {
+        if (this.closed)
+            return;
+        this.closed = true;
+        if (!this.proc.killed) {
+            try {
+                this.proc.kill("SIGTERM");
+            }
+            catch { /* ignore */ }
+        }
+        this.failAll(new Error(`MCP client "${this.serverName}" closed`));
+    }
+    // --- internals ---
+    send(method, params, timeoutMs) {
+        if (this.closed)
+            return Promise.reject(new Error(`MCP client "${this.serverName}" closed`));
+        const id = this.nextId++;
+        const req = { jsonrpc: "2.0", id, method, params };
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                this.pending.delete(id);
+                reject(new Error(`MCP "${this.serverName}" ${method} timed out after ${timeoutMs}ms`));
+            }, timeoutMs);
+            timer.unref();
+            this.pending.set(id, {
+                resolve: resolve,
+                reject,
+                timer,
+            });
+            this.proc.stdin?.write(JSON.stringify(req) + "\n");
+        });
+    }
+    notify(method, params) {
+        if (this.closed)
+            return;
+        const req = { jsonrpc: "2.0", method, params };
+        this.proc.stdin?.write(JSON.stringify(req) + "\n");
+    }
+    onStdout(chunk) {
+        this.stdoutBuffer += chunk.toString();
+        const lines = this.stdoutBuffer.split("\n");
+        this.stdoutBuffer = lines.pop() ?? "";
+        for (const line of lines) {
+            if (!line.trim())
+                continue;
+            let msg;
+            try {
+                msg = JSON.parse(line);
+            }
+            catch {
+                console.warn(`[mcp:${this.serverName}] non-JSON output: ${line.slice(0, 200)}`);
+                continue;
+            }
+            if (msg.id === undefined)
+                continue; // notification from server, ignore for now
+            const pending = this.pending.get(msg.id);
+            if (!pending)
+                continue;
+            this.pending.delete(msg.id);
+            clearTimeout(pending.timer);
+            if (msg.error) {
+                pending.reject(new Error(`MCP "${this.serverName}" error ${msg.error.code}: ${msg.error.message}`));
+            }
+            else {
+                pending.resolve(msg.result);
+            }
+        }
+    }
+    failAll(err) {
+        for (const { reject, timer } of this.pending.values()) {
+            clearTimeout(timer);
+            reject(err);
+        }
+        this.pending.clear();
+    }
+}
+//# sourceMappingURL=mcp-client.js.map

package/dist/extensions/mcp-bridge/mcp-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../../../extensions/mcp-bridge/mcp-client.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,KAAK,EAAqB,MAAM,oBAAoB,CAAC;AAwC9D,MAAM,gBAAgB,GAAG,YAAY,CAAC;AACtC,MAAM,oBAAoB,GAAG,KAAK,CAAC;AACnC,MAAM,oBAAoB,GAAG,KAAK,CAAC;AAEnC,MAAM,OAAO,SAAS;IACZ,IAAI,CAAe;IACnB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,IAAI,GAAG,EAIrB,CAAC;IACG,YAAY,GAAG,EAAE,CAAC;IAClB,WAAW,GAAG,KAAK,CAAC;IACpB,MAAM,GAAG,KAAK,CAAC;IACf,iBAAiB,GAAyB,IAAI,CAAC;IACvC,UAAU,CAAS;IAEnC,YAAY,UAAkB,EAAE,MAAuB;QACrD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE,EAAE;YACnD,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,GAAG,EAAE;YACtC,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SAChC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;QACtE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YAC7C,6EAA6E;YAC7E,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;YACrC,IAAI,IAAI;gBAAE,OAAO,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,UAAU,KAAK,IAAI,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;QAClD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YACpC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;YACnB,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,UAAU,kBAAkB,IAAI,YAAY,MAAM,GAAG,CAAC,CAAC,CAAC;QACrG,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,yCAAyC;IAC9D,CAAC;IAED,mGAAmG;IACnG,UAAU;QACR,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC/C,IAAI,IAAI,CAAC,iBAAiB;YAAE,OAAO,IAAI,CAAC,iBAAiB,CAAC;QAC1D,IAAI,CAAC,iBAAiB,GAAG,CAAC,KAAK,IAAI,EAAE;YACnC,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;gBAC5B,eAAe,EAAE,gBAAgB;gBACjC,YAAY,EAAE,EAAE;gBAChB,UAAU,EAAE,EAAE,IAAI,EAAE,uBAAuB,EAAE,OAAO,EAAE,OAAO,EAAE;aAChE,EAAE,oBAAoB,CAAC,CAAC;YACzB,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,CAAC;YACzC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAuB,YAAY,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC;QACpG,OAAO,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAY,EAAE,IAA6B;QACxD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAgB,YAAY,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAC7G,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACtB,IAAI,CAAC;gBAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,UAAU,UAAU,CAAC,CAAC,CAAC;IACpE,CAAC;IAED,oBAAoB;IAEZ,IAAI,CAAI,MAAc,EAAE,MAAe,EAAE,SAAiB;QAChE,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,UAAU,UAAU,CAAC,CAAC,CAAC;QAC5F,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,GAAG,GAAmB,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QACnE,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACxB,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,UAAU,KAAK,MAAM,oBAAoB,SAAS,IAAI,CAAC,CAAC,CAAC;YACzF,CAAC,EAAE,SAAS,CAAC,CAAC;YACd,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,OAA+B;gBACxC,MAAM;gBACN,KAAK;aACN,CAAC,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,MAAc,EAAE,MAAgB;QAC7C,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,MAAM,GAAG,GAAmB,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QAC/D,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;IACrD,CAAC;IAEO,QAAQ,CAAC,KAAa;QAC5B,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAS;YAC3B,IAAI,GAAoB,CAAC;YACzB,IAAI,CAAC;gBACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,UAAU,sBAAsB,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;gBAChF,SAAS;YACX,CAAC;YACD,IAAI,GAAG,CAAC,EAAE,KAAK,SAAS;gBAAE,SAAS,CAAC,2CAA2C;YAE/E,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACzC,IAAI,CAAC,OAAO;gBAAE,SAAS;YACvB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC5B,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAE5B,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACd,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,UAAU,WAAW,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACtG,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,GAAU;QACxB,KAAK,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YACtD,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;CACF"}

package/dist/extensions/mcp-bridge/package.json

@@ -0,0 +1,8 @@
+{
+  "name": "@shog-lab/pi-toolkit-mcp-bridge",
+  "version": "0.1.0",
+  "private": true,
+  "description": "Bridge MCP (Model Context Protocol) servers as pi-coding-agent tools. Reads mcp-servers.json, spawns servers, registers tools.",
+  "type": "module",
+  "main": "index.js"
+}

package/dist/extensions/subagent/index.d.ts

@@ -0,0 +1,20 @@
+/**
+ * pi-toolkit Sub-Agent Extension
+ *
+ * Provides spawn_subagent tool for an agent to spawn a focused sub-agent in a
+ * target directory. Sub-agents have minimal context: no memory access, no parent
+ * skills, no system prompt — just the task plus web_search (when installed
+ * alongside in toolkit).
+ *
+ * Lives in pi-toolkit (not pi-mind) because it's an agent-facing tool with no
+ * memory dependency — it just wraps spawnPi from pi-utils. Pi-mind has its own
+ * internal L2 spawn helper for memory-specific use cases.
+ *
+ * Use cases:
+ *   - Run tests / read code in an isolated context
+ *   - Generate a SKILL.md from raw observations
+ *   - Classify content without bias from the parent agent's loaded context
+ */
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+export default function subagentExtension(pi: ExtensionAPI): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/extensions/subagent/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../extensions/subagent/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAMH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAgCpE,MAAM,CAAC,OAAO,UAAU,iBAAiB,CAAC,EAAE,EAAE,YAAY,QA0EzD"}

package/dist/extensions/subagent/index.js

@@ -0,0 +1,118 @@
+/**
+ * pi-toolkit Sub-Agent Extension
+ *
+ * Provides spawn_subagent tool for an agent to spawn a focused sub-agent in a
+ * target directory. Sub-agents have minimal context: no memory access, no parent
+ * skills, no system prompt — just the task plus web_search (when installed
+ * alongside in toolkit).
+ *
+ * Lives in pi-toolkit (not pi-mind) because it's an agent-facing tool with no
+ * memory dependency — it just wraps spawnPi from pi-utils. Pi-mind has its own
+ * internal L2 spawn helper for memory-specific use cases.
+ *
+ * Use cases:
+ *   - Run tests / read code in an isolated context
+ *   - Generate a SKILL.md from raw observations
+ *   - Classify content without bias from the parent agent's loaded context
+ */
+import { existsSync, realpathSync } from "node:fs";
+import { dirname, join, normalize, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+import { Type } from "@sinclair/typebox";
+import { spawnPi } from "@shog-lab/pi-utils";
+const SubAgentParams = Type.Object({
+    cwd: Type.String({ description: "Absolute path to the working directory for the sub-agent (typically a repo or a subdirectory)." }),
+    prompt: Type.String({ description: "Task instruction for the sub-agent. Be specific about what to do and what to return." }),
+    timeout: Type.Optional(Type.Number({ description: "Timeout in seconds (default: 300, max: 600)" })),
+});
+/** Resolve the toolkit package extensions directory regardless of symlink chain. */
+function getExtensionsDir() {
+    const here = realpathSync(fileURLToPath(import.meta.url));
+    // index.ts is at <pkg-root>/extensions/subagent/index.ts → up two levels = extensions/
+    return join(dirname(here), "..");
+}
+function ok(text, details = {}) {
+    return { content: [{ type: "text", text }], details };
+}
+function err(text) {
+    return { content: [{ type: "text", text }], details: {}, isError: true };
+}
+function resolveExtensionPath(name) {
+    const p = join(getExtensionsDir(), name);
+    for (const entry of ["dist/index.js", "index.ts", "index.js"]) {
+        if (existsSync(join(p, entry)))
+            return join(p, entry);
+    }
+    return null;
+}
+export default function subagentExtension(pi) {
+    pi.registerTool({
+        name: "spawn_subagent",
+        label: "Spawn Sub-Agent",
+        description: "Spawn a focused sub-agent in a target directory to execute a specific task. " +
+            "The sub-agent has minimal context: no memory access, no parent skills, no system prompt. " +
+            "Use this for isolated tasks like running tests, reading code, classifying content.",
+        parameters: SubAgentParams,
+        async execute(_toolCallId, params) {
+            const normalized = resolve(normalize(params.cwd));
+            if (!existsSync(normalized)) {
+                return err(`Working directory does not exist: ${params.cwd}`);
+            }
+            const timeoutSec = Math.min(params.timeout ?? 300, 600);
+            const timeoutMs = timeoutSec * 1000;
+            const args = [
+                "-p",
+                "--no-extensions",
+            ];
+            if (process.env.MODEL) {
+                args.push("--model", process.env.MODEL);
+            }
+            // Load basic work tools — explicitly excludes memory (sub-agents must not touch parent memory).
+            // Browser automation belongs in pi-chrome, not pi-mind, so it isn't loaded here;
+            // a sub-agent that needs the browser should be spawned from a pi-chrome-aware caller.
+            const basicExts = ["web_search"];
+            for (const ext of basicExts) {
+                const extPath = resolveExtensionPath(ext);
+                if (extPath)
+                    args.push("-e", extPath);
+            }
+            const systemPrompt = [
+                "You are a sub-agent executing a focused task.",
+                "Rules:",
+                "- Read code, run tests, verify behavior — report findings as plain text",
+                "- Do not write code unless the task explicitly requires it",
+                "- Do not access memory, parent skills, or persistent state",
+                "- Do not modify files unless the task explicitly requires it",
+                "- Return findings, results, or errors as plain text",
+                "",
+                `Task: ${params.prompt}`,
+            ].join("\n");
+            args.push("--append-system-prompt", systemPrompt);
+            args.push(params.prompt);
+            let stdout = "";
+            let stderr = "";
+            const result = await spawnPi({
+                cwd: normalized,
+                args,
+                onStdout: (d) => { stdout += d; },
+                onStderr: (d) => { stderr += d; },
+                timeoutMs,
+            });
+            if (result.killed) {
+                return err(`Sub-agent timed out after ${timeoutSec}s`);
+            }
+            else if (result.code === 0) {
+                const text = stdout
+                    .replace(/^\*\*Output:\*\*\n?/s, "")
+                    .replace(/^.*?---\n/s, "")
+                    .replace(/^\*\*\(.+\)\*\*\n?/s, "")
+                    .trim() || "Done.";
+                return ok(text, result.tokens ? { tokens: result.tokens } : {});
+            }
+            else {
+                return err(stderr || stdout || `Sub-agent exited with code ${result.code}`);
+            }
+        },
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/extensions/subagent/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../extensions/subagent/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,IAAI,EAAe,MAAM,mBAAmB,CAAC;AAEtD,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAE7C,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;IACjC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,gGAAgG,EAAE,CAAC;IACnI,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,sFAAsF,EAAE,CAAC;IAC5H,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,6CAA6C,EAAE,CAAC,CAAC;CACpG,CAAC,CAAC;AAEH,oFAAoF;AACpF,SAAS,gBAAgB;IACvB,MAAM,IAAI,GAAG,YAAY,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1D,uFAAuF;IACvF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,EAAE,CAAC,IAAY,EAAE,UAAmC,EAAE;IAC7D,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC;AACjE,CAAC;AAED,SAAS,GAAG,CAAC,IAAY;IACvB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,IAAa,EAAE,CAAC;AAC7F,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAY;IACxC,MAAM,CAAC,GAAG,IAAI,CAAC,gBAAgB,EAAE,EAAE,IAAI,CAAC,CAAC;IACzC,KAAK,MAAM,KAAK,IAAI,CAAC,eAAe,EAAE,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;QAC9D,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAAE,OAAO,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,OAAO,UAAU,iBAAiB,CAAC,EAAgB;IACxD,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,gBAAgB;QACtB,KAAK,EAAE,iBAAiB;QACxB,WAAW,EACT,8EAA8E;YAC9E,2FAA2F;YAC3F,oFAAoF;QACtF,UAAU,EAAE,cAAc;QAC1B,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,MAAqC;YACtE,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5B,OAAO,GAAG,CAAC,qCAAqC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC;YAChE,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,GAAG,EAAE,GAAG,CAAC,CAAC;YACxD,MAAM,SAAS,GAAG,UAAU,GAAG,IAAI,CAAC;YAEpC,MAAM,IAAI,GAAG;gBACX,IAAI;gBACJ,iBAAiB;aAClB,CAAC;YACF,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;gBACtB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1C,CAAC;YAED,gGAAgG;YAChG,iFAAiF;YACjF,sFAAsF;YACtF,MAAM,SAAS,GAAG,CAAC,YAAY,CAAC,CAAC;YACjC,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;gBAC5B,MAAM,OAAO,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;gBAC1C,IAAI,OAAO;oBAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,YAAY,GAAG;gBACnB,+CAA+C;gBAC/C,QAAQ;gBACR,yEAAyE;gBACzE,4DAA4D;gBAC5D,4DAA4D;gBAC5D,8DAA8D;gBAC9D,qDAAqD;gBACrD,EAAE;gBACF,SAAS,MAAM,CAAC,MAAM,EAAE;aACzB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE,YAAY,CAAC,CAAC;YAClD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEzB,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;gBAC3B,GAAG,EAAE,UAAU;gBACf,IAAI;gBACJ,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC;gBACjC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC;gBACjC,SAAS;aACV,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,OAAO,GAAG,CAAC,6BAA6B,UAAU,GAAG,CAAC,CAAC;YACzD,CAAC;iBAAM,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,IAAI,GAAG,MAAM;qBAChB,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC;qBACnC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;qBACzB,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC;qBAClC,IAAI,EAAE,IAAI,OAAO,CAAC;gBACrB,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,CAAC,MAAM,IAAI,MAAM,IAAI,8BAA8B,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/extensions/subagent/package.json

@@ -0,0 +1,8 @@
+{
+  "name": "@shog-lab/pi-toolkit-subagent",
+  "version": "0.1.0",
+  "private": true,
+  "description": "Spawn focused sub-agents with minimal context for isolated tasks.",
+  "type": "module",
+  "main": "index.js"
+}

package/dist/extensions/understand_image/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * pi-mind Image Understanding Extension
+ *
+ * Provides image understanding via mmx vision CLI.
+ * Also intercepts base64 images in prompts and saves them to temp files,
+ * so the agent can reference them when MiniMax doesn't receive images directly.
+ */
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+export default function imageUnderstandingExtension(pi: ExtensionAPI): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/extensions/understand_image/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../extensions/understand_image/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAOH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AA4EpE,MAAM,CAAC,OAAO,UAAU,2BAA2B,CAAC,EAAE,EAAE,YAAY,QAoEnE"}

package/dist/extensions/understand_image/index.js

@@ -0,0 +1,159 @@
+/**
+ * pi-mind Image Understanding Extension
+ *
+ * Provides image understanding via mmx vision CLI.
+ * Also intercepts base64 images in prompts and saves them to temp files,
+ * so the agent can reference them when MiniMax doesn't receive images directly.
+ */
+import * as fs from "fs";
+import * as path from "path";
+import * as os from "os";
+import { Type } from "@sinclair/typebox";
+import { spawn } from "child_process";
+const MMX = "mmx";
+/** Tracks temp image files created by this extension (path -> true) */
+const tempFiles = new Map();
+/** Reads MINIMAX_API_KEY: env var preferred, fallback to disk */
+function getMinimaxKey() {
+    if (process.env.MINIMAX_API_KEY)
+        return process.env.MINIMAX_API_KEY;
+    if (process.env.MINIMAX_CN_API_KEY)
+        return process.env.MINIMAX_CN_API_KEY;
+    const mmxConfig = path.join(process.env.HOME || "", ".mmx", "config.json");
+    if (fs.existsSync(mmxConfig)) {
+        try {
+            return JSON.parse(fs.readFileSync(mmxConfig, "utf-8")).api_key;
+        }
+        catch { }
+    }
+    return undefined;
+}
+function ok(text) {
+    return { content: [{ type: "text", text }], details: {} };
+}
+function err(text) {
+    return { content: [{ type: "text", text }], details: {}, isError: true };
+}
+function execMmx(args, timeoutMs = 30000) {
+    return new Promise((resolve, reject) => {
+        const key = getMinimaxKey();
+        const env = {};
+        for (const [k, v] of Object.entries(process.env)) {
+            if (v !== undefined)
+                env[k] = v;
+        }
+        if (key)
+            env.MINIMAX_API_KEY = key;
+        const proc = spawn(MMX, args, { timeout: timeoutMs, env });
+        let stdout = "";
+        let stderr = "";
+        proc.stdout?.on("data", (d) => { stdout += d.toString(); });
+        proc.stderr?.on("data", (d) => { stderr += d.toString(); });
+        proc.on("close", (code) => {
+            if (code === 0)
+                resolve(stdout);
+            else
+                reject(new Error(stderr || stdout || `mmx exited with code ${code}`));
+        });
+        proc.on("error", reject);
+    });
+}
+function resolveImageToPath(imageInput) {
+    // URL or absolute path — use as-is
+    if (imageInput.startsWith("http://") || imageInput.startsWith("https://") || imageInput.startsWith("/")) {
+        return imageInput;
+    }
+    // Base64 data URL: data:image/png;base64,xxxx
+    if (imageInput.startsWith("data:")) {
+        const match = imageInput.match(/^data:([^;]+);base64,(.+)$/);
+        if (!match)
+            throw new Error(`Invalid base64 data URL`);
+        const mimeType = match[1];
+        const data = match[2];
+        const ext = mimeType.split("/")[1] || "png";
+        const tmpFile = path.join(os.tmpdir(), `img-${Date.now()}-${Math.random().toString(36).slice(2)}.${ext}`);
+        fs.writeFileSync(tmpFile, Buffer.from(data, "base64"));
+        tempFiles.set(tmpFile, true);
+        return tmpFile;
+    }
+    // Plain base64 (no prefix) — treat as PNG
+    const tmpFile = path.join(os.tmpdir(), `img-${Date.now()}-${Math.random().toString(36).slice(2)}.png`);
+    fs.writeFileSync(tmpFile, Buffer.from(imageInput, "base64"));
+    tempFiles.set(tmpFile, true);
+    return tmpFile;
+}
+const UnderstandImageParams = Type.Object({
+    image: Type.String({ description: "Image file path, URL, or base64 data URL (data:image/png;base64,...)" }),
+    prompt: Type.Optional(Type.String({ description: "Question about the image (default: 'Describe the image.')" })),
+});
+export default function imageUnderstandingExtension(pi) {
+    // Intercept base64 images before LLM call — save to temp files
+    // so the agent can use understand_image with file paths
+    pi.on("before_agent_start", async (event) => {
+        if (!event.images || event.images.length === 0)
+            return;
+        const imageDescs = [];
+        for (const img of event.images) {
+            if (img.type === "image" && img.data) {
+                try {
+                    const filePath = resolveImageToPath(img.data);
+                    imageDescs.push(`[图片已保存: ${filePath}]`);
+                }
+                catch {
+                    // ignore resolve errors
+                }
+            }
+        }
+        if (imageDescs.length > 0) {
+            const note = `\n\n${imageDescs.join("\n")}`;
+            return {
+                message: {
+                    customType: "image_attachments",
+                    content: `你收到了 ${imageDescs.length} 张图片，已保存为临时文件。你可以使用 understand_image 工具分析这些图片。${note}`,
+                    display: false,
+                },
+            };
+        }
+    });
+    pi.registerTool({
+        name: "understand_image",
+        label: "Understand Image",
+        description: "Understand and analyze images. Supports local file paths, URLs, and base64 data URLs.",
+        parameters: UnderstandImageParams,
+        async execute(_toolCallId, params) {
+            let tmpFile;
+            try {
+                const imagePath = resolveImageToPath(params.image);
+                tmpFile = imagePath.startsWith(os.tmpdir()) ? imagePath : undefined;
+                const args = [
+                    "vision", "describe",
+                    "--image", imagePath,
+                    "--output", "json",
+                ];
+                if (params.prompt) {
+                    args.push("--prompt", params.prompt);
+                }
+                const output = await execMmx(args);
+                let text = output.trim();
+                try {
+                    const json = JSON.parse(text);
+                    text = typeof json === "string" ? json : JSON.stringify(json, null, 2);
+                }
+                catch {
+                    // output is plain text
+                }
+                return ok(text);
+            }
+            catch (e) {
+                return err(`understand_image failed: ${e instanceof Error ? e.message : String(e)}`);
+            }
+            finally {
+                if (tmpFile && fs.existsSync(tmpFile) && tempFiles.has(tmpFile)) {
+                    fs.unlinkSync(tmpFile);
+                    tempFiles.delete(tmpFile);
+                }
+            }
+        },
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/extensions/understand_image/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../extensions/understand_image/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,EAAE,IAAI,EAAe,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AAGtC,MAAM,GAAG,GAAG,KAAK,CAAC;AAElB,uEAAuE;AACvE,MAAM,SAAS,GAAG,IAAI,GAAG,EAAmB,CAAC;AAE7C,iEAAiE;AACjE,SAAS,aAAa;IACpB,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IACpE,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IAC1E,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;IAC3E,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YAAC,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IAClF,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,EAAE,CAAC,IAAY;IACtB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;AACrE,CAAC;AACD,SAAS,GAAG,CAAC,IAAY;IACvB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,IAAa,EAAE,CAAC;AAC7F,CAAC;AAED,SAAS,OAAO,CAAC,IAAc,EAAE,SAAS,GAAG,KAAK;IAChD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;QAC5B,MAAM,GAAG,GAA2B,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACjD,IAAI,CAAC,KAAK,SAAS;gBAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,GAAG;YAAE,GAAG,CAAC,eAAe,GAAG,GAAG,CAAC;QACnC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,KAAK,CAAC;gBAAE,OAAO,CAAC,MAAM,CAAC,CAAC;;gBAC3B,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,IAAI,MAAM,IAAI,wBAAwB,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CAAC,UAAkB;IAC5C,mCAAmC;IACnC,IAAI,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACxG,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,8CAA8C;IAC9C,IAAI,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC7D,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACvD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;QAC1G,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;QACvD,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC7B,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,0CAA0C;IAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IACvG,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC7D,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC7B,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,qBAAqB,GAAG,IAAI,CAAC,MAAM,CAAC;IACxC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,sEAAsE,EAAE,CAAC;IAC3G,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,2DAA2D,EAAE,CAAC,CAAC;CACjH,CAAC,CAAC;AAEH,MAAM,CAAC,OAAO,UAAU,2BAA2B,CAAC,EAAgB;IAClE,+DAA+D;IAC/D,wDAAwD;IACxD,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;QAC1C,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEvD,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YAC/B,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;gBACrC,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBAC9C,UAAU,CAAC,IAAI,CAAC,WAAW,QAAQ,GAAG,CAAC,CAAC;gBAC1C,CAAC;gBAAC,MAAM,CAAC;oBACP,wBAAwB;gBAC1B,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,OAAO;gBACL,OAAO,EAAE;oBACP,UAAU,EAAE,mBAAmB;oBAC/B,OAAO,EAAE,QAAQ,UAAU,CAAC,MAAM,iDAAiD,IAAI,EAAE;oBACzF,OAAO,EAAE,KAAK;iBACf;aACF,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,kBAAkB;QACxB,KAAK,EAAE,kBAAkB;QACzB,WAAW,EAAE,uFAAuF;QACpG,UAAU,EAAE,qBAAqB;QACjC,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,MAA4C;YAC7E,IAAI,OAA2B,CAAC;YAChC,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACnD,OAAO,GAAG,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;gBAEpE,MAAM,IAAI,GAAG;oBACX,QAAQ,EAAE,UAAU;oBACpB,SAAS,EAAE,SAAS;oBACpB,UAAU,EAAE,MAAM;iBACnB,CAAC;gBACF,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBACvC,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;gBACnC,IAAI,IAAI,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;gBACzB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC9B,IAAI,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;gBACzE,CAAC;gBAAC,MAAM,CAAC;oBACP,uBAAuB;gBACzB,CAAC;gBACD,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,GAAG,CAAC,4BAA4B,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACvF,CAAC;oBAAS,CAAC;gBACT,IAAI,OAAO,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAChE,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;oBACvB,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/extensions/understand_image/package.json

@@ -0,0 +1,9 @@
+{
+  "name": "@shog-lab/pi-toolkit-understand-image",
+  "version": "0.1.0",
+  "private": true,
+  "description": "Image understanding via mmx vision CLI.",
+  "type": "module",
+  "main": "index.js",
+  "dependencies": {}
+}

package/dist/extensions/web_search/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * pi-mind Web Search Extension
+ *
+ * Provides web search via mmx search CLI.
+ */
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+export default function webSearchExtension(pi: ExtensionAPI): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/extensions/web_search/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../extensions/web_search/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAiDpE,MAAM,CAAC,OAAO,UAAU,kBAAkB,CAAC,EAAE,EAAE,YAAY,QAuC1D"}

package/dist/extensions/web_search/index.js

@@ -0,0 +1,100 @@
+/**
+ * pi-mind Web Search Extension
+ *
+ * Provides web search via mmx search CLI.
+ */
+import * as fs from "fs";
+import * as path from "path";
+import { Type } from "@sinclair/typebox";
+import { spawn } from "child_process";
+const MMX = "mmx";
+function ok(text) {
+    return { content: [{ type: "text", text }], details: {} };
+}
+function err(text) {
+    return { content: [{ type: "text", text }], details: {}, isError: true };
+}
+/** Reads MINIMAX_API_KEY: env var preferred, fallback to disk */
+function getMinimaxKey() {
+    if (process.env.MINIMAX_API_KEY)
+        return process.env.MINIMAX_API_KEY;
+    if (process.env.MINIMAX_CN_API_KEY)
+        return process.env.MINIMAX_CN_API_KEY;
+    const mmxConfig = path.join(process.env.HOME || "", ".mmx", "config.json");
+    if (fs.existsSync(mmxConfig)) {
+        try {
+            return JSON.parse(fs.readFileSync(mmxConfig, "utf-8")).api_key;
+        }
+        catch { }
+    }
+    return undefined;
+}
+function execMmx(args, timeoutMs = 30000) {
+    return new Promise((resolve, reject) => {
+        const key = getMinimaxKey();
+        const env = {};
+        for (const [k, v] of Object.entries(process.env)) {
+            if (v !== undefined)
+                env[k] = v;
+        }
+        if (key)
+            env.MINIMAX_API_KEY = key;
+        const proc = spawn(MMX, args, { timeout: timeoutMs, env });
+        let stdout = "";
+        let stderr = "";
+        proc.stdout?.on("data", (d) => { stdout += d.toString(); });
+        proc.stderr?.on("data", (d) => { stderr += d.toString(); });
+        proc.on("close", (code) => {
+            if (code === 0)
+                resolve(stdout);
+            else
+                reject(new Error(stderr || stdout || `mmx exited with code ${code}`));
+        });
+        proc.on("error", reject);
+    });
+}
+const SearchParams = Type.Object({
+    query: Type.String({ description: "Search query" }),
+    max_results: Type.Optional(Type.Number({ description: "Maximum number of results (default: 5)" })),
+});
+export default function webSearchExtension(pi) {
+    pi.registerTool({
+        name: "web_search",
+        label: "Web Search",
+        description: "Search the web. Returns titles, URLs, and snippets.",
+        parameters: SearchParams,
+        async execute(_toolCallId, params) {
+            try {
+                const args = [
+                    "search", "query",
+                    "--q", params.query,
+                    "--output", "json",
+                ];
+                const output = await execMmx(args);
+                let text = output.trim();
+                try {
+                    const json = JSON.parse(text);
+                    // mmx returns {organic: [...]} — also handle plain arrays and results wrappers
+                    const results = Array.isArray(json) ? json : (json.organic || json.results || json.data || []);
+                    if (results.length > 0) {
+                        text = results
+                            .slice(0, params.max_results ?? 5)
+                            .map((r, i) => `${i + 1}. ${r.title || ""}\n   ${r.url || r.link || ""}\n   ${r.snippet || r.description || ""}`)
+                            .join("\n\n");
+                    }
+                    else {
+                        text = typeof json === "string" ? json : JSON.stringify(json, null, 2);
+                    }
+                }
+                catch {
+                    // output is plain text
+                }
+                return ok(text);
+            }
+            catch (e) {
+                return err(`web_search failed: ${e instanceof Error ? e.message : String(e)}`);
+            }
+        },
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/extensions/web_search/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../extensions/web_search/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAe,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AAGtC,MAAM,GAAG,GAAG,KAAK,CAAC;AAElB,SAAS,EAAE,CAAC,IAAY;IACtB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;AACrE,CAAC;AACD,SAAS,GAAG,CAAC,IAAY;IACvB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,IAAa,EAAE,CAAC;AAC7F,CAAC;AAED,iEAAiE;AACjE,SAAS,aAAa;IACpB,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IACpE,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB;QAAE,OAAO,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IAC1E,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;IAC3E,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YAAC,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IAClF,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,OAAO,CAAC,IAAc,EAAE,SAAS,GAAG,KAAK;IAChD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;QAC5B,MAAM,GAAG,GAA2B,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACjD,IAAI,CAAC,KAAK,SAAS;gBAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,GAAG;YAAE,GAAG,CAAC,eAAe,GAAG,GAAG,CAAC;QACnC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,KAAK,CAAC;gBAAE,OAAO,CAAC,MAAM,CAAC,CAAC;;gBAC3B,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,IAAI,MAAM,IAAI,wBAAwB,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC;IAC/B,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC;IACnD,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,wCAAwC,EAAE,CAAC,CAAC;CACnG,CAAC,CAAC;AAEH,MAAM,CAAC,OAAO,UAAU,kBAAkB,CAAC,EAAgB;IACzD,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE,YAAY;QACnB,WAAW,EAAE,qDAAqD;QAClE,UAAU,EAAE,YAAY;QACxB,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,MAAmC;YACpE,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG;oBACX,QAAQ,EAAE,OAAO;oBACjB,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,UAAU,EAAE,MAAM;iBACnB,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;gBACnC,IAAI,IAAI,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;gBACzB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC9B,+EAA+E;oBAC/E,MAAM,OAAO,GACX,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;oBACjF,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACvB,IAAI,GAAG,OAAO;6BACX,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,IAAI,CAAC,CAAC;6BACjC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACZ,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,KAAK,IAAI,EAAE,QAAQ,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,IAAI,EAAE,QAAQ,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,WAAW,IAAI,EAAE,EAAE,CAClG;6BACA,IAAI,CAAC,MAAM,CAAC,CAAC;oBAClB,CAAC;yBAAM,CAAC;wBACN,IAAI,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;oBACzE,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,uBAAuB;gBACzB,CAAC;gBACD,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,GAAG,CAAC,sBAAsB,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACjF,CAAC;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}