plugmem-cli 0.1.0

plugmem command-line interface — temporal memory for LLM agents over one file.
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

//! CLI-side configuration glue. The engine [`Settings`] (config, embedder,
//! maintenance policy) come from the shared [`plugmem_host`] loader; only the
//! CLI-specific `import` batch size — `[maintenance].batch_size`, not an engine
//! knob — is read here, from the same config table.

use plugmem_host::SettingsError;

use crate::CliError;

impl From<SettingsError> for CliError {
    fn from(e: SettingsError) -> Self {
        CliError::Usage(e.to_string())
    }
}

/// The `[maintenance].batch_size` — facts per `import` batch. CLI-only (not an
/// engine knob), read from the shared [`plugmem_host::read_config`] table.
pub(crate) fn read_batch_size(table: Option<&toml::Table>) -> Option<u64> {
    table
        .and_then(|t| t.get("maintenance"))
        .and_then(toml::Value::as_table)
        .and_then(|m| m.get("batch_size"))
        .and_then(toml::Value::as_integer)
        .filter(|n| *n >= 0)
        .map(|n| n as u64)
}

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

    #[test]
    fn batch_size_reads_from_the_maintenance_table() {
        let table: toml::Table = "[maintenance]\nbatch_size = 256\n".parse().unwrap();
        assert_eq!(read_batch_size(Some(&table)), Some(256));
        // Absent section / absent key / no table → None.
        let empty: toml::Table = "[engine]\ndim = 8\n".parse().unwrap();
        assert_eq!(read_batch_size(Some(&empty)), None);
        assert_eq!(read_batch_size(None), None);
        // A negative value is rejected (treated as unset).
        let neg: toml::Table = "[maintenance]\nbatch_size = -1\n".parse().unwrap();
        assert_eq!(read_batch_size(Some(&neg)), None);
    }

    #[test]
    fn settings_error_maps_to_a_usage_error() {
        let e: CliError = SettingsError::Config("boom".into()).into();
        assert!(matches!(e, CliError::Usage(m) if m == "boom"));
    }
}