feat: 获取知识库文档列表

src/main/java/cn/com/poc/knowledge/rest/KnowledgeRest.java

@@ -65,6 +65,14 @@ public interface KnowledgeRest extends BaseRest {
      */
     List<BizKnowledgeDocumentDto> getList(@RequestParam(required = false) String search, @RequestParam(required = false) String trainStatus, PagingInfo pagingInfo) throws Exception;
 
+    /**
+     * 获取知识库的文档列表
+     *
+     * @param knowledgeInfoId
+     * @return
+     * @throws Exception
+     */
+    List<BizKnowledgeDocumentDto> getKdIdsByKnowledgeInfoId(@RequestParam Integer knowledgeInfoId) throws Exception;
 
     /**
      * 获取知识库的文档列表

src/main/java/cn/com/poc/knowledge/rest/impl/KnowledgeRestImpl.java

@@ -2,6 +2,7 @@ package cn.com.poc.knowledge.rest.impl;
 
 import cn.com.poc.common.constant.CommonConstant;
 import cn.com.poc.common.utils.BlContext;
+import cn.com.poc.common.utils.JsonUtils;
 import cn.com.poc.knowledge.aggregate.KnowledgeService;
 import cn.com.poc.knowledge.constant.KnowledgeConstant;
 import cn.com.poc.knowledge.convert.BizKnowledgeInfoConvert;
@@ -17,9 +18,11 @@ import cn.com.poc.knowledge.rest.KnowledgeRest;
 import cn.com.poc.knowledge.service.BizKnowledgeInfoService;
 import cn.com.poc.support.security.oauth.entity.UserBaseEntity;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.knowledge.GetKnowledgeChunkInfoResult;
+import cn.com.yict.framemax.core.exception.BusinessException;
 import cn.com.yict.framemax.data.model.PagingInfo;
 import cn.hutool.core.bean.BeanUtil;
 import cn.hutool.core.lang.Assert;
+import com.alibaba.fastjson.TypeReference;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Component;
@@ -99,6 +102,33 @@ public class KnowledgeRestImpl implements KnowledgeRest {
         return result;
     }
 
+    @Override
+    public List<BizKnowledgeDocumentDto> getKdIdsByKnowledgeInfoId(Integer knowledgeInfoId) throws Exception {
+        Assert.notNull(knowledgeInfoId);
+        BizKnowledgeInfoEntity bizKnowledgeInfoEntity = bizKnowledgeInfoService.get(knowledgeInfoId);
+        if (bizKnowledgeInfoEntity == null) {
+            throw new BusinessException("知识库不存在");
+        }
+        List<BizKnowledgeDocumentDto> res = new ArrayList<>();
+        String kdIds = bizKnowledgeInfoEntity.getKdIds();
+        if (StringUtils.isNotBlank(kdIds)) {
+            List<Integer> kdIdList = JsonUtils.deSerialize(kdIds, new TypeReference<List<Integer>>() {
+            }.getType());
+            if (CollectionUtils.isEmpty(kdIdList)) {
+                return res;
+            }
+            List<BizKnowledgeDocumentEntity> entities = knowledgeService.getListByKdIds(kdIdList);
+            if (CollectionUtils.isNotEmpty(entities)) {
+                res = entities.stream().map(entity -> {
+                    BizKnowledgeDocumentDto dto = new BizKnowledgeDocumentDto();
+                    BeanUtil.copyProperties(entity, dto);
+                    return dto;
+                }).collect(Collectors.toList());
+            }
+        }
+        return res;
+    }
+
     @Override
     public List<BizKnowledgeDocumentDto> getListByKdIds(List<Integer> kdIds) {
         List<BizKnowledgeDocumentDto> res = new ArrayList<>();