From 451101e29d1d48a7ed06b6b31fe3f6bdda5dd88a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 29 Sep 2025 15:45:51 +0800
Subject: [PATCH] feat(ai): 新增公司知识库初始化功能
---
src/main/java/cc/mrbird/febs/ai/entity/AiKnowledgeFile.java | 29 ++++++++++++++
src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java | 34 +++++++++++++++++
src/main/java/cc/mrbird/febs/ai/service/AiKnowledgeFileService.java | 7 +++
src/main/java/cc/mrbird/febs/ai/mapper/AiKnowledgeFileMapper.java | 7 +++
src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java | 17 ++++++++
src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java | 11 +++--
6 files changed, 101 insertions(+), 4 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/entity/AiKnowledgeFile.java b/src/main/java/cc/mrbird/febs/ai/entity/AiKnowledgeFile.java
new file mode 100644
index 0000000..ff2757b
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/entity/AiKnowledgeFile.java
@@ -0,0 +1,29 @@
+package cc.mrbird.febs.ai.entity;
+
+import cc.mrbird.febs.common.entity.AiBaseEntity;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+@Data
+@TableName("ai_knowledge_file")
+public class AiKnowledgeFile extends AiBaseEntity {
+ /**
+ * 文件ID
+ */
+ private String fileId;
+
+ /**
+ * 文件存储路径
+ */
+ private String savePath;
+
+ /**
+ * 文件名称
+ */
+ private String name;
+
+ /**
+ * 公司ID
+ */
+ private String companyId;
+}
diff --git a/src/main/java/cc/mrbird/febs/ai/mapper/AiKnowledgeFileMapper.java b/src/main/java/cc/mrbird/febs/ai/mapper/AiKnowledgeFileMapper.java
new file mode 100644
index 0000000..469407e
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/mapper/AiKnowledgeFileMapper.java
@@ -0,0 +1,7 @@
+package cc.mrbird.febs.ai.mapper;
+
+import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+
+public interface AiKnowledgeFileMapper extends BaseMapper<AiKnowledgeFile> {
+}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/AiKnowledgeFileService.java b/src/main/java/cc/mrbird/febs/ai/service/AiKnowledgeFileService.java
new file mode 100644
index 0000000..f4e7230
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/service/AiKnowledgeFileService.java
@@ -0,0 +1,7 @@
+package cc.mrbird.febs.ai.service;
+
+import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
+import com.baomidou.mybatisplus.extension.service.IService;
+
+public interface AiKnowledgeFileService extends IService<AiKnowledgeFile> {
+}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
index 1be9722..37f0506 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
@@ -6,12 +6,16 @@
import cc.mrbird.febs.ai.mapper.AiCompanyMapper;
import cc.mrbird.febs.ai.mapper.AiCompanyWorkflowMapper;
import cc.mrbird.febs.ai.service.AiCompanyService;
+import cc.mrbird.febs.ai.util.KnowledgeBaseUtil;
import cc.mrbird.febs.ai.util.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
+import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.ShareCodeUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
+import com.aliyun.bailian20231229.Client;
+import com.aliyun.bailian20231229.models.CreateIndexResponse;
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,7 @@
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
import java.util.*;
import java.util.stream.Collectors;
@@ -45,6 +50,7 @@
}
@Override
+ @Transactional
public FebsResponse add(AiCompany dto) {
Date createTime = new Date();
AiCompany entity = new AiCompany();
@@ -63,6 +69,34 @@
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REPORT.getCode(),null,createTime);
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.ADMIN_QUESTION.getCode(),null,createTime);
+ String categoryId = "";
+ try {
+ categoryId = KnowledgeBaseUtil.AddCategory(entity.getName(), null);
+ } catch (Exception e) {
+ throw new FebsException("初始化分类失败");
+ }
+
+ String knowledgeId = "";
+ try {
+ Client client = KnowledgeBaseUtil.createClient();
+ String workspaceId = KnowledgeBaseUtil.WORKSPACE_ID;
+ String name = entity.getName();
+ String sourceType = "DATA_CENTER_FILE";
+ String structureType = "unstructured";
+ String sinkType = "DEFAULT";
+ String fileId = "file_b7d0a5b2df1745b9bfb7402f5fe1054f_12629554";
+ CreateIndexResponse indexResponse = KnowledgeBaseUtil.createIndex(client, workspaceId, fileId, name, structureType, sourceType, sinkType);
+ knowledgeId = indexResponse.getBody().getData().getId();
+ } catch (Exception e) {
+ throw new FebsException("初始化知识库失败");
+ }
+
+ this.update(null,
+ Wrappers.lambdaUpdate(AiCompany.class)
+ .set(AiCompany::getKnowledgeId, knowledgeId)
+ .set(AiCompany::getCategoryId, categoryId)
+ .eq(AiCompany::getId, entity.getId()));
+
return new FebsResponse().success().message("操作成功");
}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java
new file mode 100644
index 0000000..afc19e3
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java
@@ -0,0 +1,17 @@
+package cc.mrbird.febs.ai.service.impl;
+
+import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
+import cc.mrbird.febs.ai.mapper.AiKnowledgeFileMapper;
+import cc.mrbird.febs.ai.service.AiKnowledgeFileService;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+@Slf4j
+@Service
+@RequiredArgsConstructor
+public class AiKnowledgeFileServiceImpl extends ServiceImpl<AiKnowledgeFileMapper, AiKnowledgeFile> implements AiKnowledgeFileService {
+
+ private final AiKnowledgeFileMapper aiKnowledgeFileMapper;
+}
diff --git a/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java b/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java
index 0ba8f5a..076ce8c 100644
--- a/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java
+++ b/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java
@@ -17,10 +17,10 @@
import java.util.concurrent.TimeUnit;
public class KnowledgeBaseUtil {
- private static String ACCESS_KEY_ID = "LTAI5tCyQRwhZ2eimxCFKbdq";
- private static String ACCESS_KEY_SECRET = "fs1mEwLXg2j9XuKJsFoW8ThQbJFqHl";
- private static String WORKSPACE_ID = "llm-4bcr09yfxlgz0b0t";
- private static String ENDPOINT = "bailian.cn-beijing.aliyuncs.com";
+ public static final String ACCESS_KEY_ID = "LTAI5tCyQRwhZ2eimxCFKbdq";
+ public static final String ACCESS_KEY_SECRET = "fs1mEwLXg2j9XuKJsFoW8ThQbJFqHl";
+ public static final String WORKSPACE_ID = "llm-4bcr09yfxlgz0b0t";
+ public static final String ENDPOINT = "bailian.cn-beijing.aliyuncs.com";
/**
* <b>description</b> :
@@ -44,6 +44,9 @@
* @throws Exception 当API调用失败或其他异常情况时抛出
*/
public static String AddCategory(String categoryName, String parentCategoryId) throws Exception {
+ if (StrUtil.isBlank(parentCategoryId)){
+ parentCategoryId = "";
+ }
// 创建阿里云百炼客户端
com.aliyun.bailian20231229.Client client = KnowledgeBaseUtil.createClient();
// 构造添加分类请求参数
--
Gitblit v1.9.1