"""求职助手 - 岗位简历优化 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