Method rerank

pkg/knowledge/rag.go:422–433

rerank 重排序

(ctx context.Context, query string, items []*RetrievalItem)

// rerank 重排序
func (r *RAG) rerank(ctx context.Context, query string, items []*RetrievalItem) []*RetrievalItem {
	switch r.config.RerankerType {
	case RerankerTypeCrossEncoder:
		return r.rerankWithCrossEncoder(ctx, query, items)
	case RerankerTypeBM25:
		return r.rerankWithBM25(ctx, query, items)
	case RerankerTypeHybrid:
		return r.rerankHybrid(ctx, query, items)
	default:
		return items
	}
}

// rerankWithCrossEncoder 使用交叉编码器重排序
func (r *RAG) rerankWithCrossEncoder(_ context.Context, _ string, items []*RetrievalItem) []*RetrievalItem {