Struct QueryRewriter 

pub struct QueryRewriter<'a, M: BaseChatModel> { /* private fields */ }

Rewrites queries to improve retrieval quality.

Implementations

impl<'a, M: BaseChatModel> QueryRewriter<'a, M>

pub fn new(llm: &'a M) -> Self

Creates a new query rewriter.

pub async fn rewrite(&self, query: &str) -> Result<String, RewriterError>

Rewrites the query to be more effective for document retrieval.

The rewriter generates alternative phrasings that may match documents the original query missed.

pub async fn generate_alternatives( &self, query: &str, count: usize, ) -> Result<Vec<String>, RewriterError>

Generates multiple alternative queries for broader retrieval.

Returns a list of rewritten queries including the original.