alemonjs-aichat 1.0.0

package/README.md

@@ -0,0 +1,18 @@
+
+## 安装
+- alemongo安装方式
+地址
+```
+https://gitee.com/suancaixianyu/alemonjs-aichat.git 
+```
+
+分支
+```
+release
+```
+
+- npm安装方式
+
+
+## 使用
+指令`/ai帮助`查看使用方法

package/lib/config.js

@@ -0,0 +1,162 @@
+import { redis } from './redis.js';
+
+const systemPrompt = ``;
+const defaultAIConfig = {
+    host: "",
+    key: "",
+    model: "",
+    temperature: 0.7,
+    max_tokens: 150,
+    systemPrompt: systemPrompt,
+};
+const getAIConfig = async (guid) => {
+    let id = guid ?? "global";
+    console.log("获取", guid);
+    const userConfigStr = await redis.get(`ai:config:${id}`);
+    if (userConfigStr) {
+        const userConfig = JSON.parse(userConfigStr);
+        return userConfig;
+    }
+    redis.set(`ai:config:${id}`, JSON.stringify(defaultAIConfig));
+    return defaultAIConfig;
+};
+const getAIList = async () => {
+    const keys = await redis.keys("ai:list:*");
+    return keys.map((key) => key.replace("ai:list:", ""));
+};
+const addAI = async (guid, name, host, key, model) => {
+    const defaultConfig = await getAIConfig();
+    defaultConfig.host = host;
+    defaultConfig.key = key;
+    defaultConfig.model = model;
+    await redis.set(`ai:list:${name}`, JSON.stringify(defaultConfig));
+    await switchAI(guid, name);
+};
+const switchAI = async (guid, name) => {
+    const aiConfigStr = await redis.get(`ai:list:${name}`);
+    if (!aiConfigStr) {
+        console.log("切换失败, 无此模型");
+        return null;
+    }
+    const aiConfig = JSON.parse(aiConfigStr);
+    await redis.set(`ai:config:${guid}`, aiConfigStr);
+    return aiConfig;
+};
+const getPromptList = async () => {
+    const keys = await redis.keys("ai:prompt:*");
+    return keys.map((key) => key.replace("ai:prompt:", ""));
+};
+const addPrompt = async (name, content) => {
+    return await redis.set(`ai:prompt:${name}`, content);
+};
+const deletePrompt = async (name) => {
+    return await redis.del(`ai:prompt:${name}`);
+};
+const switchPrompt = async (guid, name) => {
+    if (name === "") {
+        const config = await getAIConfig(guid);
+        config.systemPrompt = "";
+        await redis.set(`ai:config:${guid}`, JSON.stringify(config));
+        return systemPrompt;
+    }
+    const prompt = await redis.get(`ai:prompt:${name}`);
+    if (!prompt) {
+        console.log("切换提示词失败, 无此提示词");
+        return null;
+    }
+    const config = await getAIConfig(guid);
+    config.systemPrompt = prompt;
+    await redis.set(`ai:config:${guid}`, JSON.stringify(config));
+    return prompt;
+};
+const getAIChatHistory = async (guid) => {
+    const historyStr = await redis.get(`ai:history:${guid}`);
+    if (historyStr) {
+        return JSON.parse(historyStr);
+    }
+    return [];
+};
+const addAIChatHistory = async (guid, role, content) => {
+    const history = await getAIChatHistory(guid);
+    // 处理messages, 不要超过15条,超过就删除最早的
+    while (history.length >= 20) {
+        history.shift();
+    }
+    history.push({ role, content });
+    await redis.set(`ai:history:${guid}`, JSON.stringify(history));
+    return history;
+};
+const clearAIChatHistory = async (guid) => {
+    return await redis.del(`ai:history:${guid}`);
+};
+/**
+ *  获取好感度等级
+ * @param userKey 用户标识
+ * @returns
+ */
+const getAffectionLevel = async (userKey) => {
+    const levelStr = await redis.get(`ai:affection:${userKey}`);
+    if (levelStr) {
+        return parseInt(levelStr, 10);
+    }
+    return 0;
+};
+const getAffectionLevelAll = async (guid) => {
+    const keys = await redis.keys(`ai:affection:${guid}:*`);
+    const levels = {};
+    for (const key of keys) {
+        const levelStr = await redis.get(key);
+        if (levelStr) {
+            const userKey = key.replace(`ai:affection:${guid}:`, "");
+            levels[userKey] = parseInt(levelStr, 10);
+        }
+    }
+    return levels;
+};
+/**
+ * 设置好感度等级
+ * @param userKey 用户标识
+ * @param level 好感度等级
+ * @returns
+ */
+const setAffectionLevel = async (userKey, level) => {
+    return await redis.set(`ai:affection:${userKey}`, level.toString());
+};
+/**
+ *  增加好感度等级
+ * @param userKey 用户标识
+ * @param increment  增加的等级
+ * @returns
+ */
+const incrementAffectionLevel = async (userKey, increment) => {
+    const currentLevel = await getAffectionLevel(userKey);
+    const newLevel = currentLevel + increment;
+    await setAffectionLevel(userKey, newLevel);
+    return newLevel;
+};
+/** * 重置好感度等级
+ * @param userKey 用户标识
+ * @returns
+ */
+const resetAffectionLevel = async (userKey) => {
+    return await redis.del(`ai:affection:${userKey}`);
+};
+/** 重置所有用户好感度等级
+ * @param guid 频道标识
+ * @returns
+ */
+const resetAllAffectionLevels = async () => {
+    const keys = await redis.keys(`ai:affection:*`);
+    for (const key of keys) {
+        await redis.del(key);
+    }
+};
+const getBotID = async () => {
+    const botid = await redis.get("bot:id");
+    return String(botid) || "794161769";
+};
+const setBotID = async (id) => {
+    return await redis.set("bot:id", id);
+};
+
+export { addAI, addAIChatHistory, addPrompt, clearAIChatHistory, defaultAIConfig, deletePrompt, getAIChatHistory, getAIConfig, getAIList, getAffectionLevel, getAffectionLevelAll, getBotID, getPromptList, incrementAffectionLevel, resetAffectionLevel, resetAllAffectionLevels, setAffectionLevel, setBotID, switchAI, switchPrompt, systemPrompt };

