Struct MultiQueryRetriever 

pub struct MultiQueryRetriever { /* private fields */ }

MultiQueryRetriever

使用 LLM 生成多个查询变体，然后用基础检索器分别检索， 最后合并去重结果返回。

Implementations

impl MultiQueryRetriever

pub fn new<L>(llm: L, base_retriever: Arc<dyn RetrieverTrait>) -> Self

where L: BaseChatModel + Send + Sync + 'static, L::Error: Into<ProviderError>,

创建 MultiQueryRetriever(接受任意实现 BaseChatModel 的 LLM)

pub fn new_arc( llm: Arc<dyn BaseChatModel<Error = ProviderError> + Send + Sync>, base_retriever: Arc<dyn RetrieverTrait>, ) -> Self

P0-3: 从已包装的 Arc<dyn BaseChatModel<Error = ProviderError>> 构建

pub fn with_config(self, config: MultiQueryConfig) -> Self

pub fn with_num_queries(self, n: usize) -> Self

pub fn with_k_per_query(self, k: usize) -> Self

pub fn with_final_k(self, k: usize) -> Self

pub async fn retrieve_multi( &self, query: &str, ) -> Result<Vec<Document>, MultiQueryError>

pub async fn retrieve_multi_with_scores( &self, query: &str, ) -> Result<Vec<SearchResult>, MultiQueryError>

pub async fn get_generated_queries( &self, query: &str, ) -> Result<Vec<String>, MultiQueryError>