diff --git a/app/ai/models.py b/app/ai/models.py index 18ec925..ce08d31 100644 --- a/app/ai/models.py +++ b/app/ai/models.py @@ -18,6 +18,7 @@ _VOLCENGINE = (lambda: settings.volcengine_api_key, lambda: settings.volcengine_ _JIAYU = (lambda: settings.jiayu_api_key, lambda: settings.jiayu_base_url) _JIEKOU = (lambda: settings.jiekou_api_key, lambda: settings.jiekou_base_url) +_ZM = (lambda: settings.zm_api_key, lambda: settings.zm_base_url) class LLM(Enum): @@ -47,6 +48,15 @@ class LLM(Enum): JIAYU_MINIMAX_M2_5 = ("minimax-m2.5", *_JIAYU) + # ZM + ZM_GPT_5_5 = ("gpt-5.5", *_ZM) + ZM_GPT_5_4 = ("gpt-5.4", *_ZM) + ZM_GPT_5_4_MINI = ("gpt-5.4-mini", *_ZM) + ZM_GPT_5_2 = ("gpt-5.2", *_ZM) + ZM_GPT_4O = ("gpt-5.5", *_ZM) + + + def __init__(self, model_name: str, api_key_fn, base_url_fn): self.model_name = model_name self._api_key_fn = api_key_fn diff --git a/app/ai/resume_diagnoser/diagnoser.py b/app/ai/resume_diagnoser/diagnoser.py index 1026609..c0cd59e 100644 --- a/app/ai/resume_diagnoser/diagnoser.py +++ b/app/ai/resume_diagnoser/diagnoser.py @@ -14,14 +14,14 @@ from app.tool.json_helper import parse_llm_json # 诊断链(StrOutputParser 拿原始文本,再手动解析 JSON,避免 markdown 代码块导致解析失败) _diagnose_chain = ( ChatPromptTemplate.from_messages([("system", DIAGNOSE_MODULE_PROMPT), ("human", "请开始诊断。")]) - | LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0) + | LLM.ZM_GPT_5_5.create(temperature=0) | StrOutputParser() ) # 汇总评价链(纯文本输出) _summary_chain = ( ChatPromptTemplate.from_messages([("system", SUMMARY_PROMPT), ("human", "请生成整体评价。")]) - | LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3) + | LLM.ZM_GPT_5_5.create(temperature=0.3) | StrOutputParser() ) diff --git a/app/config/settings.py b/app/config/settings.py index 272ddeb..b38f3c9 100644 --- a/app/config/settings.py +++ b/app/config/settings.py @@ -46,6 +46,13 @@ class Settings(BaseSettings): jiayu_api_key: str = "Aa123123" jiayu_base_url: str = "http://192.168.31.133:8800/v1" + #zm + zm_api_key: str = "3-chatgpt-is-free-free-free" + zm_base_url: str = "http://re.94xy.cn/v1" + + + + # JWT jwt_secret: str = "Aa123123" token_expire_seconds: int = 5184000