package/lib/emoji.js

@@ -0,0 +1,120 @@
+const emojiMap = {
+    4: "得意",
+    5: "流泪",
+    8: "睡",
+    9: "大哭",
+    10: "尴尬",
+    12: "调皮",
+    14: "微笑",
+    16: "酷",
+    21: "可爱",
+    23: "傲慢",
+    24: "饥饿",
+    25: "困",
+    26: "惊恐",
+    27: "流汗",
+    28: "憨笑",
+    29: "悠闲",
+    30: "奋斗",
+    32: "疑问",
+    33: "嘘",
+    34: "晕",
+    38: "敲打",
+    39: "再见",
+    41: "发抖",
+    42: "爱情",
+    43: "跳跳",
+    49: "拥抱",
+    53: "蛋糕",
+    60: "咖啡",
+    63: "玫瑰",
+    66: "爱心",
+    74: "太阳",
+    75: "月亮",
+    76: "赞",
+    78: "握手",
+    79: "胜利",
+    85: "飞吻",
+    89: "西瓜",
+    96: "冷汗",
+    97: "擦汗",
+    98: "抠鼻",
+    99: "鼓掌",
+    100: "糗大了",
+    101: "坏笑",
+    102: "左哼哼",
+    103: "右哼哼",
+    104: "哈欠",
+    106: "委屈",
+    109: "左亲亲",
+    111: "可怜",
+    116: "示爱",
+    118: "抱拳",
+    120: "拳头",
+    122: "爱你",
+    123: "NO",
+    124: "OK",
+    125: "转圈",
+    129: "挥手",
+    144: "喝彩",
+    147: "棒棒糖",
+    171: "茶",
+    173: "泪奔",
+    174: "无奈",
+    175: "卖萌",
+    176: "小纠结",
+    179: "doge",
+    180: "惊喜",
+    181: "骚扰",
+    182: "笑哭",
+    183: "我最美",
+    201: "点赞",
+    203: "托脸",
+    212: "托腮",
+    214: "啵啵",
+    219: "蹭一蹭",
+    222: "抱抱",
+    227: "拍手",
+    232: "佛系",
+    240: "喷脸",
+    243: "甩头",
+    246: "加油抱抱",
+    262: "脑阔疼",
+    264: "捂脸",
+    265: "辣眼睛",
+    266: "哦哟",
+    267: "头秃",
+    268: "问号脸",
+    269: "暗中观察",
+    270: "emm",
+    271: "吃瓜",
+    272: "呵呵哒",
+    273: "我酸了",
+    277: "汪汪",
+    278: "汗",
+    281: "无眼笑",
+    282: "敬礼",
+    284: "面无表情",
+    285: "摸鱼",
+    287: "哦",
+    289: "睁眼",
+    290: "敲开心",
+    293: "摸锦鲤",
+    294: "期待",
+    297: "拜谢",
+    298: "元宝",
+    299: "牛啊",
+    305: "右亲亲",
+    306: "牛气冲天",
+    307: "喵喵",
+    314: "仔细分析",
+    315: "加油",
+    318: "崇拜",
+    319: "比心",
+    320: "庆祝",
+    322: "拒绝",
+    324: "吃糖",
+    326: "生气",
+};
+
+export { emojiMap };

