统一抽象 封装 模型配置配置管理

This commit is contained in:
zk
2026-05-26 15:10:21 +08:00
parent b558863207
commit 18589adf8c
10 changed files with 131 additions and 31 deletions
+2 -4
View File
@@ -11,7 +11,7 @@ import time
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from app.ai.models import LLM
from app.ai.model_config import ResumeExtractorModel
from app.ai.resume_extractor.prompts import (
OVERVIEW_PROFILE_PROMPT, OVERVIEW_EDUCATION_PROMPT, OVERVIEW_WORK_PROMPT,
OVERVIEW_PROJECT_PROMPT, OVERVIEW_COMPETITION_PROMPT,
@@ -21,14 +21,12 @@ from app.ai.resume_extractor.prompts import (
from app.core.logger import log
from app.tool.json_helper import parse_llm_json
_LLM_MODEL = LLM.DOUBAO_LITE_32K
# ==================== LLM 调用工具 ====================
def _build_chain(prompt: str):
"""构建提取链:prompt → LLM → 文本输出"""
return ChatPromptTemplate.from_messages([("system", prompt), ("human", "{text}")]) | _LLM_MODEL.create(temperature=0) | StrOutputParser()
return ChatPromptTemplate.from_messages([("system", prompt), ("human", "{text}")]) | ResumeExtractorModel.PARSE | StrOutputParser()
async def _safe_invoke(chain, inp: dict, label: str):