koishi-plugin-dedide-ercorrection 0.0.0-stage → 0.1.1

package/lib/index.d.ts

@@ -0,0 +1,9 @@
+import { Schema, Context } from 'koishi';
+export declare const name = "dedide-ercorrection";
+export declare const inject: any[];
+export interface Config {
+    maxLength?: number;
+    verbose?: boolean;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,153 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name2 in all)
+    __defProp(target, name2, { get: all[name2], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  inject: () => inject,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+var import_de_corrector = require("@azulores/de-corrector");
+var name = "dedide-ercorrection";
+var inject = [];
+var Config = import_koishi.Schema.object({
+  maxLength: import_koishi.Schema.number().description("\u8D85\u8FC7\u6B64\u957F\u5EA6\u7684\u6D88\u606F\u4E0D\u5904\u7406\uFF080=\u4E0D\u9650\u5236\uFF09").default(500).min(0),
+  verbose: import_koishi.Schema.boolean().description("\u8BB0\u5F55\u6BCF\u6B21\u4FEE\u6539").default(false)
+});
+var PLACEHOLDER_RE = /(\s|<at[^>]*\/?>|\[CQ:[^\]]+\])/g;
+function extractPlaceholders(text) {
+  const items = [];
+  let clean = "";
+  let last = 0;
+  for (const m of text.matchAll(PLACEHOLDER_RE)) {
+    clean += text.slice(last, m.index);
+    items.push({ content: m[0], pos: Array.from(clean).length });
+    last = m.index + m[0].length;
+  }
+  clean += text.slice(last);
+  return { clean, items };
+}
+function restorePlaceholders(text, items) {
+  const chars = Array.from(text);
+  for (const { content, pos } of [...items].sort((a, b) => b.pos - a.pos)) {
+    chars.splice(Math.min(pos, chars.length), 0, content);
+  }
+  return chars.join("");
+}
+function contentToText(content) {
+  if (typeof content === "string")
+    return content;
+  if (Array.isArray(content)) {
+    return content.map((p) => typeof p === "string" ? p : p?.text ?? "").join("");
+  }
+  return "";
+}
+function apply(ctx, config) {
+  const logger = ctx.logger("dedide-ercorrection");
+  const maxLength = config.maxLength ?? 500;
+  const corrector = new import_de_corrector.DeCorrector({ maxLength });
+  let ready = false;
+  corrector.warmup().then(
+    () => {
+      ready = true;
+      logger.info("model ready (local ONNX, offline)");
+    },
+    (e) => logger.warn("model load failed: %s", e?.message ?? e)
+  );
+  const stats = {
+    messages: 0,
+    // 含 的地得 且通过长度检查的消息数
+    fixed: 0,
+    // 实际改动的消息数
+    skipped: 0,
+    // 检查后无需改动
+    errors: 0
+    // 处理异常
+  };
+  const dumpStats = () => {
+    logger.info(
+      "[stats] msgs=%d fixed=%d skipped=%d err=%d",
+      stats.messages,
+      stats.fixed,
+      stats.skipped,
+      stats.errors
+    );
+  };
+  ctx.setInterval(dumpStats, 6e4);
+  ctx.on("dispose", dumpStats);
+  async function correctMessage(content) {
+    if (!content || content.length <= 2)
+      return null;
+    if (!/[的地得]/.test(content))
+      return null;
+    if (maxLength > 0 && Array.from(content).length > maxLength)
+      return null;
+    if (!ready)
+      return null;
+    stats.messages++;
+    const { clean, items } = extractPlaceholders(content);
+    if (!clean)
+      return null;
+    let fixed;
+    try {
+      fixed = await corrector.fix(clean);
+    } catch (e) {
+      stats.errors++;
+      logger.warn("correction failed: %s", e?.message ?? e);
+      return null;
+    }
+    if (fixed === clean) {
+      stats.skipped++;
+      return null;
+    }
+    const result = restorePlaceholders(fixed, items);
+    if (Array.from(result).length !== Array.from(content).length) {
+      stats.errors++;
+      logger.warn("[guard] length mismatch, dropped: %j -> %j", content, result);
+      return null;
+    }
+    if (result === content) {
+      stats.skipped++;
+      return null;
+    }
+    stats.fixed++;
+    if (config.verbose)
+      logger.info("[fix] %j -> %j", content, result);
+    return result;
+  }
+  const root = ctx.root ?? ctx;
+  ctx.dedideErcCorrect = correctMessage;
+  root["dedideErcCorrect"] = correctMessage;
+  ctx.dedideErcReady = () => ready;
+  root["dedideErcReady"] = () => ready;
+  ctx.middleware(async (session, next) => {
+    const result = await correctMessage(contentToText(session.content));
+    return result ?? next();
+  }, true);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  inject,
+  name
+});

package/package.json

@@ -1,6 +1,29 @@
 {
   "name": "koishi-plugin-dedide-ercorrection",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "description": "中文「的/地/得」混用纠错 for Koishi：本地 ONNX 模型，无需联网、无需 LLM，@人/表情占位符保留",
+  "version": "0.1.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib"
+  ],
+  "license": "MIT",
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin",
+    "chinese",
+    "nlp",
+    "的地得",
+    "correction"
+  ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json && esbuild src/index.ts --bundle --platform=node --format=cjs --target=node20 --external:koishi --external:@azulores/de-corrector --outfile=lib/index.js"
+  },
+  "dependencies": {
+    "@azulores/de-corrector": "^0.1.1"
+  },
+  "peerDependencies": {
+    "koishi": "^4.18.11"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-dedide-ercorrection
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-dedide-ercorrection?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-dedide-ercorrection)
+
+

package/README.md

@@ -1,3 +0,0 @@
-# Temporary Holding Version
-
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.