From 95672989ac50551f2b2b54e71c503c441629574a 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 14:58:04 +0800 Subject: [PATCH] Dev (#82) 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 频道名称 - 增强向量数据库管理器的日志信息 --------- Co-authored-by: K2cr2O1 --- adapters/discord_adapter.py | 2 +- core/managers/vectordb_manager.py | 15 ++++++++++++++- plugins/ai_chat.py | 8 +++++++- plugins/knowledge_base.py | 10 ++++++++-- 4 files changed, 30 insertions(+), 5 deletions(-) diff --git a/adapters/discord_adapter.py b/adapters/discord_adapter.py index c4a72de..3809872 100644 --- a/adapters/discord_adapter.py +++ b/adapters/discord_adapter.py @@ -101,7 +101,7 @@ class DiscordAdapter(discord.Client if DISCORD_AVAILABLE else object): return try: - channel_name = "neobot_cross_platform" + channel_name = "neobot_discord_send" pubsub = redis_manager.redis.pubsub() await pubsub.subscribe(channel_name) diff --git a/core/managers/vectordb_manager.py b/core/managers/vectordb_manager.py index 3be5eb7..9d19a15 100644 --- a/core/managers/vectordb_manager.py +++ b/core/managers/vectordb_manager.py @@ -78,12 +78,13 @@ class VectorDBManager(Singleton): return False try: + logger.info(f"正在将 {len(texts)} 条记忆存入向量集合 {collection_name}...") collection.add( documents=texts, metadatas=metadatas, ids=ids ) - logger.debug(f"成功向集合 {collection_name} 添加 {len(texts)} 条记录") + logger.success(f"成功将记忆存入集合 {collection_name}") return True except Exception as e: logger.error(f"向集合 {collection_name} 添加记录失败: {e}") @@ -104,11 +105,23 @@ class VectorDBManager(Singleton): return {"documents": [], "metadatas": [], "distances": []} try: + logger.info(f"正在从向量集合 {collection_name} 中检索相关记忆...") results = collection.query( query_texts=query_texts, n_results=n_results, where=where ) + + # 统计检索到的结果数量 + doc_count = 0 + if results and results.get("documents") and results["documents"][0]: + doc_count = len(results["documents"][0]) + + if doc_count > 0: + logger.success(f"成功从集合 {collection_name} 检索到 {doc_count} 条相关记忆") + else: + logger.info(f"集合 {collection_name} 中未检索到相关记忆") + return results except Exception as e: logger.error(f"查询集合 {collection_name} 失败: {e}") diff --git a/plugins/ai_chat.py b/plugins/ai_chat.py index 1e94bcf..92c7c5f 100644 --- a/plugins/ai_chat.py +++ b/plugins/ai_chat.py @@ -12,6 +12,12 @@ from core.config_loader import global_config logger = ModuleLogger("AIChat") +__plugin_meta__ = { + "name": "AI 聊天", + "description": "支持向量数据库记忆功能的 AI 聊天助手", + "usage": "/chat <内容> - 与 AI 进行对话" +} + # 尝试导入 OpenAI 客户端 try: from openai import AsyncOpenAI @@ -97,7 +103,7 @@ async def get_ai_response(user_id: int, group_id: int, user_message: str) -> str logger.error(f"AI 聊天请求失败: {e}") return f"请求失败: {str(e)}" -@matcher.command("chat", "聊天") +@matcher.command("chat") async def chat_command(event: GroupMessageEvent | PrivateMessageEvent, args: list[str]): """AI 聊天命令""" if not args: diff --git a/plugins/knowledge_base.py b/plugins/knowledge_base.py index 88dd5e2..af85440 100644 --- a/plugins/knowledge_base.py +++ b/plugins/knowledge_base.py @@ -12,7 +12,13 @@ from core.permission import Permission logger = ModuleLogger("GroupKnowledgeBase") -@matcher.command("kb_add", "添加知识库", permission=Permission.ADMIN) +__plugin_meta__ = { + "name": "群聊知识库", + "description": "基于向量数据库的群聊知识库，支持语义检索", + "usage": "/kb_add <问题> <答案> - 添加知识库条目 (仅管理员)\n/kb_search <关键词> - 搜索知识库" +} + +@matcher.command("kb_add", permission=Permission.ADMIN) async def kb_add_command(event: GroupMessageEvent, args: list[str]): """添加知识库条目""" if len(args) < 2: @@ -51,7 +57,7 @@ async def kb_add_command(event: GroupMessageEvent, args: list[str]): logger.error(f"添加知识库失败: {e}") await event.reply(f"添加失败: {str(e)}") -@matcher.command("kb_search", "搜索知识库") +@matcher.command("kb_search") async def kb_search_command(event: GroupMessageEvent, args: list[str]): """搜索知识库条目""" if not args: