Method SearchBySourceType

pkg/memory/semantic.go:287–322

SearchBySourceType 按来源类型检索记忆。

(ctx context.Context, query string, meta map[string]any, topK int, sourceTypes []SourceType)

// SearchBySourceType 按来源类型检索记忆。
func (sm *SemanticMemory) SearchBySourceType(ctx context.Context, query string, meta map[string]any, topK int, sourceTypes []SourceType) ([]vector.Hit, error) {
	if !sm.cfg.EnableProvenance {
		return sm.Search(ctx, query, meta, topK)
	}

	// 先执行标准检索
	hits, err := sm.Search(ctx, query, meta, topK*2)
	if err != nil {
		return nil, err
	}

	// 按来源类型过滤
	sourceTypeMap := make(map[SourceType]bool)
	for _, st := range sourceTypes {
		sourceTypeMap[st] = true
	}

	var filtered []vector.Hit
	for _, hit := range hits {
		provenance := FromMetadata(hit.Metadata)
		if provenance == nil {
			continue
		}

		if sourceTypeMap[provenance.SourceType] {
			filtered = append(filtered, hit)
		}
	}

	// 限制返回数量
	if len(filtered) > topK {
		filtered = filtered[:topK]
	}

	return filtered, nil
}

// PruneMemories 剪枝（删除）低置信度记忆。
// 返回被删除的记忆ID列表。