diff --git a/src/think_flow_demo/current_mind.py b/src/think_flow_demo/current_mind.py index 32d77ef7..8d71a3dc 100644 --- a/src/think_flow_demo/current_mind.py +++ b/src/think_flow_demo/current_mind.py @@ -2,7 +2,7 @@ from .outer_world import outer_world import asyncio from src.plugins.moods.moods import MoodManager from src.plugins.models.utils_model import LLM_request -from src.plugins.chat.config import global_config, BotConfig +from src.plugins.chat.config import global_config import re import time from src.plugins.schedule.schedule_generator import bot_schedule @@ -37,7 +37,7 @@ class SubHeartflow: if not self.current_mind: self.current_mind = "你什么也没想" - self.personality_info = " ".join(BotConfig.PROMPT_PERSONALITY) + self.personality_info = " ".join(global_config.PROMPT_PERSONALITY) def assign_observe(self,stream_id): self.outer_world = outer_world.get_world_by_stream_id(stream_id) diff --git a/src/think_flow_demo/heartflow.py b/src/think_flow_demo/heartflow.py index dcdbe508..4243895e 100644 --- a/src/think_flow_demo/heartflow.py +++ b/src/think_flow_demo/heartflow.py @@ -1,7 +1,7 @@ from .current_mind import SubHeartflow from src.plugins.moods.moods import MoodManager from src.plugins.models.utils_model import LLM_request -from src.plugins.chat.config import global_config, BotConfig +from src.plugins.chat.config import global_config from src.plugins.schedule.schedule_generator import bot_schedule import asyncio from src.common.logger import get_module_logger, LogConfig, HEARTFLOW_STYLE_CONFIG # noqa: E402 @@ -46,7 +46,7 @@ class Heartflow: logger.info("麦麦大脑袋转起来了") self.current_state.update_current_state_info() - personality_info = " ".join(BotConfig.PROMPT_PERSONALITY) + personality_info = " ".join(global_config.PROMPT_PERSONALITY) current_thinking_info = self.current_mind mood_info = self.current_state.mood related_memory_info = 'memory' @@ -91,7 +91,7 @@ class Heartflow: return await self.minds_summary(sub_minds) async def minds_summary(self,minds_str): - personality_info = " ".join(BotConfig.PROMPT_PERSONALITY) + personality_info = " ".join(global_config.PROMPT_PERSONALITY) mood_info = self.current_state.mood prompt = ""