Package com.bytedesk.ai.service

Class KnowledgeBaseSearchHelper

java.lang.Object

com.bytedesk.ai.service.KnowledgeBaseSearchHelper

@Component
public class KnowledgeBaseSearchHelper
extends Object

知识库检索与结果重排辅助类： - 封装全文/向量检索及来源构造 - 提供聚合/去重/重排/TopK 能力

Field Summary

Fields

Modifier and Type	Field	Description
private ChunkElasticService	chunkElasticService
private ChunkVectorService	chunkVectorService
private FaqElasticService	faqElasticService
private FaqVectorService	faqVectorService
private TextElasticService	textElasticService
private TextVectorService	textVectorService
private WebpageElasticService	webpageElasticService
private WebpageVectorService	webpageVectorService

Constructor Summary

Constructors

Constructor	Description
KnowledgeBaseSearchHelper()

Method Summary

Modifier and Type	Method	Description
void	executeFulltextSearchWithSources(String query, String kbUid, List<FaqProtobuf> searchResultList, List<RobotContent.SourceReference> sourceReferences)	执行全文搜索并填充结果与来源引用
void	executeVectorSearchWithSources(String query, String kbUid, List<FaqProtobuf> searchResultList, List<RobotContent.SourceReference> sourceReferences)	执行向量搜索并填充结果与来源引用
private String	getContentSummary(String content, int maxLength)	简单内容摘要
SearchResultWithSources	rerankMergeTopK(SearchResultWithSources raw, RobotProtobuf robot)	聚合/去重/重排并TopK输出
protected List<FaqProtobuf>	searchKnowledgeBase(String query, RobotProtobuf robot)
protected SearchResultWithSources	searchKnowledgeBaseWithSources(String query, RobotProtobuf robot)	搜索知识库并收集源引用信息

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

faqElasticService

@Autowired
private FaqElasticService faqElasticService

textElasticService

@Autowired
private TextElasticService textElasticService

chunkElasticService

@Autowired
private ChunkElasticService chunkElasticService

webpageElasticService

@Autowired
private WebpageElasticService webpageElasticService

faqVectorService

@Autowired(required=false)
private FaqVectorService faqVectorService

textVectorService

@Autowired(required=false)
private TextVectorService textVectorService

chunkVectorService

@Autowired(required=false)
private ChunkVectorService chunkVectorService

webpageVectorService

@Autowired(required=false)
private WebpageVectorService webpageVectorService

Constructor Details

KnowledgeBaseSearchHelper

public KnowledgeBaseSearchHelper()

Method Details

searchKnowledgeBase

protected List<FaqProtobuf> searchKnowledgeBase(String query,
 RobotProtobuf robot)

searchKnowledgeBaseWithSources

protected SearchResultWithSources searchKnowledgeBaseWithSources(String query,
 RobotProtobuf robot)

搜索知识库并收集源引用信息

Parameters:

query - 查询内容

robot - 机器人配置

Returns:

包含源引用信息的搜索结果

executeFulltextSearchWithSources

public void executeFulltextSearchWithSources(String query,
 String kbUid,
 List<FaqProtobuf> searchResultList,
 List<RobotContent.SourceReference> sourceReferences)

执行全文搜索并填充结果与来源引用

executeVectorSearchWithSources

public void executeVectorSearchWithSources(String query,
 String kbUid,
 List<FaqProtobuf> searchResultList,
 List<RobotContent.SourceReference> sourceReferences)

执行向量搜索并填充结果与来源引用

rerankMergeTopK

public SearchResultWithSources rerankMergeTopK(SearchResultWithSources raw,
 RobotProtobuf robot)

聚合/去重/重排并TopK输出

getContentSummary

private String getContentSummary(String content,
 int maxLength)

简单内容摘要