diff --git a/scripts/count.py b/scripts/count.py new file mode 100644 index 000000000..a0e66dd0c --- /dev/null +++ b/scripts/count.py @@ -0,0 +1,16 @@ +def 计算字符串长度(输入字符串: str) -> int: + """计算输入字符串的长度 + + 参数: + 输入字符串: 要计算长度的字符串 + + 返回: + 字符串的长度(整数) + """ + return len(输入字符串) + +if __name__ == "__main__": + # 测试代码 + 测试字符串 = '''你。''' + print(f"字符串 '{测试字符串}' 的长度是: {计算字符串长度(测试字符串)}") + diff --git a/src/heart_flow/observation.py b/src/heart_flow/observation.py index 040ba9af0..6cd6ed24e 100644 --- a/src/heart_flow/observation.py +++ b/src/heart_flow/observation.py @@ -35,6 +35,7 @@ class ChattingObservation(Observation): self.talking_message = [] self.talking_message_str = "" + self.talking_message_str_truncate = "" self.name = global_config.BOT_NICKNAME self.nick_name = global_config.BOT_ALIAS_NAMES @@ -145,6 +146,12 @@ class ChattingObservation(Observation): timestamp_mode="normal", read_mark=last_obs_time_mark, ) + self.talking_message_str_truncate = await build_readable_messages( + messages=self.talking_message, + timestamp_mode="normal", + read_mark=last_obs_time_mark, + truncate=True, + ) logger.trace( f"Chat {self.chat_id} - 压缩早期记忆：{self.mid_memory_info}\n现在聊天内容：{self.talking_message_str}" diff --git a/src/plugins/heartFC_chat/heartFC_chat.py b/src/plugins/heartFC_chat/heartFC_chat.py index a3a5594f3..00cc27cd9 100644 --- a/src/plugins/heartFC_chat/heartFC_chat.py +++ b/src/plugins/heartFC_chat/heartFC_chat.py @@ -745,7 +745,7 @@ class HeartFChatting: if is_re_planned: await observation.observe() observed_messages = observation.talking_message - observed_messages_str = observation.talking_message_str + observed_messages_str = observation.talking_message_str_truncate # --- 使用 LLM 进行决策 --- # reasoning = "默认决策或获取决策失败" diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index 1c4458846..9eb6d6fd3 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -32,8 +32,8 @@ def init_prompt(): {current_mind_info} 因为上述想法，你决定发言，原因是：{reason} -回复尽量简短一些。请注意把握聊天内容，不要回复的太有条理，可以有个性。请一次只回复一个话题，不要同时回复多个人，不用指出你回复的是谁。{prompt_ger} -请回复的平淡一些，简短一些，说中文，不要刻意突出自身学科背景，不要说你说过的话题 ，注意只输出回复内容。 +回复尽量简短一些。请注意把握聊天内容，{reply_style2}。请一次只回复一个话题，不要同时回复多个人。{prompt_ger} +{reply_style1}，说中文，不要刻意突出自身学科背景，注意只输出回复内容。 {moderation_prompt}。注意：回复不要输出多余内容(包括前后缀，冒号和引号，括号，表情包，at或 @等 )。""", "heart_flow_prompt", ) @@ -183,6 +183,7 @@ class PromptBuilder: merge_messages=False, timestamp_mode="normal", read_mark=0.0, + truncate=True, ) # 中文高手(新加的好玩功能) @@ -192,6 +193,30 @@ class PromptBuilder: if random.random() < 0.02: prompt_ger += "你喜欢用反问句" + reply_styles1 = [ + ("给出日常且口语化的回复，平淡一些", 0.4), # 40%概率 + ("给出非常简短的回复", 0.4), # 40%概率 + ("给出缺失主语的回复，简短", 0.15), # 15%概率 + ("给出带有语病的回复，朴实平淡", 0.05) # 5%概率 + ] + reply_style1_chosen = random.choices( + [style[0] for style in reply_styles1], + weights=[style[1] for style in reply_styles1], + k=1 + )[0] + + reply_styles2 = [ + ("不要回复的太有条理，可以有个性", 0.6), # 60%概率 + ("不要回复的太有条理，可以复读", 0.15), # 15%概率 + ("回复的认真一些", 0.2), # 20%概率 + ("可以回复单个表情符号", 0.05) # 5%概率 + ] + reply_style2_chosen = random.choices( + [style[0] for style in reply_styles2], + weights=[style[1] for style in reply_styles2], + k=1 + )[0] + if structured_info: structured_info_prompt = await global_prompt_manager.format_prompt( "info_from_tools", structured_info=structured_info @@ -214,6 +239,8 @@ class PromptBuilder: if chat_in_group else await global_prompt_manager.get_prompt_async("chat_target_private2"), current_mind_info=current_mind_info, + reply_style2=reply_style2_chosen, + reply_style1=reply_style1_chosen, reason=reason, prompt_ger=prompt_ger, moderation_prompt=await global_prompt_manager.get_prompt_async("moderation_prompt"), diff --git a/src/plugins/utils/chat_message_builder.py b/src/plugins/utils/chat_message_builder.py index f510365fa..7844572ab 100644 --- a/src/plugins/utils/chat_message_builder.py +++ b/src/plugins/utils/chat_message_builder.py @@ -144,7 +144,8 @@ async def _build_readable_messages_internal( messages: List[Dict[str, Any]], replace_bot_name: bool = True, merge_messages: bool = False, - timestamp_mode: str = "relative", # 新增参数控制时间戳格式 + timestamp_mode: str = "relative", + truncate: bool = False, ) -> Tuple[str, List[Tuple[float, str, str]]]: """ 内部辅助函数，构建可读消息字符串和原始消息详情列表。 @@ -154,6 +155,7 @@ async def _build_readable_messages_internal( replace_bot_name: 是否将机器人的 user_id 替换为 "我"。 merge_messages: 是否合并来自同一用户的连续消息。 timestamp_mode: 时间戳的显示模式 ('relative', 'absolute', etc.)。传递给 translate_timestamp_to_human_readable。 + truncate: 是否根据消息的新旧程度截断过长的消息内容。 Returns: 包含格式化消息的字符串和原始消息详情列表 (时间戳, 发送者名称, 内容) 的元组。 @@ -161,7 +163,7 @@ async def _build_readable_messages_internal( if not messages: return "", [] - message_details: List[Tuple[float, str, str]] = [] + message_details_raw: List[Tuple[float, str, str]] = [] # 1 & 2: 获取发送者信息并提取消息组件 for msg in messages: @@ -177,7 +179,6 @@ async def _build_readable_messages_internal( # 检查必要信息是否存在 if not all([platform, user_id, timestamp is not None]): - # logger.warning(f"Skipping message due to missing info: {msg.get('_id', 'N/A')}") continue person_id = person_info_manager.get_person_id(platform, user_id) @@ -196,12 +197,38 @@ async def _build_readable_messages_internal( else: person_name = "某人" - message_details.append((timestamp, person_name, content)) + message_details_raw.append((timestamp, person_name, content)) - if not message_details: + if not message_details_raw: return "", [] - message_details.sort(key=lambda x: x[0]) # 按时间戳(第一个元素)升序排序，越早的消息排在前面 + message_details_raw.sort(key=lambda x: x[0]) # 按时间戳(第一个元素)升序排序，越早的消息排在前面 + + # 应用截断逻辑 (如果 truncate 为 True) + message_details: List[Tuple[float, str, str]] = [] + n_messages = len(message_details_raw) + if truncate and n_messages > 0: + for i, (timestamp, name, content) in enumerate(message_details_raw): + percentile = i / n_messages # 计算消息在列表中的位置百分比 (0 <= percentile < 1) + original_len = len(content) + limit = -1 # 默认不截断 + + if percentile < 0.6: # 60% 之前的消息 (即最旧的 60%) + limit = 170 + elif percentile < 0.8: # 60% 到 80% 之前的消息 (即中间的 20%) + limit = 250 + elif percentile < 1.0: # 80% 到 100% 之前的消息 (即较新的 20%) + limit = 500 + # 最新的 20% (理论上 percentile 会趋近 1，但这里不需要显式处理，因为 limit 默认为 -1) + + truncated_content = content + if limit > 0 and original_len > limit: + truncated_content = f"{content[:limit]}......（内容太长）" + + message_details.append((timestamp, name, truncated_content)) + else: + # 如果不截断，直接使用原始列表 + message_details = message_details_raw # 3: 合并连续消息 (如果 merge_messages 为 True) merged_messages = [] @@ -250,16 +277,21 @@ async def _build_readable_messages_internal( for line in merged["content"]: stripped_line = line.strip() if stripped_line: # 过滤空行 - # 移除末尾句号，添加分号 + # 移除末尾句号，添加分号 - 这个逻辑似乎有点奇怪，暂时保留 if stripped_line.endswith("。"): stripped_line = stripped_line[:-1] - output_lines.append(f"{stripped_line};") + # 如果内容被截断，结尾已经是 ...（内容太长），不再添加分号 + if not stripped_line.endswith("（内容太长）"): + output_lines.append(f"{stripped_line};") + else: + output_lines.append(stripped_line) # 直接添加截断后的内容 output_lines.append("\n") # 在每个消息块后添加换行，保持可读性 # 移除可能的多余换行，然后合并 formatted_string = "".join(output_lines).strip() - # 返回格式化后的字符串和原始的 message_details 列表 + # 返回格式化后的字符串和 *应用截断后* 的 message_details 列表 + # 注意：如果外部调用者需要原始未截断的内容，可能需要调整返回策略 return formatted_string, message_details @@ -268,13 +300,14 @@ async def build_readable_messages_with_list( replace_bot_name: bool = True, merge_messages: bool = False, timestamp_mode: str = "relative", + truncate: bool = False, ) -> Tuple[str, List[Tuple[float, str, str]]]: """ 将消息列表转换为可读的文本格式，并返回原始(时间戳, 昵称, 内容)列表。 允许通过参数控制格式化行为。 """ formatted_string, details_list = await _build_readable_messages_internal( - messages, replace_bot_name, merge_messages, timestamp_mode + messages, replace_bot_name, merge_messages, timestamp_mode, truncate ) return formatted_string, details_list @@ -285,6 +318,7 @@ async def build_readable_messages( merge_messages: bool = False, timestamp_mode: str = "relative", read_mark: float = 0.0, + truncate: bool = False, ) -> str: """ 将消息列表转换为可读的文本格式。 @@ -294,7 +328,7 @@ async def build_readable_messages( if read_mark <= 0: # 没有有效的 read_mark，直接格式化所有消息 formatted_string, _ = await _build_readable_messages_internal( - messages, replace_bot_name, merge_messages, timestamp_mode + messages, replace_bot_name, merge_messages, timestamp_mode, truncate ) return formatted_string else: @@ -303,11 +337,13 @@ async def build_readable_messages( messages_after_mark = [msg for msg in messages if msg.get("time", 0) > read_mark] # 分别格式化 + # 注意：这里决定对已读和未读部分都应用相同的 truncate 设置 + # 如果需要不同的行为（例如只截断已读部分），需要调整这里的调用 formatted_before, _ = await _build_readable_messages_internal( - messages_before_mark, replace_bot_name, merge_messages, timestamp_mode + messages_before_mark, replace_bot_name, merge_messages, timestamp_mode, truncate ) formatted_after, _ = await _build_readable_messages_internal( - messages_after_mark, replace_bot_name, merge_messages, timestamp_mode + messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, truncate ) readable_read_mark = translate_timestamp_to_human_readable(read_mark, mode=timestamp_mode)