K2CrO4/NeoBot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Dev (#83)

Browse Source 
* 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 <indoec@163.com>
This commit is contained in:
镀铬酸钾
2026-03-24 15:18:55 +08:00
committed by GitHub
parent 95672989ac
commit ec165bbd31
3 changed files with 384 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
plugins/ai_chat.py
View File

@@ -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', '<br>')
# 渲染图片
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)

152
plugins/knowledge_base.py
View File

@@ -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]:
await event.reply("未找到相关的知识库条目。")
return
reply_msg = f"为您找到以下相关知识：\n"
for i, metadata in enumerate(results["metadatas"][0], 1):
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
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)}")

220
templates/ai_chat.html Normal file
View File

@@ -0,0 +1,220 @@
<!DOCTYPE html>
<html lang="zh-CN">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>AI 聊天回复</title>
<style>
@import url('https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;500&family=Noto+Sans+SC:wght@400;500;700&display=swap');
:root {
--bg-color: #0f172a;
--window-bg: rgba(30, 41, 59, 0.85);
--border-color: rgba(255, 255, 255, 0.08);
--accent: #6366f1;
--text-title: #f8fafc;
--text-desc: #94a3b8;
--text-content: #e2e8f0;
--user-bg: rgba(99, 102, 241, 0.15);
--ai-bg: rgba(16, 185, 129, 0.1);
--user-border: rgba(99, 102, 241, 0.3);
--ai-border: rgba(16, 185, 129, 0.3);
}
* {
margin: 0;
padding: 0;
box-sizing: border-box;
}
body {
font-family: 'Noto Sans SC', system-ui, sans-serif;
background-color: var(--bg-color);
color: var(--text-title);
display: flex;
justify-content: center;
padding: 20px;
min-height: 100vh;
-webkit-font-smoothing: antialiased;
}
.window {
width: 100%;
max-width: 800px;
background: var(--window-bg);
backdrop-filter: blur(20px);
border-radius: 16px;
border: 1px solid var(--border-color);
box-shadow: 0 25px 50px -12px rgba(0, 0, 0, 0.5);
overflow: hidden;
display: flex;
flex-direction: column;
}
.header {
padding: 20px 30px;
border-bottom: 1px solid var(--border-color);
background: rgba(255, 255, 255, 0.02);
display: flex;
justify-content: space-between;
align-items: center;
}
.dots { display: flex; gap: 8px; }
.dot { width: 12px; height: 12px; border-radius: 50%; }
.red { background: #ef4444; }
.yellow { background: #f59e0b; }
.green { background: #10b981; }
.title {
font-size: 16px;
font-weight: 700;
letter-spacing: 1px;
color: var(--text-desc);
}
.content {
padding: 30px;
display: flex;
flex-direction: column;
gap: 24px;
}
.message-card {
border-radius: 12px;
padding: 24px;
border: 1px solid;
}
.user-card {
background: var(--user-bg);
border-color: var(--user-border);
margin-right: 40px;
}
.ai-card {
background: var(--ai-bg);
border-color: var(--ai-border);
margin-left: 40px;
}
.message-header {
display: flex;
align-items: center;
gap: 12px;
margin-bottom: 16px;
padding-bottom: 12px;
border-bottom: 1px solid var(--border-color);
}
.avatar {
width: 32px;
height: 32px;
border-radius: 50%;
display: flex;
align-items: center;
justify-content: center;
font-weight: bold;
font-size: 14px;
}
.user-avatar {
background: #6366f1;
color: white;
}
.ai-avatar {
background: #10b981;
color: white;
}
.name {
font-size: 16px;
font-weight: 700;
color: var(--text-title);
}
.message-body {
font-size: 16px;
line-height: 1.6;
color: var(--text-content);
white-space: pre-wrap;
word-break: break-word;
}
/* Markdown 样式支持 */
.message-body code {
font-family: 'JetBrains Mono', monospace;
background: rgba(0, 0, 0, 0.3);
padding: 2px 6px;
border-radius: 4px;
font-size: 0.9em;
color: #a5f3fc;
}
.message-body pre {
background: rgba(0, 0, 0, 0.3);
padding: 16px;
border-radius: 8px;
overflow-x: auto;
margin: 12px 0;
}
.message-body pre code {
background: transparent;
padding: 0;
color: #e2e8f0;
}
.message-body p {
margin-bottom: 12px;
}
.message-body p:last-child {
margin-bottom: 0;
}
/* 数学公式支持 */
.math {
font-family: 'Times New Roman', Times, serif;
font-style: italic;
background: rgba(255, 255, 255, 0.05);
padding: 2px 4px;
border-radius: 4px;
}
</style>
</head>
<body>
<div class="window">
<div class="header">
<div class="dots">
<div class="dot red"></div>
<div class="dot yellow"></div>
<div class="dot green"></div>
</div>
<div class="title">AI CHAT</div>
<div style="width: 44px;"></div>
</div>
<div class="content">
<!-- 用户消息 -->
<div class="message-card user-card">
<div class="message-header">
<div class="avatar user-avatar">{{ user_name[0] if user_name else 'U' }}</div>
<div class="name">{{ user_name }}</div>
</div>
<div class="message-body">{{ user_message }}</div>
</div>
<!-- AI 回复 -->
<div class="message-card ai-card">
<div class="message-header">
<div class="avatar ai-avatar">AI</div>
<div class="name">NeoBot AI</div>
</div>
<div class="message-body">{{ ai_reply }}</div>
</div>
</div>
</div>
</body>
</html>