Struct kalosm_language::context::DocumentFolder

pub struct DocumentFolder { /* private fields */ }

A folder full of documents.

Example

use kalosm_language::prelude::*;
use std::io::Write;
use std::path::PathBuf;

#[tokio::main]
async fn main() {
    let documents = DocumentFolder::try_from(PathBuf::from("./documents")).unwrap();

    let mut database = DocumentDatabase::new(
        Bert::builder().build().unwrap(),
        ChunkStrategy::Sentence {
            sentence_count: 1,
            overlap: 0,
        },
    );
    database.extend(documents.clone()).await.unwrap();
    let mut fuzzy = FuzzySearchIndex::default();
    fuzzy.extend(documents).await.unwrap();

    loop {
        print!("Query: ");
        std::io::stdout().flush().unwrap();
        let mut user_question = String::new();
        std::io::stdin().read_line(&mut user_question).unwrap();

        println!(
            "vector: {:?}",
            database
                .search(&user_question, 5)
                .await
                .iter()
                .collect::<Vec<_>>()
        );
        println!(
            "fuzzy: {:?}",
            fuzzy
                .search(&user_question, 5)
                .await
                .iter()
                .collect::<Vec<_>>()
        );
    }
}

Trait Implementations

impl Clone for DocumentFolder

fn clone(&self) -> DocumentFolder

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DocumentFolder

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

Formats the value using the given formatter.

impl IntoDocuments for DocumentFolder

fn into_documents<'async_trait>( self ) -> Pin<Box<dyn Future<Output = Result<Vec<Document>>> + Send + 'async_trait>>

where Self: 'async_trait,

Convert the document into a Document

impl TryFrom<PathBuf> for DocumentFolder

type Error = Error

The type returned in the event of a conversion error.

fn try_from(path: PathBuf) -> Result<Self, Self::Error>

Performs the conversion.