From c8e3460c5aefedfa45dc19a7a8d5720a10d0f058 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 23 Sep 2025 15:38:44 +0800
Subject: [PATCH] feat(ai): 引入LlmApplicationAppIdEnum枚举优化工作流类型管理
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java | 46 +++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 43 insertions(+), 3 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java
index 8bead4e..4bde82d 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java
@@ -1,14 +1,18 @@
package cc.mrbird.febs.ai.service.impl;
import cc.mrbird.febs.ai.entity.AiProduct;
+import cc.mrbird.febs.ai.entity.AiProductCategory;
import cc.mrbird.febs.ai.entity.AiProductPoint;
import cc.mrbird.febs.ai.mapper.AiProductPointMapper;
+import cc.mrbird.febs.ai.service.AiProductCategoryService;
import cc.mrbird.febs.ai.service.AiProductPointLinkService;
import cc.mrbird.febs.ai.service.AiProductPointService;
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 com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
@@ -20,6 +24,9 @@
import java.util.Date;
import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
/**
* AI产品知识点 Service实现类
@@ -34,6 +41,7 @@
private final AiProductPointMapper aiProductPointMapper;
private final AiProductPointLinkService aiProductPointLinkService;
+ private final AiProductCategoryService aiProductCategoryService;
@Override
public AiProductPoint getById(String id) {
@@ -44,7 +52,24 @@
public IPage<AiProductPoint> listInPage(AiProductPoint dto, QueryRequest request) {
Page<AiProductPoint> page = new Page<>(request.getPageNum(), request.getPageSize());
LambdaQueryWrapper<AiProductPoint> query = Wrappers.lambdaQuery(AiProductPoint.class);
+ if (StrUtil.isNotEmpty(dto.getCompanyId())){
+ query.eq(AiProductPoint::getCompanyId, dto.getCompanyId());
+ }
Page<AiProductPoint> pages = aiProductPointMapper.selectPage(page, query);
+ List<AiProductPoint> records = pages.getRecords();
+ if (CollUtil.isNotEmpty( records)){
+ Set<String> collect = records.stream().map(AiProductPoint::getProductCategoryId).collect(Collectors.toSet());
+
+ if(CollUtil.isNotEmpty( collect)){
+ Map<String, AiProductCategory> map = aiProductCategoryService.selectMapByIds(collect);
+ for (AiProductPoint record : records){
+ AiProductCategory orDefault = map.getOrDefault(record.getProductCategoryId(), null);
+ if(ObjectUtil.isNotNull(orDefault)){
+ record.setProductCategoryName(orDefault.getName());
+ }
+ }
+ }
+ }
return pages;
}
@@ -52,10 +77,12 @@
public FebsResponse add(AiProductPoint dto) {
AiProductPoint entity = new AiProductPoint();
entity.setId(UUID.getSimpleUUIDString());
+ entity.setProductCategoryId(dto.getProductCategoryId());
entity.setCompanyId(dto.getCompanyId());
entity.setIsNormal(dto.getIsNormal() );
entity.setFinderUserName(dto.getFinderUserName());
entity.setFeedId(dto.getFeedId());
+ entity.setFeedImg(dto.getFeedImg());
entity.setTitle(dto.getTitle());
entity.setDescription(dto.getDescription());
entity.setCreatedTime(new Date());
@@ -71,9 +98,11 @@
this.update(null,
Wrappers.lambdaUpdate(AiProductPoint.class)
.set(AiProductPoint::getIsNormal, dto.getIsNormal())
+ .set(AiProductPoint::getProductCategoryId, dto.getProductCategoryId())
.set(AiProductPoint::getFinderUserName, dto.getFinderUserName())
.set(AiProductPoint::getFeedId, dto.getFeedId())
.set(AiProductPoint::getTitle, dto.getTitle())
+ .set(AiProductPoint::getFeedImg, dto.getFeedImg())
.set(AiProductPoint::getDescription, dto.getDescription())
.set(AiProductPoint::getUpdatedTime, new Date())
.eq(AiProductPoint::getId, id)
@@ -94,11 +123,22 @@
}
@Override
- public List<AiProductPoint> pointTree() {
-
- return aiProductPointMapper.selectList(null);
+ public List<AiProductPoint> pointTree(String companyId) {
+ LambdaQueryWrapper<AiProductPoint> query = Wrappers.lambdaQuery(AiProductPoint.class);
+ if (StrUtil.isNotEmpty(companyId)){
+ query.eq(AiProductPoint::getCompanyId, companyId);
+ }
+ return aiProductPointMapper.selectList(query);
}
+ @Override
+ public List<AiProductPoint> selectList(String companyId) {
+ LambdaQueryWrapper<AiProductPoint> query = Wrappers.lambdaQuery(AiProductPoint.class);
+ if (StrUtil.isNotEmpty(companyId)){
+ query.eq(AiProductPoint::getCompanyId, companyId);
+ }
+ return aiProductPointMapper.selectList(query);
+ }
}
--
Gitblit v1.9.1