From 33fb9a2fdb0e6cd68a987bbdcbe22ef587fa5813 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Wed, 17 Sep 2025 16:30:32 +0800 Subject: [PATCH] refactor(mallMember): 修改会员列表中管理员字段的标题 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java | 169 +++++++++++++++++++++++++++++++++++++++++++++++++++---- 1 files changed, 155 insertions(+), 14 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java index 9b23d92..06a3247 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java @@ -1,19 +1,28 @@ package cc.mrbird.febs.ai.service.impl; -import cc.mrbird.febs.ai.entity.AiMemberRole; -import cc.mrbird.febs.ai.entity.AiProductPoint; -import cc.mrbird.febs.ai.entity.AiProductQuestion; +import cc.mrbird.febs.ai.entity.*; import cc.mrbird.febs.ai.mapper.AiProductQuestionMapper; +import cc.mrbird.febs.ai.req.AiProductQuestionAiDto; +import cc.mrbird.febs.ai.service.AiProductQuestionItemService; import cc.mrbird.febs.ai.service.AiProductQuestionService; +import cc.mrbird.febs.ai.service.AiService; import cc.mrbird.febs.ai.util.UUID; import cc.mrbird.febs.common.entity.FebsResponse; import cc.mrbird.febs.common.entity.QueryRequest; +import cn.hutool.core.collection.CollUtil; import cn.hutool.core.util.ObjectUtil; +import cn.hutool.core.util.StrUtil; +import cn.hutool.json.JSONArray; +import cn.hutool.json.JSONObject; +import cn.hutool.json.JSONUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; +import com.fasterxml.jackson.databind.SerializationFeature; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; @@ -34,6 +43,8 @@ public class AiProductQuestionServiceImpl extends ServiceImpl<AiProductQuestionMapper, AiProductQuestion> implements AiProductQuestionService { private final AiProductQuestionMapper aiProductQuestionMapper; + private final AiProductQuestionItemService aiProductQuestionItemService; + private final AiService aiService; @Override public AiProductQuestion getById(String id) { @@ -45,6 +56,10 @@ public IPage<AiProductQuestion> listInPage(AiProductQuestion dto, QueryRequest request) { Page<AiProductQuestion> page = new Page<>(request.getPageNum(), request.getPageSize()); LambdaQueryWrapper<AiProductQuestion> query = Wrappers.lambdaQuery(AiProductQuestion.class); + if (StrUtil.isNotEmpty(dto.getCompanyId())){ + query.eq(AiProductQuestion::getCompanyId, dto.getCompanyId()); + } + query.ne(AiProductQuestion::getState, 2); Page<AiProductQuestion> pages = aiProductQuestionMapper.selectPage(page, query); return pages; } @@ -65,39 +80,72 @@ @Override public FebsResponse add(AiProductQuestion dto) { + List<AiProductQuestionItem> aiProductQuestionItems = dto.getAiProductQuestionItems(); + if(CollUtil.isEmpty(aiProductQuestionItems)){ + return new FebsResponse().fail().message("请添加答案选项"); + } + //stream流操作aiProductQuestionItems,返回aiProductQuestionItems是否存在中correctAnswer为1的元素 + boolean hasCorrectAnswer = aiProductQuestionItems.stream().anyMatch(item -> item != null && item.getCorrectAnswer() != null && item.getCorrectAnswer() == 1); + if (!hasCorrectAnswer) { + return new FebsResponse().fail().message("请添加一项为正确的答案"); + } + AiProductQuestion entity = new AiProductQuestion(); entity.setId(UUID.getSimpleUUIDString()); entity.setCompanyId(dto.getCompanyId()); entity.setProductCategoryId(dto.getProductCategoryId()); entity.setTitle(dto.getTitle()); - entity.setAnswer(dto.getAnswer()); - entity.setCorrectAnswer(dto.getCorrectAnswer()); - entity.setParentId(dto.getParentId()); - entity.setAnswerAnalysis(dto.getAnswerAnalysis()); entity.setDifficulty(dto.getDifficulty()); entity.setCreatedTime(new Date()); this.save(entity); + + + for (AiProductQuestionItem item : dto.getAiProductQuestionItems()){ + item.setId(UUID.getSimpleUUIDString()); + item.setProductQuestionId(entity.getId()); + item.setCompanyId(dto.getCompanyId()); + item.setTitle(entity.getTitle()); + item.setCreatedTime(new Date()); + aiProductQuestionItemService.getBaseMapper().insert(item); + } return new FebsResponse().success().message("操作成功"); } + + @Override public FebsResponse update(AiProductQuestion dto) { String id = dto.getId(); AiProductQuestion entity = this.getById(id); + + List<AiProductQuestionItem> aiProductQuestionItems = dto.getAiProductQuestionItems(); + if(CollUtil.isEmpty(aiProductQuestionItems)){ + return new FebsResponse().fail().message("请添加答案选项"); + } + //stream流操作aiProductQuestionItems,返回aiProductQuestionItems是否存在中correctAnswer为1的元素 + boolean hasCorrectAnswer = aiProductQuestionItems.stream().anyMatch(item -> item != null && item.getCorrectAnswer() != null && item.getCorrectAnswer() == 1); + if (!hasCorrectAnswer) { + return new FebsResponse().fail().message("请添加一项为正确的答案"); + } if (ObjectUtil.isNotNull( entity)){ this.update(null, Wrappers.lambdaUpdate(AiProductQuestion.class) - .set(AiProductQuestion::getCompanyId, dto.getCompanyId()) .set(AiProductQuestion::getProductCategoryId, dto.getProductCategoryId()) .set(AiProductQuestion::getTitle, dto.getTitle()) - .set(AiProductQuestion::getAnswer, dto.getAnswer()) - .set(AiProductQuestion::getCorrectAnswer, dto.getCorrectAnswer()) - .set(AiProductQuestion::getParentId, dto.getParentId()) - .set(AiProductQuestion::getAnswerAnalysis, dto.getAnswerAnalysis()) .set(AiProductQuestion::getDifficulty, dto.getDifficulty()) .set(AiProductQuestion::getUpdatedTime, new Date()) .eq(AiProductQuestion::getId, id) ); + + aiProductQuestionItemService.deleteByQuestionId(id); + + for (AiProductQuestionItem item : dto.getAiProductQuestionItems()){ + item.setId(UUID.getSimpleUUIDString()); + item.setProductQuestionId(entity.getId()); + item.setTitle(entity.getTitle()); + item.setCreatedTime(new Date()); + aiProductQuestionItemService.getBaseMapper().insert(item); + } } return new FebsResponse().success().message("操作成功"); } @@ -117,9 +165,102 @@ } @Override - public List<AiProductQuestion> questionTree() { + public List<AiProductQuestion> questionTree(String companyId) { + LambdaQueryWrapper<AiProductQuestion> query = Wrappers.lambdaQuery(AiProductQuestion.class); + if (StrUtil.isNotEmpty(companyId)){ + query.eq(AiProductQuestion::getCompanyId, companyId); + } - return aiProductQuestionMapper.selectList(null); + return aiProductQuestionMapper.selectList(query); + } + + @Override + public List<AiProductQuestion> productQuestionTree(LambdaQueryWrapper<AiProductQuestion> aiProductQuestionLambdaQueryWrapper) { + return aiProductQuestionMapper.selectList(aiProductQuestionLambdaQueryWrapper); + } + + @Override + public FebsResponse aiAdd(AiProductQuestionAiDto dto) { + + String jsonFormat = "{\"question_list\":[{\"title\": \"消费者对透明质酸的主要担忧是什么?\",\"answer_list\":[{\"answer\": \"消费者担心透明质酸维持时间短,效果不明显。\",\"type\": 1,\"analysis\": \"\"},{\"answer\": \"消费者担心透明质酸注射后可能出现移位、凹陷、馒化、僵硬以及炎症反应。\",\"type\": 1,\"analysis\": \"\"},{\"answer\": \"消费者的主要担忧包括效果不明显、维持时间短、注射后出现移位、凹陷、馒化、僵硬以及炎症反应。他们希望既达到理想效果,又避免这些副作用。\",\"type\": 2,\"analysis\": \"标准答案涵盖了消费者对透明质酸的所有主要担忧,既包括效果问题,也涵盖副作用风险,全面反映消费者心理需求。\"}]}]} "; + dto.setJsonFormat(jsonFormat); + String questionAndAnswerStr = aiService.llmInvokeNonStreaming(dto); + System.out.println(questionAndAnswerStr); + // 解析AI返回的结果 + List<JSONObject> questionList = parseAiQuestionResponse(questionAndAnswerStr); + + String productCategoryId = dto.getProductCategoryId(); + Integer difficulty = dto.getDifficulty(); + Date createdTime = new Date(); + // 处理解析后的问题列表 + for (JSONObject questionObj : questionList) { + String title = questionObj.getStr("title"); + + AiProductQuestion aiProductQuestion = new AiProductQuestion(); + aiProductQuestion.setId(UUID.getSimpleUUIDString()); + aiProductQuestion.setCompanyId(dto.getCompanyId()); + aiProductQuestion.setProductCategoryId(productCategoryId); + aiProductQuestion.setTitle(title); + aiProductQuestion.setDifficulty(difficulty); + aiProductQuestion.setCreatedTime(createdTime); + this.save(aiProductQuestion); + + JSONArray answerList = questionObj.getJSONArray("answer_list"); + System.out.println("问题: " + title); + for (int i = 0; i < answerList.size(); i++) { + JSONObject answer = answerList.getJSONObject(i); + System.out.println("答案" + (i+1) + ": " + answer.getStr("answer")); + System.out.println("类型: " + answer.getStr("type")); + System.out.println("分析: " + answer.getStr("analysis")); + AiProductQuestionItem aiProductQuestionItem = new AiProductQuestionItem(); + aiProductQuestionItem.setId(UUID.getSimpleUUIDString()); + aiProductQuestionItem.setCompanyId(dto.getCompanyId()); + aiProductQuestionItem.setProductQuestionId(aiProductQuestion.getId()); + aiProductQuestionItem.setTitle(aiProductQuestion.getTitle()); + aiProductQuestionItem.setAnswer(answer.getStr("answer")); + aiProductQuestionItem.setCorrectAnswer(answer.getInt("type") == 2 ? 1 : 0); + aiProductQuestionItem.setAnswerAnalysis(answer.getStr("analysis")); + aiProductQuestionItem.setCreatedTime(createdTime); + aiProductQuestionItemService.getBaseMapper().insert(aiProductQuestionItem); + } + } + return new FebsResponse().success().message("操作成功"); + } + + /** + * 解析AI返回的包含问题和答案的JSON字符串 + * @param aiResponse AI返回的原始响应字符串 + * @return 解析后的问题列表 + */ + public List<JSONObject> parseAiQuestionResponse(String aiResponse) { + try { + // 解析外层JSON + JSONObject outerJson = JSONUtil.parseObj(aiResponse); + + // 提取output字段 + String output = outerJson.getStr("output"); + + // 去除<start>和</start>标签,并清理多余字符 + String jsonContent = output.replace("<start>", "") + .replace("</start>", "") + .replace("\\n", "") + .replace("\\n", "") + .replace("\\\"", "\"") + .trim(); + + // 解析内部JSON + JSONObject innerJson = JSONUtil.parseObj(jsonContent); + + // 提取question_list + JSONArray questionList = innerJson.getJSONArray("question_list"); + + // 转换为List<JSONObject>返回 + return questionList.toList(JSONObject.class); + + } catch (Exception e) { + log.error("解析AI问题响应失败: ", e); + throw new RuntimeException("解析AI响应失败", e); + } } } -- Gitblit v1.9.1