Moduletrace/bun-sqlite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix vector table errors

Browse Source
This commit is contained in:
Benjamin Toby 2026-04-05 09:34:18 +01:00
parent 259a684136
commit f500c52ee5
8 changed files with 74 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dist/commands/admin/index.js vendored
View File

@ -6,6 +6,7 @@ import { select } from "@inquirer/prompts";
import { Database } from "bun:sqlite";
import listTables from "./list-tables";
import runSQL from "./run-sql";
import * as sqliteVec from "sqlite-vec";
export default function () {
return new Command("admin")
.description("View Tables and Data, Run SQL Queries, Etc.")
@ -13,6 +14,7 @@ export default function () {
const { config } = await init();
const { db_file_path } = grabDBDir({ config });
const db = new Database(db_file_path);
sqliteVec.load(db);
console.log(chalk.bold(chalk.blue("\nBun SQLite Admin\n")));
try {
while (true) {

1
dist/lib/sqlite/db-schema-manager.d.ts vendored
View File

@ -44,6 +44,7 @@ declare class SQLiteSchemaManager {
* Add a new column to existing table
*/
private addColumn;
private checkIfTableExists;
/**
* Recreate table (for complex schema changes)
*/

37
dist/lib/sqlite/db-schema-manager.js vendored
View File

@ -102,13 +102,10 @@ class SQLiteSchemaManager {
await this.createTable(table);
this.insertDbManagerTable(table.tableName);
}
else if (!table.isVector) {
else {
// Update existing table
await this.updateTable(table);
}
else {
return;
}
// Sync indexes
await this.syncIndexes(table);
}
@ -133,9 +130,9 @@ class SQLiteSchemaManager {
const columns = [];
const foreignKeys = [];
for (const field of new_table.fields) {
const columnDef = this.buildColumnDefinition(field);
const columnDef = this.buildColumnDefinition(field, table.isVector);
columns.push(columnDef);
if (field.foreignKey) {
if (field.foreignKey && !table.isVector) {
foreignKeys.push(this.buildForeignKeyConstraint(field));
}
}
@ -224,6 +221,12 @@ class SQLiteSchemaManager {
const sql = `ALTER TABLE "${tableName}" ADD COLUMN ${cleanDef}`;
this.db.run(sql);
}
checkIfTableExists(table) {
const tableExists = this.db
.query(`SELECT name FROM sqlite_master WHERE type='table' AND name=?`)
.get(table);
return Boolean(tableExists?.name);
}
/**
* Recreate table (for complex schema changes)
*/
@ -232,11 +235,14 @@ class SQLiteSchemaManager {
if (!this.recreate_vector_table) {
return;
}
console.log(`Recreating vector table: ${table.tableName}`);
const existingRows = this.db
.query(`SELECT * FROM "${table.tableName}"`)
.all();
this.db.run(`DROP TABLE "${table.tableName}"`);
const does_table_exist = this.checkIfTableExists(table.tableName);
let existingRows = [];
if (does_table_exist) {
existingRows = this.db
.query(`SELECT * FROM "${table.tableName}"`)
.all();
this.db.run(`DROP TABLE "${table.tableName}"`);
}
await this.createTable(table);
if (existingRows.length > 0) {
for (let i = 0; i < existingRows.length; i++) {
@ -274,7 +280,7 @@ class SQLiteSchemaManager {
/**
* Build column definition SQL
*/
buildColumnDefinition(field) {
buildColumnDefinition(field, is_vector) {
if (!field.fieldName) {
throw new Error("Field name is required");
}
@ -284,7 +290,12 @@ class SQLiteSchemaManager {
const parts = [fieldName];
// Data type mapping
const dataType = this.mapDataType(field);
parts.push(dataType);
if (dataType == "BLOB") {
parts.push("FLOAT[128]");
}
else {
parts.push(dataType);
}
// Primary key
if (field.primaryKey) {
parts.push("PRIMARY KEY");

3
dist/types/index.d.ts vendored
View File

@ -85,6 +85,9 @@ export interface BUN_SQLITE_TableSchemaType {
* If this is a vector table
*/
isVector?: boolean;
/**
* Type of vector. Defaults to `vec0`
*/
vectorType?: string;
}
/**

2
package.json
View File

@ -1,6 +1,6 @@
{
"name": "@moduletrace/bun-sqlite",
"version": "1.0.24",
"version": "1.0.25",
"description": "SQLite manager for Bun",
"author": "Benjamin Toby",
"main": "dist/index.js",

3
src/commands/admin/index.ts
View File

@ -6,6 +6,7 @@ import { select } from "@inquirer/prompts";
import { Database } from "bun:sqlite";
import listTables from "./list-tables";
import runSQL from "./run-sql";
import * as sqliteVec from "sqlite-vec";
export default function () {
return new Command("admin")
@ -15,6 +16,8 @@ export default function () {
const { db_file_path } = grabDBDir({ config });
const db = new Database(db_file_path);
sqliteVec.load(db);
console.log(chalk.bold(chalk.blue("\nBun SQLite Admin\n")));
try {

51
src/lib/sqlite/db-schema-manager.ts
View File

@ -5,6 +5,7 @@ import _ from "lodash";
import DbClient from ".";
import type {
BUN_SQLITE_DatabaseSchemaType,
BUN_SQLITE_DATATYPES,
BUN_SQLITE_FieldSchemaType,
BUN_SQLITE_TableSchemaType,
} from "../../types";
@ -159,11 +160,9 @@ class SQLiteSchemaManager {
// Create new table
await this.createTable(table);
this.insertDbManagerTable(table.tableName);
} else if (!table.isVector) {
} else {
// Update existing table
await this.updateTable(table);
} else {
return;
}
// Sync indexes
@ -203,10 +202,10 @@ class SQLiteSchemaManager {
const foreignKeys: string[] = [];
for (const field of new_table.fields) {
const columnDef = this.buildColumnDefinition(field);
const columnDef = this.buildColumnDefinition(field, table.isVector);
columns.push(columnDef);
if (field.foreignKey) {
if (field.foreignKey && !table.isVector) {
foreignKeys.push(this.buildForeignKeyConstraint(field));
}
}
@ -326,6 +325,16 @@ class SQLiteSchemaManager {
this.db.run(sql);
}
private checkIfTableExists(table: string) {
const tableExists = this.db
.query(
`SELECT name FROM sqlite_master WHERE type='table' AND name=?`,
)
.get(table) as any;
return Boolean(tableExists?.name);
}
/**
* Recreate table (for complex schema changes)
*/
@ -337,13 +346,18 @@ class SQLiteSchemaManager {
return;
}
console.log(`Recreating vector table: ${table.tableName}`);
const does_table_exist = this.checkIfTableExists(table.tableName);
const existingRows = this.db
.query(`SELECT * FROM "${table.tableName}"`)
.all() as { [k: string]: any }[];
let existingRows: { [k: string]: any }[] = [];
if (does_table_exist) {
existingRows = this.db
.query(`SELECT * FROM "${table.tableName}"`)
.all() as { [k: string]: any }[];
this.db.run(`DROP TABLE "${table.tableName}"`);
}
this.db.run(`DROP TABLE "${table.tableName}"`);
await this.createTable(table);
if (existingRows.length > 0) {
@ -404,7 +418,10 @@ class SQLiteSchemaManager {
/**
* Build column definition SQL
*/
private buildColumnDefinition(field: BUN_SQLITE_FieldSchemaType): string {
private buildColumnDefinition(
field: BUN_SQLITE_FieldSchemaType,
is_vector?: boolean,
): string {
if (!field.fieldName) {
throw new Error("Field name is required");
}
@ -417,7 +434,11 @@ class SQLiteSchemaManager {
// Data type mapping
const dataType = this.mapDataType(field);
parts.push(dataType);
if (dataType == "BLOB") {
parts.push("FLOAT[128]");
} else {
parts.push(dataType);
}
// Primary key
if (field.primaryKey) {
@ -459,13 +480,15 @@ class SQLiteSchemaManager {
/**
* Map DSQL data types to SQLite types
*/
private mapDataType(field: BUN_SQLITE_FieldSchemaType): string {
private mapDataType(
field: BUN_SQLITE_FieldSchemaType,
): (typeof BUN_SQLITE_DATATYPES)[number]["value"] {
const dataType = field.dataType?.toLowerCase() || "text";
const vectorSize = field.vectorSize || 1536;
// Vector Embeddings
if (field.isVector) {
return `FLOAT[${vectorSize}]`;
return `FLOAT[${vectorSize}]` as any;
}
// Integer types

3
src/types/index.ts
View File

@ -104,6 +104,9 @@ export interface BUN_SQLITE_TableSchemaType {
* If this is a vector table
*/
isVector?: boolean;
/**
* Type of vector. Defaults to `vec0`
*/
vectorType?: string;
}