koishi-plugin-satori-ai 1.2.2 → 1.2.5

package/lib/index.js

@@ -228,6 +228,8 @@ var APIClient = class {
   }
   // 发送生成用户画像请求
   async generateUserPortrait(user, messages) {
+    if (user.userid == "Alice")
+      return { content: "测试画像", error: false };
     const enableUserKey = user?.items?.["地灵殿通行证"]?.description && user.items["地灵殿通行证"].description == "on";
     let keys;
     let modle;
@@ -307,7 +309,7 @@ var APIClient = class {
         const responseMsg = { role: "assistant", content };
         if (payload.messages.some((msg) => msg === responseMsg) && content.length > 5) {
           const lastMsg = payload.messages[payload.messages.length - 1];
-          payload.messages[payload.messages.length - 1].content = lastMsg.content + "(system:注意不要重复之前的内容)";
+          payload.messages[payload.messages.length - 1].content = lastMsg.content + "(注意不要重复之前的内容)";
           logger2.warn(`返回内容与之前内容相同，重试(第${i}次)中...`);
           continue;
         }
@@ -1094,6 +1096,7 @@ var Sat = class extends import_koishi5.Service {
       visible_favorability: import_koishi5.Schema.boolean().default(true).description("是否开启好感度升降显示"),
       enable_user_portrait: import_koishi5.Schema.boolean().default(true).description("是否启用用户画像功能"),
       portrait_usage: import_koishi5.Schema.number().default(2).description("每天触发第几次对话后触发画像生成（最小为2）"),
+      max_portrait_dialogues: import_koishi5.Schema.number().default(100).description("画像生成获取的最大对话条数"),
       portrait_min_favorability: import_koishi5.Schema.number().default(100).description("触发画像生成的最小好感度"),
       prompt_0: import_koishi5.Schema.string().role("textarea").description("厌恶好感补充设定"),
       favorability_div_1: import_koishi5.Schema.number().default(15).description("厌恶-陌生分界线"),
@@ -1150,9 +1153,9 @@ var UserPortraitManager = class {
     const MemoryContent = await JSON.parse(fs5.readFileSync(memoryPath, "utf-8"));
     const dialoguesContent = MemoryContent.map((msg) => `${msg.content}${msg.role == "user" ? "(未记录时间)" : msg.role}`);
     const level = user.userlevel < 5 ? user.userlevel : 4;
-    const usageLimit = this.config.max_usage[level] || 0;
-    if (dialoguesContent.length > usageLimit) {
-      dialoguesContent.splice(0, dialoguesContent.length - usageLimit);
+    const usageLimit = this.config.max_usage[level] == 0 ? this.config.max_portrait_dialogues : this.config.max_usage[level];
+    if (dialoguesContent.length >= usageLimit || dialoguesContent.length >= this.config.max_portrait_dialogues) {
+      dialoguesContent.splice(0, dialoguesContent.length - Math.min(usageLimit, this.config.max_portrait_dialogues));
     }
     return dialoguesContent;
   }
@@ -1177,13 +1180,14 @@ ${existingPortrait ? "无" : existingPortrait}
 ${history}
 
 注意事项：
-·需要综合历史画像和用户发言记录，避免重复信息或遗漏信息
+·用户一定是人类，其他情况是角色扮演
+·需要综合历史画像和用户发言记录，避免重复信息或遗漏信息，并根据发言修正历史画像内容
 ·基于事实推断，避免盲目相信或主观臆测
 ·使用简洁的条目式表达，但内容要全面
-·保留不确定性的表述（如"可能"、"似乎"）
+·保留不确定性的表述（如"可能"、"似乎"、"用户自称"）
 ·保持中立和客观，避免带有个人情感色彩的描述，不要添加评价或建议
 ·仅给出画像内容，不要添加额外的描述、建议、评价、注解等
-·不必使用标记语言，直接书写即可`
+·不使用markdown等标记语言，直接书写即可`
     }];
   }
   // 执行画像生成
@@ -1199,13 +1203,12 @@ ${history}
     const userlevel = getFavorabilityLevel(user, this.getFavorabilityConfig());
     const messages = this.buildMessage(userlevel, dialogues.join("\n"), existing);
     logger5.info(`用户 ${user.userid} 画像生成中...`);
-    logger5.info(`${messages[0].content}`);
     try {
       const response = await apiClient.generateUserPortrait(user, messages);
       if (response && !response.error) {
         this.savePortrait(user, response.content);
-        if (user.usage > 0)
-          session.send(`用户画像更新成功。`);
+        if (user.usage > this.config.portrait_usage - 1)
+          session.send("用户画像更新成功。");
         logger5.success(`用户 ${user.userid} 画像更新成功`);
       }
     } catch (error) {
@@ -1437,7 +1440,7 @@ var SAT = class extends Sat {
   }
   // 前置检查
   performPreChecks(session, prompt) {
-    if (prompt === "")
+    if (!prompt || prompt.length == 0)
       return session.text("commands.sat.messages.no-prompt");
     if (prompt.length > this.config.max_tokens)
       return session.text("commands.sat.messages.tooLong");
@@ -1642,10 +1645,10 @@ ${payload}`);
     const userId = options.id || session.userId;
     const level = options.level || 1;
     const user = await ensureUserExists(this.ctx, userId, session.username);
-    await updateUserLevel(this.ctx, user, level);
     const enableUserKey = user?.items?.["地灵殿通行证"]?.description && user.items["地灵殿通行证"].description == "on";
     if (enableUserKey)
       await this.portraitManager.generatePortrait(session, user, this.apiClient);
+    await updateUserLevel(this.ctx, user, level);
     return session.text("commands.sat.messages.update_level_succeed", [level]);
   }
   // 处理查询用户使用次数
@@ -1653,7 +1656,18 @@ ${payload}`);
     const user = await ensureUserExists(this.ctx, session.userId, session.username);
     const userUsage = user.usage || 0;
     const maxUsage = this.config.max_usage[user.userlevel || 0] || 0;
-    return session.text("commands.sat.messages.usage", [userUsage, maxUsage]);
+    const enableUserKey = user?.items?.["地灵殿通行证"]?.description && user.items["地灵殿通行证"].description == "on";
+    let result = "";
+    result += "用户等级：" + user.userlevel + "\n";
+    if (enableUserKey)
+      result += session.text("commands.sat.messages.usage", [userUsage, "∞"]) + "\n地灵殿通行证生效中\n";
+    else
+      result += session.text("commands.sat.messages.usage", [userUsage, maxUsage]) + "\n";
+    if (this.portraitManager.hasPortrait(user.userid))
+      result += "用户画像生效中\n";
+    else
+      result += "用户画像未生效\n";
+    return result;
   }
   // 随机中间件转接
   async handleRandomMiddleware(session, prompt) {

package/lib/types.d.ts

@@ -199,6 +199,7 @@ export declare namespace Sat {
         offset_of_fafavorability: number;
         visible_favorability: boolean;
         enable_user_portrait: boolean;
+        max_portrait_dialogues: number;
         portrait_usage: number;
         portrait_min_favorability: number;
         prompt_0: string;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-satori-ai",
-  "description": "基于 Satori 适配器的智能对话插件，具备长期记忆、常识管理和好感度系统",
-  "version": "1.2.2",
+  "description": "基于 Satori 适配器的智能对话插件，具备用户画像、长期记忆、常识管理和好感度系统",
+  "version": "1.2.5",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "types": "lib/index.d.ts",