添加AI推荐列表

This commit is contained in:
zk
2026-04-24 15:38:27 +08:00
parent 3db75be486
commit f8bf2727f4
7 changed files with 176 additions and 2 deletions
@@ -4,6 +4,7 @@ import lombok.AllArgsConstructor;
import org.jiayunet.pojo.PageResult;
import org.jiayunet.pojo.dto.job.JobDetailDto;
import org.jiayunet.pojo.dto.job.JobDto;
import org.jiayunet.pojo.dto.job.JobAgentRecommendDto;
import org.jiayunet.pojo.param.job.*;
import org.jiayunet.pojo.vo.JobApplyCountVo;
import org.jiayunet.pojo.vo.JobFavoriteCountVo;
@@ -139,4 +140,13 @@ public class JobController {
Long userId = UserSecurityTool.getUserId();
return jobService.listAgentTasks(param, userId);
}
/**
* 求职助手岗位推荐(AI精筛)
*/
@PostMapping("/agent/recommend")
public JobAgentRecommendDto recommendJobs(@RequestBody JobAgentRecommendParam param) {
Long userId = UserSecurityTool.getUserId();
return jobService.recommendJobs(param, userId);
}
}
@@ -0,0 +1,20 @@
package org.jiayunet.pojo.dto.job;
import lombok.Data;
import java.util.List;
/**
* 求职助手岗位推荐出参
*
* @author zk
*/
@Data
public class JobAgentRecommendDto {
/** 推荐说明(20字以内) */
private String summary;
/** 推荐的岗位列表(8-10个,完整列表数据) */
private List<JobDto> list;
}
@@ -0,0 +1,20 @@
package org.jiayunet.pojo.param.job;
import lombok.Data;
import java.util.List;
/**
* 求职助手岗位推荐入参
*
* @author zk
*/
@Data
public class JobAgentRecommendParam {
/** 用户偏好描述,如"更喜欢技术一点的产品" */
private String preference;
/** 排除已推荐过的岗位ID列表 */
private List<Long> excludeJobIds;
}
@@ -30,6 +30,9 @@ public class JobQueryParam extends PageParam {
/** 指定岗位ID列表(用于收藏列表) */
private List<Long> jobIds;
/** 排除岗位ID列表(用于推荐时排除已推荐过的) */
private List<Long> excludeJobIds;
/** 岗位状态过滤(0=有效 1=已下架 2=已过期,可多选,null或空=查所有) */
private List<Integer> statusFilter;
}
@@ -1,5 +1,6 @@
package org.jiayunet.service;
import org.jiayunet.ai.AiChatAbility;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;
@@ -8,11 +9,16 @@ import org.jiayunet.pojo.PageResult;
import org.jiayunet.pojo.dto.job.JobDetailDto;
import org.jiayunet.pojo.dto.job.JobDto;
import org.jiayunet.pojo.dto.job.JobMatchScoreDto;
import org.jiayunet.pojo.dto.job.JobAgentRecommendDto;
import org.jiayunet.pojo.param.job.JobAgentRecommendParam;
import org.jiayunet.pojo.param.job.JobAgentTaskQueryParam;
import org.jiayunet.pojo.param.job.JobApplyQueryParam;
import org.jiayunet.pojo.param.job.JobFavoriteQueryParam;
import org.jiayunet.pojo.param.job.JobQueryParam;
import org.jiayunet.pojo.po.*;
import org.jiayunet.tool.HttpTool;
import org.jiayunet.exception.BusinessException;
import org.jiayunet.exception.BusinessExpCodeEnum;
import org.jiayunet.pojo.vo.JobApplyCountVo;
import org.jiayunet.pojo.vo.JobFavoriteCountVo;
import org.jiayunet.pojo.vo.JobListItemVo;
@@ -20,6 +26,7 @@ import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.fasterxml.jackson.databind.JsonNode;
import java.time.Instant;
import java.util.*;
import java.util.stream.Collectors;
@@ -66,6 +73,12 @@ public class JobService {
@Autowired
private JobMatchService jobMatchService;
@Autowired
private UserJobIntentionMapper userJobIntentionMapper;
@Autowired
private AiChatAbility aiChatAbility;
/**
* 岗位列表查询
* <p>方法逻辑流程:</p>
@@ -96,7 +109,10 @@ public class JobService {
List<UserJobDislike> dislikes = userJobDislikeMapper.selectList(new LambdaQueryWrapper<UserJobDislike>().eq(UserJobDislike::getUserId, userId));
// 3. 提取排除列表
List<Long> excludeJobIds = dislikes.stream().map(UserJobDislike::getJobId).filter(Objects::nonNull).distinct().collect(Collectors.toList());
List<Long> excludeJobIds = new ArrayList<>(dislikes.stream().map(UserJobDislike::getJobId).filter(Objects::nonNull).distinct().collect(Collectors.toList()));
if (param.getExcludeJobIds() != null && !param.getExcludeJobIds().isEmpty()) {
excludeJobIds.addAll(param.getExcludeJobIds());
}
List<Long> excludeCompanyIds = dislikes.stream().map(UserJobDislike::getCompanyId).filter(Objects::nonNull).distinct().collect(Collectors.toList());
// 4. 提取不感兴趣的地区(直接使用,不扩展子级)
@@ -561,4 +577,104 @@ public class JobService {
return vo;
}
/**
* 求职助手岗位推荐
* <p>1. 查求职意向构造筛选条件(无意向则不设条件) 2. 排除已推荐的+已投递的 3. 取前35条候选 4. AI精筛返回8-10个</p>
*/
public JobAgentRecommendDto recommendJobs(JobAgentRecommendParam param, Long userId) {
// 1. 查求职意向
UserJobIntention intention = userJobIntentionMapper.selectOne(new LambdaQueryWrapper<UserJobIntention>().eq(UserJobIntention::getUserId, userId));
// 2. 构造查询参数
JobQueryParam queryParam = new JobQueryParam();
queryParam.setPageNum(1);
queryParam.setPageSize(35);
if (intention != null) {
queryParam.setCategoryIds(intention.getCategoryIds());
queryParam.setRegionCodes(intention.getRegionCodes());
queryParam.setIndustryIds(intention.getIndustryIds());
queryParam.setEmploymentType(intention.getEmploymentType());
}
// 3. 合并排除列表:入参传的 + 已投递/待投递的
List<Long> excludeIds = new ArrayList<>();
if (param.getExcludeJobIds() != null) {
excludeIds.addAll(param.getExcludeJobIds());
}
List<UserJobApplication> applications = userJobApplicationMapper.selectList(new LambdaQueryWrapper<UserJobApplication>().eq(UserJobApplication::getUserId, userId));
applications.forEach(a -> excludeIds.add(a.getJobId()));
queryParam.setExcludeJobIds(excludeIds);
// 4. 查询候选岗位(完整列表数据)
PageResult<JobDto> candidates = listJobs(queryParam, userId);
if (candidates.getList().isEmpty()) {
JobAgentRecommendDto dto = new JobAgentRecommendDto();
dto.setSummary("暂无合适的岗位推荐");
dto.setList(Collections.emptyList());
return dto;
}
// 5. 批量查岗位详情(title + description + requirement
List<Long> candidateJobIds = candidates.getList().stream().map(JobDto::getId).collect(Collectors.toList());
List<Job> jobDetails = jobMapper.selectList(new LambdaQueryWrapper<Job>().in(Job::getId, candidateJobIds).select(Job::getId, Job::getTitle, Job::getDescription, Job::getRequirement));
Map<Long, Job> jobDetailMap = jobDetails.stream().collect(Collectors.toMap(Job::getId, j -> j));
// 6. 构造候选岗位映射(供AI返回后过滤使用)
Map<Long, JobDto> candidateMap = candidates.getList().stream().collect(Collectors.toMap(JobDto::getId, d -> d));
// 7. 构造AI输入
StringBuilder jobInfo = new StringBuilder();
for (JobDto dto : candidates.getList()) {
Job detail = jobDetailMap.get(dto.getId());
jobInfo.append("ID:").append(dto.getId())
.append("\n标题:").append(dto.getTitle())
.append("\n公司:").append(dto.getCompanyName())
.append("\n岗位职责:").append(detail != null ? detail.getDescription() : "")
.append("\n任职要求:").append(detail != null ? detail.getRequirement() : "")
.append("\n---\n");
}
String preferenceInfo = param.getPreference() != null ? param.getPreference() : "无特殊偏好";
String systemPrompt = "你是一个求职推荐助手。根据用户的偏好,从候选岗位中选出8-10个最合适的岗位。" +
"返回JSON格式:{\"summary\":\"推荐说明(20字以内)\",\"jobIds\":[岗位ID列表]}。" +
"只返回JSON,不要其他内容。";
String userMessage = "【用户偏好】\n" + preferenceInfo + "\n\n【候选岗位】\n" + jobInfo;
// 8. 调用AI
String aiResponse = aiChatAbility.chat("job-recommend", systemPrompt, userMessage);
String json = cleanAiResponse(aiResponse);
// 9. 解析AI返回,过滤出选中的岗位
try {
JsonNode root = HttpTool.objectMapper.readTree(json);
String summary = root.path("summary").asText("为你精选了合适的岗位");
List<Long> selectedIds = new ArrayList<>();
root.path("jobIds").forEach(node -> selectedIds.add(node.asLong()));
List<JobDto> selectedJobs = selectedIds.stream()
.map(candidateMap::get)
.filter(Objects::nonNull)
.collect(Collectors.toList());
JobAgentRecommendDto result = new JobAgentRecommendDto();
result.setSummary(summary);
result.setList(selectedJobs);
return result;
} catch (Exception e) {
log.error("AI推荐结果解析失败, response={}", json, e);
throw new BusinessException(BusinessExpCodeEnum.UNKNOWN_ERROR, "AI推荐结果解析失败");
}
}
/** 清理AI返回的markdown代码块和控制字符 */
private String cleanAiResponse(String response) {
String json = response.trim();
if (json.startsWith("```")) {
json = json.replaceAll("^```\\w*\\n?", "").replaceAll("\\n?```$", "").trim();
}
json = json.replaceAll("[\\x00-\\x08\\x0B\\x0C\\x0E-\\x1F]", "");
return json;
}
}
@@ -88,6 +88,11 @@ app:
base-url: ${AI_BASE_URL:https://ark.cn-beijing.volces.com/api/v3}
api-key: ${AI_API_KEY:fd065993-bee2-4f31-8bf2-56d5d3012c02}
model: ${AI_MODEL:doubao-seed-2-0-lite-260215}
job-recommend:
base-url: ${AI_BASE_URL:https://ark.cn-beijing.volces.com/api/v3}
api-key: ${AI_API_KEY:fd065993-bee2-4f31-8bf2-56d5d3012c02}
model: ${AI_MODEL:doubao-1-5-pro-32k-250115}
# 岗位清洗配置
job-clean:
@@ -20,7 +20,7 @@ public class PageParam {
/** 每页条数 */
@Min(value = 1, message = "每页条数最小为1")
@Max(value = 100, message = "每页条数最大为100")
@Max(value = 1000, message = "每页条数最大为100")
private Integer pageSize = 10;
/**