27 lines
726 B
TypeScript
27 lines
726 B
TypeScript
import AppData from "@/data/app-data";
|
|
import grabDirNames from "@/utils/grab-dir-names";
|
|
import { Database } from "bun:sqlite";
|
|
import path from "node:path";
|
|
import * as sqliteVec from "sqlite-vec";
|
|
|
|
const { ROOT_DIR } = grabDirNames();
|
|
|
|
const DBFilePath = path.join(ROOT_DIR, AppData["DbName"]);
|
|
const DBVecPluginFilePath = path.join(ROOT_DIR, AppData["DbVecPluginName"]);
|
|
|
|
const DbClient = new Database(DBFilePath, {
|
|
create: true,
|
|
});
|
|
|
|
// DbClient.loadExtension(DBVecPluginFilePath);
|
|
|
|
sqliteVec.load(DbClient);
|
|
|
|
// Test if it's working
|
|
// const { vec_version } = DbClient.prepare(
|
|
// "select vec_version() as vec_version",
|
|
// ).get();
|
|
// console.log(`sqlite-vec version: ${vec_version}`);
|
|
|
|
export default DbClient;
|