修改模型配置

This commit is contained in:
zk
2026-04-23 15:19:25 +08:00
parent af3ba45558
commit 49ca025b95
5 changed files with 25 additions and 17 deletions
+3 -3
View File
@@ -23,14 +23,14 @@ def _parse_json(text: str) -> dict:
# 诊断链(StrOutputParser 拿原始文本,再手动解析 JSON,避免 markdown 代码块导致解析失败)
_diagnose_chain = (
ChatPromptTemplate.from_messages([("system", DIAGNOSE_MODULE_PROMPT), ("human", "请开始诊断。")])
| LLM.CLAUDE_SONNET_4.create(temperature=0)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0)
| StrOutputParser()
)
# 汇总评价链(纯文本输出)
_summary_chain = (
ChatPromptTemplate.from_messages([("system", SUMMARY_PROMPT), ("human", "请生成整体评价。")])
| LLM.CLAUDE_SONNET_4.create(temperature=0.3)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3)
| StrOutputParser()
)
@@ -64,7 +64,7 @@ async def generate_summary(grade: str, urgent_total: int, important_total: int,
_polish_chain = (
ChatPromptTemplate.from_messages([("system", POLISH_PROMPT), ("human", "请开始优化。")])
| LLM.CLAUDE_SONNET_4.create(temperature=0.3)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3)
| StrOutputParser()
)