Commit 952a4792 authored by alex yao's avatar alex yao

feat:问答知识库分片 - 分片是否开启字段

parent 131c21e4
package cn.com.poc.knowledge.convert;
import cn.com.poc.knowledge.dto.QAKnowledgeChunkDto;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult;
/**
* @author alex.yao
* @date 2025/3/5
*/
public class QAKnowledgeChunkConvert {
public static QAKnowledgeChunkDto result2Dto(QAKnowledgeChunkResult result) {
QAKnowledgeChunkDto qAKnowledgeChunkDto = new QAKnowledgeChunkDto();
qAKnowledgeChunkDto.setTotalChunk(result.getTotalChunk());
qAKnowledgeChunkDto.setKey(result.getKey());
qAKnowledgeChunkDto.setChunk(result.getChunk());
return qAKnowledgeChunkDto;
}
}
package cn.com.poc.knowledge.dto;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.Chunk;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAChunkKey;
import java.util.List;
/**
* @author alex.yao
* @date 2025/3/5
*/
public class QAKnowledgeChunkDto {
private List<QAChunkKey> key;
private List<Chunk> chunk;
private Integer totalChunk;
public Integer getTotalChunk() {
return totalChunk;
}
public void setTotalChunk(Integer totalChunk) {
this.totalChunk = totalChunk;
}
public List<QAChunkKey> getKey() {
return key;
}
public void setKey(List<QAChunkKey> key) {
this.key = key;
}
public List<Chunk> getChunk() {
return chunk;
}
public void setChunk(List<Chunk> chunk) {
this.chunk = chunk;
}
}
package cn.com.poc.knowledge.rest; package cn.com.poc.knowledge.rest;
import cn.com.poc.knowledge.dto.QAKnowledgeChunkDto;
import cn.com.poc.knowledge.dto.UpsertChunkInfoDto; import cn.com.poc.knowledge.dto.UpsertChunkInfoDto;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult; import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult;
import cn.com.yict.framemax.core.rest.BaseRest; import cn.com.yict.framemax.core.rest.BaseRest;
...@@ -41,7 +42,7 @@ public interface QAKnowledgeRest extends BaseRest { ...@@ -41,7 +42,7 @@ public interface QAKnowledgeRest extends BaseRest {
/** /**
* 获取问答知识库分片信息 * 获取问答知识库分片信息
*/ */
QAKnowledgeChunkResult getQAKnowledgeChunks(@RequestParam Integer kdId, @RequestParam(required = false) String query, PagingInfo pagingInfo); QAKnowledgeChunkDto getQAKnowledgeChunks(@RequestParam Integer kdId, @RequestParam(required = false) String query, PagingInfo pagingInfo);
/** /**
* 更新知识库结构信息 * 更新知识库结构信息
......
package cn.com.poc.knowledge.rest.impl; package cn.com.poc.knowledge.rest.impl;
import cn.com.poc.knowledge.aggregate.QAKnowledgeService; import cn.com.poc.knowledge.aggregate.QAKnowledgeService;
import cn.com.poc.knowledge.convert.QAKnowledgeChunkConvert;
import cn.com.poc.knowledge.dto.QAKnowledgeChunkDto;
import cn.com.poc.knowledge.dto.UpsertChunkInfoDto; import cn.com.poc.knowledge.dto.UpsertChunkInfoDto;
import cn.com.poc.knowledge.rest.QAKnowledgeRest; import cn.com.poc.knowledge.rest.QAKnowledgeRest;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult; import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult;
...@@ -41,8 +43,8 @@ public class QAKnowledgeRestImpl implements QAKnowledgeRest { ...@@ -41,8 +43,8 @@ public class QAKnowledgeRestImpl implements QAKnowledgeRest {
} }
@Override @Override
public QAKnowledgeChunkResult getQAKnowledgeChunks(Integer kdId, String query, PagingInfo pagingInfo) { public QAKnowledgeChunkDto getQAKnowledgeChunks(Integer kdId, String query, PagingInfo pagingInfo) {
return qaKnowledgeService.getChunkInfo(kdId, query, pagingInfo); return QAKnowledgeChunkConvert.result2Dto(qaKnowledgeService.getChunkInfo(kdId, query, pagingInfo));
} }
@Override @Override
......
...@@ -12,8 +12,18 @@ public class Chunk { ...@@ -12,8 +12,18 @@ public class Chunk {
private Integer chunkSort; private Integer chunkSort;
private String isOpen;
private List<ChunkInfo> chunkInfo; private List<ChunkInfo> chunkInfo;
public String getIsOpen() {
return isOpen;
}
public void setIsOpen(String isOpen) {
this.isOpen = isOpen;
}
public Integer getChunkSort() { public Integer getChunkSort() {
return chunkSort; return chunkSort;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment