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

feat:问答知识库结构查询

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