kwaak 0.16.2

Run a team of autonomous agents on your code, right from your terminal
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64

//! Builds various storage providers for kwaak
//!
//! Handled as statics to avoid multiple instances of the same storage provider
//!
//!
//! Currently there are 3 tables:
//! - project itself (indexing/retrieval) (uuid, path, chunk, embeddings)
//! - cache (for caching in indexing/retrieval) (uuid, path)
//! - runtime settings (for storing runtime settings) (key, value)
//!
//! Right now, these are relatively simple. Friendly reminder for future me and others to consider
//! a decent migration strategy if these tables change.

use std::sync::OnceLock;

use anyhow::{Context, Result};
use swiftide::{indexing::EmbeddedField, integrations::duckdb::Duckdb};

use crate::repository::Repository;

static DUCK_DB: OnceLock<Duckdb> = OnceLock::new();

/// Retrieves a static duckdb
///
/// # Panics
///
/// Panics if it cannot setup duckdb
pub fn get_duckdb(repository: &Repository) -> Duckdb {
    DUCK_DB
        .get_or_init(|| build_duckdb(repository).expect("Failed to build duckdb"))
        .to_owned()
}

// Probably should just be on the repository/config, cloned from there.
// This sucks in tests
pub(crate) fn build_duckdb(repository: &Repository) -> Result<Duckdb> {
    let config = repository.config();
    let path = config.cache_dir().join("duck.db3");

    tracing::debug!("Building Duckdb: {}", path.display());

    let embedding_provider = config.embedding_provider();

    let connection =
        duckdb::Connection::open(&path).context("Failed to open connection to duckdb")?;
    Duckdb::builder()
        .connection(connection)
        .with_vector(
            EmbeddedField::Combined,
            embedding_provider.vector_size().try_into()?,
        )
        .table_name(normalize_table_name(&config.project_name))
        .cache_table(format!(
            "cache_{}",
            normalize_table_name(&config.project_name)
        ))
        .build()
        .context("Failed to build Duckdb")
}

// Is this enough?
fn normalize_table_name(name: &str) -> String {
    name.replace('-', "_")
}