@shog-lab/pi-toolkit 0.1.1 → 0.1.2

package/README.md

@@ -8,13 +8,12 @@ A drop-in package adding several commonly-used external tool integrations to any
 
 | Extension | Tool name(s) | Backend | Required env |
 |---|---|---|---|
-| `jimeng` | `jimeng_generate` | Volcengine Jimeng T2I API | `JIMENG_ACCESS_KEY`, `JIMENG_SECRET_KEY` |
 | `web_search` | `web_search` | mmx CLI | (mmx config) |
 | `understand_image` | `understand_image` | mmx vision CLI | (mmx config) |
 | `mcp-bridge` | `<server>_<tool>` per MCP server | Any MCP server | `mcp-servers.json` config |
 | `subagent` | `spawn_subagent` | child pi process | (none) |
 
-Each extension silently skips registration if its required env / config is missing — install pi-toolkit even if you only use some.
+`mcp-bridge` silently skips registration when no `mcp-servers.json` exists, so install pi-toolkit even if you only use some extensions.
 
 Plus `agent-browser` CLI is shipped as a dependency. Its SKILL.md is symlinked from upstream so the agent knows how to use it via Bash.
 
@@ -28,13 +27,6 @@ npm i -D @shog-lab/pi-toolkit
 
 ## Configure
 
-### Jimeng (image generation)
-
-```bash
-export JIMENG_ACCESS_KEY="AKLT..."
-export JIMENG_SECRET_KEY="..."
-```
-
 ### MCP servers (figma, filesystem, etc.)
 
 Create `mcp-servers.json` (or `.pi/mcp-servers.json`) at the host repo root:

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@shog-lab/pi-toolkit",
-  "version": "0.1.1",
-  "description": "Common pi-coding-agent extensions and CLI tools: image generation (Jimeng), web search and image understanding (mmx), browser automation (agent-browser).",
+  "version": "0.1.2",
+  "description": "Common pi-coding-agent extensions: web search + image understanding (mmx), MCP server bridge, sub-agent spawn, browser automation (agent-browser).",
   "type": "module",
   "license": "MIT",
   "scripts": {
@@ -25,7 +25,6 @@
   },
   "dependencies": {
     "@sinclair/typebox": "^0.34.0",
-    "@volcengine/openapi": "^1.0.0",
     "agent-browser": "^0.27.0",
     "@shog-lab/pi-utils": "*"
   },

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

@@ -1,11 +0,0 @@
-/**
- * Jimeng (即梦) Image Generation Extension for pi-coding-agent
- *
- * Registers a `jimeng_generate` tool that generates images via
- * Volcengine's Jimeng T2I API (text-to-image).
- *
- * Requires JIMENG_ACCESS_KEY and JIMENG_SECRET_KEY environment variables.
- */
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-export default function jimengExtension(pi: ExtensionAPI): void;
-//# sourceMappingURL=index.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../extensions/jimeng/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAwDpE,MAAM,CAAC,OAAO,UAAU,eAAe,CAAC,EAAE,EAAE,YAAY,QA6GvD"}

package/dist/extensions/jimeng/index.js

