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 | 117 +++++++++++++++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 110 insertions(+), 7 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 cdea318..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,22 +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.AiProductQuestionItem; +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; @@ -38,6 +44,7 @@ private final AiProductQuestionMapper aiProductQuestionMapper; private final AiProductQuestionItemService aiProductQuestionItemService; + private final AiService aiService; @Override public AiProductQuestion getById(String id) { @@ -49,6 +56,9 @@ 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; @@ -93,6 +103,7 @@ 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); @@ -119,7 +130,6 @@ 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::getDifficulty, dto.getDifficulty()) @@ -155,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