岗位自动投递agent 对话功能

app/services/job_agent_chat_service.py

@@ -0,0 +1,85 @@
+"""求职助手 Agent 对话 Service
+
+主要功能：查询简历数据，调用 AI 模块完成对话。
+依赖：UserResume + 5张子表 ORM、job_agent.chat AI 模块
+使用表：bg_user_resume、bg_user_resume_education/work/internship/project/competition（查询简历）
+"""
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.ai.job_agent.chat import agent_chat
+from app.models.user_resume import UserResume
+from app.models.user_resume_education import UserResumeEducation
+from app.models.user_resume_work import UserResumeWork
+from app.models.user_resume_internship import UserResumeInternship
+from app.models.user_resume_project import UserResumeProject
+from app.models.user_resume_competition import UserResumeCompetition
+
+
+class JobAgentChatService:
+
+    def __init__(self, session: AsyncSession):
+        self.session = session
+
+    async def chat(self, user_id: int, resume_id: int, message: str,
+                   history: list[dict], job_categories: list[str],
+                   regions: list[str], industries: list[str]) -> dict:
+        """求职助手对话：查简历 → 序列化 → 调 AI 模块"""
+        resume = await self._get_resume(resume_id, user_id)
+        edu, work, intern, proj, comp = await self._load_sub_tables(resume_id)
+        resume_text = self._build_resume_text(resume, edu, work, intern, proj, comp)
+        return await agent_chat(resume_text, message, history, job_categories, regions, industries)
+
+    async def _get_resume(self, resume_id: int, user_id: int) -> UserResume:
+        """查指定简历"""
+        result = await self.session.execute(select(UserResume).where(UserResume.id == resume_id, UserResume.user_id == user_id))
+        resume = result.scalar_one_or_none()
+        if not resume:
+            raise ValueError("简历不存在")
+        return resume
+
+    async def _load_sub_tables(self, resume_id: int):
+        """查询简历5张子表"""
+        edu = (await self.session.execute(select(UserResumeEducation).where(UserResumeEducation.resume_id == resume_id))).scalars().all()
+        work = (await self.session.execute(select(UserResumeWork).where(UserResumeWork.resume_id == resume_id))).scalars().all()
+        intern = (await self.session.execute(select(UserResumeInternship).where(UserResumeInternship.resume_id == resume_id))).scalars().all()
+        proj = (await self.session.execute(select(UserResumeProject).where(UserResumeProject.resume_id == resume_id))).scalars().all()
+        comp = (await self.session.execute(select(UserResumeCompetition).where(UserResumeCompetition.resume_id == resume_id))).scalars().all()
+        return edu, work, intern, proj, comp
+
+    @staticmethod
+    def _build_resume_text(resume: UserResume, edu_list, work_list, intern_list, proj_list, comp_list) -> str:
+        """将简历数据序列化为文本供 AI 使用"""
+        parts = []
+        if resume.name:
+            parts.append(f"姓名：{resume.name}")
+        if resume.target_position:
+            parts.append(f"目标岗位：{resume.target_position}")
+        if resume.skills:
+            parts.append(f"技能：{'、'.join(resume.skills)}")
+        if resume.certificates:
+            parts.append(f"证书：{'、'.join(resume.certificates)}")
+        if resume.summary:
+            parts.append(f"个人概述：{resume.summary}")
+        if edu_list:
+            parts.append("教育经历：")
+            for r in edu_list:
+                parts.append(f"  - {r.school or ''} {r.major or ''} {r.degree or ''}")
+        if work_list:
+            parts.append("工作经历：")
+            for r in work_list:
+                parts.append(f"  - {r.company_name or ''} {r.position or ''}")
+        if intern_list:
+            parts.append("实习经历：")
+            for r in intern_list:
+                parts.append(f"  - {r.company_name or ''} {r.position or ''}")
+        if proj_list:
+            parts.append("项目经历：")
+            for r in proj_list:
+                parts.append(f"  - {r.project_name or ''} {r.role or ''}")
+        if comp_list:
+            parts.append("竞赛经历：")
+            for r in comp_list:
+                parts.append(f"  - {r.competition_name or ''} {r.award or ''}")
+        return "\n".join(parts) if parts else "暂无简历信息"