gitsem 0.5.4

Semantic search and spatial navigation for Git repositories — map, get, and grep for AI coding agents
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

use anyhow::Result;

pub use crate::embeddings::config::{EmbeddingConfig, EmbeddingProviderType};
pub use crate::embeddings::{create_provider, EmbeddingProvider};

pub fn generate_embedding(text: &str) -> Result<Vec<f32>> {
    generate_embedding_with_config(text, None)
}

pub fn generate_embedding_with_config(
    text: &str,
    config: Option<EmbeddingConfig>,
) -> Result<Vec<f32>> {
    let config = config.unwrap_or_else(|| EmbeddingConfig::load_or_default().unwrap_or_default());
    let mut provider = create_provider(&config)?;
    provider.init()?;
    provider.generate_embedding(text)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_generate_embedding_with_default_config() {
        let result = generate_embedding("test text");
        assert!(result.is_ok());
    }
}