feat：Agent对话 返回知识库和插件信息

src/main/java/cn/com/poc/agent_application/aggregate/AgentApplicationInfoService.java

@@ -4,6 +4,7 @@ import cn.com.poc.agent_application.domain.FunctionResult;
 import cn.com.poc.agent_application.entity.AgentResultEntity;
 import cn.com.poc.agent_application.entity.BizAgentApplicationInfoEntity;
 import cn.com.poc.agent_application.entity.CreateAgentTitleAndDescEntity;
+import cn.com.poc.thirdparty.resource.demand.ai.constants.KnowledgeSearchTypeEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Message;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Tool;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.function.FunctionCallResult;
@@ -38,45 +39,53 @@ public interface AgentApplicationInfoService {
     /**
      * Agent应用对话
      *
-     * @param agentId           应用ID
-     * @param identifier        对话唯一标识
-     * @param largeModel        模型
-     * @param agentSystem       应用角色指令
-     * @param knowledgeIds      知识库ID
-     * @param communicationTurn 对话轮数
-     * @param topP              模型参数topP
-     * @param temperature       模型参数temperature
-     * @param messages          对话消息
-     * @param tools             插件配置
-     * @param fileUrls          文件URLs
-     * @param imageUrls         图片URLs
-     * @param stream            是否流式传输
+     * @param agentId             应用ID
+     * @param identifier          对话唯一标识
+     * @param largeModel          模型
+     * @param agentSystem         应用角色指令
+     * @param knowledgeIds        知识库ID
+     * @param communicationTurn   对话轮数
+     * @param topP                模型参数topP
+     * @param temperature         模型参数temperature
+     * @param messages            对话消息
+     * @param tools               插件配置
+     * @param fileUrls            文件URLs
+     * @param imageUrls           图片URLs
+     * @param stream              是否流式传输
+     * @param score               知识库参数score
+     * @param topK                知识库参数topK
+     * @param knowledgeSearchType 知识库参数知识搜索类型
      */
     AgentResultEntity callAgentApplication(String agentId, String identifier, String largeModel, String agentSystem,
                                            Integer[] knowledgeIds, Integer communicationTurn, Float topP, Float temperature,
-                                           List<Message> messages, List<Tool> tools, List<String> fileUrls, boolean stream, List<String> imageUrls, HttpServletResponse httpServletResponse) throws Exception;
+                                           List<Message> messages, List<Tool> tools, List<String> fileUrls, boolean stream, List<String> imageUrls,
+                                           Double score, Integer topK, KnowledgeSearchTypeEnum knowledgeSearchType, HttpServletResponse httpServletResponse) throws Exception;
 
     /**
      * Agent应用对话
      *
-     * @param agentId            应用ID
-     * @param identifier         对话唯一标识
-     * @param largeModel         模型
-     * @param agentSystem        应用角色指令
-     * @param knowledgeIds       知识库ID
-     * @param communicationTurn  对话轮数
-     * @param topP               模型参数topP
-     * @param temperature        模型参数temperature
-     * @param messages           对话消息
-     * @param tools              插件配置
-     * @param functionCallResult 插件回调结果
-     * @param fileUrls           文件URLs
-     * @param imageUrls          图片URLs
-     * @param stream             是否流式传输
+     * @param agentId             应用ID
+     * @param identifier          对话唯一标识
+     * @param largeModel          模型
+     * @param agentSystem         应用角色指令
+     * @param knowledgeIds        知识库ID
+     * @param communicationTurn   对话轮数
+     * @param topP                模型参数topP
+     * @param temperature         模型参数temperature
+     * @param messages            对话消息
+     * @param tools               插件配置
+     * @param functionCallResult  插件回调结果
+     * @param fileUrls            文件URLs
+     * @param imageUrls           图片URLs
+     * @param stream              是否流式传输
+     * @param score               知识库参数score
+     * @param topK                知识库参数topK
+     * @param knowledgeSearchType 知识库参数知识搜索类型
      */
     AgentResultEntity callAgentApplication(String agentId, String identifier, String largeModel, String agentSystem,
-                                Integer[] knowledgeIds, Integer communicationTurn, Float topP, Float temperature,
-                                List<Message> messages, List<Tool> tools, FunctionCallResult functionCallResult, List<String> fileUrls, boolean stream, List<String> imageUrls, HttpServletResponse httpServletResponse) throws Exception;
+                                           Integer[] knowledgeIds, Integer communicationTurn, Float topP, Float temperature,
+                                           List<Message> messages, List<Tool> tools, FunctionCallResult functionCallResult, List<String> fileUrls, boolean stream, List<String> imageUrls,
+                                           Double score, Integer topK, KnowledgeSearchTypeEnum knowledgeSearchType, HttpServletResponse httpServletResponse) throws Exception;
 
     /**
      * 应用下架

src/main/java/cn/com/poc/agent_application/dto/KnowledgeContentResult.java

+package cn.com.poc.agent_application.dto;
+
+/**
+ * @author alex.yao
+ * @date 2025/2/28
+ */
+public class KnowledgeContentResult {
+
+    private String content;
+
+    private String knowledgeId;
+
+    private String knowledgeName;
+
+    private Integer kdId;
+    private String documentName;
+    private Double score;
+
+    public Integer getKdId() {
+        return kdId;
+    }
+
+    public void setKdId(Integer kdId) {
+        this.kdId = kdId;
+    }
+
+    public String getDocumentName() {
+        return documentName;
+    }
+
+    public void setDocumentName(String documentName) {
+        this.documentName = documentName;
+    }
+
+    public String getContent() {
+        return content;
+    }
+
+    public void setContent(String content) {
+        this.content = content;
+    }
+
+    public String getKnowledgeId() {
+        return knowledgeId;
+    }
+
+    public void setKnowledgeId(String knowledgeId) {
+        this.knowledgeId = knowledgeId;
+    }
+
+    public String getKnowledgeName() {
+        return knowledgeName;
+    }
+
+    public void setKnowledgeName(String knowledgeName) {
+        this.knowledgeName = knowledgeName;
+    }
+
+    public Double getScore() {
+        return score;
+    }
+
+    public void setScore(Double score) {
+        this.score = score;
+    }
+}

