From f14f07affa651a0aed36b7c2c3ec56077003b483 Mon Sep 17 00:00:00 2001
From: zk <sim18502043706@163.com>
Date: Fri, 8 May 2026 16:00:12 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/ai/models.py                     | 10 ++++++++++
 app/ai/resume_diagnoser/diagnoser.py |  4 ++--
 app/config/settings.py               |  7 +++++++
 3 files changed, 19 insertions(+), 2 deletions(-)

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