修改模型配置

2026-04-23 15:19:25 +08:00
parent af3ba45558
commit 49ca025b95
5 changed files with 25 additions and 17 deletions
@@ -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
@@ -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()
 )

@@ -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()
    )

@@ -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()
 )

@@ -38,14 +38,14 @@ class Settings(BaseSettings):
    volcengine_api_key: str = "fd065993-bee2-4f31-8bf2-56d5d3012c02"
    volcengine_base_url: str = "https://ark.cn-beijing.volces.com/api/v3"

-    # 心缘
-    cardiacBrder_api_key: str = "sk-8NxoLe7ZTJveGSmtPENBm4NwN9ai4YLGw8y6fqueZrPTo4Uu"
-    cardiacBrder_base_url: str = "https://api-i.xykjy.com/v1"
-
    # jiekou
    jiekou_api_key: str = "sk_4BXi4_yvwx-NW4_zuVt2LufrrlGyhs2q9qMN5Vq_j7Q"
    jiekou_base_url: str = "https://api.jiekou.ai/openai"

+    # jiayu
+    jiayu_api_key: str = "Aa123123"
+    jiayu_base_url: str = "http://192.168.31.133:8800/v1"
+
    # JWT
    jwt_secret: str = "Aa123123"
    token_expire_seconds: int = 5184000