Method Search

api/service/rag/search.go:173–335

Search 统一搜索接口

(eid int64, req *SearchRequest, userID *int64)

// Search 统一搜索接口
func (s *SearchService) Search(eid int64, req *SearchRequest, userID *int64) (*SearchResponse, error) {
	startTime := time.Now()

	if req == nil {
		return nil, fmt.Errorf("搜索请求不能为空")
	}

	// 设置默认值
	if req.TopK <= 0 {
		req.TopK = 10
	}
	if req.SearchType == "" {
		req.SearchType = "vector"
	}

	originalReq := cloneSearchRequest(req)
	if originalReq == nil {
		originalReq = &SearchRequest{}
	}
	searchTrace := newSearchTimingRecorder()
	originalReq.trace = searchTrace
	configService := NewChunkConfigService(s.db)
	narrowedReq := cloneSearchRequest(originalReq)
	if narrowedReq == nil {
		narrowedReq = &SearchRequest{}
	}
	entityMeta := &entityScopeNarrowMeta{}
	scopeNarrowStart := time.Now()
	if !narrowedReq.SkipEntityScopeNarrowing {
		var narrowErr error
		entityMeta, narrowErr = s.applyEntityScopeNarrowingWithMeta(eid, narrowedReq)
		if narrowErr != nil {
			logger.SysDebugf("【实体范围】实体收窄失败，回退原始请求: eid=%d, err=%v", eid, narrowErr)
			narrowedReq = cloneSearchRequest(originalReq)
			if narrowedReq == nil {
				narrowedReq = &SearchRequest{}
			}
			entityMeta = &entityScopeNarrowMeta{}
		}
	} else {
		logger.SysDebugf("【实体范围】跳过实体收窄: eid=%d, query=%q, file_count=%d",
			eid, truncateForDebug(narrowedReq.Query, 256), len(narrowedReq.FileIDs))
	}
	searchTrace.add("scope_narrowing_ms", time.Since(scopeNarrowStart))
	if entityMeta == nil {
		entityMeta = &entityScopeNarrowMeta{}
	}

	effectiveReq := normalizeSearchRequestForExecution(narrowedReq)
	if effectiveReq == nil {
		effectiveReq = &SearchRequest{}
	}

	logger.SysDebugf(
		"【实体范围】摘要: eid=%d, 种子实体数=%d, 分片候选数=%d, 请求内容=%q, 检索类型=%s, 取回数量=%d, 文件数=%d, 分片数=%d",
		eid,
		len(entityMeta.SeedEntities),
		entityMeta.ChunkCandidateCount,