abu-rag 0.2.0

Retrieval-Augmented Generation system
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

pub trait Float :
      num_traits::Float
    + std::ops::AddAssign
    + std::ops::SubAssign
    + std::ops::MulAssign
    + std::ops::DivAssign
    + std::iter::Sum
    + Send + Sync
{
}

impl Float for f32 {}
impl Float for f64 {}

pub trait Vector: Sized {
    type Scalar: Float;

    fn scalars(self) -> impl IntoIterator<Item = Self::Scalar>;
    
    fn into_iter(self) -> impl Iterator<Item = Self::Scalar> {
        self.scalars().into_iter()
    }
    
    fn to_vec(self) -> Vec<Self::Scalar> {
        self.into_iter().collect()
    }
    
    fn len(&self) -> usize;
}

impl<F: Float> Vector for Vec<F> {
    type Scalar = F;
    
    fn scalars(self) -> impl IntoIterator<Item = Self::Scalar> {
        self
    }

    fn to_vec(self) -> Vec<Self::Scalar> {
        self
    }
    
    fn len(&self) -> usize {
        self.len()
    }
}

impl<F: Float> Vector for &Vec<F> {
    type Scalar = F;
    
    fn scalars(self) -> impl IntoIterator<Item = Self::Scalar> {
        self.iter().cloned()
    }

    fn to_vec(self) -> Vec<Self::Scalar> {
        self.clone()
    }
    
    fn len(&self) -> usize {
        Vec::len(&self)
    }
}

impl<F: Float, const N: usize> Vector for [F; N] {
    type Scalar = F;

    fn scalars(self) -> impl IntoIterator<Item = Self::Scalar> {
        self
    }

    fn len(&self) -> usize {
        N
    }
}

impl<F: Float> Vector for &[F] {
    type Scalar = F;

    fn scalars(self) -> impl IntoIterator<Item = Self::Scalar> {
        self.iter().cloned()
    }

    fn to_vec(self) -> Vec<Self::Scalar> {
        self.to_owned()
    }

    fn len(&self) -> usize {
        <[F]>::len(self)
    }
}