Skip to content

Commit 99eabed

Browse files
committed
fix: 修复并行搜索中WebSearchStateService缺失问题
1 parent 6a123f4 commit 99eabed

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

internal/application/service/chat_pipline/search.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ func (p *PluginSearch) tryDirectChunkLoading(ctx context.Context, tenantID uint6
491491

492492
// searchWebIfEnabled executes web search when enabled and returns converted results
493493
func (p *PluginSearch) searchWebIfEnabled(ctx context.Context, chatManage *types.ChatManage) []*types.SearchResult {
494-
if !chatManage.WebSearchEnabled || p.webSearchService == nil || p.tenantService == nil || chatManage.TenantID <= 0 {
494+
if !chatManage.WebSearchEnabled || p.webSearchService == nil || p.tenantService == nil {
495495
return nil
496496
}
497497
tenant := ctx.Value(types.TenantInfoContextKey).(*types.Tenant)

internal/application/service/chat_pipline/search_parallel.go

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -40,19 +40,21 @@ func NewPluginSearchParallel(
4040
webSearchService interfaces.WebSearchService,
4141
tenantService interfaces.TenantService,
4242
sessionService interfaces.SessionService,
43+
webSearchStateService interfaces.WebSearchStateService,
4344
graphRepository interfaces.RetrieveGraphRepository,
4445
chunkRepository interfaces.ChunkRepository,
4546
knowledgeRepository interfaces.KnowledgeRepository,
4647
) *PluginSearchParallel {
4748
// Create internal plugins without registering them
4849
searchPlugin := &PluginSearch{
49-
knowledgeBaseService: knowledgeBaseService,
50-
knowledgeService: knowledgeService,
51-
chunkService: chunkService,
52-
config: config,
53-
webSearchService: webSearchService,
54-
tenantService: tenantService,
55-
sessionService: sessionService,
50+
knowledgeBaseService: knowledgeBaseService,
51+
knowledgeService: knowledgeService,
52+
chunkService: chunkService,
53+
config: config,
54+
webSearchService: webSearchService,
55+
tenantService: tenantService,
56+
sessionService: sessionService,
57+
webSearchStateService: webSearchStateService,
5658
}
5759

5860
searchEntityPlugin := &PluginSearchEntity{

0 commit comments

Comments
 (0)