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 |   27 ++++++++++++++++++++++-----
 1 files changed, 22 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 2a934fb..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
@@ -116,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("操作成功");
@@ -138,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)
             );
@@ -162,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
@@ -254,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