diff --git a/pom.xml b/pom.xml
index 9db7b5154..4613650e0 100644
--- a/pom.xml
+++ b/pom.xml
@@ -23,6 +23,7 @@
yudao-module-erp
+ yudao-module-llm
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/KnowledgeBaseController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/KnowledgeBaseController.java
new file mode 100644
index 000000000..670b99494
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/KnowledgeBaseController.java
@@ -0,0 +1,93 @@
+package cn.iocoder.yudao.module.llm.controller.admin.knowledgebase;
+
+import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
+import cn.iocoder.yudao.framework.common.pojo.CommonResult;
+import cn.iocoder.yudao.framework.common.pojo.PageParam;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
+import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBasePageReqVO;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBaseRespVO;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBaseSaveReqVO;
+import cn.iocoder.yudao.module.llm.dal.dataobject.knowledgebase.KnowledgeBaseDO;
+import cn.iocoder.yudao.module.llm.service.knowledgebase.KnowledgeBaseService;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import org.springframework.security.access.prepost.PreAuthorize;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.web.bind.annotation.*;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import javax.validation.Valid;
+import java.io.IOException;
+import java.util.List;
+
+import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
+import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
+
+@Tag(name = "管理后台 - 知识库")
+@RestController
+@RequestMapping("/llm/knowledge-base")
+@Validated
+public class KnowledgeBaseController {
+
+ @Resource
+ private KnowledgeBaseService knowledgeBaseService;
+
+ @PostMapping("/create")
+ @Operation(summary = "创建知识库")
+ @PreAuthorize("@ss.hasPermission('llm:knowledge-base:create')")
+ public CommonResult createKnowledgeBase(@Valid @RequestBody KnowledgeBaseSaveReqVO createReqVO) {
+ return success(knowledgeBaseService.createKnowledgeBase(createReqVO));
+ }
+
+ @PutMapping("/update")
+ @Operation(summary = "更新知识库")
+ @PreAuthorize("@ss.hasPermission('llm:knowledge-base:update')")
+ public CommonResult updateKnowledgeBase(@Valid @RequestBody KnowledgeBaseSaveReqVO updateReqVO) {
+ knowledgeBaseService.updateKnowledgeBase(updateReqVO);
+ return success(true);
+ }
+
+ @DeleteMapping("/delete")
+ @Operation(summary = "删除知识库")
+ @Parameter(name = "id", description = "编号", required = true)
+ @PreAuthorize("@ss.hasPermission('llm:knowledge-base:delete')")
+ public CommonResult deleteKnowledgeBase(@RequestParam("id") Long id) {
+ knowledgeBaseService.deleteKnowledgeBase(id);
+ return success(true);
+ }
+
+ @GetMapping("/get")
+ @Operation(summary = "获得知识库")
+ @Parameter(name = "id", description = "编号", required = true, example = "1024")
+ @PreAuthorize("@ss.hasPermission('llm:knowledge-base:query')")
+ public CommonResult getKnowledgeBase(@RequestParam("id") Long id) {
+ KnowledgeBaseDO knowledgeBase = knowledgeBaseService.getKnowledgeBase(id);
+ return success(BeanUtils.toBean(knowledgeBase, KnowledgeBaseRespVO.class));
+ }
+
+ @GetMapping("/page")
+ @Operation(summary = "获得知识库分页")
+ @PreAuthorize("@ss.hasPermission('llm:knowledge-base:query')")
+ public CommonResult> getKnowledgeBasePage(@Valid KnowledgeBasePageReqVO pageReqVO) {
+ PageResult pageResult = knowledgeBaseService.getKnowledgeBasePage(pageReqVO);
+ return success(BeanUtils.toBean(pageResult, KnowledgeBaseRespVO.class));
+ }
+
+ @GetMapping("/export-excel")
+ @Operation(summary = "导出知识库 Excel")
+ @PreAuthorize("@ss.hasPermission('llm:knowledge-base:export')")
+ @ApiAccessLog(operateType = EXPORT)
+ public void exportKnowledgeBaseExcel(@Valid KnowledgeBasePageReqVO pageReqVO,
+ HttpServletResponse response) throws IOException {
+ pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
+ List list = knowledgeBaseService.getKnowledgeBasePage(pageReqVO).getList();
+ // 导出 Excel
+ ExcelUtils.write(response, "知识库.xls", "数据", KnowledgeBaseRespVO.class,
+ BeanUtils.toBean(list, KnowledgeBaseRespVO.class));
+ }
+
+}
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBasePageReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBasePageReqVO.java
new file mode 100644
index 000000000..bd24de002
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBasePageReqVO.java
@@ -0,0 +1,43 @@
+package cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo;
+
+import lombok.*;
+import java.util.*;
+import io.swagger.v3.oas.annotations.media.Schema;
+import cn.iocoder.yudao.framework.common.pojo.PageParam;
+import org.springframework.format.annotation.DateTimeFormat;
+import java.time.LocalDateTime;
+
+import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
+
+@Schema(description = "管理后台 - 知识库分页 Request VO")
+@Data
+@EqualsAndHashCode(callSuper = true)
+@ToString(callSuper = true)
+public class KnowledgeBasePageReqVO extends PageParam {
+
+ @Schema(description = "知识库名称", example = "张三")
+ private String knowledgeBaseName;
+
+ @Schema(description = "知识库简介")
+ private String knowledgeBaseIntro;
+
+ @Schema(description = "知识库图标")
+ private String knowledgeBaseLogo;
+
+ @Schema(description = "创建时间")
+ @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
+ private LocalDateTime[] createTime;
+
+ @Schema(description = "topK")
+ private Integer topK;
+
+ @Schema(description = "Score阈值")
+ private Integer score;
+
+ @Schema(description = "知识长度")
+ private Integer knowledgeLength;
+
+ @Schema(description = "文件引用")
+ private String knowledgeFile;
+
+}
\ No newline at end of file
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBaseRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBaseRespVO.java
new file mode 100644
index 000000000..71ce8b88e
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBaseRespVO.java
@@ -0,0 +1,51 @@
+package cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.*;
+import java.util.*;
+import org.springframework.format.annotation.DateTimeFormat;
+import java.time.LocalDateTime;
+import com.alibaba.excel.annotation.*;
+
+@Schema(description = "管理后台 - 知识库 Response VO")
+@Data
+@ExcelIgnoreUnannotated
+public class KnowledgeBaseRespVO {
+
+ @Schema(description = "知识库ID", requiredMode = Schema.RequiredMode.REQUIRED, example = "25984")
+ @ExcelProperty("知识库ID")
+ private Long id;
+
+ @Schema(description = "知识库名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "张三")
+ @ExcelProperty("知识库名称")
+ private String knowledgeBaseName;
+
+ @Schema(description = "知识库简介")
+ @ExcelProperty("知识库简介")
+ private String knowledgeBaseIntro;
+
+ @Schema(description = "知识库图标")
+ @ExcelProperty("知识库图标")
+ private String knowledgeBaseLogo;
+
+ @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED)
+ @ExcelProperty("创建时间")
+ private LocalDateTime createTime;
+
+ @Schema(description = "topK")
+ @ExcelProperty("topK")
+ private Integer topK;
+
+ @Schema(description = "Score阈值")
+ @ExcelProperty("Score阈值")
+ private Integer score;
+
+ @Schema(description = "知识长度")
+ @ExcelProperty("知识长度")
+ private Integer knowledgeLength;
+
+ @Schema(description = "文件引用")
+ @ExcelProperty("文件引用")
+ private String knowledgeFile;
+
+}
\ No newline at end of file
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBaseSaveReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBaseSaveReqVO.java
new file mode 100644
index 000000000..e315a742f
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/knowledgebase/vo/KnowledgeBaseSaveReqVO.java
@@ -0,0 +1,37 @@
+package cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.*;
+import java.util.*;
+import javax.validation.constraints.*;
+
+@Schema(description = "管理后台 - 知识库新增/修改 Request VO")
+@Data
+public class KnowledgeBaseSaveReqVO {
+
+ @Schema(description = "知识库ID", requiredMode = Schema.RequiredMode.REQUIRED, example = "25984")
+ private Long id;
+
+ @Schema(description = "知识库名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "张三")
+ @NotEmpty(message = "知识库名称不能为空")
+ private String knowledgeBaseName;
+
+ @Schema(description = "知识库简介")
+ private String knowledgeBaseIntro;
+
+ @Schema(description = "知识库图标")
+ private String knowledgeBaseLogo;
+
+ @Schema(description = "topK")
+ private Integer topK;
+
+ @Schema(description = "Score阈值")
+ private Integer score;
+
+ @Schema(description = "知识长度")
+ private Integer knowledgeLength;
+
+ @Schema(description = "文件引用")
+ private String knowledgeFile;
+
+}
\ No newline at end of file
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/knowledgebase/KnowledgeBaseDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/knowledgebase/KnowledgeBaseDO.java
new file mode 100644
index 000000000..6f1b957b3
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/knowledgebase/KnowledgeBaseDO.java
@@ -0,0 +1,58 @@
+package cn.iocoder.yudao.module.llm.dal.dataobject.knowledgebase;
+
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.*;
+
+/**
+ * 知识库 DO
+ *
+ * @author 华大大模型
+ */
+@TableName("llm_knowledge_base")
+@KeySequence("llm_knowledge_base_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库,可不写。
+@Data
+@EqualsAndHashCode(callSuper = true)
+@ToString(callSuper = true)
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class KnowledgeBaseDO extends BaseDO {
+
+ /**
+ * 知识库ID
+ */
+ @TableId
+ private Long id;
+ /**
+ * 知识库名称
+ */
+ private String knowledgeBaseName;
+ /**
+ * 知识库简介
+ */
+ private String knowledgeBaseIntro;
+ /**
+ * 知识库图标
+ */
+ private String knowledgeBaseLogo;
+ /**
+ * topK
+ */
+ private Integer topK;
+ /**
+ * Score阈值
+ */
+ private Integer score;
+ /**
+ * 知识长度
+ */
+ private Integer knowledgeLength;
+ /**
+ * 文件引用
+ */
+ private String knowledgeFile;
+
+}
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/knowledgebase/KnowledgeBaseMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/knowledgebase/KnowledgeBaseMapper.java
new file mode 100644
index 000000000..911c3af04
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/knowledgebase/KnowledgeBaseMapper.java
@@ -0,0 +1,31 @@
+package cn.iocoder.yudao.module.llm.dal.mysql.knowledgebase;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBasePageReqVO;
+import cn.iocoder.yudao.module.llm.dal.dataobject.knowledgebase.KnowledgeBaseDO;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * 知识库 Mapper
+ *
+ * @author 华大大模型
+ */
+@Mapper
+public interface KnowledgeBaseMapper extends BaseMapperX {
+
+ default PageResult selectPage(KnowledgeBasePageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(KnowledgeBaseDO::getKnowledgeBaseName, reqVO.getKnowledgeBaseName())
+ .eqIfPresent(KnowledgeBaseDO::getKnowledgeBaseIntro, reqVO.getKnowledgeBaseIntro())
+ .eqIfPresent(KnowledgeBaseDO::getKnowledgeBaseLogo, reqVO.getKnowledgeBaseLogo())
+ .betweenIfPresent(KnowledgeBaseDO::getCreateTime, reqVO.getCreateTime())
+ .eqIfPresent(KnowledgeBaseDO::getTopK, reqVO.getTopK())
+ .eqIfPresent(KnowledgeBaseDO::getScore, reqVO.getScore())
+ .eqIfPresent(KnowledgeBaseDO::getKnowledgeLength, reqVO.getKnowledgeLength())
+ .eqIfPresent(KnowledgeBaseDO::getKnowledgeFile, reqVO.getKnowledgeFile())
+ .orderByDesc(KnowledgeBaseDO::getId));
+ }
+
+}
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseService.java
new file mode 100644
index 000000000..9ca89f434
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseService.java
@@ -0,0 +1,55 @@
+package cn.iocoder.yudao.module.llm.service.knowledgebase;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBasePageReqVO;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBaseSaveReqVO;
+import cn.iocoder.yudao.module.llm.dal.dataobject.knowledgebase.KnowledgeBaseDO;
+
+import javax.validation.Valid;
+
+/**
+ * 知识库 Service 接口
+ *
+ * @author 华大大模型
+ */
+public interface KnowledgeBaseService {
+
+ /**
+ * 创建知识库
+ *
+ * @param createReqVO 创建信息
+ * @return 编号
+ */
+ Long createKnowledgeBase(@Valid KnowledgeBaseSaveReqVO createReqVO);
+
+ /**
+ * 更新知识库
+ *
+ * @param updateReqVO 更新信息
+ */
+ void updateKnowledgeBase(@Valid KnowledgeBaseSaveReqVO updateReqVO);
+
+ /**
+ * 删除知识库
+ *
+ * @param id 编号
+ */
+ void deleteKnowledgeBase(Long id);
+
+ /**
+ * 获得知识库
+ *
+ * @param id 编号
+ * @return 知识库
+ */
+ KnowledgeBaseDO getKnowledgeBase(Long id);
+
+ /**
+ * 获得知识库分页
+ *
+ * @param pageReqVO 分页查询
+ * @return 知识库分页
+ */
+ PageResult getKnowledgeBasePage(KnowledgeBasePageReqVO pageReqVO);
+
+}
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java
new file mode 100644
index 000000000..c67de9e80
--- /dev/null
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java
@@ -0,0 +1,68 @@
+package cn.iocoder.yudao.module.llm.service.knowledgebase;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBasePageReqVO;
+import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBaseSaveReqVO;
+import cn.iocoder.yudao.module.llm.dal.dataobject.knowledgebase.KnowledgeBaseDO;
+import cn.iocoder.yudao.module.llm.dal.mysql.knowledgebase.KnowledgeBaseMapper;
+import org.springframework.stereotype.Service;
+import org.springframework.validation.annotation.Validated;
+
+import javax.annotation.Resource;
+
+/**
+ * 知识库 Service 实现类
+ *
+ * @author 华大大模型
+ */
+@Service
+@Validated
+public class KnowledgeBaseServiceImpl implements KnowledgeBaseService {
+
+ @Resource
+ private KnowledgeBaseMapper knowledgeBaseMapper;
+
+ @Override
+ public Long createKnowledgeBase(KnowledgeBaseSaveReqVO createReqVO) {
+ // 插入
+ KnowledgeBaseDO knowledgeBase = BeanUtils.toBean(createReqVO, KnowledgeBaseDO.class);
+ knowledgeBaseMapper.insert(knowledgeBase);
+ // 返回
+ return knowledgeBase.getId();
+ }
+
+ @Override
+ public void updateKnowledgeBase(KnowledgeBaseSaveReqVO updateReqVO) {
+ // 校验存在
+ validateKnowledgeBaseExists(updateReqVO.getId());
+ // 更新
+ KnowledgeBaseDO updateObj = BeanUtils.toBean(updateReqVO, KnowledgeBaseDO.class);
+ knowledgeBaseMapper.updateById(updateObj);
+ }
+
+ @Override
+ public void deleteKnowledgeBase(Long id) {
+ // 校验存在
+ validateKnowledgeBaseExists(id);
+ // 删除
+ knowledgeBaseMapper.deleteById(id);
+ }
+
+ private void validateKnowledgeBaseExists(Long id) {
+ if (knowledgeBaseMapper.selectById(id) == null) {
+// throw exception(KNOWLEDGE_BASE_NOT_EXISTS);
+ }
+ }
+
+ @Override
+ public KnowledgeBaseDO getKnowledgeBase(Long id) {
+ return knowledgeBaseMapper.selectById(id);
+ }
+
+ @Override
+ public PageResult getKnowledgeBasePage(KnowledgeBasePageReqVO pageReqVO) {
+ return knowledgeBaseMapper.selectPage(pageReqVO);
+ }
+
+}
diff --git a/yudao-server/pom.xml b/yudao-server/pom.xml
index efd53c84a..f43126ed4 100644
--- a/yudao-server/pom.xml
+++ b/yudao-server/pom.xml
@@ -115,6 +115,12 @@
+
+ cn.iocoder.boot
+ yudao-module-llm-biz
+ ${revision}
+
+
org.springframework.boot