koishi-plugin-satori-ai 1.2.1 → 1.2.4

package/lib/index.d.ts

@@ -4,6 +4,7 @@ export declare class SAT extends Sat {
     config: Sat.Config;
     private apiClient;
     private memoryManager;
+    private portraitManager;
     private ChannelParallelCount;
     private onlineUsers;
     constructor(ctx: Context, config: Sat.Config);

package/lib/index.js

@@ -44,8 +44,8 @@ __export(src_exports, {
   default: () => src_default
 });
 module.exports = __toCommonJS(src_exports);
-var import_koishi6 = require("koishi");
-var path5 = __toESM(require("path"));
+var import_koishi7 = require("koishi");
+var path6 = __toESM(require("path"));
 
 // src/api.ts
 var import_koishi2 = require("koishi");
@@ -131,6 +131,8 @@ function isErrorWithMessage(error) {
 }
 __name(isErrorWithMessage, "isErrorWithMessage");
 function processPrompt(prompt) {
+  if (!prompt)
+    return "";
   if (prompt.includes(":poke"))
     return "戳戳";
   return prompt.replace(/<[^>]*?name="([^\"]*)"[^>]*>/g, (_, name) => `@${name}`);
@@ -206,19 +208,28 @@ var APIClient = class {
   // 发送辅助聊天请求
   async auxiliaryChat(messages) {
     const AuxiliaryPayload = this.createAuxiliaryPayload(messages, this.config.auxiliary_LLM);
-    for (let i = 0; i < this.config.keys.length; i++) {
+    const url = `${trimSlash(this.config.auxiliary_LLM_URL)}/chat/completions`;
+    const headers = this.createHeaders(this.config.auxiliary_LLM_key);
+    let content;
+    for (let i = 1; i <= this.config.maxRetryTimes; i++) {
       try {
-        return await this.tryRequest(this.config.auxiliary_LLM_URL, AuxiliaryPayload, this.config.auxiliary_LLM_key);
+        const response = await this.ctx.http.post(url, AuxiliaryPayload, { headers, timeout: 36e5 });
+        content = response.choices[0].message.content;
+        return { content, error: false };
       } catch (error) {
-        if (i == this.config.keys.length - 1) {
+        if (i == this.config.maxRetryTimes) {
           return this.handleAPIError(error);
         }
-        this.rotateKey();
+        logger2.warn(`辅助模型API请求失败(${error})，重试(第${i}次)中...`);
+        continue;
       }
     }
+    return { content: "unreachable", error: true };
   }
   // 发送生成用户画像请求
   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;
@@ -234,16 +245,24 @@ var APIClient = class {
       baseURL = this.config.baseURL;
     }
     const payload = this.createAuxiliaryPayload(messages, modle);
-    for (let i = 0; i < keys.length; i++) {
+    const url = `${trimSlash(baseURL)}/chat/completions`;
+    const headers = this.createHeaders(keys);
+    let content;
+    for (let i = 1; i <= this.config.maxRetryTimes; i++) {
       try {
-        return await this.tryRequest(baseURL, payload, keys);
+        const response = await this.ctx.http.post(url, payload, { headers, timeout: 36e5 });
+        content = response.choices[0].message.content;
+        return { content, error: false };
       } catch (error) {
-        if (i == keys.length - 1) {
+        if (i == this.config.maxRetryTimes) {
           return this.handleAPIError(error);
         }
-        this.rotateKey();
+        await new Promise((resolve2) => setTimeout(resolve2, this.config.retry_delay_time || 5e3));
+        logger2.warn(`生成画像时API请求失败(${error})，重试(第${i}次)中...`);
+        continue;
       }
     }
+    return { content: "unreachable", error: true };
   }
   // 生成请求体
   createPayload(messages, model) {
@@ -290,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;
         }
@@ -362,9 +381,101 @@ var APIClient = class {
 var import_koishi3 = require("koishi");
 var fs2 = __toESM(require("fs"));
 var path2 = __toESM(require("path"));
+
+// src/database.ts
+async function isTargetIdNotExists(ctx, userId) {
+  const users = await ctx.database.get("p_system", { userid: userId });
+  return users.length === 0;
+}
+__name(isTargetIdNotExists, "isTargetIdNotExists");
+async function createUser(ctx, user) {
+  await ctx.database.create("p_system", {
+    userid: user.userid || "",
+    usersname: user.usersname || "",
+    p: user.p || 0,
+    favorability: user.favorability || 0,
+    userlevel: user.userlevel || 0,
+    usage: user.usage || 0,
+    lastChatTime: user.lastChatTime || (/* @__PURE__ */ new Date()).getDate(),
+    items: user.items || {}
+  });
+}
+__name(createUser, "createUser");
+async function updateFavorability(ctx, user, adjustment) {
+  if (!user)
+    return;
+  let newValue;
+  if (typeof adjustment === "number")
+    newValue = user.favorability + adjustment;
+  else
+    newValue = user.favorability;
+  await ctx.database.set("p_system", { userid: user.userid }, { favorability: newValue });
+}
+__name(updateFavorability, "updateFavorability");
+async function updateUserLevel(ctx, user, level) {
+  if (!user)
+    return;
+  await ctx.database.set("p_system", { userid: user.userid }, { userlevel: level });
+}
+__name(updateUserLevel, "updateUserLevel");
+async function updateUserUsage(ctx, user, adjustment = 1) {
+  if (!user)
+    return;
+  if (user.lastChatTime && (/* @__PURE__ */ new Date()).getDate() !== user.lastChatTime)
+    user.usage = 0;
+  await ctx.database.set("p_system", { userid: user.userid }, { usage: user.usage + adjustment });
+  await ctx.database.set("p_system", { userid: user.userid }, { lastChatTime: (/* @__PURE__ */ new Date()).getDate() });
+  return user.usage + adjustment;
+}
+__name(updateUserUsage, "updateUserUsage");
+async function updateUserItems(ctx, user) {
+  if (!user)
+    return;
+  await ctx.database.set("p_system", { userid: user.userid }, { items: user.items });
+}
+__name(updateUserItems, "updateUserItems");
+async function getUser(ctx, userId) {
+  const users = await ctx.database.get("p_system", { userid: userId });
+  return users[0] || null;
+}
+__name(getUser, "getUser");
+async function ensureUserExists(ctx, userId, username) {
+  const notExists = await isTargetIdNotExists(ctx, userId);
+  if (notExists) {
+    await createUser(ctx, {
+      userid: userId,
+      usersname: username,
+      p: 0,
+      favorability: 0,
+      userlevel: 0,
+      usage: 0,
+      lastChatTime: (/* @__PURE__ */ new Date()).getDate(),
+      items: {}
+    });
+  }
+  return getUser(ctx, userId);
+}
+__name(ensureUserExists, "ensureUserExists");
+function extendDatabase(ctx) {
+  ctx.model.extend("p_system", {
+    id: "unsigned",
+    userid: "string",
+    usersname: "string",
+    p: "integer",
+    favorability: "integer",
+    userlevel: "integer",
+    usage: "integer",
+    lastChatTime: "integer",
+    items: "object"
+  }, { autoInc: true });
+}
+__name(extendDatabase, "extendDatabase");
+
+// src/memory.ts
 var logger3 = new import_koishi3.Logger("satori-memory");
 var MemoryManager = class {
-  constructor(config) {
+  constructor(ctx, config) {
+    this.ctx = ctx;
     this.config = config;
   }
   static {
@@ -379,7 +490,8 @@ var MemoryManager = class {
     this.updateChannelMemory(session, prompt, config, response.content);
     prompt = prompt.replace("我", session.username);
     const date = ` (对话日期和时间：${(/* @__PURE__ */ new Date()).toLocaleString()})`;
-    if (this.shouldRemember(prompt)) {
+    const userFavourbility = await (await getUser(this.ctx, session.userId)).favorability;
+    if (this.shouldRemember(prompt, userFavourbility)) {
       await this.saveLongTermMemory(session, [{
         role: date,
         content: prompt
@@ -387,8 +499,8 @@ var MemoryManager = class {
     }
   }
   // 是否应当记忆
-  shouldRemember(content) {
-    return (content.length >= this.config.remember_min_length || content.includes("记住")) && !this.config.memory_block_words.some((word) => content.includes(word));
+  shouldRemember(content, userFavourbility) {
+    return (content.length >= this.config.remember_min_length || content.includes("记住") && userFavourbility >= 50) && !this.config.memory_block_words.some((word) => content.includes(word));
   }
   // 更新频道对话
   async updateChannelDialogue(session, prompt, name) {
@@ -595,95 +707,6 @@ ${matched.map((entry) => entry.content).join("\n")}
   }
 };
 
-// src/database.ts
-async function isTargetIdNotExists(ctx, userId) {
-  const users = await ctx.database.get("p_system", { userid: userId });
-  return users.length === 0;
-}
-__name(isTargetIdNotExists, "isTargetIdNotExists");
-async function createUser(ctx, user) {
-  await ctx.database.create("p_system", {
-    userid: user.userid || "",
-    usersname: user.usersname || "",
-    p: user.p || 0,
-    favorability: user.favorability || 0,
-    userlevel: user.userlevel || 0,
-    usage: user.usage || 0,
-    lastChatTime: user.lastChatTime || (/* @__PURE__ */ new Date()).getDate(),
-    items: user.items || {}
-  });
-}
-__name(createUser, "createUser");
-async function updateFavorability(ctx, user, adjustment) {
-  if (!user)
-    return;
-  let newValue;
-  if (typeof adjustment === "number")
-    newValue = user.favorability + adjustment;
-  else
-    newValue = user.favorability;
-  await ctx.database.set("p_system", { userid: user.userid }, { favorability: newValue });
-}
-__name(updateFavorability, "updateFavorability");
-async function updateUserLevel(ctx, user, level) {
-  if (!user)
-    return;
-  await ctx.database.set("p_system", { userid: user.userid }, { userlevel: level });
-}
-__name(updateUserLevel, "updateUserLevel");
-async function updateUserUsage(ctx, user, adjustment = 1) {
-  if (!user)
-    return;
-  if (user.lastChatTime && (/* @__PURE__ */ new Date()).getDate() !== user.lastChatTime)
-    user.usage = 0;
-  await ctx.database.set("p_system", { userid: user.userid }, { usage: user.usage + adjustment });
-  await ctx.database.set("p_system", { userid: user.userid }, { lastChatTime: (/* @__PURE__ */ new Date()).getDate() });
-  return user.usage + adjustment;
-}
-__name(updateUserUsage, "updateUserUsage");
-async function updateUserItems(ctx, user) {
-  if (!user)
-    return;
-  await ctx.database.set("p_system", { userid: user.userid }, { items: user.items });
-}
-__name(updateUserItems, "updateUserItems");
-async function getUser(ctx, userId) {
-  const users = await ctx.database.get("p_system", { userid: userId });
-  return users[0] || null;
-}
-__name(getUser, "getUser");
-async function ensureUserExists(ctx, userId, username) {
-  const notExists = await isTargetIdNotExists(ctx, userId);
-  if (notExists) {
-    await createUser(ctx, {
-      userid: userId,
-      usersname: username,
-      p: 0,
-      favorability: 0,
-      userlevel: 0,
-      usage: 0,
-      lastChatTime: (/* @__PURE__ */ new Date()).getDate(),
-      items: {}
-    });
-  }
-  return getUser(ctx, userId);
-}
-__name(ensureUserExists, "ensureUserExists");
-function extendDatabase(ctx) {
-  ctx.model.extend("p_system", {
-    id: "unsigned",
-    userid: "string",
-    usersname: "string",
-    p: "integer",
-    favorability: "integer",
-    userlevel: "integer",
-    usage: "integer",
-    lastChatTime: "integer",
-    items: "object"
-  }, { autoInc: true });
-}
-__name(extendDatabase, "extendDatabase");
-
 // src/fixed-dialogues.ts
 var fs3 = __toESM(require("fs"));
 var path3 = __toESM(require("path"));
@@ -1059,13 +1082,11 @@ var Sat = class extends import_koishi5.Service {
       randnum: import_koishi5.Schema.number().role("slider").min(0).max(1).step(0.01).default(0).description("在群聊中随机触发对话的概率，如需关闭可设置为 0"),
       sentences_divide: import_koishi5.Schema.boolean().default(true).description("是否分句发送"),
       time_interval: import_koishi5.Schema.number().default(1e3).description("每句话的时间间隔"),
-      reply_pointing: import_koishi5.Schema.boolean().default(true).description("是否在与多人同时对话时显示回复指向"),
-      enable_user_portrait: import_koishi5.Schema.boolean().default(true).description("是否启用用户画像功能"),
-      portrait_min_dialogues: import_koishi5.Schema.number().default(100).description("触发画像生成的最小对话条数")
+      reply_pointing: import_koishi5.Schema.boolean().default(true).description("是否在与多人同时对话时显示回复指向")
     }).description("对话设置"),
     import_koishi5.Schema.object({
       enable_favorability: import_koishi5.Schema.boolean().default(false).description("是否开启好感度系统(每次对话默认+1好感度)"),
-      max_favorability_perday: import_koishi5.Schema.number().default(100).description("每日有效(引发好感度变化)对话次数上限"),
+      max_favorability_perday: import_koishi5.Schema.number().default(100).description("每日有效(引发好感度增长)对话次数上限"),
       input_censor_favorability: import_koishi5.Schema.boolean().default(false).description("是否开启好感度审查(通过输入屏蔽词扣除好感)"),
       value_of_input_favorability: import_koishi5.Schema.number().default(15).description("输入触发屏蔽词每次扣除的好感度"),
       output_censor_favorability: import_koishi5.Schema.boolean().default(false).description("通过输出屏蔽词扣除好感,在dataDir中的output_censor.txt修改)"),
@@ -1073,6 +1094,10 @@ var Sat = class extends import_koishi5.Service {
       enable_auxiliary_LLM: import_koishi5.Schema.boolean().default(false).description("是否使用辅助大模型判断好感度增减(量与输入屏蔽词每次扣除的好感度相关,不稳定，慎用)"),
       offset_of_fafavorability: import_koishi5.Schema.number().default(3.5).description("辅助大模型好感度偏移量(越大越容易扣好感度)"),
       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("厌恶-陌生分界线"),
       prompt_1: import_koishi5.Schema.string().role("textarea").description("陌生好感补充设定"),
@@ -1090,8 +1115,155 @@ var Sat = class extends import_koishi5.Service {
   ]);
 })(Sat || (Sat = {}));
 
+// src/userportrait.ts
+var import_koishi6 = require("koishi");
+var fs5 = __toESM(require("fs"));
+var path5 = __toESM(require("path"));
+var logger5 = new import_koishi6.Logger("satori-portrait");
+var UserPortraitManager = class {
+  constructor(ctx, config) {
+    this.ctx = ctx;
+    this.config = config;
+  }
+  static {
+    __name(this, "UserPortraitManager");
+  }
+  // 获取用户印象路径
+  getPortraitPath(userId) {
+    return path5.join(this.config.dataDir, "UserPortrait", `${userId}.txt`);
+  }
+  // 检查用户印象文件
+  hasPortrait(userId) {
+    return fs5.existsSync(this.getPortraitPath(userId));
+  }
+  // 读取用户印象
+  readPortrait(userId) {
+    try {
+      return fs5.readFileSync(this.getPortraitPath(userId), "utf-8");
+    } catch {
+      return "";
+    }
+  }
+  // 获取长期记忆对话
+  async getDialogues(user) {
+    const userId = user.userid;
+    const memoryPath = path5.join(this.config.dataDir, "dialogues", `${userId}.txt`);
+    if (!fs5.existsSync(memoryPath))
+      return [];
+    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 ? 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;
+  }
+  // 生成提示词模板
+  buildMessage(level, history, existingPortrait) {
+    return [{
+      role: "user",
+      content: `你是一个角色扮演智能体工作流中的一环，请根据以下信息生成用户画像。需包括：
+1. 基本信息（性别、年龄、生日等）
+2. 生活习惯（作息、爱好等）
+3. 性格特征（用几句话描述）
+4. 交流偏好（常用语气词、话题倾向）
+5. 敏感点（明确表示反感的内容）
+6. 对AI的期望（从对话历史中推断）
+7. 其他你认为需要记录的信息（如有）
+
+你与用户的关系：${level}
+历史画像：
+${existingPortrait ? "无" : existingPortrait}
+
+近期的用户发言记录：
+${history}
+
+注意事项：
+·用户一定是人类，其他情况是角色扮演
+·需要综合历史画像和用户发言记录，避免重复信息或遗漏信息，并根据发言修正历史画像内容
+·基于事实推断，避免盲目相信或主观臆测
+·使用简洁的条目式表达，但内容要全面
+·保留不确定性的表述（如"可能"、"似乎"、"用户自称"）
+·保持中立和客观，避免带有个人情感色彩的描述，不要添加评价或建议
+·仅给出画像内容，不要添加额外的描述、建议、评价、注解等
+·不使用markdown等标记语言，直接书写即可`
+    }];
+  }
+  // 执行画像生成
+  async generatePortrait(session, user, apiClient) {
+    if (!this.config.enable_favorability)
+      return;
+    if (!this.config.enable_user_portrait)
+      return;
+    if (user.favorability < this.config.portrait_min_favorability)
+      return;
+    const dialogues = await this.getDialogues(user);
+    const existing = this.readPortrait(user.userid);
+    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 > this.config.portrait_usage - 1)
+          session.send("用户画像更新成功。");
+        logger5.success(`用户 ${user.userid} 画像更新成功`);
+      }
+    } catch (error) {
+      logger5.error(`画像生成失败：${error.message}`);
+    }
+  }
+  savePortrait(user, portrait) {
+    const filePath = this.getPortraitPath(user.userid);
+    this.ensurePortraitFile(user.userid);
+    fs5.writeFileSync(filePath, portrait);
+  }
+  ensurePortraitFile(userId) {
+    const filePath = this.getPortraitPath(userId);
+    if (!fs5.existsSync(filePath)) {
+      fs5.mkdirSync(path5.dirname(filePath), { recursive: true });
+      fs5.writeFileSync(filePath, "");
+    }
+  }
+  // 获取用户画像
+  getUserPortrait(session) {
+    if (!this.config.enable_user_portrait)
+      return "";
+    const portrait = this.readPortrait(session.userId);
+    return portrait ? `
+以下是当前用户的补充信息：{${portrait}
+}
+` : "";
+  }
+  getFavorabilityConfig() {
+    return {
+      enable_favorability: this.config.enable_favorability,
+      dataDir: this.config.dataDir,
+      max_favorability_perday: this.config.max_favorability_perday,
+      input_censor_favorability: this.config.input_censor_favorability,
+      value_of_input_favorability: this.config.value_of_input_favorability,
+      output_censor_favorability: this.config.output_censor_favorability,
+      value_of_output_favorability: this.config.value_of_output_favorability,
+      enable_auxiliary_LLM: this.config.enable_auxiliary_LLM,
+      offset_of_fafavorability: this.config.offset_of_fafavorability,
+      prompt_0: this.config.prompt_0,
+      favorability_div_1: this.config.favorability_div_1,
+      prompt_1: this.config.prompt_1,
+      favorability_div_2: this.config.favorability_div_2,
+      prompt_2: this.config.prompt_2,
+      favorability_div_3: this.config.favorability_div_3,
+      prompt_3: this.config.prompt_3,
+      favorability_div_4: this.config.favorability_div_4,
+      prompt_4: this.config.prompt_4
+    };
+  }
+};
+
 // src/index.tsx
-var logger5 = new import_koishi6.Logger("satori-ai");
+var logger6 = new import_koishi7.Logger("satori-ai");
 var SAT = class extends Sat {
   // 重写构造函数
   constructor(ctx, config) {
@@ -1100,7 +1272,8 @@ var SAT = class extends Sat {
     ctx.i18n.define("zh", require_zh());
     extendDatabase(ctx);
     this.apiClient = new APIClient(ctx, this.getAPIConfig());
-    this.memoryManager = new MemoryManager(this.getMemoryConfig());
+    this.memoryManager = new MemoryManager(ctx, this.getMemoryConfig());
+    this.portraitManager = new UserPortraitManager(ctx, config);
     ensureCensorFileExists(this.config.dataDir);
     ctx.middleware(createMiddleware(ctx, this, this.getMiddlewareConfig()));
     this.registerCommands(ctx);
@@ -1110,7 +1283,7 @@ var SAT = class extends Sat {
   }
   apiClient;
   memoryManager;
-  //private portraitManager: UserPortraitManager
+  portraitManager;
   ChannelParallelCount = /* @__PURE__ */ new Map();
   onlineUsers = [];
   getAPIConfig() {
@@ -1231,6 +1404,8 @@ var SAT = class extends Sat {
     const response = await this.generateResponse(session, processedPrompt);
     const auxiliaryResult = await this.handleAuxiliaryDialogue(session, processedPrompt, response);
     await this.memoryManager.updateMemories(session, processedPrompt, this.getMemoryConfig(), response);
+    if (user.usage == this.config.portrait_usage - 1)
+      this.portraitManager.generatePortrait(session, user, this.apiClient);
     return await this.formatResponse(session, response.content, auxiliaryResult);
   }
   // 处理辅助判断
@@ -1250,9 +1425,9 @@ var SAT = class extends Sat {
       const messages = generateAuxiliaryPrompt(prompt, response.content, user, this.getFavorabilityConfig());
       const result = await this.apiClient.auxiliaryChat(messages);
       if (result.error) {
-        logger5.error(`辅助判断失败：${result.content}`);
+        logger6.error(`辅助判断失败：${result.content}`);
       } else {
-        logger5.info(`辅助判断：${result.content}`);
+        logger6.info(`辅助判断：${result.content}`);
         return handleAuxiliaryResult(this.ctx, session, this.getFavorabilityConfig(), result.content);
       }
     }
@@ -1266,7 +1441,7 @@ var SAT = class extends Sat {
   }
   // 前置检查
   performPreChecks(session, prompt) {
-    if (!prompt)
+    if (prompt === "")
       return session.text("commands.sat.messages.no-prompt");
     if (prompt.length > this.config.max_tokens)
       return session.text("commands.sat.messages.tooLong");
@@ -1312,7 +1487,7 @@ var SAT = class extends Sat {
     const usage = await updateUserUsage(this.ctx, user, adjustment);
     if (user?.items?.["地灵殿通行证"]?.description && user.items["地灵殿通行证"].description === "on")
       return null;
-    const level = user.userlevel || 0;
+    const level = user.userlevel < 5 ? user.userlevel : 4;
     const usageLimit = this.config.max_usage[level] || 0;
     if (usage && usageLimit != 0 && usage > usageLimit) {
       return session.text("commands.sat.messages.exceeds");
@@ -1339,13 +1514,13 @@ var SAT = class extends Sat {
       await inputContentCheck(this.ctx, censored, session.userId, this.getFavorabilityConfig(), session);
     }
     if (this.config.log_ask_response)
-      logger5.info(`用户 ${session.username}：${processedPrompt}`);
+      logger6.info(`用户 ${session.username}：${processedPrompt}`);
     return censored;
   }
   // 生成回复
   async generateResponse(session, prompt) {
     if (this.getChannelParallelCount(session) > this.config.max_parallel_count) {
-      logger5.info(`频道 ${session.channelId} 并发数过高(${this.getChannelParallelCount(session)})，${session.username}等待中...`);
+      logger6.info(`频道 ${session.channelId} 并发数过高(${this.getChannelParallelCount(session)})，${session.username}等待中...`);
     }
     while (this.getChannelParallelCount(session) > this.config.max_parallel_count) {
       this.updateChannelParallelCount(session, -1);
@@ -1353,11 +1528,11 @@ var SAT = class extends Sat {
       this.updateChannelParallelCount(session, 1);
     }
     const messages = this.buildMessages(session, prompt);
-    logger5.info(`频道 ${session.channelId} 处理：${session.userId},剩余${this.getChannelParallelCount(session)}并发`);
+    logger6.info(`频道 ${session.channelId} 处理：${session.userId},剩余${this.getChannelParallelCount(session)}并发`);
     const user = await getUser(this.ctx, session.userId);
     const response = await this.apiClient.chat(user, await messages);
     if (this.config.log_ask_response)
-      logger5.info(`Satori AI：${response.content}`);
+      logger6.info(`Satori AI：${response.content}`);
     return response;
   }
   // 构建消息
@@ -1382,7 +1557,7 @@ var SAT = class extends Sat {
     });
     let payload = messages.map((msg) => msg.role + ":" + msg.content).join("\n");
     if (this.config.log_system_prompt)
-      logger5.info(`系统提示：
+      logger6.info(`系统提示：
 ${payload}`);
     return messages;
   }
@@ -1396,6 +1571,7 @@ ${payload}`);
     systemPrompt += channelDialogue;
     systemPrompt += userMemory;
     const user = await getUser(this.ctx, session.userId);
+    systemPrompt += this.portraitManager.getUserPortrait(session);
     const nickName = user.items["情侣合照"]?.metadata?.userNickName;
     systemPrompt += `用户的名字是：${session.username}, id是：${session.userId}`;
     if (nickName)
@@ -1430,7 +1606,7 @@ ${payload}`);
       response = `@${session.username} ` + response;
     }
     if (this.config.sentences_divide && response.length > 10) {
-      const sentences = splitSentences(response).map((text) => import_koishi6.h.text(text));
+      const sentences = splitSentences(response).map((text) => import_koishi7.h.text(text));
       for (const sentence of sentences) {
         await session.send(sentence);
         await new Promise((resolve2) => setTimeout(resolve2, this.config.time_interval));
@@ -1458,7 +1634,7 @@ ${payload}`);
   async addCommonSense(session, content) {
     if (!content)
       return session.text("commands.sat.common_sense.messages.no-prompt");
-    const filePath = path5.join(this.config.dataDir, "common_sense.txt");
+    const filePath = path6.join(this.config.dataDir, "common_sense.txt");
     await this.memoryManager.saveLongTermMemory(session, [{
       role: "user",
       content
@@ -1470,6 +1646,9 @@ ${payload}`);
     const userId = options.id || session.userId;
     const level = options.level || 1;
     const user = await ensureUserExists(this.ctx, userId, session.username);
+    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]);
   }
@@ -1478,7 +1657,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/memory.d.ts

@@ -1,10 +1,11 @@
-import { Session } from 'koishi';
+import { Session, Context } from 'koishi';
 import { MemoryEntry, MemoryConfig } from './types';
 export declare class MemoryManager {
+    private ctx;
     private config;
     private channelMemories;
     private channelDialogues;
-    constructor(config: MemoryConfig);
+    constructor(ctx: Context, config: MemoryConfig);
     updateMemories(session: Session, prompt: string, config: MemoryConfig, response: {
         content: string;
         error: boolean;

package/lib/types.d.ts

@@ -189,8 +189,6 @@ export declare namespace Sat {
         time_interval: number;
         max_parallel_count: number;
         reply_pointing: boolean;
-        enable_user_portrait: boolean;
-        portrait_min_dialogues: number;
         enable_favorability: boolean;
         max_favorability_perday: number;
         input_censor_favorability: boolean;
@@ -200,6 +198,10 @@ export declare namespace Sat {
         enable_auxiliary_LLM: boolean;
         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;
         favorability_div_1: number;
         prompt_1: string;

package/lib/userportrait.d.ts

@@ -8,9 +8,11 @@ export declare class UserPortraitManager {
     private getPortraitPath;
     hasPortrait(userId: string): boolean;
     readPortrait(userId: string): string;
-    private processDialogues;
-    private buildPrompt;
+    private getDialogues;
+    private buildMessage;
     generatePortrait(session: Session, user: User, apiClient: APIClient): Promise<void>;
-    enhanceSystemPrompt(session: Session): string;
+    private savePortrait;
+    private ensurePortraitFile;
+    getUserPortrait(session: Session): string;
     private getFavorabilityConfig;
 }

package/package.json

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