sqlite-graphrag 1.0.0

Local GraphRAG memory for LLMs in a single SQLite 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

use crate::errors::AppError;
use rusqlite::{params, Connection};

#[allow(clippy::too_many_arguments)]
pub fn insert_version(
    conn: &Connection,
    memory_id: i64,
    version: i64,
    name: &str,
    memory_type: &str,
    description: &str,
    body: &str,
    metadata: &str,
    changed_by: Option<&str>,
    change_reason: &str,
) -> Result<(), AppError> {
    conn.execute(
        "INSERT INTO memory_versions
         (memory_id, version, name, type, description, body, metadata, changed_by, change_reason)
         VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9)",
        params![
            memory_id,
            version,
            name,
            memory_type,
            description,
            body,
            metadata,
            changed_by,
            change_reason
        ],
    )?;
    Ok(())
}

pub fn next_version(conn: &Connection, memory_id: i64) -> Result<i64, AppError> {
    let v: i64 = conn.query_row(
        "SELECT COALESCE(MAX(version), 0) + 1 FROM memory_versions WHERE memory_id = ?1",
        params![memory_id],
        |r| r.get(0),
    )?;
    Ok(v)
}