kbolt-core 0.1.7

Core engine for kbolt local-first retrieval
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

use std::path::Path;

use crate::Result;

#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) enum ModelFileRequirement {
    ExactPath {
        path: String,
        config_field: &'static str,
    },
    SingleExtension {
        extension: &'static str,
        config_field: &'static str,
    },
    SingleTokenizerJson {
        config_field: &'static str,
    },
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct ModelDownloadRequest {
    pub model_id: String,
    pub requirements: Vec<ModelFileRequirement>,
}

pub(crate) trait ModelArtifactProvider {
    fn download_model(&self, request: &ModelDownloadRequest, target_dir: &Path) -> Result<u64>;
}