58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
# consistency_checker.py
|
|
# -*- coding: utf-8 -*-
|
|
from langchain_openai import ChatOpenAI
|
|
|
|
CONSISTENCY_PROMPT = """\
|
|
请检查下面的小说设定与最新章节是否存在明显冲突或不一致之处，如有请列出：
|
|
- 小说设定：
|
|
{novel_setting}
|
|
|
|
- 角色状态（可能包含重要信息）：
|
|
{character_state}
|
|
|
|
- 全局摘要：
|
|
{global_summary}
|
|
|
|
- 最新章节内容：
|
|
{chapter_text}
|
|
|
|
如果存在冲突或不一致，请说明；否则请返回“无明显冲突”。
|
|
"""
|
|
|
|
def check_consistency(
|
|
novel_setting: str,
|
|
character_state: str,
|
|
global_summary: str,
|
|
chapter_text: str,
|
|
api_key: str,
|
|
base_url: str,
|
|
model_name: str,
|
|
temperature: float = 0.3
|
|
) -> str:
|
|
"""
|
|
调用模型做简单的一致性检查。可扩展更多提示或校验规则。
|
|
"""
|
|
prompt = CONSISTENCY_PROMPT.format(
|
|
novel_setting=novel_setting,
|
|
character_state=character_state,
|
|
global_summary=global_summary,
|
|
chapter_text=chapter_text
|
|
)
|
|
model = ChatOpenAI(
|
|
model=model_name,
|
|
api_key=api_key,
|
|
base_url=base_url,
|
|
temperature=temperature
|
|
)
|
|
# 调试日志
|
|
print("\n[ConsistencyChecker] Prompt >>>", prompt)
|
|
|
|
response = model.invoke(prompt)
|
|
if not response:
|
|
return "审校Agent无回复"
|
|
|
|
# 调试日志
|
|
print("[ConsistencyChecker] Response <<<", response.content.strip())
|
|
|
|
return response.content.strip()
|