Commit 4703870a authored by alex yao's avatar alex yao

feat:问答知识库结构查询

parent 96cf4b4c
......@@ -3,6 +3,7 @@ package cn.com.poc.knowledge.aggregate;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.ChunkInfo;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeConfig;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeStructResult;
import cn.com.yict.framemax.data.model.PagingInfo;
import java.util.List;
......@@ -74,7 +75,7 @@ public interface QAKnowledgeService {
* @param kdId 文档ID
* @return
*/
List<QAKnowledgeConfig> getKnowledgeStruct(Integer kdId);
QAKnowledgeStructResult getKnowledgeStruct(Integer kdId);
/**
* 更新分片排序
......
......@@ -7,6 +7,7 @@ import cn.com.poc.thirdparty.resource.demand.ai.aggregate.DemandQAKnowledgeServi
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.ChunkInfo;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeConfig;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeStructResult;
import cn.com.yict.framemax.data.model.PagingInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
......@@ -73,7 +74,7 @@ public class QAKnowledgeServiceImpl implements QAKnowledgeService {
}
@Override
public List<QAKnowledgeConfig> getKnowledgeStruct(Integer kdId) {
public QAKnowledgeStructResult getKnowledgeStruct(Integer kdId) {
logger.info("getKnowledgeStruct: kdId={}", kdId);
String knowledgeId = getKnowledgeId(kdId);
return demandQAKnowledgeService.getKnowledgeStruct(knowledgeId);
......
package cn.com.poc.knowledge.dto;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeConfig;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAChunkKey;
import java.util.List;
......@@ -10,13 +10,13 @@ import java.util.List;
*/
public class QAKnowledgeStructDto {
private List<QAKnowledgeConfig> struct;
private List<QAChunkKey> key;
public List<QAKnowledgeConfig> getStruct() {
return struct;
public List<QAChunkKey> getKey() {
return key;
}
public void setStruct(List<QAKnowledgeConfig> struct) {
this.struct = struct;
public void setKey(List<QAChunkKey> key) {
this.key = key;
}
}
......@@ -8,6 +8,7 @@ import cn.com.poc.knowledge.dto.UpsertChunkInfoDto;
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.QAKnowledgeConfig;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeStructResult;
import cn.com.yict.framemax.data.model.PagingInfo;
import org.springframework.stereotype.Component;
......@@ -68,9 +69,9 @@ public class QAKnowledgeRestImpl implements QAKnowledgeRest {
@Override
public QAKnowledgeStructDto getKnowledgeStruct(Integer kdId) {
List<QAKnowledgeConfig> knowledgeStruct = qaKnowledgeService.getKnowledgeStruct(kdId);
QAKnowledgeStructResult knowledgeStruct = qaKnowledgeService.getKnowledgeStruct(kdId);
QAKnowledgeStructDto dto = new QAKnowledgeStructDto();
dto.setStruct(knowledgeStruct);
dto.setKey(knowledgeStruct.getKey());
return dto;
}
}
......@@ -3,6 +3,7 @@ package cn.com.poc.thirdparty.resource.demand.ai.aggregate;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.ChunkInfo;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeChunkResult;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeConfig;
import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeStructResult;
import cn.com.yict.framemax.data.model.PagingInfo;
import java.util.List;
......@@ -86,6 +87,6 @@ public interface DemandQAKnowledgeService {
*
* @param knowledgeId 知识库ID
*/
List<QAKnowledgeConfig> getKnowledgeStruct(String knowledgeId);
QAKnowledgeStructResult getKnowledgeStruct(String knowledgeId);
}
......@@ -114,12 +114,11 @@ public class DemandQAKnowledgeServiceImpl implements DemandQAKnowledgeService {
}
@Override
public List<QAKnowledgeConfig> getKnowledgeStruct(String knowledgeId) {
public QAKnowledgeStructResult getKnowledgeStruct(String knowledgeId) {
Assert.notBlank(knowledgeId);
QAKnowledgeConfigRequest request = new QAKnowledgeConfigRequest();
QAKnowledgeStructRequest request = new QAKnowledgeStructRequest();
request.setKnowledgeId(knowledgeId);
QAKnowledgeConfigResult qaKnowledgeConfigResult = dgToolsAbstractHttpClient.doRequest(DgtoolsApiRoute.DgtoolsAI.GET_QA_KNOWLEDGE_STRUCT, request, getHeaders());
return qaKnowledgeConfigResult.getQaKnowledgeConfigs();
return dgToolsAbstractHttpClient.doRequest(DgtoolsApiRoute.DgtoolsAI.GET_QA_KNOWLEDGE_STRUCT, request, getHeaders());
}
private List<Header> getHeaders() {
......
......@@ -8,7 +8,7 @@ import java.io.Serializable;
* @author alex.yao
* @date 2025/2/26
*/
public class QAKnowledgeConfigRequest extends AbstractRequest<QAKnowledgeConfigResult> implements Serializable {
public class QAKnowledgeStructRequest extends AbstractRequest<QAKnowledgeStructResult> implements Serializable {
private String knowledgeId;
public String getKnowledgeId() {
......
......@@ -8,15 +8,15 @@ import java.util.List;
* @author alex.yao
* @date 2025/2/26
*/
public class QAKnowledgeConfigResult extends AbstractResult {
public class QAKnowledgeStructResult extends AbstractResult {
private List<QAKnowledgeConfig> qaKnowledgeConfigs;
private List<QAChunkKey> key;
public List<QAKnowledgeConfig> getQaKnowledgeConfigs() {
return qaKnowledgeConfigs;
public List<QAChunkKey> getKey() {
return key;
}
public void setQaKnowledgeConfigs(List<QAKnowledgeConfig> qaKnowledgeConfigs) {
this.qaKnowledgeConfigs = qaKnowledgeConfigs;
public void setKey(List<QAChunkKey> key) {
this.key = key;
}
}
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