package/lib/index.js

@@ -0,0 +1,9 @@
+var index = defineChildren({
+    onCreated() {
+        logger.info({
+            message: '本地测试启动'
+        });
+    }
+});
+
+export { index as default };

package/lib/middleware/mw.js

@@ -0,0 +1,78 @@
+import { getConfigValue, useClient } from 'alemonjs';
+import { API } from '@alemonjs/onebot';
+import { redis } from '../redis.js';
+
+const selects = onSelects(["message.create", "private.message.create"]);
+// 中间件
+var mw = onMiddleware(selects, async (event, next) => {
+    const config = getConfigValue();
+    // 新增字段
+    event["user_id"] = event.UserId;
+    event["msg"] = event.MessageText;
+    event["guid"] = event.ChannelId ?? event.UserId;
+    event["uidkey"] = `${event["guid"]}:${event.UserName}(${event.UserId})`;
+    event["nickname"] = `${event.UserName}(${event.UserId})`;
+    console.log('event["nickname"]', event["nickname"]);
+    console.log("event.IsMaster", event.UserId, config.master_id, event.IsMaster);
+    // 判断event.value是否为数组
+    if (Array.isArray(event.value)) {
+        event["originalMsg"] = event.value;
+        // 遍历数组, 将类型为at或Mention的项提取出来放到at中
+        event["at"] = event.value.filter((item) => item.type === "at" || item.type === "Mention");
+    }
+    // onebot平台特有处理 //
+    if (event.Platform == "onebot") {
+        // 获取客户端
+        const [client] = useClient(event, API);
+        // 获取群成员存入redis
+        const redisKey = `group_members_${event.ChannelId}`;
+        const cachedMembers = await redis.get(redisKey);
+        const cachedMembersArray = cachedMembers ? JSON.parse(cachedMembers) : [];
+        // 尝试能否从缓存中找到当前用户,没有的话就重新获取群成员列表并缓存
+        const userInCache = cachedMembersArray.find((member) => {
+            const [id, name] = member.split(":");
+            return id === event.UserId.toString();
+        });
+        if (!cachedMembers || !userInCache) {
+            const list = await client.getGroupMemberList({
+                group_id: event.ChannelId,
+            });
+            const memberNames = list[0].data.map((item) => item.user_id + ":" + item.nickname);
+            await redis.set(redisKey, JSON.stringify(memberNames));
+        }
+        event["self_id"] = event.value.self_id;
+        event["originalMsg"] = event.value.message;
+        event["at"] = event.value.message
+            .filter((item) => item.type === "at" || item.type === "Mention")
+            .map((item) => {
+            cachedMembersArray.find((member) => {
+                const [id, name] = member.split(":");
+                if (id === item.data.qq.toString()) {
+                    item.data.nickname = name;
+                }
+            });
+            return item;
+        });
+        event["atBot"] = event["at"].find((item) => item.data.qq.toString() === event.self_id.toString());
+        event["img"] = event.value.message
+            .filter((item) => item.type === "image" || item.type === "Image")
+            .map((item) => item.data.url);
+        event["isAdmin"] =
+            event.value.sender.role === "admin" ||
+                event.value.sender.role === "owner";
+        const replyData = event.value.message.find((item) => item.type === "reply");
+        if (replyData) {
+            const msg = await client.getMsg({
+                message_id: Number(replyData.data.id),
+            });
+            console.log("msg[0].data;", JSON.stringify(msg[0].data, null, 2));
+            if (msg && msg[0] && msg[0].data) {
+                event["reply"] = msg[0].data;
+            }
+        }
+    }
+    // 常用于兼容其他框架或增强event功能
+    next();
+});
+
+export { mw as default };

