diff --git a/src/plugins/chat/auto_speak.py b/src/plugins/chat/auto_speak.py new file mode 100644 index 000000000..38ca7d9f2 --- /dev/null +++ b/src/plugins/chat/auto_speak.py @@ -0,0 +1,172 @@ +import time +import asyncio +import random +from random import random as random_float +from typing import Dict +from ..config.config import global_config +from .message import MessageSending, MessageThinking, MessageSet, MessageRecv +from .message_base import UserInfo, Seg +from .message_sender import message_manager +from ..moods.moods import MoodManager +from .llm_generator import ResponseGenerator +from src.common.logger import get_module_logger +from src.think_flow_demo.heartflow import subheartflow_manager +from ...common.database import db +logger = get_module_logger("auto_speak") + +class AutoSpeakManager: + def __init__(self): + self._last_auto_speak_time: Dict[str, float] = {} # 记录每个聊天流上次自主发言的时间 + self.mood_manager = MoodManager.get_instance() + self.gpt = ResponseGenerator() # 添加gpt实例 + self._started = False + self._check_task = None + self.db = db + + async def get_chat_info(self, chat_id: str) -> dict: + """从数据库获取聊天流信息""" + chat_info = await self.db.chat_streams.find_one({"stream_id": chat_id}) + return chat_info + + async def start_auto_speak_check(self): + """启动自动发言检查任务""" + if not self._started: + self._check_task = asyncio.create_task(self._periodic_check()) + self._started = True + logger.success("自动发言检查任务已启动") + + async def _periodic_check(self): + """定期检查是否需要自主发言""" + while True and global_config.enable_think_flow: + + # 获取所有活跃的子心流 + active_subheartflows = [] + for chat_id, subheartflow in subheartflow_manager._subheartflows.items(): + if subheartflow.is_active and subheartflow.current_state.willing > 0: # 只考虑活跃且意愿值大于0.5的子心流 + active_subheartflows.append((chat_id, subheartflow)) + logger.debug(f"发现活跃子心流 - 聊天ID: {chat_id}, 意愿值: {subheartflow.current_state.willing:.2f}") + + if not active_subheartflows: + logger.debug("当前没有活跃的子心流") + await asyncio.sleep(20) # 添加异步等待 + continue + + # 随机选择一个活跃的子心流 + chat_id, subheartflow = random.choice(active_subheartflows) + logger.info(f"随机选择子心流 - 聊天ID: {chat_id}, 意愿值: {subheartflow.current_state.willing:.2f}") + + # 检查是否应该自主发言 + if await self.check_auto_speak(subheartflow): + logger.info(f"准备自主发言 - 聊天ID: {chat_id}") + # 生成自主发言 + bot_user_info = UserInfo( + user_id=global_config.BOT_QQ, + user_nickname=global_config.BOT_NICKNAME, + platform="qq", # 默认使用qq平台 + ) + + # 创建一个空的MessageRecv对象作为上下文 + message = MessageRecv({ + "message_info": { + "user_info": { + "user_id": chat_id, + "user_nickname": "", + "platform": "qq" + }, + "group_info": None, + "platform": "qq", + "time": time.time() + }, + "processed_plain_text": "", + "raw_message": "", + "is_emoji": False + }) + + await self.generate_auto_speak(subheartflow, message, bot_user_info, message.message_info["user_info"], message.message_info) + else: + logger.debug(f"不满足自主发言条件 - 聊天ID: {chat_id}") + + # 每分钟检查一次 + await asyncio.sleep(20) + + # await asyncio.sleep(5) # 发生错误时等待5秒再继续 + + async def check_auto_speak(self, subheartflow) -> bool: + """检查是否应该自主发言""" + if not subheartflow: + return False + + current_time = time.time() + chat_id = subheartflow.observe_chat_id + + # 获取上次自主发言时间 + if chat_id not in self._last_auto_speak_time: + self._last_auto_speak_time[chat_id] = 0 + last_speak_time = self._last_auto_speak_time.get(chat_id, 0) + + # 如果距离上次自主发言不到5分钟，不发言 + if current_time - last_speak_time < 30: + logger.debug(f"距离上次发言时间太短 - 聊天ID: {chat_id}, 剩余时间: {30 - (current_time - last_speak_time):.1f}秒") + return False + + # 获取当前意愿值 + current_willing = subheartflow.current_state.willing + + if current_willing > 0.1 and random_float() < 0.5: + self._last_auto_speak_time[chat_id] = current_time + logger.info(f"满足自主发言条件 - 聊天ID: {chat_id}, 意愿值: {current_willing:.2f}") + return True + + logger.debug(f"不满足自主发言条件 - 聊天ID: {chat_id}, 意愿值: {current_willing:.2f}") + return False + + async def generate_auto_speak(self, subheartflow, message, bot_user_info: UserInfo, userinfo, messageinfo): + """生成自主发言内容""" + thinking_time_point = round(time.time(), 2) + think_id = "mt" + str(thinking_time_point) + thinking_message = MessageThinking( + message_id=think_id, + chat_stream=None, # 不需要chat_stream + bot_user_info=bot_user_info, + reply=message, + thinking_start_time=thinking_time_point, + ) + + message_manager.add_message(thinking_message) + + # 生成自主发言内容 + response, raw_content = await self.gpt.generate_response(message) + + if response: + message_set = MessageSet(None, think_id) # 不需要chat_stream + mark_head = False + + for msg in response: + message_segment = Seg(type="text", data=msg) + bot_message = MessageSending( + message_id=think_id, + chat_stream=None, # 不需要chat_stream + bot_user_info=bot_user_info, + sender_info=userinfo, + message_segment=message_segment, + reply=message, + is_head=not mark_head, + is_emoji=False, + thinking_start_time=thinking_time_point, + ) + if not mark_head: + mark_head = True + message_set.add_message(bot_message) + + message_manager.add_message(message_set) + + # 更新情绪和关系 + stance, emotion = await self.gpt._get_emotion_tags(raw_content, message.processed_plain_text) + self.mood_manager.update_mood_from_emotion(emotion, global_config.mood_intensity_factor) + + return True + + return False + +# 创建全局AutoSpeakManager实例 +auto_speak_manager = AutoSpeakManager() \ No newline at end of file diff --git a/src/plugins/chat/bot.py b/src/plugins/chat/bot.py index 1c28422c1..490003c8f 100644 --- a/src/plugins/chat/bot.py +++ b/src/plugins/chat/bot.py @@ -113,16 +113,24 @@ class ChatBot: # 根据话题计算激活度 topic = "" + await self.storage.store_message(message, chat, topic[0] if topic else None) + + interested_rate = 0 interested_rate = await HippocampusManager.get_instance().get_activate_from_text( message.processed_plain_text, fast_retrieval=True ) - # interested_rate = 0.1 - # logger.info(f"对{message.processed_plain_text}的激活度:{interested_rate}") - # logger.info(f"\033[1;32m[主题识别]\033[0m 使用{global_config.topic_extract}主题: {topic}") - - await self.storage.store_message(message, chat, topic[0] if topic else None) - is_mentioned = is_mentioned_bot_in_message(message) + + if global_config.enable_think_flow: + current_willing_old = willing_manager.get_willing(chat_stream=chat) + current_willing_new = (subheartflow_manager.get_subheartflow(chat.stream_id).current_state.willing - 5) / 4 + print(f"旧回复意愿：{current_willing_old}，新回复意愿：{current_willing_new}") + current_willing = (current_willing_old + current_willing_new) / 2 + else: + current_willing = willing_manager.get_willing(chat_stream=chat) + + willing_manager.set_willing(chat.stream_id, current_willing) + reply_probability = await willing_manager.change_reply_willing_received( chat_stream=chat, is_mentioned_bot=is_mentioned, @@ -131,10 +139,6 @@ class ChatBot: interested_rate=interested_rate, sender_id=str(message.message_info.user_info.user_id), ) - current_willing_old = willing_manager.get_willing(chat_stream=chat) - current_willing_new = (subheartflow_manager.get_subheartflow(chat.stream_id).current_state.willing - 5) / 4 - print(f"旧回复意愿：{current_willing_old}，新回复意愿：{current_willing_new}") - current_willing = (current_willing_old + current_willing_new) / 2 logger.info( f"[{current_time}][{chat.group_info.group_name if chat.group_info else '私聊'}]" diff --git a/src/plugins/memory_system/Hippocampus.py b/src/plugins/memory_system/Hippocampus.py index bdb2a50b1..6a59db581 100644 --- a/src/plugins/memory_system/Hippocampus.py +++ b/src/plugins/memory_system/Hippocampus.py @@ -1225,7 +1225,7 @@ class Hippocampus: total_nodes = len(self.memory_graph.G.nodes()) # activated_nodes = len(activate_map) activation_ratio = total_activation / total_nodes if total_nodes > 0 else 0 - activation_ratio = activation_ratio*40 + activation_ratio = activation_ratio*60 logger.info(f"总激活值: {total_activation:.2f}, 总节点数: {total_nodes}, 激活: {activation_ratio}") return activation_ratio diff --git a/src/plugins/schedule/offline_llm.py b/src/plugins/schedule/offline_llm.py deleted file mode 100644 index 5276f3802..000000000 --- a/src/plugins/schedule/offline_llm.py +++ /dev/null @@ -1,70 +0,0 @@ -import asyncio -import os - -import aiohttp -from src.common.logger import get_module_logger - -logger = get_module_logger("offline_llm") - - -class LLMModel: - def __init__(self, model_name="deepseek-ai/DeepSeek-V3", **kwargs): - self.model_name = model_name - self.params = kwargs - self.api_key = os.getenv("SILICONFLOW_KEY") - self.base_url = os.getenv("SILICONFLOW_BASE_URL") - - if not self.api_key or not self.base_url: - raise ValueError("环境变量未正确加载：SILICONFLOW_KEY 或 SILICONFLOW_BASE_URL 未设置") - - logger.info(f"API URL: {self.base_url}") # 使用 logger 记录 base_url - - async def generate_response_async(self, prompt: str) -> str: - """异步方式根据输入的提示生成模型的响应""" - headers = {"Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json"} - - # 构建请求体 - data = { - "model": self.model_name, - "messages": [{"role": "user", "content": prompt}], - "temperature": 0.7, - **self.params, - } - - # 发送请求到完整的 chat/completions 端点 - api_url = f"{self.base_url.rstrip('/')}/chat/completions" - logger.info(f"Request URL: {api_url}") # 记录请求的 URL - - max_retries = 3 - base_wait_time = 15 - - async with aiohttp.ClientSession() as session: - for retry in range(max_retries): - try: - async with session.post(api_url, headers=headers, json=data) as response: - if response.status == 429: - wait_time = base_wait_time * (2**retry) # 指数退避 - logger.warning(f"遇到请求限制(429)，等待{wait_time}秒后重试...") - await asyncio.sleep(wait_time) - continue - - response.raise_for_status() # 检查其他响应状态 - - result = await response.json() - if "choices" in result and len(result["choices"]) > 0: - content = result["choices"][0]["message"]["content"] - reasoning_content = result["choices"][0]["message"].get("reasoning_content", "") - return content, reasoning_content - return "没有返回结果", "" - - except Exception as e: - if retry < max_retries - 1: # 如果还有重试机会 - wait_time = base_wait_time * (2**retry) - logger.error(f"[回复]请求失败，等待{wait_time}秒后重试... 错误: {str(e)}") - await asyncio.sleep(wait_time) - else: - logger.error(f"请求失败: {str(e)}") - return f"请求失败: {str(e)}", "" - - logger.error("达到最大重试次数，请求仍然失败") - return "达到最大重试次数，请求仍然失败", "" diff --git a/src/plugins/schedule/schedule_generator.py b/src/plugins/schedule/schedule_generator.py index b07a16296..54b470d8c 100644 --- a/src/plugins/schedule/schedule_generator.py +++ b/src/plugins/schedule/schedule_generator.py @@ -72,7 +72,7 @@ class ScheduleGenerator: self.print_schedule() while True: - print(self.get_current_num_task(1, True)) + # print(self.get_current_num_task(1, True)) current_time = datetime.datetime.now() diff --git a/src/plugins/utils/statistic.py b/src/plugins/utils/statistic.py index 1071b29b0..b9efafd03 100644 --- a/src/plugins/utils/statistic.py +++ b/src/plugins/utils/statistic.py @@ -223,7 +223,7 @@ class LLMStatistics: logger.exception("统计数据处理失败") # 等待5分钟 - for _ in range(300): # 5分钟 = 300秒 + for _ in range(30): # 5分钟 = 300秒 if not self.running: break time.sleep(1) diff --git a/src/think_flow_demo/sub_heartflow.py b/src/think_flow_demo/sub_heartflow.py index 805218d5a..d394a0205 100644 --- a/src/think_flow_demo/sub_heartflow.py +++ b/src/think_flow_demo/sub_heartflow.py @@ -48,6 +48,8 @@ class SubHeartflow: self.current_mind = "你什么也没想" self.personality_info = " ".join(global_config.PROMPT_PERSONALITY) + + self.is_active = False def assign_observe(self,stream_id): self.outer_world = outer_world.get_world_by_stream_id(stream_id) @@ -58,8 +60,10 @@ class SubHeartflow: current_time = time.time() if current_time - self.last_reply_time > 180: # 3分钟 = 180秒 # print(f"{self.observe_chat_id}麦麦已经3分钟没有回复了，暂时停止思考") + self.is_active = False await asyncio.sleep(60) # 每30秒检查一次 else: + self.is_active = True await self.do_a_thinking() await self.judge_willing() await asyncio.sleep(60) @@ -88,7 +92,7 @@ class SubHeartflow: else: related_memory_info = '' - print(f"相关记忆：{related_memory_info}") + # print(f"相关记忆：{related_memory_info}") schedule_info = bot_schedule.get_current_num_task(num = 1,time_info = False)