src/main/java/cn/com/poc/agent_application/rest/impl/AgentApplicationInfoRestImpl.java

@@ -24,6 +24,7 @@ import cn.com.poc.equity.entity.BizPointDeductionRulesEntity;
 import cn.com.poc.equity.service.BizPointDeductionRulesService;
 import cn.com.poc.knowledge.aggregate.KnowledgeService;
 import cn.com.poc.support.security.oauth.entity.UserBaseEntity;
+import cn.com.poc.thirdparty.resource.demand.ai.constants.KnowledgeSearchTypeEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Tool;
 import cn.com.poc.thirdparty.resource.demand.ai.function.long_memory.AgentLongMemoryEntity;
 import cn.com.poc.thirdparty.resource.demand.ai.function.long_memory.LongMemory;
@@ -35,6 +36,8 @@ import cn.hutool.core.collection.ListUtil;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.collections4.MapUtils;
 import org.apache.commons.lang3.StringUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Component;
 import org.springframework.util.Assert;
 
@@ -48,6 +51,8 @@ import java.util.stream.Collectors;
 @Component
 public class AgentApplicationInfoRestImpl implements AgentApplicationInfoRest {
 
+    private Logger logger = LoggerFactory.getLogger(AgentApplicationInfoRest.class);
+
     @Resource
     private BizAgentApplicationInfoService bizAgentApplicationInfoService;
 
@@ -256,11 +261,11 @@ public class AgentApplicationInfoRestImpl implements AgentApplicationInfoRest {
             Float topP = dto.getTopP() == null ? infoEntity.getTopP() : dto.getTopP();
             Float temperature = dto.getTemperature() == null ? infoEntity.getTemperature() : dto.getTemperature();
             String agentSystem = StringUtils.isBlank(dto.getAgentSystem()) ? infoEntity.getAgentSystem() : dto.getAgentSystem();
-            Integer communicationTurn = dto.getCommunicationTurn() == null ? infoEntity.getCommunicationTurn(): dto.getCommunicationTurn();
+            Integer communicationTurn = dto.getCommunicationTurn() == null ? infoEntity.getCommunicationTurn() : dto.getCommunicationTurn();
 
             // 判断是否调用function
             //计算扣分数
-            CheckPluginUseEntity checkPluginUseEntity = AgentApplicationTools.checkPluginUse(dto.getMessages(), tools,fileUrls,imageUrls);
+            CheckPluginUseEntity checkPluginUseEntity = AgentApplicationTools.checkPluginUse(dto.getMessages(), tools, fileUrls, imageUrls);
             Long pointDeductionNum = pointDeductionRulesService.calculatePointDeductionNum(model, communicationTurn, checkPluginUseEntity.getDeductionTools());
             AgentUseModifyEventInfo agentUseModifyEventInfo = new AgentUseModifyEventInfo();
             agentUseModifyEventInfo.setAgentId(agentId);
@@ -269,7 +274,9 @@ public class AgentApplicationInfoRestImpl implements AgentApplicationInfoRest {
             //调用应用服务
             agentApplicationInfoService.callAgentApplication(agentId, dialogueId, model,
                     agentSystem, kdIds.toArray(new Integer[0]), communicationTurn, topP,
-                    temperature, dto.getMessages(), tools, checkPluginUseEntity.getFunctionCallResult(), dto.getFileUrls(), true, imageUrls, httpServletResponse);
+                    temperature, dto.getMessages(), tools, checkPluginUseEntity.getFunctionCallResult(), dto.getFileUrls(), true, imageUrls,
+                    infoEntity.getKnowledgeSimilarity(), infoEntity.getKnowledgeNResult(), KnowledgeSearchTypeEnum.valueOf(infoEntity.getKnowledgeSearchType()),
+                    httpServletResponse);
             //数据采集
             if (StringUtils.isBlank(dto.getChannel())) {
                 dto.setChannel(DataAnalyzeChannelEnum.preview.getChannel());
@@ -281,7 +288,9 @@ public class AgentApplicationInfoRestImpl implements AgentApplicationInfoRest {
             writer.write("data: {\"code\":-1,\"message\":\"" + e.getLocalizedMessage() + "\"} \n\n");
             writer.write("data: [DONE]\n\n");
             writer.flush();
+            writer.close();
             memberEquityService.rollbackPoint(reduceSn);
+            logger.error("preview error", e);
         }
     }
 

src/main/java/cn/com/poc/expose/aggregate/impl/AgentApplicationApiServiceImpl.java

@@ -18,6 +18,7 @@ import cn.com.poc.equity.domain.modifyEquityInfo.AgentUseModifyEventInfo;
 import cn.com.poc.expose.aggregate.AgentApplicationApiService;
 import cn.com.poc.knowledge.aggregate.KnowledgeService;
 import cn.com.poc.support.security.oauth.constants.OauthConstants;
+import cn.com.poc.thirdparty.resource.demand.ai.constants.KnowledgeSearchTypeEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.constants.LLMRoleEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Message;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Tool;
@@ -164,7 +165,9 @@ public class AgentApplicationApiServiceImpl implements AgentApplicationApiServic
         try {
             AgentResultEntity agentResultEntity = agentApplicationInfoService.callAgentApplication(agentId, conversationId, infoEntity.getLargeModel(),
                     infoEntity.getAgentSystem(), kdIdList.toArray(new Integer[0]), infoEntity.getCommunicationTurn(),
-                    infoEntity.getTopP(), infoEntity.getTemperature(), messages, tools, checkPluginUseEntity.getFunctionCallResult(), fileUrls, stream, imageUrls, httpServletResponse);
+                    infoEntity.getTopP(), infoEntity.getTemperature(), messages, tools, checkPluginUseEntity.getFunctionCallResult(), fileUrls, stream, imageUrls,
+                    infoEntity.getKnowledgeSimilarity(), infoEntity.getKnowledgeNResult(), KnowledgeSearchTypeEnum.valueOf(infoEntity.getKnowledgeSearchType()),
+                    httpServletResponse);
             saveRecord(conversationId, query, agentId, profileEntity, inputTimestamp, infoEntity, agentResultEntity.getMessage());
         } catch (Exception e) {
             memberEquityService.rollbackPoint(reduceSn);

src/main/java/cn/com/poc/expose/aggregate/impl/AgentApplicationServiceImpl.java

@@ -22,6 +22,7 @@ import cn.com.poc.equity.domain.modifyEquityInfo.AgentUseModifyEventInfo;
 import cn.com.poc.expose.aggregate.AgentApplicationService;
 import cn.com.poc.knowledge.aggregate.KnowledgeService;
 import cn.com.poc.support.security.oauth.entity.UserBaseEntity;
+import cn.com.poc.thirdparty.resource.demand.ai.constants.KnowledgeSearchTypeEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.constants.LLMRoleEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Message;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.MultiContent;
@@ -182,7 +183,9 @@ public class AgentApplicationServiceImpl implements AgentApplicationService {
             //对话
             AgentResultEntity agentResultEntity = agentApplicationInfoService.callAgentApplication(agentId, dialogsId, infoEntity.getLargeModel(),
                     infoEntity.getAgentSystem(), kdIdList.toArray(new Integer[0]), infoEntity.getCommunicationTurn(),
-                    infoEntity.getTopP(), infoEntity.getTemperature(), messages, tools, checkPluginUseEntity.getFunctionCallResult(), fileUrls, true, imageUrls, httpServletResponse);
+                    infoEntity.getTopP(), infoEntity.getTemperature(), messages, tools, checkPluginUseEntity.getFunctionCallResult(), fileUrls, true, imageUrls,
+                    infoEntity.getKnowledgeSimilarity(), infoEntity.getKnowledgeNResult(), KnowledgeSearchTypeEnum.valueOf(infoEntity.getKnowledgeSearchType()),
+                    httpServletResponse);
 
             //保存对话记录
             outputRecord.setContent(agentResultEntity.getMessage());

src/main/java/cn/com/poc/knowledge/query/KnowledgeDocumentRelationQuery.sql

+select
+    bkd.kd_id,
+    bki.knowledge_name,
+    bkd.document_name,
+    bkd.knowledge_id
+from
+    biz_knowledge_document bkd
+left join biz_knowledge_info bki on
+    JSON_CONTAINS(bki.kd_ids , json_array(bkd.kd_id))
+where
+bkd.is_deleted = 'N'
+<<and    bkd.knowledge_id in(:knowledgeIds)>>
\ No newline at end of file

src/main/java/cn/com/poc/knowledge/query/KnowledgeDocumentRelationQueryCondition.java

+package cn.com.poc.knowledge.query;
+
+import java.io.Serializable;
+import java.util.List;
+
+/**
+ * Query Condition class for KnowledgeDocumentRelationQuery
+ */
+public class KnowledgeDocumentRelationQueryCondition implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+
+    private List<String> knowledgeIds;
+
+    public List<String> getKnowledgeIds() {
+        return knowledgeIds;
+    }
+
+    public void setKnowledgeIds(List<String> knowledgeIds) {
+        this.knowledgeIds = knowledgeIds;
+    }
+}
\ No newline at end of file

src/main/java/cn/com/poc/knowledge/query/KnowledgeDocumentRelationQueryItem.java

+package cn.com.poc.knowledge.query;
+
+import cn.com.yict.framemax.data.model.BaseItemClass;
+
+import javax.persistence.Column;
+import javax.persistence.Entity;
+import java.io.Serializable;
+
+/**
+ * Query Item class for KnowledgeDocumentRelationQuery
+ */
+@Entity
+public class KnowledgeDocumentRelationQueryItem extends BaseItemClass implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    /**
+     * kd_id
+     * kd_id
+     */
+    private Integer kdId;
+
+    @Column(name = "kd_id")
+    public Integer getKdId() {
+        return this.kdId;
+    }
+
+    public void setKdId(Integer kdId) {
+        this.kdId = kdId;
+    }
+
+    /**
+     * knowledge_id
+     */
+    private String knowledgeId;
+
+    @Column(name = "knowledge_id")
+    public String getKnowledgeId() {
+        return knowledgeId;
+    }
+
+    public void setKnowledgeId(String knowledgeId) {
+        this.knowledgeId = knowledgeId;
+    }
+
+    /**
+     * knowledge_name
+     * knowledge_name
+     */
+    private String knowledgeName;
+
+    @Column(name = "knowledge_name")
+    public String getKnowledgeName() {
+        return this.knowledgeName;
+    }
+
+    public void setKnowledgeName(String knowledgeName) {
+        this.knowledgeName = knowledgeName;
+    }
+
+    /**
+     * document_name
+     * document_name
+     */
+    private String documentName;
+
+    @Column(name = "document_name")
+    public String getDocumentName() {
+        return this.documentName;
+    }
+
+    public void setDocumentName(String documentName) {
+        this.documentName = documentName;
+    }
+
+} 
\ No newline at end of file

