koishi-plugin-satori-ai 1.0.3 → 1.0.4

package/lib/index.js

@@ -122,7 +122,7 @@ var fs = __toESM(require("fs"));
 var path = __toESM(require("path"));
 var name = "satori-ai";
 var logger = new import_koishi2.Logger(name);
-var debug = 0;
+var debug = 1;
 var SAt = class extends Sat {
   static {
     __name(this, "SAt");
@@ -232,7 +232,7 @@ var SAt = class extends Sat {
         return session.text("commands.sat.messages.duplicate-dialogue");
       }
       this.personality["人格"][0].content += "\n这是刚刚的对话内容：{\n" + recentDialogues.map((msg) => `${msg.role}: ${msg.content}`).join("\n") + "\n}";
-      const charactersToRemove = ["的", "一", "是", "了", "什", "么", "我", "谁", "不", "人", "在", "他", "有", "这", "个", "上", "们", "来", "到", "时", "大", "地", "为", "子", "中", "你", "说", "生", "国", "年", "着", "就", "那", "和", "要", "她", "出", "也", "得", "里", "后", "自", "以", "会"];
+      const charactersToRemove = ["的", "一", "是", "了", "什", "么", "我", "谁", "不", "人", "在", "他", "有", "这", "个", "上", "们", "来", "到", "时", "大", "地", "为", "子", "中", "你", "说", "生", "国", "年", "着", "就", "那", "和", "要", "她", "出", "也", "得", "里", "后", "自", "以", "会", "id=", "1", "2", "3", "4", "5", "6", "7", "8", "9", "0"];
       const filePath = path.join(this.pluginConfig.dataDir, "dialogues", `${session.userId}.txt`);
       const notExists = await isTargetIdExists(this.ctx, session.userId);
       let tmp;
@@ -240,10 +240,11 @@ var SAt = class extends Sat {
         const user = await this.ctx.database.get("p_system", { userid: session.userId });
         tmp = (user[0].usersname + prompt).split("");
       } else {
-        tmp = prompt.split("");
+        tmp = (session.username + prompt).split("");
       }
       const keywords = tmp.filter((word) => !charactersToRemove.includes(word));
       const fs2 = require("fs");
+      const USERID = [session.userId];
       let sortedMatches = searchKeywordsInFile(filePath, keywords);
       if (sortedMatches.length > 0) {
         this.personality["人格"][0].content += appendTopMatches(JSON.parse(fs2.readFileSync(filePath, "utf-8")), sortedMatches, 10, "这是你可能用到的较久之前的对话内容：");
@@ -258,7 +259,7 @@ var SAt = class extends Sat {
       }
       sortedMatches = searchKeywordsInFile(commonSenseFilePath, keywords);
       if (sortedMatches.length > 0) {
-        this.personality["人格"][0].content += appendTopMatches(commonSenseContent, sortedMatches, 5, "这是你需要知道的信息：");
+        this.personality["人格"][0].content += appendTopMatches(commonSenseContent, sortedMatches, 10, "这是你需要知道的信息：");
         if (sortedMatches.length > 4) {
           for (let i = 0; i < 5; i++) {
             const matchContent = commonSenseContent[sortedMatches[i].index].content.split("").filter((word) => !charactersToRemove.includes(word));
@@ -291,44 +292,40 @@ var SAt = class extends Sat {
             level = levels[0];
             this.personality["人格"][0].content += `
  ${this.pluginConfig.prompt_0} 
-我的名字: ${user[0].usersname}`;
+`;
           } else if (favorability < this.pluginConfig.favorability_div_2) {
             level = levels[1];
             this.personality["人格"][0].content += `
  ${this.pluginConfig.prompt_1} 
-我的名字: ${user[0].usersname}`;
+`;
           } else if (favorability < this.pluginConfig.favorability_div_3) {
             level = levels[2];
             this.personality["人格"][0].content += `
  ${this.pluginConfig.prompt_2} 
-我的名字: ${user[0].usersname}`;
+`;
           } else if (favorability < this.pluginConfig.favorability_div_4) {
             level = levels[3];
             this.personality["人格"][0].content += `
  ${this.pluginConfig.prompt_3} 
-我的名字: ${user[0].usersname}`;
+`;
           } else {
             level = levels[4];
-            const isPrivate = session.channelId.match(new RegExp("private", "g"))?.includes("private");
-            if (isPrivate) {
-              this.personality["人格"][0].content += `
+            this.personality["人格"][0].content += `
  ${this.pluginConfig.prompt_4} 
 `;
-            } else {
-              this.personality["人格"][0].content += `
- ${this.pluginConfig.prompt_3} 
-`;
-            }
-            this.personality["人格"][0].content += `我的名字: ${user[0].usersname}`;
           }
-          logger.info(`名字: ${user[0].usersname}, 关系: ${level}`);
+          this.personality["人格"][0].content += `发言者的名字: ${session.username}，发言者的id：${session.userId}`;
+          logger.info(`名字: ${session.username}, 关系: ${level}`);
         } else {
           this.personality["人格"][0].content += `
  ${this.pluginConfig.prompt_1} 
-我的名字: ${session.username}`;
+发言者的名字: ${session.username}，发言者的id：${session.userId}`;
           logger.info(`名字: ${session.username}, 关系: 陌生`);
         }
       }
+      sortedMatches = searchKeywordsInFile(commonSenseFilePath, USERID);
+      if (sortedMatches.length > 0)
+        this.personality["人格"][0].content += appendTopMatches(commonSenseContent, sortedMatches, 1, "这是发言者的真实身份：");
       if (debug)
         logger.info(this.personality["人格"][0].content.slice(-1500));
       return await this.chat(censored_prompt, session.userId, session);
@@ -423,7 +420,6 @@ var SAt = class extends Sat {
       this.channelDialogues[channelId] = [];
     }
     this.channelDialogues[channelId].push({ "role": session.username, "content": msg });
-    this.channelDialogues[channelId].push({ "role": "回复", "content": message });
     if (this.channelDialogues[channelId].length > this.pluginConfig.message_max_length) {
       this.channelDialogues[channelId].shift();
     }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-satori-ai",
-  "description": "适用于satori的ai对话，请务必使用最新版本",
-  "version": "1.0.3",
+  "description": "适用于satori的ai对话，具有长期记忆、常识写入和好感度功能，请务必使用最新版本",
+  "version": "1.0.4",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "types": "lib/index.d.ts",