Struct TextRerank

pub struct TextRerank {
    pub tokenizer: Tokenizer,
    /* private fields */
}

Fields

tokenizer: Tokenizer

Implementations

impl TextRerank

pub fn get_model_info(model: &RerankerModel) -> RerankerModelInfo

pub fn list_supported_models() -> Vec<RerankerModelInfo>

pub fn try_new(options: RerankInitOptions) -> Result<TextRerank>

pub fn try_new_from_user_defined( model: UserDefinedRerankingModel, options: RerankInitOptionsUserDefined, ) -> Result<Self>

Create a TextRerank instance from model files provided by the user.

This can be used for ‘bring your own’ reranking models

pub fn rerank<S: AsRef<str> + Send + Sync>( &self, query: S, documents: Vec<S>, return_documents: bool, batch_size: Option<usize>, ) -> Result<Vec<RerankResult>>

Reranks documents using the reranker model and returns the results sorted by score in descending order.

Trait Implementations

impl Debug for TextRerank

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.