src/main/java/cn/com/poc/knowledge/service/BizKnowledgeDocumentService.java

 package cn.com.poc.knowledge.service;
 
 import cn.com.poc.knowledge.entity.BizKnowledgeDocumentEntity;
+import cn.com.poc.knowledge.query.KnowledgeDocumentRelationQueryItem;
 import cn.com.poc.knowledge.query.KnowledgeQueryItem;
 import cn.com.yict.framemax.core.service.BaseService;
 import cn.com.yict.framemax.data.model.PagingInfo;
@@ -20,4 +21,6 @@ public interface BizKnowledgeDocumentService extends BaseService {
     Boolean deleted(Integer kdId);
 
     List<KnowledgeQueryItem> searchKnowledge(String document, String trainStatus, String memberId, String documentName, List<Integer> kdIds, PagingInfo pagingInfo);
+
+    List<KnowledgeDocumentRelationQueryItem> knowledgeDocumentRelationQuery(List<String> knowledgeIds, PagingInfo pagingInfo);
 }
\ No newline at end of file

src/main/java/cn/com/poc/knowledge/service/impl/BizKnowledgeDocumentServiceImpl.java

@@ -6,6 +6,8 @@ import cn.com.poc.common.utils.JsonUtils;
 import cn.com.poc.knowledge.convert.KnowledgeDocumentConvert;
 import cn.com.poc.knowledge.entity.BizKnowledgeDocumentEntity;
 import cn.com.poc.knowledge.model.BizKnowledgeDocumentModel;
