spire-ai 0.1.3

AI-native SDK for SpireDB — RAG, code search, agents
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

//! Reranking strategies for RAG pipelines.

use async_trait::async_trait;

use crate::error::Result;
use crate::rag::ScoredChunk;

/// Trait for reranking strategies.
#[async_trait]
pub trait RerankerFn: Send + Sync {
    /// Rerank chunks for better relevance ordering.
    async fn rerank(
        &self,
        query: &str,
        chunks: Vec<ScoredChunk>,
        top_k: usize,
    ) -> Result<Vec<ScoredChunk>>;
}

/// No-op reranker that passes through results unchanged.
pub struct NoReranker;

#[async_trait]
impl RerankerFn for NoReranker {
    async fn rerank(
        &self,
        _query: &str,
        mut chunks: Vec<ScoredChunk>,
        top_k: usize,
    ) -> Result<Vec<ScoredChunk>> {
        chunks.truncate(top_k);
        Ok(chunks)
    }
}