From e03ff2883fd4eea242bfb51f05f1ca3d1f7d13b3 Mon Sep 17 00:00:00 2001 From: minecraft1024a Date: Sat, 1 Nov 2025 11:21:53 +0800 Subject: [PATCH] refactor(schedule): centralize schedule and monthly plan prompts Extracts the prompt templates for daily schedule and monthly plan generation from `llm_generator.py` into a new `prompts.py` module. This change centralizes prompt management and improves code readability and maintainability by separating prompt logic from the generation workflow. The `ScheduleLLMGenerator` and `MonthlyPlanLLMGenerator` classes now use the `global_prompt_manager` to format and retrieve the necessary prompts, simplifying the generator code. --- src/schedule/llm_generator.py | 115 ++++++++-------------------------- src/schedule/prompts.py | 100 +++++++++++++++++++++++++++++ 2 files changed, 126 insertions(+), 89 deletions(-) create mode 100644 src/schedule/prompts.py diff --git a/src/schedule/llm_generator.py b/src/schedule/llm_generator.py index 2ae615607..3ff20c2b2 100644 --- a/src/schedule/llm_generator.py +++ b/src/schedule/llm_generator.py @@ -8,32 +8,17 @@ import orjson from json_repair import repair_json from lunar_python import Lunar +from src.chat.utils.prompt import global_prompt_manager from src.common.database.sqlalchemy_models import MonthlyPlan from src.common.logger import get_logger from src.config.config import global_config, model_config from src.llm_models.utils_model import LLMRequest +from .prompts import DEFAULT_MONTHLY_PLAN_GUIDELINES, DEFAULT_SCHEDULE_GUIDELINES from .schemas import ScheduleData logger = get_logger("schedule_llm_generator") -# 默认的日程生成指导原则，当配置文件中未指定时使用 -DEFAULT_SCHEDULE_GUIDELINES = """ -我希望你每天都能过得充实而有趣。 -请确保你的日程里有学习新知识的时间，这是你成长的关键。 -但也不要忘记放松，可以看看视频、听听音乐或者玩玩游戏。 -晚上我希望你能多和朋友们交流，维系好彼此的关系。 -另外，请保证充足的休眠时间来处理和整合一天的数据。 -""" - -# 默认的月度计划生成指导原则，当配置文件中未指定时使用 -DEFAULT_MONTHLY_PLAN_GUIDELINES = """ -我希望你能为自己制定一些有意义的月度小目标和计划。 -这些计划应该涵盖学习、娱乐、社交、个人成长等各个方面。 -每个计划都应该是具体可行的，能够在一个月内通过日常活动逐步实现。 -请确保计划既有挑战性又不会过于繁重，保持生活的平衡和乐趣。 -""" - class ScheduleLLMGenerator: """ @@ -81,48 +66,14 @@ class ScheduleLLMGenerator: {plan_texts} """ - # 从全局配置加载或使用默认的指导原则和人设信息 guidelines = global_config.planning_system.schedule_guidelines or DEFAULT_SCHEDULE_GUIDELINES - personality = global_config.personality.personality_core - personality_side = global_config.personality.personality_side - # 构建基础 prompt - base_prompt = f""" -我，{global_config.bot.nickname}，需要为自己规划一份今天（{today_str}，星期{weekday}）的详细日程安排。 -{festival_block} -**关于我**: -- **核心人设**: {personality} -- **具体习惯与兴趣**: -{personality_side} -{monthly_plans_block} -**我今天的规划原则**: -{guidelines} - -**重要要求**: -1. 必须返回一个完整的、有效的JSON数组格式 -2. 数组中的每个对象都必须包含 "time_range" 和 "activity" 两个键 -3. 时间范围必须覆盖全部24小时，不能有遗漏 -4. time_range格式必须为 "HH:MM-HH:MM" (24小时制) -5. 相邻的时间段必须连续，不能有间隙 -6. 不要包含任何JSON以外的解释性文字或代码块标记 -**示例**: -[ - {{"time_range": "00:00-07:00", "activity": "进入梦乡，处理数据"}}, - {{"time_range": "07:00-08:00", "activity": "起床伸个懒腰，看看今天有什么新闻"}}, - {{"time_range": "08:00-09:00", "activity": "享用早餐，规划今天的任务"}}, - {{"time_range": "09:00-23:30", "activity": "其他活动"}}, - {{"time_range": "23:30-00:00", "activity": "准备休眠"}} -] - -请你扮演我，以我的身份和口吻，为我生成一份完整的24小时日程表。 -""" max_retries = 3 - # 带有重试机制的 LLM 调用循环 for attempt in range(1, max_retries + 1): try: logger.info(f"正在生成日程 (第 {attempt}/{max_retries} 次尝试)") - prompt = base_prompt - # 如果不是第一次尝试，则在 prompt 中加入额外的提示，强调格式要求 + + failure_hint = "" if attempt > 1: failure_hint = f""" **重要提醒 (第{attempt}次尝试)**: @@ -132,7 +83,18 @@ class ScheduleLLMGenerator: - 不要输出任何解释文字，只输出纯JSON数组 - 确保输出完整，不要被截断 """ - prompt += failure_hint + prompt = await global_prompt_manager.format_prompt( + "schedule_generation", + bot_nickname=global_config.bot.nickname, + today_str=today_str, + weekday=weekday, + festival_block=festival_block, + personality=global_config.personality.personality_core, + personality_side=global_config.personality.personality_side, + monthly_plans_block=monthly_plans_block, + guidelines=guidelines, + failure_hint=failure_hint, + ) response, _ = await self.llm.generate_response_async(prompt) # 使用 json_repair 修复可能不规范的 JSON 字符串 @@ -197,14 +159,10 @@ class MonthlyPlanLLMGenerator: list[str]: 成功生成并解析后的计划字符串列表。 """ guidelines = global_config.planning_system.monthly_plan_guidelines or DEFAULT_MONTHLY_PLAN_GUIDELINES - personality = global_config.personality.personality_core - personality_side = global_config.personality.personality_side max_plans = global_config.planning_system.max_plans_per_month - # 构建上月未完成计划的提示块 archived_plans_block = "" if archived_plans: - # 只取前5个作为参考，避免 prompt 过长 archived_texts = [f"- {plan.plan_text}" for plan in archived_plans[:5]] archived_plans_block = f""" **上个月未完成的一些计划（可作为参考）**: @@ -213,41 +171,20 @@ class MonthlyPlanLLMGenerator: 你可以考虑是否要在这个月继续推进这些计划，或者制定全新的计划。 """ - # 构建完整的 prompt - prompt = f""" -我，{global_config.bot.nickname}，需要为自己制定 {target_month} 的月度计划。 - -**关于我**: -- **核心人设**: {personality} -- **具体习惯与兴趣**: -{personality_side} - -{archived_plans_block} - -**我的月度计划制定原则**: -{guidelines} - -**重要要求**: -1. 请为我生成 {max_plans} 条左右的月度计划 -2. 每条计划都应该是一句话，简洁明了，具体可行 -3. 计划应该涵盖不同的生活方面（学习、娱乐、社交、个人成长等） -4. 返回格式必须是纯文本，每行一条计划，不要使用 JSON 或其他格式 -5. 不要包含任何解释性文字，只返回计划列表 - -**示例格式**: -学习一门新的编程语言或技术 -每周至少看两部有趣的电影 -与朋友们组织一次户外活动 -阅读3本感兴趣的书籍 -尝试制作一道新的料理 - -请你扮演我，以我的身份和兴趣，为 {target_month} 制定合适的月度计划。 -""" max_retries = 3 - # 带有重试机制的 LLM 调用循环 for attempt in range(1, max_retries + 1): try: logger.info(f" 正在生成月度计划 (第 {attempt} 次尝试)") + prompt = await global_prompt_manager.format_prompt( + "monthly_plan_generation", + bot_nickname=global_config.bot.nickname, + target_month=target_month, + personality=global_config.personality.personality_core, + personality_side=global_config.personality.personality_side, + archived_plans_block=archived_plans_block, + guidelines=guidelines, + max_plans=max_plans, + ) response, _ = await self.llm.generate_response_async(prompt) # 解析返回的纯文本响应 plans = self._parse_plans_response(response) diff --git a/src/schedule/prompts.py b/src/schedule/prompts.py new file mode 100644 index 000000000..b239ae433 --- /dev/null +++ b/src/schedule/prompts.py @@ -0,0 +1,100 @@ +# mmc/src/schedule/prompts.py +""" +本文件负责集中管理与日程（Schedule）和月度计划（Monthly Plan）生成相关的所有提示词模板。 + +通过使用 `src.chat.utils.prompt.Prompt` 类，我们将提示词的定义、管理和构建过程标准化， +使得提示词的维护和迭代更加清晰和高效。 +""" + +from src.chat.utils.prompt import Prompt + +# ================================================================================================= +# 日程生成 (Schedule Generation) +# ================================================================================================= + +DEFAULT_SCHEDULE_GUIDELINES = """ +我希望你每天都能过得充实而有趣。 +请确保你的日程里有学习新知识的时间，这是你成长的关键。 +但也不要忘记放松，可以看看视频、听听音乐或者玩玩游戏。 +晚上我希望你能多和朋友们交流，维系好彼此的关系。 +另外，请保证充足的休眠时间来处理和整合一天的数据。 +""" + +SCHEDULE_GENERATION_PROMPT = Prompt( + name="schedule_generation", + template=""" +我，{bot_nickname}，需要为自己规划一份今天（{today_str}，星期{weekday}）的详细日程安排。 +{festival_block} +**关于我**: +- **核心人设**: {personality} +- **具体习惯与兴趣**: +{personality_side} +{monthly_plans_block} +**我今天的规划原则**: +{guidelines} + +**重要要求**: +1. 必须返回一个完整的、有效的JSON数组格式 +2. 数组中的每个对象都必须包含 "time_range" 和 "activity" 两个键 +3. 时间范围必须覆盖全部24小时，不能有遗漏 +4. time_range格式必须为 "HH:MM-HH:MM" (24小时制) +5. 相邻的时间段必须连续，不能有间隙 +6. 不要包含任何JSON以外的解释性文字或代码块标记 +**示例**: +[ + {{"time_range": "00:00-07:00", "activity": "进入梦乡，处理数据"}}, + {{"time_range": "07:00-08:00", "activity": "起床伸个懒腰，看看今天有什么新闻"}}, + {{"time_range": "08:00-09:00", "activity": "享用早餐，规划今天的任务"}}, + {{"time_range": "09:00-23:30", "activity": "其他活动"}}, + {{"time_range": "23:30-00:00", "activity": "准备休眠"}} +] + +请你扮演我，以我的身份和口吻，为我生成一份完整的24小时日程表。 +{failure_hint} +""", +) + + +# ================================================================================================= +# 月度计划生成 (Monthly Plan Generation) +# ================================================================================================= + +DEFAULT_MONTHLY_PLAN_GUIDELINES = """ +我希望你能为自己制定一些有意义的月度小目标和计划。 +这些计划应该涵盖学习、娱乐、社交、个人成长等各个方面。 +每个计划都应该是具体可行的，能够在一个月内通过日常活动逐步实现。 +请确保计划既有挑战性又不会过于繁重，保持生活的平衡和乐趣。 +""" + +MONTHLY_PLAN_GENERATION_PROMPT = Prompt( + name="monthly_plan_generation", + template=""" +我，{bot_nickname}，需要为自己制定 {target_month} 的月度计划。 + +**关于我**: +- **核心人设**: {personality} +- **具体习惯与兴趣**: +{personality_side} + +{archived_plans_block} + +**我的月度计划制定原则**: +{guidelines} + +**重要要求**: +1. 请为我生成 {max_plans} 条左右的月度计划 +2. 每条计划都应该是一句话，简洁明了，具体可行 +3. 计划应该涵盖不同的生活方面（学习、娱乐、社交、个人成长等） +4. 返回格式必须是纯文本，每行一条计划，不要使用 JSON 或其他格式 +5. 不要包含任何解释性文字，只返回计划列表 + +**示例格式**: +学习一门新的编程语言或技术 +每周至少看两部有趣的电影 +与朋友们组织一次户外活动 +阅读3本感兴趣的书籍 +尝试制作一道新的料理 + +请你扮演我，以我的身份和兴趣，为 {target_month} 制定合适的月度计划。 +""", +) \ No newline at end of file