From ec165bbd3169c92bc12b1b56f9724558aa2e921e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=95=80=E9=93=AC=E9=85=B8=E9=92=BE?= <148796996+K2cr2O1@users.noreply.github.com> Date: Tue, 24 Mar 2026 15:18:55 +0800 Subject: [PATCH] Dev (#83) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * fix(discord): 修复 WebSocket 连接检测并增强跨平台文件处理 修复 Discord WebSocket 连接检测逻辑，使用正确的属性检查连接状态 为跨平台消息处理添加文件类型支持，并增加详细的调试日志 优化附件处理逻辑，确保所有文件类型都能正确识别和转发 * feat(跨平台): 优化消息处理并添加纯文本提取功能 添加 extract_text_only 函数过滤非文本标记 修改翻译逻辑仅处理纯文本内容 完善附件处理和消息内容拼接 修复仅包含表情时的消息处理问题 * refactor(discord-cross): 使用模块专用日志记录器替换全局日志记录器 将各模块中的全局日志记录器替换为模块专用日志记录器，以提供更清晰的日志来源标识 同时在适配器中添加会话状态检查和重连机制，提升消息发送的可靠性 * feat(翻译): 改进翻译功能，同时显示原文和译文 修改翻译功能，不再替换原文而是同时显示原文和翻译内容，方便用户对照 更新 DeepSeek API 配置为官方地址和模型 优化 Discord 适配器的重连逻辑，直接关闭 WebSocket 触发重连 修复 Discord 频道 ID 转换逻辑，简化处理流程 * feat(cross-platform): 添加跨平台功能支持及配置优化 - 新增跨平台配置模型和全局配置支持 - 优化 Discord 适配器的连接管理和错误处理 - 添加 watchdog 和 discord.py 依赖 - 创建 DeepSeek API 配置文档 - 移除重复的同步帮助图片代码 - 改进跨平台插件配置加载逻辑 * fix(jrcd): 修正群组ID检查条件 删除不再使用的示例插件文件 * feat: 改进配置加载逻辑并更新项目配置 当配置文件不存在时自动生成示例配置 添加pyproject.toml作为项目构建配置 更新.gitignore忽略更多文件类型 删除不再使用的反向WebSocket示例文件 * docs: 更新架构文档和项目结构说明 添加反向WebSocket连接模式说明 补充核心管理器文档 更新项目结构文件 在文档首页添加特色功能说明 * fix(discord): 修复WebSocket连接检查并添加错误日志 refactor(config): 更新配置文件的网络和认证信息 feat(cross-platform): 为跨平台消息处理添加异常捕获和日志 * fix(discord-cross): 修复跨平台消息处理和附件下载问题 修复QQ群消息处理中的非群消息过滤问题 优化Discord附件下载逻辑，使用aiohttp替代requests 修复Redis订阅任务重复创建问题 调整消息格式化的embed字段处理逻辑 * feat(vectordb): 添加向量数据库支持及集成功能 新增向量数据库管理器模块，支持文本的存储、检索和相似度查询 添加知识库插件和AI聊天插件，利用向量数据库实现记忆功能 优化跨平台翻译模块，集成向量数据库存储历史翻译记录 改进消息处理逻辑，优先使用用户显示名称 * feat(plugins): add furry_assistant plugin by Calgau - Add furry assistant plugin with 7 commands - Include furry greetings, fortunes, jokes, and advice - Add plugin metadata and README documentation - Implement plugin lifecycle methods - Created by Calgau (furry AI assistant) * fix: 调整昵称和用户名的获取优先级 修改QQ群消息处理中昵称获取顺序，优先使用昵称而非群名片 移除Discord消息转换中global_name的检查，直接使用用户名 * refactor(插件): 优化插件元信息和命令配置 - 为 AI 聊天和知识库插件添加元信息配置 - 简化插件命令配置，移除冗余别名 - 更新 Discord 适配器的 Redis 频道名称 - 增强向量数据库管理器的日志信息 * feat(ai_chat): 添加Markdown渲染和图片生成功能 支持将AI回复的Markdown内容转换为HTML并渲染为美观的图片格式返回，提升聊天体验 ``` ```msg feat(knowledge_base): 扩展知识库支持个人和群聊独立记忆 - 新增个人知识库功能，支持独立记忆 - 添加清除个人/群聊记忆命令 - 优化知识搜索逻辑，优先搜索个人记忆 - 更新插件帮助信息 --------- Co-authored-by: K2cr2O1 --- plugins/ai_chat.py | 36 ++++++- plugins/knowledge_base.py | 154 +++++++++++++++++++++----- templates/ai_chat.html | 220 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 384 insertions(+), 26 deletions(-) create mode 100644 templates/ai_chat.html diff --git a/plugins/ai_chat.py b/plugins/ai_chat.py index 92c7c5f..3ff1c39 100644 --- a/plugins/ai_chat.py +++ b/plugins/ai_chat.py @@ -4,9 +4,12 @@ AI 聊天插件，支持向量数据库记忆功能 """ import time import uuid +import markdown from core.managers.command_manager import matcher from models.events.message import GroupMessageEvent, PrivateMessageEvent +from models.message import MessageSegment from core.managers.vectordb_manager import vectordb_manager +from core.managers.image_manager import image_manager from core.utils.logger import ModuleLogger from core.config_loader import global_config @@ -113,7 +116,38 @@ async def chat_command(event: GroupMessageEvent | PrivateMessageEvent, args: lis user_message = " ".join(args) user_id = event.user_id group_id = getattr(event, 'group_id', 0) + user_name = event.sender.nickname or event.sender.card or str(user_id) await event.reply("正在思考中...") reply = await get_ai_response(user_id, group_id, user_message) - await event.reply(reply) + + # 将 Markdown 转换为 HTML + try: + # 启用扩展以支持代码块、表格等 + html_reply = markdown.markdown(reply, extensions=['fenced_code', 'tables', 'nl2br']) + except Exception as e: + logger.error(f"Markdown 转换失败: {e}") + html_reply = reply.replace('\n', '
') + + # 渲染图片 + try: + template_data = { + "user_name": user_name, + "user_message": user_message, + "ai_reply": html_reply + } + + base64_img = await image_manager.render_template_to_base64( + template_name="ai_chat.html", + data=template_data, + output_name=f"chat_{user_id}_{int(time.time())}.png", + image_type="png" + ) + + if base64_img: + await event.reply(MessageSegment.image(f"base64://{base64_img}")) + else: + await event.reply("图片生成失败，返回文本：\n" + reply) + except Exception as e: + logger.error(f"渲染聊天图片失败: {e}") + await event.reply("图片生成失败，返回文本：\n" + reply) diff --git a/plugins/knowledge_base.py b/plugins/knowledge_base.py index af85440..71db0c5 100644 --- a/plugins/knowledge_base.py +++ b/plugins/knowledge_base.py @@ -5,7 +5,7 @@ import time import uuid from core.managers.command_manager import matcher -from models.events.message import GroupMessageEvent +from models.events.message import GroupMessageEvent, PrivateMessageEvent from core.managers.vectordb_manager import vectordb_manager from core.utils.logger import ModuleLogger from core.permission import Permission @@ -13,24 +13,62 @@ from core.permission import Permission logger = ModuleLogger("GroupKnowledgeBase") __plugin_meta__ = { - "name": "群聊知识库", - "description": "基于向量数据库的群聊知识库，支持语义检索", - "usage": "/kb_add <问题> <答案> - 添加知识库条目 (仅管理员)\n/kb_search <关键词> - 搜索知识库" + "name": "知识库", + "description": "基于向量数据库的知识库，支持个人和群聊独立记忆", + "usage": "/kb_add <问题> <答案> - 添加个人知识库\n/kb_add_group <问题> <答案> - 添加群聊知识库 (仅管理员)\n/kb_search <关键词> - 搜索知识库\n/kb_remove_person - 清除个人所有记忆\n/kb_remove_group - 清除群聊所有记忆 (仅管理员)" } -@matcher.command("kb_add", permission=Permission.ADMIN) -async def kb_add_command(event: GroupMessageEvent, args: list[str]): - """添加知识库条目""" +@matcher.command("kb_add") +async def kb_add_person_command(event: GroupMessageEvent | PrivateMessageEvent, args: list[str]): + """添加个人知识库条目""" if len(args) < 2: await event.reply("用法: /kb_add <问题> <答案>") return + question = args[0] + answer = " ".join(args[1:]) + user_id = event.user_id + + try: + collection_name = f"knowledge_base_user_{user_id}" + doc_id = str(uuid.uuid4()) + + text_to_embed = f"问题: {question}\n答案: {answer}" + metadata = { + "user_id": user_id, + "question": question, + "answer": answer, + "timestamp": int(time.time()) + } + + success = vectordb_manager.add_texts( + collection_name=collection_name, + texts=[text_to_embed], + metadatas=[metadata], + ids=[doc_id] + ) + + if success: + await event.reply(f"个人知识库条目添加成功！\n问题: {question}") + else: + await event.reply("个人知识库条目添加失败，请查看日志。") + except Exception as e: + logger.error(f"添加个人知识库失败: {e}") + await event.reply(f"添加失败: {str(e)}") + +@matcher.command("kb_add_group", permission=Permission.ADMIN) +async def kb_add_group_command(event: GroupMessageEvent, args: list[str]): + """添加群聊知识库条目""" + if len(args) < 2: + await event.reply("用法: /kb_add_group <问题> <答案>") + return + question = args[0] answer = " ".join(args[1:]) group_id = event.group_id try: - collection_name = f"knowledge_base_{group_id}" + collection_name = f"knowledge_base_group_{group_id}" doc_id = str(uuid.uuid4()) text_to_embed = f"问题: {question}\n答案: {answer}" @@ -50,43 +88,109 @@ async def kb_add_command(event: GroupMessageEvent, args: list[str]): ) if success: - await event.reply(f"知识库条目添加成功！\n问题: {question}") + await event.reply(f"群聊知识库条目添加成功！\n问题: {question}") else: - await event.reply("知识库条目添加失败，请查看日志。") + await event.reply("群聊知识库条目添加失败，请查看日志。") except Exception as e: - logger.error(f"添加知识库失败: {e}") + logger.error(f"添加群聊知识库失败: {e}") await event.reply(f"添加失败: {str(e)}") @matcher.command("kb_search") -async def kb_search_command(event: GroupMessageEvent, args: list[str]): - """搜索知识库条目""" +async def kb_search_command(event: GroupMessageEvent | PrivateMessageEvent, args: list[str]): + """搜索知识库条目（优先搜索个人，再搜索群聊）""" if not args: await event.reply("用法: /kb_search <关键词>") return query = " ".join(args) - group_id = event.group_id + user_id = event.user_id + group_id = getattr(event, 'group_id', None) try: - collection_name = f"knowledge_base_{group_id}" + reply_msg = f"为您找到以下相关知识：\n" + found = False - results = vectordb_manager.query_texts( - collection_name=collection_name, + # 1. 搜索个人知识库 + person_collection = f"knowledge_base_user_{user_id}" + person_results = vectordb_manager.query_texts( + collection_name=person_collection, query_texts=[query], - n_results=3 + n_results=2 ) - if not results or not results.get("documents") or not results["documents"][0]: + if person_results and person_results.get("documents") and person_results["documents"][0]: + reply_msg += "\n【个人记忆】" + for i, metadata in enumerate(person_results["metadatas"][0], 1): + question = metadata.get("question", "") + answer = metadata.get("answer", "") + reply_msg += f"\n{i}. Q: {question}\n A: {answer}" + found = True + + # 2. 搜索群聊知识库 + if group_id: + group_collection = f"knowledge_base_group_{group_id}" + group_results = vectordb_manager.query_texts( + collection_name=group_collection, + query_texts=[query], + n_results=2 + ) + + if group_results and group_results.get("documents") and group_results["documents"][0]: + reply_msg += "\n\n【群聊记忆】" + for i, metadata in enumerate(group_results["metadatas"][0], 1): + question = metadata.get("question", "") + answer = metadata.get("answer", "") + reply_msg += f"\n{i}. Q: {question}\n A: {answer}" + found = True + + if not found: await event.reply("未找到相关的知识库条目。") return - reply_msg = f"为您找到以下相关知识：\n" - for i, metadata in enumerate(results["metadatas"][0], 1): - question = metadata.get("question", "") - answer = metadata.get("answer", "") - reply_msg += f"\n{i}. Q: {question}\n A: {answer}" - await event.reply(reply_msg) except Exception as e: logger.error(f"搜索知识库失败: {e}") await event.reply(f"搜索失败: {str(e)}") + +@matcher.command("kb_remove_person") +async def kb_remove_person_command(event: GroupMessageEvent | PrivateMessageEvent): + """清除个人所有记忆""" + user_id = event.user_id + collection_name = f"knowledge_base_user_{user_id}" + + try: + # ChromaDB 不支持直接删除整个 collection 的所有数据，最简单的方法是删除 collection + if vectordb_manager._client: + try: + vectordb_manager._client.delete_collection(collection_name) + if collection_name in vectordb_manager._collections: + del vectordb_manager._collections[collection_name] + await event.reply("已成功清除您的所有个人记忆。") + except ValueError: + await event.reply("您还没有任何个人记忆。") + else: + await event.reply("向量数据库未初始化。") + except Exception as e: + logger.error(f"清除个人记忆失败: {e}") + await event.reply(f"清除失败: {str(e)}") + +@matcher.command("kb_remove_group", permission=Permission.ADMIN) +async def kb_remove_group_command(event: GroupMessageEvent): + """清除群聊所有记忆""" + group_id = event.group_id + collection_name = f"knowledge_base_group_{group_id}" + + try: + if vectordb_manager._client: + try: + vectordb_manager._client.delete_collection(collection_name) + if collection_name in vectordb_manager._collections: + del vectordb_manager._collections[collection_name] + await event.reply("已成功清除本群的所有群聊记忆。") + except ValueError: + await event.reply("本群还没有任何群聊记忆。") + else: + await event.reply("向量数据库未初始化。") + except Exception as e: + logger.error(f"清除群聊记忆失败: {e}") + await event.reply(f"清除失败: {str(e)}") diff --git a/templates/ai_chat.html b/templates/ai_chat.html new file mode 100644 index 0000000..19576cf --- /dev/null +++ b/templates/ai_chat.html @@ -0,0 +1,220 @@ + + + + + + AI 聊天回复 + + + +
+
+
+
+
+
+
+
AI CHAT
+
+
+ +
+ +
+
+
{{ user_name[0] if user_name else 'U' }}
+
{{ user_name }}
+
+
{{ user_message }}
+
+ + +
+
+
AI
+
NeoBot AI
+
+
{{ ai_reply }}
+
+
+
+ + \ No newline at end of file