添加AI 优化接口

app/api/resume_diagnose.py

@@ -3,7 +3,7 @@
 from fastapi import APIRouter
 from pydantic import BaseModel, Field
 
-from app.ai.resume_diagnoser.diagnoser import diagnose_all, generate_summary
+from app.ai.resume_diagnoser.diagnoser import diagnose_all, generate_summary, polish_content
 from app.core.context import RequestContext
 from app.core.database import get_db
 from app.services.resume_diagnose_service import ResumeDiagnoseService, aggregate_results
@@ -83,3 +83,25 @@ async def feedback_issue(issue_id: int, param: FeedbackParam):
     async for session in get_db():
         service = ResumeDiagnoseService(session)
         await service.update_feedback(issue_id, user_id, param.user_feedback)
+
+
+class PolishParam(BaseModel):
+    content: list[str] = Field(..., description="用户编辑后的文本段落数组")
+
+
+@router.post("/issue/{issue_id}/polish", summary="AI润色用户编辑的文本")
+async def polish_issue_content(issue_id: int, param: PolishParam):
+    """基于诊断问题上下文，AI润色用户编辑后的文本"""
+    user_id = RequestContext.user_id.get()
+
+    async for session in get_db():
+        service = ResumeDiagnoseService(session)
+        ctx = await service.get_issue_for_polish(issue_id, user_id)
+
+    result = await polish_content(
+        module_type=ctx["module_label"],
+        reference_content=ctx["optimized_content"],
+        user_content=param.content,
+        is_summary=ctx["is_summary"],
+    )
+    return {"content": result}