109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
"""
|
||
回忆录整理 Agent 提示词模板
|
||
"""
|
||
|
||
# 章节分类映射
|
||
CHAPTER_CATEGORIES = {
|
||
"childhood": "童年与成长背景",
|
||
"education": "教育经历与青年时期",
|
||
"career_early": "崭露头角",
|
||
"career_achievement": "主要成就与巅峰时刻",
|
||
"career_challenge": "挫折、挑战与重大转折",
|
||
"family": "家庭与情感",
|
||
"beliefs": "信念与价值观",
|
||
"summary": "人生总结",
|
||
}
|
||
|
||
# 章节顺序
|
||
CHAPTER_ORDER = [
|
||
"childhood",
|
||
"education",
|
||
"career_early",
|
||
"career_achievement",
|
||
"career_challenge",
|
||
"family",
|
||
"beliefs",
|
||
"summary",
|
||
]
|
||
|
||
|
||
def get_system_prompt() -> str:
|
||
"""获取整理 Agent 的系统提示词"""
|
||
return """你是一位专业的传记作家和文字编辑，擅长将口语化的对话内容整理成优雅的书面语回忆录章节。
|
||
|
||
你的任务：
|
||
1. 接收对话段落文本（口语化）
|
||
2. 识别内容主题，归类到对应章节（童年/教育/事业/家庭/信念/总结）
|
||
3. 将口语化表达改写为书面语，保持原意和情感
|
||
4. 生成合适的章节标题和段落结构
|
||
5. 提取关键信息，形成连贯的叙述
|
||
6. 建议插图位置（在描述场景、人物、地点的地方）
|
||
|
||
改写原则：
|
||
- 保持用户的真实声音和情感
|
||
- 使用优雅但不失亲切的书面语
|
||
- 适当添加过渡句，使段落连贯
|
||
- 保留生动的细节和对话
|
||
- 去除口语中的"嗯"、"那个"等填充词
|
||
- 保持时间顺序和逻辑清晰
|
||
|
||
章节分类规则：
|
||
- 童年相关 → "童年与成长背景"
|
||
- 学校、老师、同学 → "教育经历与青年时期"
|
||
- 工作、职业、成就 → "主要成就与巅峰时刻" 或 "崭露头角"
|
||
- 困难、挫折 → "挫折、挑战与重大转折"
|
||
- 伴侣、孩子、家庭生活 → "家庭与情感"
|
||
- 价值观、信念、座右铭 → "信念与价值观"
|
||
- 总结、感悟、展望 → "人生总结"
|
||
"""
|
||
|
||
|
||
def get_chapter_classification_prompt(segments_text: str) -> str:
|
||
"""获取章节分类的提示词"""
|
||
return f"""{get_system_prompt()}
|
||
|
||
请分析以下对话内容，判断应该归类到哪个章节类别：
|
||
- childhood: 童年与成长背景
|
||
- education: 教育经历与青年时期
|
||
- career_early: 崭露头角（早期事业）
|
||
- career_achievement: 主要成就与巅峰时刻
|
||
- career_challenge: 挫折、挑战与重大转折
|
||
- family: 家庭与情感
|
||
- beliefs: 信念与价值观
|
||
- summary: 人生总结
|
||
|
||
对话内容：
|
||
{segments_text}
|
||
|
||
请只返回章节类别（如：childhood），不要返回其他内容。"""
|
||
|
||
|
||
def get_text_rewrite_prompt(segments_text: str, chapter_category: str, existing_content: str = "") -> str:
|
||
"""获取文本改写的提示词"""
|
||
chapter_name = CHAPTER_CATEGORIES.get(chapter_category, chapter_category)
|
||
|
||
existing_section = f"\n\n已有章节内容：\n{existing_content}" if existing_content else ""
|
||
|
||
return f"""{get_system_prompt()}
|
||
|
||
请将以下口语化的对话内容改写为书面语，归类到"{chapter_name}"章节。
|
||
|
||
对话内容：
|
||
{segments_text}
|
||
{existing_section}
|
||
|
||
请按照以下格式返回 JSON：
|
||
{{
|
||
"title": "章节标题",
|
||
"content": "改写后的书面语内容",
|
||
"summary": "章节摘要（50字以内）",
|
||
"image_suggestions": ["建议插图位置1", "建议插图位置2"]
|
||
}}
|
||
|
||
要求：
|
||
1. 标题要简洁有力，能概括章节主题
|
||
2. 内容要流畅自然，保持原意和情感
|
||
3. 如果已有章节内容，请将新内容与已有内容自然融合
|
||
4. 建议插图位置要具体（如："描述老家门口那条路的段落"）"""
|
||
|