添加AI 优化接口

app/ai/resume_diagnoser/diagnoser.py

@@ -9,7 +9,7 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate
 
 from app.ai.models import LLM
-from app.ai.resume_diagnoser.prompts import DIAGNOSE_MODULE_PROMPT, SUMMARY_PROMPT
+from app.ai.resume_diagnoser.prompts import DIAGNOSE_MODULE_PROMPT, SUMMARY_PROMPT, POLISH_PROMPT
 from app.core.logger import log
 
 
@@ -62,6 +62,45 @@ async def generate_summary(grade: str, urgent_total: int, important_total: int,
         return "简历诊断已完成，请查看各模块的详细诊断结果。"
 
 
+_polish_chain = (
+    ChatPromptTemplate.from_messages([("system", POLISH_PROMPT), ("human", "请开始优化。")])
+    | LLM.CLAUDE_SONNET_4.create(temperature=0.3)
+    | StrOutputParser()
+)
+
+
+async def polish_content(module_type: str, reference_content: list[dict] | str | None,
+                         user_content: list[str], is_summary: bool) -> list[str]:
+    """润色用户编辑后的文本"""
+    ref_text = ""
+    if reference_content:
+        if isinstance(reference_content, list):
+            ref_text = "\n".join(
+                item.get("text", "") if isinstance(item, dict) else str(item)
+                for item in reference_content
+            )
+        else:
+            ref_text = str(reference_content)
+    if not ref_text:
+        ref_text = "无"
+
+    inp = {
+        "module_type": module_type,
+        "reference_content": ref_text,
+        "user_content": "\n".join(user_content),
+        "summary_constraint": "- 注意：此模块只能输出一个段落，数组只能有一个元素" if is_summary else "",
+    }
+    try:
+        raw = await _polish_chain.ainvoke(inp)
+        result = _parse_json(raw)
+        if isinstance(result, list):
+            return [str(item) for item in result]
+        return [str(result)]
+    except Exception as e:
+        log.warning(f"AI润色失败: {e}")
+        return user_content
+
+
 async def _safe_invoke(task: dict) -> dict:
     """单条记录诊断，失败返回空结果"""
     raw = ""

app/ai/resume_diagnoser/prompts.py

@@ -77,3 +77,26 @@ SUMMARY_PROMPT = """你是一位资深简历顾问。请根据以下简历诊断
 4. 一句鼓励或行动建议
 
 直接输出评价文本，不要输出JSON或其他格式标记。控制在300字以内。"""
+
+POLISH_PROMPT = """你是一位资深简历顾问。请对用户提供的简历描述文本进行润色优化，让语言更精练、更专业。
+
+## 模块类型
+{module_type}
+
+## AI 之前的优化版本（仅供参考）
+{reference_content}
+
+## 用户提交的文本（以此为主进行优化）
+{user_content}
+
+## 优化要求
+- 以用户提交的文本为主体进行润色，AI之前的版本仅作参考
+- 让语言更精练、更专业，去除冗余表达
+- 尽量使用数据量化成果
+- 保持原意不变，不凭空捏造内容
+- 输出为 JSON 数组格式，每个元素是一个段落的纯文本
+{summary_constraint}
+
+## 输出格式
+严格输出 JSON 数组，不要输出其他内容：
+["优化后的段落1", "优化后的段落2"]"""