+import cn.com.poc.knowledge.query.KnowledgeDocumentRelationQueryCondition;
+import cn.com.poc.knowledge.query.KnowledgeDocumentRelationQueryItem;
 import cn.com.poc.knowledge.query.KnowledgeQueryCondition;
 import cn.com.poc.knowledge.query.KnowledgeQueryItem;
 import cn.com.poc.knowledge.repository.BizKnowledgeDocumentRepository;
@@ -95,4 +97,11 @@ public class BizKnowledgeDocumentServiceImpl extends BaseServiceImpl
         condition.setKdIds(kdIds);
         return this.sqlDao.query(condition, KnowledgeQueryItem.class, pagingInfo);
     }
+
+    @Override
+    public List<KnowledgeDocumentRelationQueryItem> knowledgeDocumentRelationQuery(List<String> knowledgeIds, PagingInfo pagingInfo) {
+        KnowledgeDocumentRelationQueryCondition condition = new KnowledgeDocumentRelationQueryCondition();
+        condition.setKnowledgeIds(knowledgeIds);
+        return this.sqlDao.query(condition, KnowledgeDocumentRelationQueryItem.class, pagingInfo);
+    }
 }
\ No newline at end of file

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/aggregate/DemandKnowledgeService.java

@@ -2,6 +2,7 @@ package cn.com.poc.thirdparty.resource.demand.ai.aggregate;
 
 import cn.com.poc.thirdparty.resource.demand.ai.constants.KnowledgeSearchTypeEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.knowledge.GetKnowledgeChunkInfoResult;
