koishi-plugin-chat-analyse 1.7.6 → 1.7.8

package/lib/Analyse.d.ts

@@ -13,7 +13,7 @@ export declare class Analyse {
     private ctx;
     private config;
     private renderer;
-    private readonly jieba;
+    private readonly segmentit;
     constructor(ctx: Context, config: Config);
     /**
      * @public @method registerCommands

package/lib/index.js

@@ -2417,8 +2417,7 @@ ${commandOutput}`;
 
 // src/Analyse.ts
 var import_koishi6 = require("koishi");
-var import_jieba = require("@node-rs/jieba");
-var import_dict = require("@node-rs/jieba/dict");
+var import_segmentit = require("segmentit");
 function cosineSimilarity(vecA, vecB) {
   let dotProduct = 0;
   let magA = 0;
@@ -2439,7 +2438,9 @@ var Analyse = class {
     this.ctx = ctx;
     this.config = config;
     this.renderer = new Renderer(ctx);
-    if (config.enableWordCloud) this.jieba = import_jieba.Jieba.withDict(import_dict.dict);
+    if (config.enableWordCloud) {
+      this.segmentit = (0, import_segmentit.useDefault)(new import_segmentit.Segment());
+    }
     if (this.config.enableOriRecord && this.config.cacheRetentionDays > 0) {
       this.ctx.cron("0 0 * * *", async () => {
         const cutoffDate = new Date(Date.now() - this.config.cacheRetentionDays * import_koishi6.Time.day);
@@ -2451,7 +2452,7 @@ var Analyse = class {
     __name(this, "Analyse");
   }
   renderer;
-  jieba = null;
+  segmentit = null;
   /**
    * @public @method registerCommands
    * @description 在主命令下注册子命令。
@@ -2461,7 +2462,7 @@ var Analyse = class {
     if (this.config.enableWordCloud) {
       cmd.subcommand("wordcloud", "生成词云").usage("基于聊天记录生成词云图，可指定范围，默认当前群组。").option("guild", "-g <guildId:string> 指定群组").option("user", "-u <user:string> 指定用户").option("hours", "-t <hours:number> 指定时长", { fallback: 24 }).action(async ({ session, options }) => {
         try {
-          if (!this.jieba) return "Jieba 分词服务未就绪";
+          if (!this.segmentit) return "分词服务未就绪";
           const scope = await parseQueryScope(this.ctx, session, options);
           if (scope.error) return scope.error;
           scope.uids ??= (await this.ctx.database.get("analyse_user", {}, ["uid"])).map((u) => u.uid);
@@ -2472,7 +2473,7 @@ var Analyse = class {
           const excludeWords = new Set(this.config.excludeWords.split(",").map((w) => w.trim().toLowerCase()).filter(Boolean));
           const exclusionRegex = /\[(face|file|forward|img|gif|audio|video|json|rps|markdown|dice|at:.*?)\]/g;
           const allText = records.map((r) => r.content.replace(exclusionRegex, "")).join(" ");
-          const words = this.jieba.cut(allText).filter((w) => {
+          const words = this.segmentit.doSegment(allText).map((res) => String(res.w)).filter((w) => {
             const trimmedWord = w.trim();
             if (trimmedWord.length <= 1) return false;
             if (/^\d+$/.test(trimmedWord)) return false;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chat-analyse",
   "description": "强大而全面的聊天数据分析插件。支持多维度统计（命令、发言、消息类型、活跃度），可生成发言排行、词云图，并提供完善的数据管理和 AI 分析功能。",
-  "version": "1.7.6",
+  "version": "1.7.8",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],
@@ -38,6 +38,6 @@
     "koishi": "^4.18.8"
   },
   "dependencies": {
-    "@node-rs/jieba": "^2.0.1"
+    "segmentit": "^2.0.3"
   }
 }