better:优化关系构建

This commit is contained in:
SengokuCola
2025-07-16 11:29:11 +08:00
parent e54e4ced26
commit e0a6474416
3 changed files with 8 additions and 14 deletions

View File

@@ -237,7 +237,7 @@ class HeartFChatting:
if if_think: if if_think:
self.energy_value *= 1.1 / (global_config.chat.focus_value + 0.2) self.energy_value *= 1.1 / (global_config.chat.focus_value + 0.2)
logger.info(f"{self.log_prefix} 麦麦进行了思考能量值增加1当前能量值{self.energy_value}") logger.info(f"{self.log_prefix} 麦麦进行了思考能量值增加1当前能量值{self.energy_value}")
return False return True
await asyncio.sleep(1) await asyncio.sleep(1)

View File

@@ -302,7 +302,7 @@ class DefaultReplyer:
traceback.print_exc() traceback.print_exc()
return False, None return False, None
async def build_relation_info(self, reply_data=None, chat_history=None): async def build_relation_info(self, reply_data=None):
if not global_config.relationship.enable_relationship: if not global_config.relationship.enable_relationship:
return "" return ""
@@ -321,7 +321,7 @@ class DefaultReplyer:
logger.warning(f"{self.log_prefix} 未找到用户 {sender} 的ID跳过信息提取") logger.warning(f"{self.log_prefix} 未找到用户 {sender} 的ID跳过信息提取")
return f"你完全不认识{sender}不理解ta的相关信息。" return f"你完全不认识{sender}不理解ta的相关信息。"
return await relationship_fetcher.build_relation_info(person_id, text, chat_history) return await relationship_fetcher.build_relation_info(person_id, points_num=5)
async def build_expression_habits(self, chat_history, target): async def build_expression_habits(self, chat_history, target):
if not global_config.expression.enable_expression: if not global_config.expression.enable_expression:
@@ -619,7 +619,7 @@ class DefaultReplyer:
self.build_expression_habits(chat_talking_prompt_short, target), "build_expression_habits" self.build_expression_habits(chat_talking_prompt_short, target), "build_expression_habits"
), ),
self._time_and_run_task( self._time_and_run_task(
self.build_relation_info(reply_data, chat_talking_prompt_short), "build_relation_info" self.build_relation_info(reply_data), "build_relation_info"
), ),
self._time_and_run_task(self.build_memory_block(chat_talking_prompt_short, target), "build_memory_block"), self._time_and_run_task(self.build_memory_block(chat_talking_prompt_short, target), "build_memory_block"),
self._time_and_run_task( self._time_and_run_task(
@@ -806,7 +806,7 @@ class DefaultReplyer:
# 并行执行2个构建任务 # 并行执行2个构建任务
expression_habits_block, relation_info = await asyncio.gather( expression_habits_block, relation_info = await asyncio.gather(
self.build_expression_habits(chat_talking_prompt_half, target), self.build_expression_habits(chat_talking_prompt_half, target),
self.build_relation_info(reply_data, chat_talking_prompt_half), self.build_relation_info(reply_data),
) )
keywords_reaction_prompt = await self.build_keywords_reaction_prompt(target) keywords_reaction_prompt = await self.build_keywords_reaction_prompt(target)

View File

@@ -96,7 +96,7 @@ class RelationshipFetcher:
if not self.info_fetched_cache[person_id]: if not self.info_fetched_cache[person_id]:
del self.info_fetched_cache[person_id] del self.info_fetched_cache[person_id]
async def build_relation_info(self, person_id, target_message, chat_history): async def build_relation_info(self, person_id, points_num = 3):
# 清理过期的信息缓存 # 清理过期的信息缓存
self._cleanup_expired_cache() self._cleanup_expired_cache()
@@ -124,13 +124,13 @@ class RelationshipFetcher:
# 按时间排序forgotten_points # 按时间排序forgotten_points
current_points.sort(key=lambda x: x[2]) current_points.sort(key=lambda x: x[2])
# 按权重加权随机抽取最多3个不重复的pointspoint[1]的值在1-10之间权重越高被抽到概率越大 # 按权重加权随机抽取最多3个不重复的pointspoint[1]的值在1-10之间权重越高被抽到概率越大
if len(current_points) > 3: if len(current_points) > points_num:
# point[1] 取值范围1-10直接作为权重 # point[1] 取值范围1-10直接作为权重
weights = [max(1, min(10, int(point[1]))) for point in current_points] weights = [max(1, min(10, int(point[1]))) for point in current_points]
# 使用加权采样不放回,保证不重复 # 使用加权采样不放回,保证不重复
indices = list(range(len(current_points))) indices = list(range(len(current_points)))
points = [] points = []
for _ in range(3): for _ in range(points_num):
if not indices: if not indices:
break break
sub_weights = [weights[i] for i in indices] sub_weights = [weights[i] for i in indices]
@@ -143,12 +143,6 @@ class RelationshipFetcher:
# 构建points文本 # 构建points文本
points_text = "\n".join([f"{point[2]}{point[0]}" for point in points]) points_text = "\n".join([f"{point[2]}{point[0]}" for point in points])
# info_type = await self._build_fetch_query(person_id, target_message, chat_history)
# if info_type:
# await self._extract_single_info(person_id, info_type, person_name)
# relation_info = self._organize_known_info()
nickname_str = "" nickname_str = ""
if person_name != nickname_str: if person_name != nickname_str:
nickname_str = f"(ta在{platform}上的昵称是{nickname_str})" nickname_str = f"(ta在{platform}上的昵称是{nickname_str})"