zk
86 lines
4.2 KiB
Python
86 lines
4.2 KiB
Python
"""求职助手 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 "暂无简历信息"
|