vectus 0.1.38

A vector database implemented in Rust for learning purposes.
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
35
36
37
38
39
40
41
42
43
44

pub mod splitters;

use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use uuid::Uuid;

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct Document {
    pub id: Uuid,
    pub page_content: String,
    pub metadata: HashMap<String, String>,
}

pub struct DocBuilder {
    page_content: String,
    metadata: HashMap<String, String>,
}

impl DocBuilder {
    pub fn new() -> Self {
        DocBuilder {
            page_content: String::new(),
            metadata: HashMap::new(),
        }
    }

    pub fn with_page_content(mut self, page_content: &str) -> Self {
        self.page_content = page_content.to_string();
        self
    }

    pub fn add_metadata(mut self, key: &str, value: &str) -> Self {
        self.metadata.insert(key.to_string(), value.to_string());
        self
    }

    pub fn build(self) -> Document {
        Document {
            id: Uuid::new_v4(),
            page_content: self.page_content,
            metadata: self.metadata,
        }
    }
}