fix:1.修复mcp提示词没有加载用户提示词模板问题

This commit is contained in:
xiamuceer
2025-11-30 11:46:26 +08:00
parent 117aedbc31
commit 19e08d99b3
3 changed files with 70 additions and 59 deletions
+25 -51
View File
@@ -83,23 +83,14 @@ async def world_building_generator(
if available_tools:
yield await SSEResponse.send_progress("🔍 尝试使用MCP工具收集参考资料...", 18)
# 构建资料收集提示词
planning_prompt = f"""你正在为小说《{title}》设计世界观。
【小说信息】
- 题材:{genre}
- 主题:{theme}
- 简介:{description}
【任务】
请使用可用工具搜索相关背景资料,帮助构建更真实、更有深度的世界观设定。
你可以查询:
1. 历史背景(如果是历史题材)
2. 地理环境和文化特征
3. 相关领域的专业知识
4. 类似作品的设定参考
请查询最关键的1个问题(不要超过1个)。"""
mcp_template = await PromptService.get_template("MCP_WORLD_BUILDING_PLANNING", user_id, db)
planning_prompt = PromptService.format_prompt(
mcp_template,
title=title,
genre=genre,
theme=theme,
description=description
)
# 调用MCP增强的AI(非流式,最多1轮工具调用,避免超时)
planning_result = await user_ai_service.generate_text_with_mcp(
@@ -357,24 +348,15 @@ async def characters_generator(
if available_tools:
yield await SSEResponse.send_progress("🔍 尝试使用MCP工具收集角色参考资料...", 8)
# 构建角色资料收集提示词
planning_prompt = f"""你正在为小说《{project.title}》设计角色。
【小说信息】
- 题材:{genre or project.genre}
- 主题:{theme or project.theme}
- 时代背景:{world_context.get('time_period', '未设定')}
- 地理位置:{world_context.get('location', '未设定')}
【任务】
请使用可用工具搜索相关参考资料,帮助设计更真实、更有深度的角色。
你可以查询:
1. 该时代/地域的真实历史人物特征
2. 文化背景和社会习俗
3. 职业特点和生活方式
4. 相关领域的人物原型
请查询最关键的1个问题(不要超过1个)。"""
mcp_template = await PromptService.get_template("MCP_CHARACTER_PLANNING", user_id, db)
planning_prompt = PromptService.format_prompt(
mcp_template,
title=project.title,
genre=genre or project.genre,
theme=theme or project.theme,
time_period=world_context.get('time_period', '未设定'),
location=world_context.get('location', '未设定')
)
# 调用MCP增强的AI(非流式,最多1轮工具调用,避免超时)
planning_result = await user_ai_service.generate_text_with_mcp(
@@ -1185,22 +1167,14 @@ async def world_building_regenerate_generator(
if available_tools:
yield await SSEResponse.send_progress("🔍 尝试使用MCP工具收集参考资料...", 18)
planning_prompt = f"""你正在为小说《{project.title}》重新设计世界观。
【小说信息】
- 题材:{project.genre}
- 主题:{project.theme}
- 简介:{project.description or '未设定'}
【任务】
请使用可用工具搜索相关背景资料,帮助构建更真实、更有深度的世界观设定。
你可以查询:
1. 历史背景(如果是历史题材)
2. 地理环境和文化特征
3. 相关领域的专业知识
4. 类似作品的设定参考
请查询最关键的1个问题(不要超过1个)。"""
mcp_template = await PromptService.get_template("MCP_WORLD_BUILDING_PLANNING", user_id, db)
planning_prompt = PromptService.format_prompt(
mcp_template,
title=project.title,
genre=project.genre,
theme=project.theme,
description=project.description or '未设定'
)
planning_result = await user_ai_service.generate_text_with_mcp(
prompt=planning_prompt,