@@ -1,130 +0,0 @@
-/**
- * Jimeng (即梦) Image Generation Extension for pi-coding-agent
- *
- * Registers a `jimeng_generate` tool that generates images via
- * Volcengine's Jimeng T2I API (text-to-image).
- *
- * Requires JIMENG_ACCESS_KEY and JIMENG_SECRET_KEY environment variables.
- */
-import { Type } from "@sinclair/typebox";
-import { Signer } from "@volcengine/openapi";
-import { mkdirSync, writeFileSync } from "node:fs";
-import { dirname, join } from "node:path";
-const API_HOST = "visual.volcengineapi.com";
-const API_VERSION = "2022-08-31";
-const MODEL_KEY = "jimeng_t2i_v40";
-const MAX_POLL_ATTEMPTS = 60;
-const POLL_INTERVAL_MS = 2000;
-function signRequest(accessKey, secretKey, action, body) {
-    const requestData = {
-        method: "POST",
-        region: "cn-north-1",
-        params: { Action: action, Version: API_VERSION },
-        headers: {
-            "Content-Type": "application/json",
-            Host: API_HOST,
-        },
-        body,
-    };
-    const signer = new Signer(requestData, "cv");
-    signer.addAuthorization({ accessKeyId: accessKey, secretKey });
-    return requestData.headers;
-}
-async function apiCall(accessKey, secretKey, action, body) {
-    const headers = signRequest(accessKey, secretKey, action, body);
-    const url = `https://${API_HOST}/?Action=${action}&Version=${API_VERSION}`;
-    const res = await fetch(url, { method: "POST", headers, body });
-    if (!res.ok)
-        throw new Error(`Jimeng API error: ${res.status}`);
-    return (await res.json());
-}
-export default function jimengExtension(pi) {
-    const accessKey = process.env.JIMENG_ACCESS_KEY;
-    const secretKey = process.env.JIMENG_SECRET_KEY;
-    if (!accessKey || !secretKey) {
-        // Silently skip if not configured
-        return;
-    }
-    pi.registerTool({
-        name: "jimeng_generate",
-        label: "Jimeng Image",
-        description: "Generate an image from a text prompt using Jimeng (即梦) AI. Returns the file path of the generated image.",
-        promptSnippet: "Generate AI images from text descriptions (Chinese prompts work best)",
-        parameters: Type.Object({
-            prompt: Type.String({ description: "Image description (Chinese recommended for best results)" }),
-            width: Type.Optional(Type.Number({ description: "Image width in pixels (default: 1024)" })),
-            height: Type.Optional(Type.Number({ description: "Image height in pixels (default: 1024)" })),
-            filename: Type.Optional(Type.String({ description: "Output filename (default: jimeng-{timestamp}.png)" })),
-        }),
-        async execute(_toolCallId, params, signal, onUpdate) {
-            const width = params.width || 1024;
-            const height = params.height || 1024;
-            const filename = params.filename || `jimeng-${Date.now()}.png`;
-            const outputPath = join(process.env.PI_MIND_DIR || (process.cwd() + "/.pi-mind"), filename);
-            // Step 1: Submit task
-            onUpdate?.({ content: [{ type: "text", text: "提交生成任务..." }] });
-            const submitBody = JSON.stringify({
-                req_key: MODEL_KEY,
-                prompt: params.prompt,
-                width,
-                height,
-                scale: 0.5,
-                force_single: true,
-            });
-            const submitResult = await apiCall(accessKey, secretKey, "CVSync2AsyncSubmitTask", submitBody);
-            if (submitResult.code !== 10000) {
-                return { content: [{ type: "text", text: `生成失败: ${submitResult.message || "Unknown error"}` }] };
-            }
-            const taskId = submitResult.data?.task_id;
-            if (!taskId) {
-                return { content: [{ type: "text", text: "生成失败: 未返回 task_id" }] };
-            }
-            // Step 2: Poll for result
-            for (let attempt = 0; attempt < MAX_POLL_ATTEMPTS; attempt++) {
-                if (signal?.aborted) {
-                    return { content: [{ type: "text", text: "已取消" }] };
-                }
-                const queryBody = JSON.stringify({
-                    req_key: MODEL_KEY,
-                    task_id: taskId,
-                    req_json: JSON.stringify({ return_url: true, logo_info: { add_logo: false } }),
-                });
-                const result = await apiCall(accessKey, secretKey, "CVSync2AsyncGetResult", queryBody);
-                if (result.code === 10000) {
-                    const status = result.data?.status;
-                    if (status === "done") {
-                        const urls = result.data?.image_urls;
-                        if (!urls?.length) {
-                            return { content: [{ type: "text", text: "生成完成但未返回图片 URL" }] };
-                        }
-                        // Step 3: Download image
-                        onUpdate?.({ content: [{ type: "text", text: "下载图片..." }] });
-                        const imgRes = await fetch(urls[0]);
-                        if (!imgRes.ok) {
-                            return { content: [{ type: "text", text: `图片下载失败: ${imgRes.status}` }] };
-                        }
-                        const buffer = Buffer.from(await imgRes.arrayBuffer());
-                        mkdirSync(dirname(outputPath), { recursive: true });
-                        writeFileSync(outputPath, buffer);
-                        return {
-                            content: [{ type: "text", text: `图片已生成并保存到 ${outputPath} (${width}x${height}, ${buffer.length} bytes)。\n图片URL: ${urls[0]}\n请使用 send_image 工具将图片发送给用户。` }],
-                        };
-                    }
-                    if (status === "not_found" || status === "expired") {
-                        return { content: [{ type: "text", text: `生成失败: 任务 ${status}` }] };
-                    }
-                    // Still generating
-                    if (attempt % 5 === 0) {
-                        onUpdate?.({ content: [{ type: "text", text: `生成中... (${attempt * 2}s)` }] });
-                    }
-                }
-                else {
-                    return { content: [{ type: "text", text: `查询失败: ${result.message || "Unknown error"}` }] };
-                }
-                await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
-            }
-            return { content: [{ type: "text", text: "生成超时（120s）" }] };
-        },
-    });
-}
-//# sourceMappingURL=index.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../extensions/jimeng/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAE1C,MAAM,QAAQ,GAAG,0BAA0B,CAAC;AAC5C,MAAM,WAAW,GAAG,YAAY,CAAC;AACjC,MAAM,SAAS,GAAG,gBAAgB,CAAC;AACnC,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAY9B,SAAS,WAAW,CAClB,SAAiB,EACjB,SAAiB,EACjB,MAAc,EACd,IAAY;IAEZ,MAAM,WAAW,GAAG;QAClB,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,YAAY;QACpB,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE;QAChD,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,IAAI,EAAE,QAAQ;SACW;QAC3B,IAAI;KACL,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC7C,MAAM,CAAC,gBAAgB,CAAC,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAC/D,OAAO,WAAW,CAAC,OAAO,CAAC;AAC7B,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,SAAiB,EACjB,SAAiB,EACjB,MAAc,EACd,IAAY;IAEZ,MAAM,OAAO,GAAG,WAAW,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IAChE,MAAM,GAAG,GAAG,WAAW,QAAQ,YAAY,MAAM,YAAY,WAAW,EAAE,CAAC;IAC3E,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAChE,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;IAChE,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuB,CAAC;AAClD,CAAC;AAED,MAAM,CAAC,OAAO,UAAU,eAAe,CAAC,EAAgB;IACtD,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IAChD,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IAEhD,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,EAAE,CAAC;QAC7B,kCAAkC;QAClC,OAAO;IACT,CAAC;IAED,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,iBAAiB;QACvB,KAAK,EAAE,cAAc;QACrB,WAAW,EACT,0GAA0G;QAC5G,aAAa,EACX,uEAAuE;QACzE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;YACtB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,0DAA0D,EAAE,CAAC;YAChG,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,uCAAuC,EAAE,CAAC,CAAC;YAC3F,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,wCAAwC,EAAE,CAAC,CAAC;YAC7F,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,mDAAmD,EAAE,CAAC,CAAC;SAC3G,CAAC;QAEF,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,MAAW,EAAE,MAAW,EAAE,QAAa;YACxE,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC;YACnC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC;YACrC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,UAAU,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;YAC/D,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,EAAE,QAAQ,CAAC,CAAC;YAE5F,sBAAsB;YACtB,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC;gBAChC,OAAO,EAAE,SAAS;gBAClB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,KAAK;gBACL,MAAM;gBACN,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,wBAAwB,EAAE,UAAU,CAAC,CAAC;YAC/F,IAAI,YAAY,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;gBAChC,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,YAAY,CAAC,OAAO,IAAI,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC;YACnG,CAAC;YAED,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC;YAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,EAAE,CAAC;YACpE,CAAC;YAED,0BAA0B;YAC1B,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,iBAAiB,EAAE,OAAO,EAAE,EAAE,CAAC;gBAC7D,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBACpB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;gBACtD,CAAC;gBAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC/B,OAAO,EAAE,SAAS;oBAClB,OAAO,EAAE,MAAM;oBACf,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,CAAC;iBAC/E,CAAC,CAAC;gBAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,uBAAuB,EAAE,SAAS,CAAC,CAAC;gBAEvF,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;oBAC1B,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC;oBAEnC,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;wBACtB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC;wBACrC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;4BAClB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC;wBACjE,CAAC;wBAED,yBAAyB;wBACzB,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;wBAE7D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;wBACpC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;4BACf,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;wBAC3E,CAAC;wBAED,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;wBACvD,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;wBACpD,aAAa,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;wBAElC,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,UAAU,KAAK,KAAK,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,oBAAoB,IAAI,CAAC,CAAC,CAAC,8BAA8B,EAAE,CAAC;yBAC1J,CAAC;oBACJ,CAAC;oBAED,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;wBACnD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;oBACrE,CAAC;oBAED,mBAAmB;oBACnB,IAAI,OAAO,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBACtB,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,OAAO,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;oBAChF,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,MAAM,CAAC,OAAO,IAAI,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC7F,CAAC;gBAED,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC;YAC5D,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;QAC7D,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/extensions/jimeng/package.json

@@ -1,12 +0,0 @@
-{
-  "name": "@shog-lab/pi-toolkit-jimeng",
-  "version": "0.1.0",
-  "private": true,
-  "description": "Jimeng (即梦) image generation via Volcengine API.",
-  "type": "module",
-  "main": "index.js",
-  "dependencies": {
-    "@volcengine/openapi": "^1.0.0",
-    "@sinclair/typebox": "^0.34.0"
-  }
-}