refactor(插件): 优化插件元信息和命令配置
- 为 AI 聊天和知识库插件添加元信息配置 - 简化插件命令配置,移除冗余别名 - 更新 Discord 适配器的 Redis 频道名称 - 增强向量数据库管理器的日志信息
This commit is contained in:
@@ -101,7 +101,7 @@ class DiscordAdapter(discord.Client if DISCORD_AVAILABLE else object):
|
|||||||
return
|
return
|
||||||
|
|
||||||
try:
|
try:
|
||||||
channel_name = "neobot_cross_platform"
|
channel_name = "neobot_discord_send"
|
||||||
pubsub = redis_manager.redis.pubsub()
|
pubsub = redis_manager.redis.pubsub()
|
||||||
await pubsub.subscribe(channel_name)
|
await pubsub.subscribe(channel_name)
|
||||||
|
|
||||||
|
|||||||
@@ -78,12 +78,13 @@ class VectorDBManager(Singleton):
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
logger.info(f"正在将 {len(texts)} 条记忆存入向量集合 {collection_name}...")
|
||||||
collection.add(
|
collection.add(
|
||||||
documents=texts,
|
documents=texts,
|
||||||
metadatas=metadatas,
|
metadatas=metadatas,
|
||||||
ids=ids
|
ids=ids
|
||||||
)
|
)
|
||||||
logger.debug(f"成功向集合 {collection_name} 添加 {len(texts)} 条记录")
|
logger.success(f"成功将记忆存入集合 {collection_name}")
|
||||||
return True
|
return True
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"向集合 {collection_name} 添加记录失败: {e}")
|
logger.error(f"向集合 {collection_name} 添加记录失败: {e}")
|
||||||
@@ -104,11 +105,23 @@ class VectorDBManager(Singleton):
|
|||||||
return {"documents": [], "metadatas": [], "distances": []}
|
return {"documents": [], "metadatas": [], "distances": []}
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
logger.info(f"正在从向量集合 {collection_name} 中检索相关记忆...")
|
||||||
results = collection.query(
|
results = collection.query(
|
||||||
query_texts=query_texts,
|
query_texts=query_texts,
|
||||||
n_results=n_results,
|
n_results=n_results,
|
||||||
where=where
|
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
|
return results
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"查询集合 {collection_name} 失败: {e}")
|
logger.error(f"查询集合 {collection_name} 失败: {e}")
|
||||||
|
|||||||
@@ -12,6 +12,12 @@ from core.config_loader import global_config
|
|||||||
|
|
||||||
logger = ModuleLogger("AIChat")
|
logger = ModuleLogger("AIChat")
|
||||||
|
|
||||||
|
__plugin_meta__ = {
|
||||||
|
"name": "AI 聊天",
|
||||||
|
"description": "支持向量数据库记忆功能的 AI 聊天助手",
|
||||||
|
"usage": "/chat <内容> - 与 AI 进行对话"
|
||||||
|
}
|
||||||
|
|
||||||
# 尝试导入 OpenAI 客户端
|
# 尝试导入 OpenAI 客户端
|
||||||
try:
|
try:
|
||||||
from openai import AsyncOpenAI
|
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}")
|
logger.error(f"AI 聊天请求失败: {e}")
|
||||||
return f"请求失败: {str(e)}"
|
return f"请求失败: {str(e)}"
|
||||||
|
|
||||||
@matcher.command("chat", "聊天")
|
@matcher.command("chat")
|
||||||
async def chat_command(event: GroupMessageEvent | PrivateMessageEvent, args: list[str]):
|
async def chat_command(event: GroupMessageEvent | PrivateMessageEvent, args: list[str]):
|
||||||
"""AI 聊天命令"""
|
"""AI 聊天命令"""
|
||||||
if not args:
|
if not args:
|
||||||
|
|||||||
@@ -12,7 +12,13 @@ from core.permission import Permission
|
|||||||
|
|
||||||
logger = ModuleLogger("GroupKnowledgeBase")
|
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]):
|
async def kb_add_command(event: GroupMessageEvent, args: list[str]):
|
||||||
"""添加知识库条目"""
|
"""添加知识库条目"""
|
||||||
if len(args) < 2:
|
if len(args) < 2:
|
||||||
@@ -51,7 +57,7 @@ async def kb_add_command(event: GroupMessageEvent, args: list[str]):
|
|||||||
logger.error(f"添加知识库失败: {e}")
|
logger.error(f"添加知识库失败: {e}")
|
||||||
await event.reply(f"添加失败: {str(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]):
|
async def kb_search_command(event: GroupMessageEvent, args: list[str]):
|
||||||
"""搜索知识库条目"""
|
"""搜索知识库条目"""
|
||||||
if not args:
|
if not args:
|
||||||
|
|||||||
Reference in New Issue
Block a user