csm-duckdb 0.3.8

Optional analytics, Parquet export, and SQL inspection for chaotic_semantic_memory
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
51
52
53
54
55
56
57
58
59
60
61
62
63

pub const SCHEMA_DDL: &str = r#"
CREATE TABLE IF NOT EXISTS concepts (
    id            VARCHAR PRIMARY KEY,
    text          VARCHAR,
    namespace     VARCHAR,
    created_at_us BIGINT,
    updated_at_us BIGINT,
    expires_at_us BIGINT,
    metadata_json JSON
);

CREATE TABLE IF NOT EXISTS associations (
    src_id   VARCHAR,
    dst_id   VARCHAR,
    strength DOUBLE,
    PRIMARY KEY (src_id, dst_id)
);

CREATE TABLE IF NOT EXISTS concept_versions (
    id          VARCHAR,
    version     INTEGER,
    text        VARCHAR,
    created_us  BIGINT,
    PRIMARY KEY (id, version)
);

CREATE TABLE IF NOT EXISTS benchmarks (
    suite     VARCHAR,
    name      VARCHAR,
    commit    VARCHAR,
    run_at_us BIGINT,
    p50_us    DOUBLE,
    p95_us    DOUBLE,
    p99_us    DOUBLE,
    extras    JSON
);
"#;

#[cfg(test)]
mod tests {
    #![allow(clippy::unwrap_used, clippy::expect_used, clippy::panic)]
    use super::*;
    use duckdb::Connection;

    #[test]
    fn test_schema_init() {
        let conn = Connection::open_in_memory().unwrap();
        conn.execute_batch(SCHEMA_DDL).unwrap();

        let tables: Vec<String> = conn
            .prepare("SELECT name FROM sqlite_master WHERE type='table'")
            .unwrap()
            .query_map([], |row| row.get(0))
            .unwrap()
            .collect::<std::result::Result<Vec<_>, _>>()
            .unwrap();

        assert!(tables.contains(&"concepts".to_string()));
        assert!(tables.contains(&"associations".to_string()));
        assert!(tables.contains(&"concept_versions".to_string()));
        assert!(tables.contains(&"benchmarks".to_string()));
    }
}