koishi-plugin-satori-ai 1.0.2 → 1.0.4

package/lib/index.d.ts

@@ -1,6 +1,5 @@
 import { Context, segment, Element, Session, Next, Fragment } from 'koishi';
 import { Sat } from './type';
-export declare const usage = "\u4F7F\u7528\u8BF4\u660E\u89C1\u63D2\u4EF6\u4E3B\u9875";
 type ChatCallback = (session: Session, session_of_id: Sat.Msg[]) => Promise<string>;
 declare module 'koishi' {
     interface Context {

package/lib/index.js

@@ -40,8 +40,7 @@ var require_zh = __commonJS({
 // src/index.tsx
 var src_exports = {};
 __export(src_exports, {
-  default: () => src_default,
-  usage: () => usage
+  default: () => src_default
 });
 module.exports = __toCommonJS(src_exports);
 var import_koishi2 = require("koishi");
@@ -121,10 +120,9 @@ var Sat = class extends import_koishi.Service {
 // src/index.tsx
 var fs = __toESM(require("fs"));
 var path = __toESM(require("path"));
-var usage = `使用说明见插件主页`;
 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");
@@ -234,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;
@@ -242,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, "这是你可能用到的较久之前的对话内容：");
@@ -260,13 +259,13 @@ 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));
             const reSortedMatches = searchKeywordsInFile(commonSenseFilePath, matchContent);
             if (reSortedMatches.length > 0) {
-              this.personality["人格"][0].content += appendTopMatches(commonSenseContent, reSortedMatches, 3, `这是第${i + 1}条信息的补充信息：`);
+              this.personality["人格"][0].content += appendTopMatches(commonSenseContent, reSortedMatches, 3, `这是第${i + 1}条信息的补充信息：`, 1);
             }
           }
         }
@@ -293,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);
@@ -425,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();
     }
@@ -537,8 +531,8 @@ function searchKeywordsInFile(filePath, keywords) {
   return sortedMatches;
 }
 __name(searchKeywordsInFile, "searchKeywordsInFile");
-function appendTopMatches(dialogues, sortedMatches, topN, prefix) {
-  const topMatches = sortedMatches.slice(0, topN);
+function appendTopMatches(dialogues, sortedMatches, topN, prefix, begin = 0) {
+  const topMatches = sortedMatches.slice(begin, topN);
   const personalityContent = `
 ${prefix}{
 ` + topMatches.map((item) => dialogues[item.index].content).join("\n") + "\n}";
@@ -567,7 +561,3 @@ async function getTimeOfDay() {
 }
 __name(getTimeOfDay, "getTimeOfDay");
 var src_default = SAt;
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  usage
-});

package/lib/type.d.ts

@@ -1,4 +1,5 @@
 import { Context, Dict, Schema, Service } from 'koishi';
+export declare const usage = "\u4F7F\u7528\u8BF4\u660E\u89C1\u63D2\u4EF6\u4E3B\u9875";
 export declare class Sat extends Service {
     static inject: {
         required: string[];

package/package.json

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