zk
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
"""求职助手 - 岗位简历优化 AI 引擎
|
|
|
|
针对目标岗位并发优化简历（summary + 经历子表，按单条记录粒度并发）。
|
|
依赖：LLM 枚举、job_agent/prompts、parse_llm_json
|
|
"""
|
|
|
|
import time
|
|
|
|
from langchain_core.output_parsers import StrOutputParser
|
|
from langchain_core.prompts import ChatPromptTemplate
|
|
|
|
from app.ai.job_agent.prompts import RESUME_SUMMARY_OPTIMIZE_PROMPT, RESUME_EXPERIENCE_OPTIMIZE_PROMPT
|
|
from app.ai.model_config import JobAgentModel
|
|
from app.core.logger import log
|
|
from app.tool.json_helper import parse_llm_json
|
|
|
|
# ===== summary 优化 =====
|
|
|
|
_summary_chain = (
|
|
ChatPromptTemplate.from_messages([("system", RESUME_SUMMARY_OPTIMIZE_PROMPT), ("human", "请开始优化。")])
|
|
| JobAgentModel.SUMMARY
|
|
| StrOutputParser()
|
|
)
|
|
|
|
|
|
async def optimize_summary(job_title: str, job_description: str, original_summary: str) -> str:
|
|
"""针对岗位优化个人概述"""
|
|
t0 = time.monotonic()
|
|
try:
|
|
result = await _summary_chain.ainvoke({
|
|
"job_title": job_title, "job_description": job_description or "",
|
|
"original_summary": original_summary or "暂无",
|
|
})
|
|
log.info(f"岗位简历summary优化完成 ({round(time.monotonic() - t0, 2)}s)")
|
|
return result
|
|
except Exception as e:
|
|
log.warning(f"岗位简历summary优化失败: {e} ({round(time.monotonic() - t0, 2)}s)")
|
|
return original_summary
|
|
|
|
|
|
# ===== 单条经历优化 =====
|
|
|
|
_experience_chain = (
|
|
ChatPromptTemplate.from_messages([("system", RESUME_EXPERIENCE_OPTIMIZE_PROMPT), ("human", "请开始优化。")])
|
|
| JobAgentModel.EXPERIENCE
|
|
| StrOutputParser()
|
|
)
|
|
|
|
|
|
async def optimize_experience_record(job_title: str, job_description: str, record_json: str) -> dict | None:
|
|
"""针对岗位优化单条经历记录，返回修改后的记录数据"""
|
|
t0 = time.monotonic()
|
|
try:
|
|
raw = await _experience_chain.ainvoke({
|
|
"job_title": job_title, "job_description": job_description or "",
|
|
"original_module_data": record_json,
|
|
})
|
|
result = parse_llm_json(raw)
|
|
log.info(f"岗位简历经历记录优化完成 ({round(time.monotonic() - t0, 2)}s)")
|
|
return result
|
|
except Exception as e:
|
|
log.warning(f"岗位简历经历记录优化失败: {e} ({round(time.monotonic() - t0, 2)}s)")
|
|
return None
|