From e1583cd7e9c87f8b008757372debc5217e4e3d6e Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 22 Jan 2026 15:49:04 +0800
Subject: [PATCH] feat(ai): 添加AI产品对话轮数功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java | 39 ++++++++++++++++++++++++++++++++++-----
1 files changed, 34 insertions(+), 5 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
index 51e1572..def2c3f 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
@@ -55,6 +55,9 @@
if (StrUtil.isNotEmpty(dto.getProductCategoryId())){
query.eq(AiProduct::getProductCategoryId, dto.getProductCategoryId());
}
+ if (StrUtil.isNotEmpty(dto.getCompanyId())){
+ query.eq(AiProduct::getCompanyId, dto.getCompanyId());
+ }
query.ne(AiProduct::getState, 2);
query.orderByDesc(AiProduct::getHotState);
query.orderByAsc(AiProduct::getSort);
@@ -113,6 +116,7 @@
entity.setTarget(dto.getTarget());
entity.setDescription(dto.getDescription());
entity.setQuestionCount(dto.getQuestionCount());
+ entity.setTalkCnt(dto.getTalkCnt());
entity.setCreatedTime(new Date());
this.save(entity);
return new FebsResponse().success().message("操作成功");
@@ -135,6 +139,7 @@
.set(AiProduct::getTarget, dto.getTarget())
.set(AiProduct::getDescription, dto.getDescription())
.set(AiProduct::getQuestionCount, dto.getQuestionCount())
+ .set(AiProduct::getTalkCnt, dto.getTalkCnt())
.set(AiProduct::getUpdatedTime, new Date())
.eq(AiProduct::getId, id)
);
@@ -159,11 +164,12 @@
@Override
public List<AiProduct> selectList(String companyId) {
- return aiProductMapper.selectList(
- Wrappers.lambdaQuery(AiProduct.class)
- .eq(AiProduct::getCompanyId, companyId)
- .ne(AiProduct::getState, 2)
- );
+ LambdaQueryWrapper<AiProduct> query = Wrappers.lambdaQuery(AiProduct.class);
+ if (StrUtil.isNotEmpty(companyId)){
+ query.eq(AiProduct::getCompanyId, companyId);
+ }
+ query.ne(AiProduct::getState, 2);
+ return aiProductMapper.selectList(query);
}
@Override
@@ -173,15 +179,18 @@
List<String> chooseIds = dto.getChooseIds();
AiProduct aiProduct = this.getById(chooseId);
if (ObjectUtil.isNotNull(aiProduct)) {
+ String companyId = aiProduct.getCompanyId();
aiProductPointLinkService.deleteByQuery(
Wrappers.lambdaQuery(AiProductPointLink.class)
.eq(AiProductPointLink::getProductId,chooseId)
+ .eq(AiProductPointLink::getCompanyId,companyId)
);
if(CollUtil.isNotEmpty(chooseIds)){
Date createdTime = new Date();
for (String item : chooseIds){
AiProductPointLink entity = new AiProductPointLink();
entity.setId(UUID.getSimpleUUIDString());
+ entity.setCompanyId(companyId);
entity.setProductId(chooseId);
entity.setProductPointId(item);
entity.setCreatedTime(createdTime);
@@ -199,15 +208,18 @@
List<String> chooseIds = dto.getChooseIds();
AiProduct aiProduct = this.getById(chooseId);
if (ObjectUtil.isNotNull(aiProduct)) {
+ String companyId = aiProduct.getCompanyId();
aiProductRoleLinkService.deleteByQuery(
Wrappers.lambdaQuery(AiProductRoleLink.class)
.eq(AiProductRoleLink::getProductId,chooseId)
+ .eq(AiProductRoleLink::getCompanyId,companyId)
);
if(CollUtil.isNotEmpty(chooseIds)){
Date createdTime = new Date();
for (String item : chooseIds){
AiProductRoleLink entity = new AiProductRoleLink();
entity.setId(UUID.getSimpleUUIDString());
+ entity.setCompanyId(companyId);
entity.setProductId(chooseId);
entity.setProductRoleId(item);
entity.setCreatedTime(createdTime);
@@ -224,15 +236,18 @@
List<String> chooseIds = dto.getChooseIds();
AiProduct aiProduct = this.getById(chooseId);
if (ObjectUtil.isNotNull(aiProduct)) {
+ String companyId = aiProduct.getCompanyId();
aiProductQuestionLinkService.deleteByQuery(
Wrappers.lambdaQuery(AiProductQuestionLink.class)
.eq(AiProductQuestionLink::getProductId,chooseId)
+ .eq(AiProductQuestionLink::getCompanyId,companyId)
);
if(CollUtil.isNotEmpty(chooseIds)){
Date createdTime = new Date();
for (String item : chooseIds){
AiProductQuestionLink entity = new AiProductQuestionLink();
entity.setId(UUID.getSimpleUUIDString());
+ entity.setCompanyId(companyId);
entity.setProductId(chooseId);
entity.setProductQuestionId(item);
entity.setCreatedTime(createdTime);
@@ -242,4 +257,18 @@
}
return new FebsResponse().success().message("操作成功");
}
+
+ @Override
+ public List<AiProduct> listByCategory(AiProduct dto) {
+ LambdaQueryWrapper<AiProduct> query = Wrappers.lambdaQuery(AiProduct.class);
+ if (StrUtil.isNotEmpty(dto.getProductCategoryId())){
+ query.eq(AiProduct::getProductCategoryId, dto.getProductCategoryId());
+ }
+ if (StrUtil.isNotEmpty(dto.getCompanyId())){
+ query.eq(AiProduct::getCompanyId, dto.getCompanyId());
+ }
+ query.ne(AiProduct::getState, 2);
+ query.orderByAsc(AiProduct::getSort);
+ return aiProductMapper.selectList(query);
+ }
}
--
Gitblit v1.9.1