lsp-ai 0.7.0

LSP-AI is an open-source language server that serves as a backend for AI-powered functionality, designed to assist and empower software engineers, not replace them.
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

use crate::config::ValidEmbeddingModel;

mod ollama;

fn normalize(mut vector: Vec<f32>) -> Vec<f32> {
    let magnitude = (vector.iter().map(|&x| x * x).sum::<f32>()).sqrt();

    if magnitude != 0.0 {
        for element in &mut vector {
            *element /= magnitude;
        }
    }

    vector
}

#[derive(Clone, Copy)]
pub(crate) enum EmbeddingPurpose {
    Storage,
    Retrieval,
}

#[async_trait::async_trait]
pub(crate) trait EmbeddingModel {
    async fn embed(
        &self,
        batch: Vec<&str>,
        purpose: EmbeddingPurpose,
    ) -> anyhow::Result<Vec<Vec<f32>>>;
}

impl TryFrom<ValidEmbeddingModel> for Box<dyn EmbeddingModel + Send + Sync> {
    type Error = anyhow::Error;

    fn try_from(value: ValidEmbeddingModel) -> Result<Self, Self::Error> {
        match value {
            ValidEmbeddingModel::Ollama(config) => Ok(Box::new(ollama::Ollama::new(config))),
        }
    }
}