小优化

This commit is contained in:
YILING0013
2025-02-01 18:49:52 +08:00
parent ed7e74583d
commit eb627fbec9
5 changed files with 107 additions and 34 deletions
+9 -2
View File
@@ -2,6 +2,7 @@
# -*- coding: utf-8 -*-
from langchain_openai import ChatOpenAI
# ============== 增加对“剧情要点/未解决冲突”进行检查的可选引导 ==============
CONSISTENCY_PROMPT = """\
请检查下面的小说设定与最新章节是否存在明显冲突或不一致之处,如有请列出:
- 小说设定:
@@ -13,10 +14,13 @@ CONSISTENCY_PROMPT = """\
- 全局摘要:
{global_summary}
- 已记录的未解决冲突或剧情要点:
{plot_arcs} # 若为空可能不输出
- 最新章节内容:
{chapter_text}
如果存在冲突或不一致,请说明;否则请返回“无明显冲突”。
如果存在冲突或不一致,请说明;如果在未解决冲突中有被忽略或需要推进的地方,也请提及;否则请返回“无明显冲突”。
"""
def check_consistency(
@@ -27,15 +31,18 @@ def check_consistency(
api_key: str,
base_url: str,
model_name: str,
temperature: float = 0.3
temperature: float = 0.3,
plot_arcs: str = "" # 新增参数,默认空字符串
) -> str:
"""
调用模型做简单的一致性检查。可扩展更多提示或校验规则。
新增: 会额外检查对“未解决冲突或剧情要点”(plot_arcs)的衔接情况。
"""
prompt = CONSISTENCY_PROMPT.format(
novel_setting=novel_setting,
character_state=character_state,
global_summary=global_summary,
plot_arcs=plot_arcs,
chapter_text=chapter_text
)
model = ChatOpenAI(