cartog 0.5.1

Code graph indexer for LLM coding agents. Map your codebase, navigate by graph.
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

use anyhow::{Context, Result};

use super::embeddings::EmbeddingEngine;
use super::model_cache_dir;
use super::reranker::CrossEncoderEngine;

/// Result of the setup operation.
#[derive(Debug, serde::Serialize)]
pub struct SetupResult {
    pub model_dir: String,
}

/// Download the embedding model by initializing the fastembed engine.
///
/// fastembed automatically downloads the ONNX model from HuggingFace on first use.
/// This function eagerly triggers that download so the user sees progress.
pub fn download_model() -> Result<SetupResult> {
    let cache_dir = model_cache_dir();

    let _engine =
        EmbeddingEngine::new_with_progress().context("Failed to download embedding model")?;

    Ok(SetupResult {
        model_dir: cache_dir.display().to_string(),
    })
}

/// Download the cross-encoder re-ranker model.
///
/// fastembed automatically downloads the ONNX model from HuggingFace on first use.
pub fn download_cross_encoder() -> Result<SetupResult> {
    let cache_dir = model_cache_dir();

    let _engine = CrossEncoderEngine::load_with_progress()
        .context("Failed to download cross-encoder model")?;

    Ok(SetupResult {
        model_dir: cache_dir.display().to_string(),
    })
}