修改模型配置

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
+13 -5
View File
@@ -15,7 +15,7 @@ from app.config import settings
# 供应商连接配置
_VOLCENGINE = (lambda: settings.volcengine_api_key, lambda: settings.volcengine_base_url)
_CARDIAC = (lambda: settings.cardiacBrder_api_key, lambda: settings.cardiacBrder_base_url)
_JIAYU = (lambda: settings.jiayu_api_key, lambda: settings.jiayu_base_url)
_JIEKOU = (lambda: settings.jiekou_api_key, lambda: settings.jiekou_base_url)
@@ -31,12 +31,20 @@ class LLM(Enum):
DOUBAO_SEED_LITE = ("doubao-seed-2-0-lite-260215", *_VOLCENGINE)
DOUBAO_SEED_PRO = ("doubao-seed-2-0-pro-260215", *_VOLCENGINE)
# 心缘
GPT_4O = ("gpt-4o", *_CARDIAC)
GPT_4O_MINI = ("gpt-4o-mini", *_CARDIAC)
CLAUDE_SONNET_4 = ("claude-sonnet-4-20250514", *_JIEKOU)
# jiekou
GPT_4O = ("gpt-4o", *_JIAYU)
GPT_4O_MINI = ("gpt-4o-mini", *_JIEKOU)
GEMINI_FLASH = ("gemini-2.5-flash", *_JIEKOU)
# 加鱼
JIAYU_CLAUDE_SONNET_4_5 = ("claude-sonnet-4.5", *_JIAYU)
JIAYU_CLAUDE_HAIKU_4_5 = ("claude-haiku-4.5", *_JIAYU)
JIAYU_DEEPSEEK_3_2 = ("deepseek-3.2", *_JIAYU)
JIAYU_GLM_5 = ("glm-5", *_JIAYU)
JIAYU_QWEN3_CODER_NEXT = ("qwen3-coder-next", *_JIAYU)
JIAYU_MINIMAX_M2_5 = ("minimax-m2.5", *_JIAYU)
def __init__(self, model_name: str, api_key_fn, base_url_fn):
self.model_name = model_name
self._api_key_fn = api_key_fn
+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()
)
+1 -1
View File
@@ -17,7 +17,7 @@ def _build_chain(prompt: str):
"""构建单个提取链:prompt → LLM → JSON解析"""
return (
ChatPromptTemplate.from_messages([("system", prompt), ("human", "{text}")])
| LLM.CLAUDE_SONNET_4.create(temperature=0)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0)
| JsonOutputParser()
)
+4 -4
View File
@@ -52,7 +52,7 @@ async def analyze_skill_gap(skill_tags: list[str], resume_json: str) -> list[str
_summary_optimize_chain = (
ChatPromptTemplate.from_messages([("system", SUMMARY_OPTIMIZE_PROMPT), ("human", "请开始优化。")])
| LLM.GPT_4O.create(temperature=0.3)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3)
| StrOutputParser()
)
@@ -73,7 +73,7 @@ async def optimize_summary(job_title: str, add_skills: list[str], original_summa
_experience_optimize_chain = (
ChatPromptTemplate.from_messages([("system", EXPERIENCE_OPTIMIZE_PROMPT), ("human", "请开始优化。")])
| LLM.GPT_4O.create(temperature=0.3)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3)
| StrOutputParser()
)
@@ -120,7 +120,7 @@ async def plan_edit(job_title: str, job_description: str, resume_json: str,
_record_edit_chain = (
ChatPromptTemplate.from_messages([("system", AGENT_MODULE_EDIT_PROMPT), ("human", "请执行修改。")])
| LLM.GPT_4O.create(temperature=0.3)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3)
| StrOutputParser()
)
@@ -145,7 +145,7 @@ async def execute_record_edit(job_title: str, job_description: str, instruction:
_record_add_chain = (
ChatPromptTemplate.from_messages([("system", AGENT_MODULE_ADD_PROMPT), ("human", "请生成新记录。")])
| LLM.GPT_4O.create(temperature=0.3)
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3)
| StrOutputParser()
)