+import cn.com.poc.thirdparty.resource.demand.ai.entity.knowledge.SearchKnowledgeResult;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.knowledge.SegmentationConfigRequest;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.qaknowledge.QAKnowledgeConfig;
 import cn.com.yict.framemax.data.model.PagingInfo;
@@ -50,10 +51,11 @@ public interface DemandKnowledgeService {
      * @param query          查询文本
      * @param knowledgeIds   知识库id
      * @param topK           返回个数
+     * @param score          分数阈值
      * @param searchTypeEnum 查询类型
      * @return 查询结果
      */
-    List<String> searchKnowledge(String query, List<String> knowledgeIds, Integer topK, KnowledgeSearchTypeEnum searchTypeEnum);
+    SearchKnowledgeResult searchKnowledge(String query, List<String> knowledgeIds, Integer topK, Double score, KnowledgeSearchTypeEnum searchTypeEnum);
 
     /**
      * 获取知识库分片

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/aggregate/impl/DemandKnowledgeServiceImpl.java

@@ -15,6 +15,8 @@ import cn.hutool.core.lang.Assert;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.http.Header;
 import org.apache.http.message.BasicHeader;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
@@ -24,6 +26,8 @@ import java.util.List;
 @Service
 public class DemandKnowledgeServiceImpl implements DemandKnowledgeService {
 
+
+    private Logger logger = LoggerFactory.getLogger(DemandKnowledgeService.class);
     @Resource
     private DgtoolsAbstractHttpClient dgToolsAbstractHttpClient;
 
@@ -44,7 +48,7 @@ public class DemandKnowledgeServiceImpl implements DemandKnowledgeService {
     }
 
     @Override
-    public String trainKnowledgeEvent(String fileURL, String knowledgeType, SegmentationConfigRequest segmentationConfig, List<QAKnowledgeConfig> qaKnowledgeConfigs){
+    public String trainKnowledgeEvent(String fileURL, String knowledgeType, SegmentationConfigRequest segmentationConfig, List<QAKnowledgeConfig> qaKnowledgeConfigs) {
         Assert.notBlank(fileURL);
         TrainKnowledgeRequest request = new TrainKnowledgeRequest();
         request.setDocumentUrl(fileURL);
@@ -82,10 +86,14 @@ public class DemandKnowledgeServiceImpl implements DemandKnowledgeService {
     }
 
     @Override
-    public List<String> searchKnowledge(String query, List<String> knowledgeIds, Integer topK, KnowledgeSearchTypeEnum searchTypeEnum) {
+    public SearchKnowledgeResult searchKnowledge(String query, List<String> knowledgeIds, Integer topK, Double score, KnowledgeSearchTypeEnum searchTypeEnum) {
         Assert.notBlank(query);
+        SearchKnowledgeResult result = new SearchKnowledgeResult();
+        result.setDocuments(new ArrayList<>());
+        result.setScore(new ArrayList<>());
+        result.setKnowledgeIds(new ArrayList<>());
         if (CollectionUtils.isEmpty(knowledgeIds)) {
-            return new ArrayList<String>();
+            return new SearchKnowledgeResult();
         }
         if (topK == null) {
             topK = 3;
@@ -94,18 +102,16 @@ public class DemandKnowledgeServiceImpl implements DemandKnowledgeService {
         searchKnowledgeRequest.setQuery(query);
         searchKnowledgeRequest.setKnowLedgeIds(knowledgeIds);
         searchKnowledgeRequest.setTopK(topK);
+        searchKnowledgeRequest.setScore(score);
         VectorSearchConfig vectorSearchConfig = new VectorSearchConfig();
         vectorSearchConfig.setSearchType(searchTypeEnum);
         vectorSearchConfig.setApiKey(Config.get("large-model.apikey"));
         searchKnowledgeRequest.setVectorSearchConfig(vectorSearchConfig);
-        SearchKnowledgeResult searchKnowledgeResult = dgToolsAbstractHttpClient.doRequest(DgtoolsApiRoute.DgtoolsAI.SEARCH_KNOWLEDGE, searchKnowledgeRequest, getHeaders());
-        if (null == searchKnowledgeResult) {
-            throw new I18nMessageException("exception/query.knowledge.base.exception");
-        }
-        if (CollectionUtils.isEmpty(searchKnowledgeResult.getDocuments())) {
-            return new ArrayList<String>();
+        result = dgToolsAbstractHttpClient.doRequest(DgtoolsApiRoute.DgtoolsAI.SEARCH_KNOWLEDGE, searchKnowledgeRequest, getHeaders());
+        if (null == result) {
+            logger.warn("query knowledge base exception");
         }
-        return searchKnowledgeResult.getDocuments();
+        return result;
     }
 
     @Override

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/entity/dialogue/ToolFunction.java

@@ -5,6 +5,16 @@ public class ToolFunction {
 
     private String arguments;
 
+    private String result;
+
+    public String getResult() {
+        return result;
+    }
+
+    public void setResult(String result) {
+        this.result = result;
+    }
+
     public String getName() {
         return name;
     }

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/entity/knowledge/SearchKnowledgeRequest.java

@@ -14,6 +14,16 @@ public class SearchKnowledgeRequest extends AbstractRequest<SearchKnowledgeResul
 
     private VectorSearchConfig vectorSearchConfig;
 
+    private Double score;
+
+    public Double getScore() {
+        return score;
+    }
+
+    public void setScore(Double score) {
+        this.score = score;
+    }
+
     public String getQuery() {
         return query;
     }

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/entity/knowledge/SearchKnowledgeResult.java

@@ -5,9 +5,13 @@ import cn.com.poc.thirdparty.resource.demand.dgTools.result.AbstractResult;
 import java.util.List;
 
 public class SearchKnowledgeResult extends AbstractResult {
-
     private List<String> documents;
 
+    private List<Double> score;
+
+    private List<String> knowledgeIds;
+
+
     public List<String> getDocuments() {
         return documents;
     }
@@ -15,4 +19,20 @@ public class SearchKnowledgeResult extends AbstractResult {
     public void setDocuments(List<String> documents) {
         this.documents = documents;
     }
+
+    public List<Double> getScore() {
+        return score;
+    }
+
+    public void setScore(List<Double> score) {
+        this.score = score;
+    }
+
+    public List<String> getKnowledgeIds() {
+        return knowledgeIds;
+    }
+
+    public void setKnowledgeIds(List<String> knowledgeIds) {
+        this.knowledgeIds = knowledgeIds;
+    }
 }

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/entity/largemodel/LargeModelDemandResult.java

 package cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel;
 
+import cn.com.poc.agent_application.dto.KnowledgeContentResult;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Usage;
 import cn.com.poc.thirdparty.resource.demand.dgTools.result.AbstractResult;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.ToolFunction;
 
 import java.io.Serializable;
+import java.util.List;
 
 public class LargeModelDemandResult extends AbstractResult implements Serializable {
 
@@ -16,10 +18,20 @@ public class LargeModelDemandResult extends AbstractResult implements Serializab
 
     private ToolFunction function;
 
+    private List<KnowledgeContentResult> knowledgeContentResult;
+
     private String finish_reason;
 
     private Usage usage;
 
+    public List<KnowledgeContentResult> getKnowledgeContentResult() {
+        return knowledgeContentResult;
+    }
+
+    public void setKnowledgeContentResult(List<KnowledgeContentResult> knowledgeContentResult) {
+        this.knowledgeContentResult = knowledgeContentResult;
+    }
+
     public String getCode() {
         return code;
     }

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/entity/qaknowledge/QAKnowledgeChunkResult.java

@@ -15,6 +15,15 @@ public class QAKnowledgeChunkResult extends AbstractResult {
 
     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;