1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use std::error::Error;

use async_trait::async_trait;

use super::Document;

#[async_trait]
pub trait Retriever: Sync + Send {
    async fn get_relevant_documents(&self, query: &str) -> Result<Vec<Document>, Box<dyn Error>>;
}

impl<R> From<R> for Box<dyn Retriever>
where
    R: Retriever + 'static,
{
    fn from(retriever: R) -> Self {
        Box::new(retriever)
    }
}