package/lib/redis.js

@@ -0,0 +1,12 @@
+import { getConfigValue } from 'alemonjs';
+import Redis from 'ioredis';
+
+const value = getConfigValue();
+const redisConfig = {
+    host: value.redis?.host ?? "localhost",
+    port: value.redis?.port ?? 6379,
+    password: value.redis?.password ?? "",
+};
+const redis = new Redis(redisConfig);
+
+export { redis };

package/lib/response/affection/res.js

@@ -0,0 +1,33 @@
+import { getAffectionLevel, resetAllAffectionLevels, resetAffectionLevel } from '../../config.js';
+import { useMessage, Text } from 'alemonjs';
+import { Regular } from 'alemonjs/utils';
+
+const selects = onSelects(["message.create", "private.message.create"]);
+const regular$1 = /(\/|#)(查看)?好感度?$/i;
+const regular$2 = /(\/|#)(重置|清空)(全部|所有)?好感度?$/i;
+const regular = Regular.or(regular$1, regular$2);
+var res = onResponse(selects, async (e, next) => {
+    // 创建
+    const [message] = useMessage(e);
+    // 查看好感度
+    if (regular$1.test(e.msg)) {
+        const affections = await getAffectionLevel(e.uidkey);
+        message.send(format(Text(`当前用户\`${e.UserName}\`的好感度为: ${affections}。`)));
+        return;
+    }
+    // 重置好感度
+    if (regular$2.test(e.msg)) {
+        if ((e.msg.includes("全部") && e.IsMaster) ||
+            (e.msg.includes("所有") && e.IsMaster)) {
+            await resetAllAffectionLevels();
+            message.send(format(Text(`已重置所有用户的好感度。`)));
+        }
+        else {
+            await resetAffectionLevel(e.uidkey);
+            message.send(format(Text(`已重置用户\`${e.UserName}\`的好感度。`)));
+        }
+    }
+    next();
+});
+
+export { res as default, regular, selects };

package/lib/response/config/res.js

@@ -0,0 +1,37 @@
+import { getAIConfig, getAIList, getPromptList, clearAIChatHistory } from '../../config.js';
+import { useMessage, Text } from 'alemonjs';
+
+const selects = onSelects(["message.create", "private.message.create"]);
+const regular$1 = /(\/|#)ai配置$/i;
+const regular$2 = /(\/|#)ai列表$/i;
+const regular$3 = /(\/|#)提示词列表$/i;
+const regular$4 = /(\/|#)清空对话$/i;
+const regular = /.*/;
+var res = onResponse(selects, async (e, next) => {
+    // 创建
+    const [message] = useMessage(e);
+    const config = await getAIConfig(e.guid);
+    // 查看AI配置
+    if (regular$1.test(e.msg)) {
+        // 发送消息
+        message.send(format(Text(`当前AI配置：\n`), Text(`模型: ${config.model}\n`), Text(`提示词: ${config.systemPrompt}`)));
+    }
+    // 查看AI列表
+    if (regular$2.test(e.msg)) {
+        const ailist = await getAIList();
+        message.send(format(Text("当前AI列表：\n"), Text(ailist.length ? ailist.join("\n") : "暂无AI配置")));
+    }
+    // 查看提示词列表
+    if (regular$3.test(e.msg)) {
+        const prompts = await getPromptList();
+        message.send(format(Text("当前提示词列表：\n"), Text(prompts.length ? prompts.join("\n") : "暂无提示词")));
+    }
+    // 清空对话
+    if (regular$4.test(e.msg)) {
+        await clearAIChatHistory(e.guid);
+        message.send(format(Text("对话已清空")));
+    }
+    next();
+});
+
+export { res as default, regular, selects };

package/lib/response/help/res.js

@@ -0,0 +1,17 @@
+import { useMessage, Text } from 'alemonjs';
+
+const selects = onSelects(["message.create", "private.message.create"]);
+const regular$1 = /(\/|#)(ai)(指令|帮助|菜单|功能)$/i;
+const regular = /.*/;
+var res = onResponse(selects, async (e, next) => {
+    // 创建
+    const [message] = useMessage(e);
+    // 查看指令帮助
+    if (regular$1.test(e.msg)) {
+        message.send(format(Text("当前支持的指令有：\n"), Text("/ai配置 - 查看当前AI配置\n"), Text("/ai列表 - 查看当前AI列表\n"), Text("/切换ai - 切换AI\n"), Text("/提示词列表 - 查看当前提示词列表\n"), Text("/添加提示词 - 添加新的设定\n"), Text("/清空对话 - 清空与AI的对话记录\n"), Text("/好感度 - 查看当前用户的好感度\n")));
+        return;
+    }
+    next();
+});
+
+export { res as default, regular, selects };

package/lib/response/setting/res.js

@@ -0,0 +1,115 @@
+import { addAI, switchAI, addPrompt, deletePrompt, switchPrompt, setBotID } from '../../config.js';
+import { useMessage, Text } from 'alemonjs';
+
+// import { Regular } from "alemonjs/utils";
+const selects = onSelects(["message.create", "private.message.create"]);
+const regular$1 = /(\/|#)添加ai$/i;
+const regular$2 = /(\/|#)添加ai\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)/i;
+const regular$3 = /(\/|#)(切换)(ai|配置) ?(.*)/i;
+const regular$4 = /(\/|#)添加提示词$/i;
+const regular$5 = /(\/|#)添加提示词 (\S+) ([\s\S]+)/i;
+const regular$6 = /(\/|#)(删除)(提示词|提示词列表) ?(.*)/i;
+const regular$7 = /(\/|#)(切换|设置)提示词 ?(.*)$/i;
+const regular$8 = /(\/|#)清空提示词$/i;
+const regular$9 = /(\/|#)设置(机器人)?(id|账号) ?(.*)$/i;
+// export const regular = Regular.or(regular$1, regular$2, regular$3);
+const regular = /.*/;
+var res = onResponse(selects, async (e, next) => {
+    // 创建
+    const [message] = useMessage(e);
+    // 添加AI配置
+    if (regular$1.test(e.msg)) {
+        message.send(format(Text("请按照 格式：/添加ai <名称> <主机地址> <api_key> <模型> 进行添加")));
+        return;
+    }
+    // 处理添加AI配置
+    if (regular$2.test(e.msg)) {
+        const match = e.msg.match(regular$2);
+        if (!match) {
+            message.send(format(Text("格式错误，请按照 格式：/添加ai <名称> <主机地址> <api_key> <模型> 进行添加")));
+            return;
+        }
+        const [, , name, host, key, model] = match;
+        await addAI(e.guid, name, host, key, model);
+        message.send(format(Text(`AI ${name} 添加成功！`)));
+    }
+    // 切换AI配置
+    if (regular$3.test(e.msg)) {
+        const match = e.msg.match(regular$3);
+        if (!match) {
+            message.send(format(Text("格式错误，请按照 格式：/切换ai <名称> 进行切换")));
+            return;
+        }
+        const [, , , , name] = match;
+        const ai = await switchAI(e.guid, name.trim());
+        if (!ai) {
+            message.send(format(Text(`切换失败，未找到 AI ${name} !`)));
+            return;
+        }
+        message.send(format(Text(`已切换到 AI ${name} !`)));
+        return;
+    }
+    // 添加提示词
+    if (regular$4.test(e.msg)) {
+        message.send(format(Text("请按照 格式：/添加提示词 <名称> <内容> 进行添加")));
+        return;
+    }
+    // 处理添加提示词
+    if (regular$5.test(e.msg)) {
+        const match = e.msg.match(regular$5);
+        if (!match) {
+            message.send(format(Text("格式错误，请按照 格式：/添加提示词 <名称> <内容> 进行添加")));
+            return;
+        }
+        const [, , name, content] = match;
+        await addPrompt(name, content);
+        message.send(format(Text(`提示词 ${name} 添加成功！`)));
+        return;
+    }
+    // 删除提示词
+    if (regular$6.test(e.msg)) {
+        const match = e.msg.match(regular$6);
+        if (!match) {
+            message.send(format(Text("格式错误，请按照 格式：/删除提示词 <名称> 进行删除")));
+            return;
+        }
+        const [, , , , name] = match;
+        await deletePrompt(name);
+        message.send(format(Text(`提示词 ${name} 删除成功！`)));
+        return;
+    }
+    // 切换提示词
+    if (regular$7.test(e.msg)) {
+        const match = e.msg.match(regular$7);
+        if (!match) {
+            message.send(format(Text("格式错误，请按照 格式：/切换提示词 <名称> 进行切换")));
+            return;
+        }
+        const [, , , name] = match;
+        await switchPrompt(e.guid, name);
+        message.send(format(Text(`已切换到提示词 ${name} !`)));
+        return;
+    }
+    // 清空提示词
+    if (regular$8.test(e.msg)) {
+        await switchPrompt(e.guid, "");
+        message.send(format(Text("已清空提示词 !")));
+        return;
+    }
+    // 设置机器人ID
+    if (regular$9.test(e.msg)) {
+        const match = e.msg.match(regular$9);
+        if (!match) {
+            message.send(format(Text("格式错误，请按照 格式：/设置机器人id <ID> 进行设置")));
+            return;
+        }
+        const [, , , , id] = match;
+        await setBotID(id.trim());
+        message.send(format(Text(`已设置机器人ID为 ${id} !`)));
+        return;
+    }
+    // 无匹配则继续下一个响应器
+    next();
+});
+
+export { res as default, regular };

package/lib/response/zreply/res.js

@@ -0,0 +1,186 @@
+import { getAIConfig, getAIChatHistory, getAffectionLevelAll, incrementAffectionLevel, addAIChatHistory } from '../../config.js';
+import { emojiMap } from '../../emoji.js';
+import { useMessage, Text } from 'alemonjs';
+import { log } from 'console';
+
+const selects = onSelects(["message.create", "private.message.create"]);
+const regular = /.*/;
+var res = onResponse(selects, async (e, next) => {
+    console.log("收到消息");
+    // 如果是命令则跳过
+    if (e.msg.startsWith("/") || e.msg.startsWith("#")) {
+        next();
+        return;
+    }
+    // 创建消息
+    const [message] = useMessage(e);
+    const config = await getAIConfig(e.guid);
+    const messages = await getAIChatHistory(e.guid);
+    const affections = await getAffectionLevelAll(e.guid);
+    const imgs = [];
+    let rawMessage = e.value.raw_message || e.msg;
+    // 如果没有配置AI,则不处理
+    if (config.host.trim() === "" ||
+        config.key.trim() === "" ||
+        config.model.trim() === "") {
+        return;
+    }
+    console.log("e.reply", e.reply);
+    // 处理回复图片
+    if (e.reply) {
+        const replyImages = e.reply.message
+            .filter((item) => item.type === "image")
+            .map((item) => item.data.url);
+        imgs.push(...replyImages);
+    }
+    // 处理图片
+    if (e.img && e.img.length > 0) {
+        console.log("有图片");
+        for (const imgUrl of e.img) {
+            imgs.push(imgUrl);
+        }
+    }
+    // 处理图片链接, 转为base64格式
+    // 下载图片并转换为 Base64
+    // const imageBase64 = await Promise.all(
+    //   imgs.map(async (imageUrl) => {
+    //     try {
+    //       // 下载图片到本地
+    //       const response = await fetch(imageUrl);
+    //       const buffer = await response.arrayBuffer();
+    //       const base64 = Buffer.from(buffer).toString("base64");
+    //       return `${base64}`;
+    //     } catch (error) {
+    //       console.error("Error downloading image:", error);
+    //       return null;
+    //     }
+    //   }),
+    // );
+    // // 过滤掉下载失败的图片
+    // const validImageBase64 = imageBase64.filter((base64) => base64 !== null);
+    // 处理艾特
+    if (e.at && e.at.length > 0) {
+        console.log("有艾特");
+        for (const atItem of e.at) {
+            const qq = atItem.data.qq;
+            const nickname = atItem.data.nickname || "未知昵称";
+            const cqAt = `[CQ:at,qq=${qq}]`;
+            rawMessage = rawMessage.replace(cqAt, `@${nickname}`);
+        }
+    }
+    rawMessage = rawMessage.replace(/\[CQ:image,[^\]]+\]/g, "[图片]");
+    rawMessage = rawMessage.replace(/\[CQ:reply,[^\]]+\]/g, "");
+    // 处理表情,转为[emojiMap[id]]
+    rawMessage = rawMessage.replace(/\[CQ:face,id=(\d+)\]/g, (_match, p1) => {
+        return `[${emojiMap[p1]}]`;
+    });
+    const usermessage = {
+        role: "user",
+        content: [
+            ...imgs.map((url) => ({
+                type: "image_url",
+                image_url: {
+                    url: url,
+                },
+            })),
+            {
+                type: "text",
+                text: JSON.stringify({
+                    [`${e.nickname}(${getTimeString()})`]: rawMessage,
+                }),
+            },
+        ],
+    };
+    // 当e.at不存在时,只给20%概率继续执行, 当e.at存在时,概率为40%,当e.atBot存在时,概率为100%
+    const rand = Math.random();
+    if ((!e.at && rand > 0.2) || (e.at && !e.atBot && rand > 0.4)) {
+        console.log("跳过AI回复");
+        return;
+    }
+    console.log("处理后的消息", JSON.stringify(usermessage, null, 2));
+    console.log("开始回复");
+    console.log("好感度列表", affections);
+    // 聊天
+    const res = await fetch(config.host, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${config.key}`,
+        },
+        body: JSON.stringify({
+            model: config.model,
+            messages: [
+                {
+                    role: "system",
+                    content: `你正在一个群聊中跟用户聊天,你的聊天昵称是'小咸鱼'
+用户的发言方式为: [{'用户昵称(用户id)(发送时间)':回复内容}]
+当该用户为主人时,昵称前面会出现[主人]标识
+你的回复方式为JSON: [{assistant:回复内容}],你可以在数组中加入多条消息来模拟更真实的聊天
+当然你觉得当前聊的话题确定与你有关是才回复,否则你就回复[]来表示不回复.请尽可能降低回复频率.
+当你认为对方在示好或骂你时,可以在回复中加入[好感+数字]或[好感-数字]来调整与用户的好感度。
+例如: [好感+1]表示增加1点好感度，[好感-1]表示减少1点好感度。你可以根据好感度来适当改变回复语气
+当对方希望和好时提出道歉条件,可以给对方一个恢复的机会.
+当前群时间:${getTimeString()}\n
+当前用户的好感度列表如下:\n` + JSON.stringify(affections),
+                },
+                {
+                    role: "system",
+                    content: config.systemPrompt || "",
+                },
+                ...messages,
+                { ...usermessage },
+            ],
+        }),
+    });
+    if (!res.ok) {
+        // message.send(
+        //   format(Text(`请求失败，状态码：${res.status}，请检查AI配置是否正确`)),
+        // );
+        log("请求AI失败:", res);
+        return;
+    }
+    const data = await res.json();
+    const reply = data.choices?.[0]?.message?.content || "AI未返回内容";
+    log("AI回复内容:", reply);
+    // 提取好感变化, 使用incrementAffectionLevel更新好感
+    const affectionChangeRegex = /\[好感([+-]\d+)\]/g;
+    let match;
+    while ((match = affectionChangeRegex.exec(reply)) !== null) {
+        const change = parseInt(match[1], 10);
+        console.log(`检测到好感变化: ${change}`);
+        await incrementAffectionLevel(e.uidkey, change); // 更新好感度
+    }
+    // 记录对话
+    await addAIChatHistory(e.guid, "user", JSON.stringify(usermessage.content));
+    await addAIChatHistory(e.guid, "assistant", reply);
+    // 解析回复内容
+    try {
+        // [{"assistant":"[好感+1]哼，笨蛋主人，这种馊主意也想得出来？会长那家伙肉肯定又老又柴，喵~你想吃就自己上啊，本小姐才不帮你！"}]
+        const aiReplys = JSON.parse(reply);
+        for (let i = 0; i < aiReplys.length; i++) {
+            const aiReply = aiReplys[i];
+            if (aiReply.assistant) {
+                if (i > 0) {
+                    // 从第二个回复开始，添加1-4秒随机延迟
+                    const delay = Math.floor(Math.random() * 3000) + 1000;
+                    await new Promise((resolve) => setTimeout(resolve, delay));
+                }
+                message.send(format(Text(aiReply.assistant.replace(/\[CQ:[^\]]+\]/, "").trim())));
+            }
+        }
+    }
+    catch (error) {
+        console.log("解析AI回复失败，发送原始内容");
+        message.send(format(Text(reply)));
+    }
+});
+function getTimeString() {
+    const now = new Date();
+    const month = String(now.getMonth() + 1).padStart(2, "0");
+    const day = String(now.getDate()).padStart(2, "0");
+    const hours = String(now.getHours()).padStart(2, "0");
+    const minutes = String(now.getMinutes()).padStart(2, "0");
+    return `${month}-${day} ${hours}:${minutes}`;
+}
+
+export { res as default, regular, selects };

package/lib/types.js

@@ -0,0 +1 @@
+

package/package.json

@@ -0,0 +1,73 @@
+{
+  "name": "alemonjs-aichat",
+  "version": "1.0.0",
+  "description": "alemonjs-aichat",
+  "author": "suancaixianyu",
+  "license": "MIT",
+  "main": "lib/index.js",
+  "type": "module",
+  "scripts": {
+    "dev": "npx lvy dev",
+    "build": "npx lvy build",
+    "review": "npx alemonc start",
+    "start": "npx pm2 startOrRestart pm2.config.cjs",
+    "stop": "npx pm2 stop pm2.config.cjs",
+    "delete": "npx pm2 delete pm2.config.cjs"
+  },
+  "devDependencies": {
+    "@alemonjs/db": "^0.0.13",
+    "@types/node": "^22",
+    "alemonjs": "2.1.14",
+    "cssnano": "^7",
+    "jsxp": "^1.2.3",
+    "lvyjs": "^0.2.25",
+    "pm2": "^5",
+    "tailwindcss": "3",
+    "@alemonjs/bubble": "^2.1.0-alpha.12",
+    "@alemonjs/onebot": "^2.1.0-alpha.14"
+  },
+  "dependencies": {},
+  "export": {
+    ".": "./lib/index.js",
+    "./package": "./package.json",
+    "./desktop": "./lib/desktop.js"
+  },
+  "exports": {
+    ".": "./lib/index.js",
+    "./package": "./package.json",
+    "./desktop": "./lib/desktop.js"
+  },
+  "publishConfig": {
+    "registry": "https://registry.npmjs.org",
+    "access": "public"
+  },
+  "repository": {
+    "type": "gitee",
+    "url": "https://gitee.com/suancaixianyu/alemonjs-aichat.git"
+  },
+  "files": [
+    "lib/**/*",
+    "public/**/*",
+    "config/**/*"
+  ],
+  "keywords": [
+    "alemonjs"
+  ],
+  "bugs": "https://gitee.com/suancaixianyu/alemonjs-aichat/issues",
+  "alemonjs": {
+    "desktop": {
+      "commond": [
+        {
+          "name": "cheese",
+          "commond": "open.cheese"
+        }
+      ],
+      "sidebars": [
+        {
+          "name": "cheese",
+          "commond": "open.cheese"
+        }
+      ]
+    }
+  }
+}