BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

还原神秘小修改

pull/937/head
Bakadax 2025-05-16 16:09:48 +08:00
parent 7d92b00d3e
commit c8b99a55ed
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/chat/heart_flow/subheartflow_manager.py
View File

@ -401,7 +401,7 @@ class SubHeartflowManager:
_mai_state_description = f"你当前状态: {current_mai_state.value}。" _mai_state_description = f"你当前状态: {current_mai_state.value}。"
individuality = Individuality.get_instance() individuality = Individuality.get_instance()
personality_prompt = individuality.get_prompt(x_person=2, level=3) personality_prompt = individuality.get_prompt(x_person=2, level=2)
prompt_personality = f"你正在扮演名为{individuality.name}的人类，{personality_prompt}" prompt_personality = f"你正在扮演名为{individuality.name}的人类，{personality_prompt}"
# --- 修改：在 prompt 中加入当前聊天计数和群名信息 (条件显示) --- # --- 修改：在 prompt 中加入当前聊天计数和群名信息 (条件显示) ---

6
src/chat/knowledge/src/qa_manager.py
View File

@ -84,9 +84,9 @@ class QAManager:
relation_search_res, paragraph_search_res, self.embed_manager relation_search_res, paragraph_search_res, self.embed_manager
) )
part_end_time = time.perf_counter() part_end_time = time.perf_counter()
logger.infoinfo(f"RAG检索用时：{part_end_time - part_start_time:.5f}s") logger.info(f"RAG检索用时：{part_end_time - part_start_time:.5f}s")
else: else:
logger.infoinfo("未找到相关关系，将使用文段检索结果") logger.info("未找到相关关系，将使用文段检索结果")
result = paragraph_search_res result = paragraph_search_res
ppr_node_weights = None ppr_node_weights = None
@ -95,7 +95,7 @@ class QAManager:
for res in result: for res in result:
raw_paragraph = self.embed_manager.paragraphs_embedding_store.store[res[0]].str raw_paragraph = self.embed_manager.paragraphs_embedding_store.store[res[0]].str
logger.info(f"找到相关文段，相关系数：{res[1]:.8f}\n{raw_paragraph}\n\n") print(f"找到相关文段，相关系数：{res[1]:.8f}\n{raw_paragraph}\n\n")
return result, ppr_node_weights return result, ppr_node_weights
else: else: