refactor(插件): 优化插件元信息和命令配置
- 为 AI 聊天和知识库插件添加元信息配置 - 简化插件命令配置,移除冗余别名 - 更新 Discord 适配器的 Redis 频道名称 - 增强向量数据库管理器的日志信息
This commit is contained in:
@@ -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}")
|
||||
|
||||
Reference in New Issue
Block a user