From 574bc2e8dba9d1d11be124f2054ba2243fca125c Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 03 Sep 2025 13:43:21 +0800
Subject: [PATCH] feat(ai): 优化 prompt 构建逻辑并添加思考内容输出- 在 AliLlmStrategyServiceImpl 中添加了对思考内容的输出- 在 ApiMemberTalkStreamServiceImpl 中优化了 prompt 的构建逻辑 - 在 TestController 中调整了测试用例，使用枚举生成 prompt- 在 TestController 中添加了对思考内容的输出
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java | 106 ++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 100 insertions(+), 6 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java
index 44a88fd..5ba0dd9 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java
@@ -9,6 +9,7 @@
import cc.mrbird.febs.ai.service.*;
import cc.mrbird.febs.ai.utils.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
+import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.LoginUserUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateUnit;
@@ -53,15 +54,15 @@
return aiMemberAnswerMapper.selectOne(aiMemberAnswerLambdaQueryWrapper);
}
-
@Override
- public FebsResponse startAnswer(ApiMemberAnswerDto dto) {
+ public FebsResponse startAnswerV2(ApiMemberAnswerDto dto) {
String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ ApiMemberAnswerVoV2 apiMemberAnswerVoV2 = new ApiMemberAnswerVoV2();
String productId = dto.getId();
AiProduct aiProduct = aiProductService.getById(productId);
if(ObjectUtil.isNull(aiProduct)){
- throw new RuntimeException("产品不存在");
+ throw new FebsException("产品不存在");
}
//验证产品分类下是否有题目
String productCategoryId = aiProduct.getProductCategoryId();
@@ -72,7 +73,80 @@
List<AiProductQuestion> aiProductQuestions = aiProductQuestionService.getListByQuery(questionLambdaQueryWrapper);
if(CollUtil.isEmpty(aiProductQuestions)){
// todo 没有题目采用AI生成题目
- throw new RuntimeException("产品没有题目");
+ throw new FebsException("产品没有题目");
+ }
+ //判断该产品是否有正在进行的练习
+ Date createdTime = new Date();
+ LambdaQueryWrapper<AiMemberAnswer> aiMemberAnswerLambdaQueryWrapper = Wrappers.lambdaQuery(AiMemberAnswer.class);
+ aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getMemberId,memberUuid);
+ aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getProductId,aiProduct.getId());
+ aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getState,0);
+ AiMemberAnswer aiMemberAnswer = this.getOneByQuery(aiMemberAnswerLambdaQueryWrapper);
+ if(ObjectUtil.isNull(aiMemberAnswer)){
+ Integer questionCount = aiProduct.getQuestionCount();
+ aiMemberAnswer = new AiMemberAnswer();
+ aiMemberAnswer.setId(UUID.getSimpleUUIDString());
+ aiMemberAnswer.setCreatedTime(createdTime);
+ aiMemberAnswer.setUpdatedTime(createdTime);
+ aiMemberAnswer.setProductId(aiProduct.getId());
+ aiMemberAnswer.setMemberId(memberUuid);
+ aiMemberAnswer.setState(0);
+ aiMemberAnswer.setScore(0);
+ aiMemberAnswer.setQuestionCnt(questionCount);
+ aiMemberAnswer.setDoneCnt(0);
+ aiMemberAnswer.setCorrectCnt(0);
+ aiMemberAnswerMapper.insert(aiMemberAnswer);
+ List<AiMemberAnswerItem> aiMemberAnswerItems = new ArrayList<>();
+ for (int i = 1; i <= questionCount; i++){
+ //初始化答题记录
+ AiProductQuestion aiProductQuestion = aiProductQuestionService.createQuestion(aiProduct.getId());
+ AiMemberAnswerItem aiMemberAnswerItem = new AiMemberAnswerItem();
+ aiMemberAnswerItem.setId(UUID.getSimpleUUIDString());
+ aiMemberAnswerItem.setCreatedTime(createdTime);
+ aiMemberAnswerItem.setAnswerId(aiMemberAnswer.getId());
+ aiMemberAnswerItem.setProductId(aiMemberAnswer.getProductId());
+ aiMemberAnswerItem.setMemberId(memberUuid);
+ aiMemberAnswerItem.setProductQuestionId(aiProductQuestion.getId());
+ aiMemberAnswerItem.setTitle(aiProductQuestion.getTitle());
+ aiMemberAnswerItem.setDifficulty(aiProductQuestion.getDifficulty());
+ aiMemberAnswerItem.setMemberAnswerState(0);
+ aiMemberAnswerItem.setSort( i );
+ aiMemberAnswerItems.add(aiMemberAnswerItem);
+ }
+ aiMemberAnswerItemService.saveBatch(aiMemberAnswerItems);
+ apiMemberAnswerVoV2.setCurrent(1);
+ }else{
+ if (aiMemberAnswer.getQuestionCnt() > aiMemberAnswer.getDoneCnt()){
+ apiMemberAnswerVoV2.setCurrent(aiMemberAnswer.getDoneCnt() + 1);
+ }else{
+ apiMemberAnswerVoV2.setCurrent(aiMemberAnswer.getDoneCnt());
+ }
+ }
+ apiMemberAnswerVoV2.setMemberAnswerId(aiMemberAnswer.getId());
+
+ return new FebsResponse().success().data(apiMemberAnswerVoV2);
+ }
+
+
+ @Override
+ public FebsResponse startAnswer(ApiMemberAnswerDto dto) {
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+
+ String productId = dto.getId();
+ AiProduct aiProduct = aiProductService.getById(productId);
+ if(ObjectUtil.isNull(aiProduct)){
+ throw new FebsException("产品不存在");
+ }
+ //验证产品分类下是否有题目
+ String productCategoryId = aiProduct.getProductCategoryId();
+ LambdaQueryWrapper<AiProductQuestion> questionLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestion.class);
+ questionLambdaQueryWrapper.eq(AiProductQuestion::getProductCategoryId,productCategoryId);
+ questionLambdaQueryWrapper.eq(AiProductQuestion::getState,1);
+ questionLambdaQueryWrapper.last("limit 1");
+ List<AiProductQuestion> aiProductQuestions = aiProductQuestionService.getListByQuery(questionLambdaQueryWrapper);
+ if(CollUtil.isEmpty(aiProductQuestions)){
+ // todo 没有题目采用AI生成题目
+ throw new FebsException("产品没有题目");
}
//判断该产品是否有正在进行的练习
Date createdTime = new Date();
@@ -173,7 +247,7 @@
AiMemberAnswer aiMemberAnswer = this.getById(memberAnswerId);
if(ObjectUtil.isNull(aiMemberAnswer)){
- throw new RuntimeException("答题记录不存在");
+ throw new FebsException("答题记录不存在");
}
Date createdTime = new Date();
@@ -203,7 +277,7 @@
}else{
aiMemberAnswerItem.setIsCollected(0);
}
- aiMemberAnswerItemService.save(aiMemberAnswerItem);
+ aiMemberAnswerItemService.updateById(aiMemberAnswerItem);
aiMemberAnswer.setDoneCnt(aiMemberAnswer.getDoneCnt() + 1);
if (aiMemberAnswerItem.getMemberAnswerState() == 1){
@@ -240,6 +314,15 @@
long between = DateUtil.between(createdTime, updatedTime, DateUnit.MINUTE);
apiMemberAnswerInsureVo.setUseTime(between);
+ apiMemberAnswerInsureVo.setProductId(aiMemberAnswer.getProductId());
+
+ aiMemberAnswerMapper.update(null,
+ Wrappers.lambdaUpdate(AiMemberAnswer.class)
+ .set(AiMemberAnswer::getUpdatedTime, new Date())
+ .set(AiMemberAnswer::getState, 1)
+ .set(AiMemberAnswer::getScore, percentage.intValue())
+ .eq(AiMemberAnswer::getId, memberAnswerId)
+ );
return new FebsResponse().success().data(apiMemberAnswerInsureVo);
}
@@ -296,4 +379,15 @@
dto.setMemberUuid(memberUuid);
return aiMemberAnswerItemService.wrongPage(dto);
}
+
+ @Override
+ public FebsResponse myWork(ApiMemberProductWorkPageDto dto) {
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ dto.setMemberUuid(memberUuid);
+ // 创建分页对象，传入当前页和每页大小
+ Page<ApiMemberProductWorkVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ Page<ApiMemberProductWorkVo> pageListByQuery = this.getBaseMapper().getPageMyWorkListByQuery(page, dto);
+
+ return new FebsResponse().success().data(pageListByQuery);
+ }
}
--
Gitblit v1.9.1