抽象json提取风格

This commit is contained in:
zk
2026-04-23 17:49:06 +08:00
parent fa7d87e435
commit 8fbb1f6fca
6 changed files with 34 additions and 37 deletions
+2 -10
View File
@@ -1,9 +1,7 @@
"""简历并行提取:将完整简历文本拆分为5个AI任务并行提取"""
import asyncio
import re
from json_repair import repair_json
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
@@ -13,13 +11,7 @@ from app.ai.resume_extractor.prompts import (
PROJECT_PROMPT, COMPETITION_PROMPT,
)
from app.core.logger import log
def _parse_json(text: str) -> dict:
"""解析 AI 输出的 JSON,自动去除 markdown 代码块包裹,容错处理"""
cleaned = re.sub(r"^```(?:json)?\s*\n?", "", text.strip())
cleaned = re.sub(r"\n?```\s*$", "", cleaned)
return repair_json(cleaned, return_objects=True)
from app.tool.json_helper import parse_llm_json
def _build_chain(prompt: str):
@@ -65,7 +57,7 @@ async def _safe_invoke(chain, inp: dict, label: str):
"""单个链调用,失败返回空"""
try:
raw = await chain.ainvoke(inp)
return _parse_json(raw)
return parse_llm_json(raw)
except Exception as e:
log.warning(f"AI提取[{label}]失败: {e}")
return {} if "个人信息" in label else []