@cydm/pie 0.1.0 → 1.0.0

package/README.md

@@ -15,7 +15,9 @@ npm install -g @cydm/pie
 设置 API Key：
 
 ```bash
-export KIMI_API_KEY="your-api-key"
+export KIMI_API_KEY="your-kimi-api-key"
+# 或
+export BIGMODEL_API_KEY="your-bigmodel-api-key"
 ```
 
 或创建配置文件 `~/.pie/config.json`：
@@ -23,11 +25,12 @@ export KIMI_API_KEY="your-api-key"
 ```json
 {
   "provider": "kimi-cn",
-  "modelId": "kimi-k2.5",
-  "apiKey": "your-api-key"
+  "modelId": "kimi-k2.5"
 }
 ```
 
+也可以在 `~/.pie/models.json` 的 `providers.<provider>.apiKey` 中为特定 provider 配置 API Key。
+
 ### 命令
 
 #### Print 模式（默认）

package/dist/builtin/extensions/ask-user/index.js

@@ -203,11 +203,11 @@ var TypeBoxError = class extends Error {
 };
 
 // ../../node_modules/@sinclair/typebox/build/esm/type/symbols/symbols.mjs
-var TransformKind = /* @__PURE__ */ Symbol.for("TypeBox.Transform");
-var ReadonlyKind = /* @__PURE__ */ Symbol.for("TypeBox.Readonly");
-var OptionalKind = /* @__PURE__ */ Symbol.for("TypeBox.Optional");
-var Hint = /* @__PURE__ */ Symbol.for("TypeBox.Hint");
-var Kind = /* @__PURE__ */ Symbol.for("TypeBox.Kind");
+var TransformKind = Symbol.for("TypeBox.Transform");
+var ReadonlyKind = Symbol.for("TypeBox.Readonly");
+var OptionalKind = Symbol.for("TypeBox.Optional");
+var Hint = Symbol.for("TypeBox.Hint");
+var Kind = Symbol.for("TypeBox.Kind");
 
 // ../../node_modules/@sinclair/typebox/build/esm/type/guard/kind.mjs
 function IsReadonly(value) {

package/dist/builtin/extensions/document-attachments/index.js

@@ -0,0 +1,144 @@
+import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
+
+// builtin/extensions/document-attachments/index.ts
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { spawnSync } from "node:child_process";
+var MAX_TEXT_CHARS = 24e3;
+var PYTHON_CANDIDATES = [
+  process.env.PYTHON,
+  "/opt/anaconda3/bin/python3",
+  "python3"
+].filter((value) => Boolean(value));
+function isTextCapable(model) {
+  return (model.input ?? []).includes("text");
+}
+function isSupportedDocument(attachment) {
+  return attachment.mimeType === "application/pdf" || attachment.mimeType === "text/html" || attachment.mimeType === "text/plain" || attachment.mimeType === "text/markdown" || attachment.mimeType === "application/json" || attachment.name.endsWith(".html") || attachment.name.endsWith(".htm") || attachment.name.endsWith(".md") || attachment.name.endsWith(".txt") || attachment.name.endsWith(".json") || attachment.name.endsWith(".pdf");
+}
+function truncateText(text) {
+  const normalized = text.replace(/\r\n/g, "\n").replace(/\n{3,}/g, "\n\n").trim();
+  if (normalized.length <= MAX_TEXT_CHARS) return normalized;
+  return `${normalized.slice(0, MAX_TEXT_CHARS)}
+
+[Truncated after ${MAX_TEXT_CHARS} characters]`;
+}
+function runPythonScript(script, fallbackMessage) {
+  const failures = [];
+  for (const pythonBin of PYTHON_CANDIDATES) {
+    const result = spawnSync(pythonBin, ["-c", script], { encoding: "utf8" });
+    if (result.status === 0) {
+      return result.stdout;
+    }
+    const detail = (result.stderr || result.stdout || `exit ${result.status}`).trim();
+    failures.push(`${pythonBin}: ${detail}`);
+  }
+  throw new Error(`${fallbackMessage}
+${failures.join("\n")}`);
+}
+function extractHtmlText(filePath) {
+  const script = `
+from pathlib import Path
+from html.parser import HTMLParser
+
+class Extractor(HTMLParser):
+    def __init__(self):
+        super().__init__()
+        self.parts = []
+        self.skip_depth = 0
+    def handle_starttag(self, tag, attrs):
+        if tag in ("script", "style"):
+            self.skip_depth += 1
+    def handle_endtag(self, tag):
+        if tag in ("script", "style") and self.skip_depth > 0:
+            self.skip_depth -= 1
+    def handle_data(self, data):
+        if self.skip_depth:
+            return
+        data = data.strip()
+        if data:
+            self.parts.append(data)
+
+p = Path(${JSON.stringify(filePath)})
+parser = Extractor()
+parser.feed(p.read_text("utf-8"))
+print("\\n".join(parser.parts))
+`;
+  return runPythonScript(script, "Failed to extract HTML text");
+}
+function extractPdfText(filePath) {
+  const script = `
+reader = None
+errors = []
+
+for module_name in ("pypdf", "PyPDF2"):
+    try:
+        module = __import__(module_name, fromlist=["PdfReader"])
+        reader = module.PdfReader(${JSON.stringify(filePath)})
+        break
+    except Exception as exc:
+        errors.append(f"{module_name}: {exc}")
+
+if reader is None:
+    raise RuntimeError(" | ".join(errors) or "No supported PDF reader module found")
+
+parts = []
+for page in reader.pages:
+    try:
+        text = page.extract_text() or ""
+    except Exception:
+        text = ""
+    if text.strip():
+        parts.append(text)
+print("\\n\\n".join(parts))
+`;
+  return runPythonScript(script, "Failed to extract PDF text");
+}
+function extractPlainText(filePath) {
+  return fs.readFileSync(filePath, "utf8");
+}
+function extractDocumentText(attachment) {
+  const filePath = attachment.path;
+  if (!filePath || !fs.existsSync(filePath)) {
+    throw new Error(`Local attachment path is unavailable for ${attachment.name}`);
+  }
+  const ext = path.extname(filePath).toLowerCase();
+  if (attachment.mimeType === "application/pdf" || ext === ".pdf") {
+    return extractPdfText(filePath);
+  }
+  if (attachment.mimeType === "text/html" || ext === ".html" || ext === ".htm") {
+    return extractHtmlText(filePath);
+  }
+  return extractPlainText(filePath);
+}
+function documentAttachmentsExtension(ctx) {
+  ctx.log("Document attachment extension loaded");
+  ctx.registerAttachmentHandler({
+    name: "local-document-preprocess",
+    priority: 20,
+    matches(model, attachment) {
+      return isTextCapable(model) && isSupportedDocument(attachment) && Boolean(attachment.path);
+    },
+    async prepare(_model, attachment, prepareCtx) {
+      const extracted = extractDocumentText(attachment);
+      const text = truncateText(extracted);
+      prepareCtx.log(`Document handler extracted ${text.length} chars from ${attachment.name}`);
+      const prefix = [
+        `[Document: ${attachment.name}]`,
+        `MIME: ${attachment.mimeType}`,
+        ""
+      ].join("\n");
+      return {
+        items: [
+          {
+            type: "text",
+            text: `${prefix}${text}`
+          }
+        ]
+      };
+    }
+  });
+}
+export {
+  documentAttachmentsExtension as default
+};

package/dist/builtin/extensions/kimi-attachments/index.js

@@ -0,0 +1,46 @@
+import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
+
+// builtin/extensions/kimi-attachments/index.ts
+function isKimiProvider(provider) {
+  return provider === "kimi-cn" || provider === "kimi-api" || provider === "kimi";
+}
+function getModality(mimeType) {
+  if (mimeType.startsWith("image/")) return "image";
+  if (mimeType.startsWith("video/")) return "video";
+  return "file";
+}
+function kimiAttachmentsExtension(ctx) {
+  ctx.log("Kimi attachment extension loaded");
+  ctx.registerAttachmentHandler({
+    name: "kimi-native-attachments",
+    priority: 100,
+    matches(model, attachment) {
+      return isKimiProvider(model.provider) && Boolean(attachment.fileId) && getModality(attachment.mimeType) !== "file";
+    },
+    async prepare(_model, attachment, prepareCtx) {
+      const fileId = attachment.fileId;
+      if (!fileId) {
+        return {
+          items: [{ type: "text", text: `[${attachment.id}]` }],
+          warnings: ["\u9644\u4EF6\u7F3A\u5C11 fileId\uFF0C\u65E0\u6CD5\u8D70 Kimi \u539F\u751F\u5F15\u7528"]
+        };
+      }
+      const modality = getModality(attachment.mimeType);
+      const url = `ms://${fileId}`;
+      prepareCtx.log(`Kimi handler prepared ${modality} reference for ${attachment.id}`);
+      return {
+        items: [
+          {
+            type: "file-ref",
+            mimeType: attachment.mimeType,
+            url,
+            modality
+          }
+        ]
+      };
+    }
+  });
+}
+export {
+  kimiAttachmentsExtension as default
+};

package/dist/builtin/extensions/questionnaire/index.js

@@ -203,11 +203,11 @@ var TypeBoxError = class extends Error {
 };
 
 // ../../node_modules/@sinclair/typebox/build/esm/type/symbols/symbols.mjs
-var TransformKind = /* @__PURE__ */ Symbol.for("TypeBox.Transform");
-var ReadonlyKind = /* @__PURE__ */ Symbol.for("TypeBox.Readonly");
-var OptionalKind = /* @__PURE__ */ Symbol.for("TypeBox.Optional");
-var Hint = /* @__PURE__ */ Symbol.for("TypeBox.Hint");
-var Kind = /* @__PURE__ */ Symbol.for("TypeBox.Kind");
+var TransformKind = Symbol.for("TypeBox.Transform");
+var ReadonlyKind = Symbol.for("TypeBox.Readonly");
+var OptionalKind = Symbol.for("TypeBox.Optional");
+var Hint = Symbol.for("TypeBox.Hint");
+var Kind = Symbol.for("TypeBox.Kind");
 
 // ../../node_modules/@sinclair/typebox/build/esm/type/guard/kind.mjs
 function IsReadonly(value) {