fix:1.修复部分提示词模板键名不匹配问题

This commit is contained in:
xiamuceer
2025-11-30 09:57:49 +08:00
parent ad6e99d868
commit ead49b7227
6 changed files with 28 additions and 9 deletions
+1 -1
View File
@@ -178,7 +178,7 @@ class ChapterRegenerator:
) -> str:
"""构建完整的重新生成提示词"""
# 获取自定义提示词模板
template = await PromptService.get_template("CHAPTER_REGENERATION", user_id, db)
template = await PromptService.get_template("CHAPTER_REGENERATION_SYSTEM", user_id, db)
# 格式化提示词
return PromptService.format_prompt(
template,
+20 -1
View File
@@ -2081,7 +2081,26 @@ class PromptService:
# 2. 降级到系统默认模板
logger.info(f"⚪ 使用系统默认提示词: user_id={user_id}, template_key={template_key} (未找到自定义模板)")
return getattr(cls, template_key, None)
# 特殊处理灵感模式的提示词(存储在INSPIRATION_PROMPTS字典中)
if template_key.startswith("INSPIRATION_"):
# 提取步骤名称(如 INSPIRATION_TITLE -> title
step = template_key.replace("INSPIRATION_", "").lower()
inspiration_prompt = cls.INSPIRATION_PROMPTS.get(step)
if inspiration_prompt:
# 返回JSON格式的提示词
return json.dumps(inspiration_prompt, ensure_ascii=False)
# 如果是INSPIRATION_QUICK_COMPLETE
if template_key == "INSPIRATION_QUICK_COMPLETE":
return cls.INSPIRATION_QUICK_COMPLETE
# 其他模板直接从类属性获取
template_content = getattr(cls, template_key, None)
if template_content is None:
logger.warning(f"⚠️ 未找到系统默认模板: {template_key}")
return template_content
@classmethod
def get_all_system_templates(cls) -> list: