AI_NovelGenerator/consistency_checker.py

# 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()