From 7171b917dab0b0b9a250ba8946f0f9c1d73f6bd4 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 08 Apr 2026 10:12:11 +0800
Subject: [PATCH] feat(ai): 添加AI代理及分类实体和服务
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiAgentServiceImpl.java | 88 +++++++++++++++++++++++++++++++++++++++++++
1 files changed, 87 insertions(+), 1 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiAgentServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiAgentServiceImpl.java
index ece9c39..f600f6d 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiAgentServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiAgentServiceImpl.java
@@ -1,15 +1,101 @@
package cc.mrbird.febs.ai.service.impl;
-import cc.mrbird.febs.ai.entity.AiAgent;
+import cc.mrbird.febs.ai.entity.*;
+import cc.mrbird.febs.ai.enumerates.AiCommonEnum;
+import cc.mrbird.febs.ai.enumerates.ProductCategoryLevelEnum;
+import cc.mrbird.febs.ai.mapper.AiAgentCategoryMapper;
import cc.mrbird.febs.ai.mapper.AiAgentMapper;
+import cc.mrbird.febs.ai.mapper.AiAgentStartQuestionMapper;
+import cc.mrbird.febs.ai.req.agent.AiAgentInitDto;
+import cc.mrbird.febs.ai.req.agent.ApiAgentCategoryAllDto;
+import cc.mrbird.febs.ai.req.agent.ApiAgentPageDto;
+import cc.mrbird.febs.ai.res.agent.AiAgentInitVo;
+import cc.mrbird.febs.ai.res.agent.ApiAgentCategoryVo;
+import cc.mrbird.febs.ai.res.agent.ApiAgentVo;
+import cc.mrbird.febs.ai.res.product.ApiProductVo;
+import cc.mrbird.febs.ai.res.productCategory.ApiProductCategoryVo;
import cc.mrbird.febs.ai.service.AiAgentService;
+import cc.mrbird.febs.common.entity.FebsResponse;
+import cc.mrbird.febs.common.exception.FebsException;
+import cn.hutool.core.bean.BeanUtil;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
+import java.util.ArrayList;
+import java.util.List;
+
@Slf4j
@Service
@RequiredArgsConstructor
public class AiAgentServiceImpl extends ServiceImpl<AiAgentMapper, AiAgent> implements AiAgentService {
+
+ private final AiAgentMapper aiAgentMapper;
+ private final AiAgentCategoryMapper aiAgentCategoryMapper;
+ private final AiAgentStartQuestionMapper aiAgentStartQuestionMapper;
+
+ @Override
+ public FebsResponse allCategoryList(ApiAgentCategoryAllDto dto) {
+ List<ApiAgentCategoryVo> list = new ArrayList<>();
+ LambdaQueryWrapper<AiAgentCategory> query = Wrappers.lambdaQuery(AiAgentCategory.class);
+ if (StrUtil.isEmpty(dto.getCompanyId())){
+ dto.setCompanyId(AiCommonEnum.COMPANY_ID.getPrompt());
+ }
+ query.eq(AiAgentCategory::getCompanyId, dto.getCompanyId());
+ query.eq(AiAgentCategory::getState, 1);
+ query.orderByAsc(AiAgentCategory::getSort);
+ List<AiAgentCategory> listByQuery = aiAgentCategoryMapper.selectList(query);
+ if (CollUtil.isNotEmpty(listByQuery)){
+ for (AiAgentCategory entity : listByQuery){
+ ApiAgentCategoryVo vo = new ApiAgentCategoryVo();
+ vo.setId(entity.getId());
+ vo.setName(entity.getName());
+ list.add(vo);
+ }
+
+ }
+ return new FebsResponse().success().data(list);
+ }
+
+ @Override
+ public FebsResponse agentList(ApiAgentPageDto dto) {
+ // 创建分页对象，传入当前页和每页大小
+ Page<ApiAgentVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ Page<ApiAgentVo> pageListByQuery = aiAgentMapper.getPageListByQuery(page, dto);
+ return new FebsResponse().success().data(pageListByQuery);
+ }
+
+ @Override
+ public FebsResponse initAgent(AiAgentInitDto dto) {
+ String id = dto.getId();
+
+ AiAgent aiAgent = aiAgentMapper.selectById(id);
+ if (aiAgent == null) {
+ throw new FebsException("智能体异常");
+ }
+ AiAgentInitVo vo = new AiAgentInitVo();
+ //将chatWebPlugin复制给apiInitPluginVo
+ BeanUtil.copyProperties(aiAgent, vo);
+
+ List<AiAgentStartQuestion> aiAgentStartQuestions = aiAgentStartQuestionMapper.selectList(
+ Wrappers.lambdaQuery(AiAgentStartQuestion.class)
+ .select(AiAgentStartQuestion::getTitle)
+ .eq(AiAgentStartQuestion::getAgentId, id)
+ );
+ if (CollUtil.isNotEmpty(aiAgentStartQuestions)){
+ List<String> items = new ArrayList<>();
+ for (AiAgentStartQuestion aiAgentStartQuestion : aiAgentStartQuestions) {
+ items.add(aiAgentStartQuestion.getTitle());
+ }
+ vo.setItems( items);
+ }
+
+ return new FebsResponse().success().data(vo);
+ }
}
--
Gitblit v1.9.1