新增功能:自定义知识库、生成指导

允许用户上传自己的知识库(建议在线版API,或上下文长的用户使用)

允许在生成下一章节时加入提前指导

模型Temperature参数支持
This commit is contained in:
YILING0013
2025-01-31 13:50:07 +08:00
parent aebaf8efc1
commit 2fe5fe1b85
17 changed files with 461 additions and 80 deletions
+12 -6
View File
@@ -1,7 +1,5 @@
"""
演示多Agent思路中的“审校Agent”,对最新章节进行简单的一致性或逻辑冲突检查。
可根据需要进行扩展。
"""
# consistency_checker.py
# -*- coding: utf-8 -*-
from langchain_openai import ChatOpenAI
CONSISTENCY_PROMPT = """\
@@ -28,7 +26,8 @@ def check_consistency(
chapter_text: str,
api_key: str,
base_url: str,
model_name: str
model_name: str,
temperature: float = 0.3
) -> str:
"""
调用模型做简单的一致性检查。可扩展更多提示或校验规则。
@@ -43,9 +42,16 @@ def check_consistency(
model=model_name,
api_key=api_key,
base_url=base_url,
temperature=0.3
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()