@infilectai/aira 0.0.0-stage → 0.1.0

package/dist/vectors.d.ts

@@ -0,0 +1,10 @@
+import type { VectorHit, VectorRecord, VectorsOptions } from './types.js';
+export declare class Vectors {
+    private readonly db;
+    private readonly dim;
+    constructor(opts: VectorsOptions);
+    addVector(record: VectorRecord): void;
+    searchVector(query: Float32Array | number[], topK?: number): VectorHit[];
+    close(): void;
+}
+//# sourceMappingURL=vectors.d.ts.map

package/dist/vectors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vectors.d.ts","sourceRoot":"","sources":["../src/vectors.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAMzE,qBAAa,OAAO;IAClB,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAc;IACjC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAQ;gBAEhB,IAAI,EAAE,cAAc;IAoBhC,SAAS,CAAC,MAAM,EAAE,YAAY,GAAG,IAAI;IA2BrC,YAAY,CAAC,KAAK,EAAE,YAAY,GAAG,MAAM,EAAE,EAAE,IAAI,SAAK,GAAG,SAAS,EAAE;IA4BpE,KAAK,IAAI,IAAI;CAGd"}

package/dist/vectors.js

@@ -0,0 +1,66 @@
+import { DatabaseSync } from 'node:sqlite';
+import * as sqliteVec from 'sqlite-vec';
+const DEFAULT_DIM = 768;
+// In-process vector store backed by SQLite + the sqlite-vec extension.
+// Independent of AiraEngine — instantiate it wherever you need persistence.
+export class Vectors {
+    db;
+    dim;
+    constructor(opts) {
+        this.dim = opts.dim ?? DEFAULT_DIM;
+        this.db = new DatabaseSync(opts.path, { allowExtension: true });
+        this.db.loadExtension(sqliteVec.getLoadablePath());
+        this.db.exec(`
+      CREATE VIRTUAL TABLE IF NOT EXISTS vec_items USING vec0(
+        embedding float[${this.dim}]
+      );
+      CREATE TABLE IF NOT EXISTS items (
+        rowid INTEGER PRIMARY KEY,
+        id TEXT UNIQUE NOT NULL,
+        metadata TEXT
+      );
+    `);
+    }
+    // Insert (or upsert by `id`) a single vector. Re-adding the same id
+    // replaces both the embedding and the metadata.
+    addVector(record) {
+        if (record.embedding.length !== this.dim) {
+            throw new Error(`embedding length ${record.embedding.length} does not match configured dim ${this.dim}`);
+        }
+        const blob = Buffer.from(new Float32Array(record.embedding).buffer);
+        const meta = record.metadata ? JSON.stringify(record.metadata) : null;
+        const upsertMeta = this.db.prepare(`INSERT INTO items(id, metadata) VALUES (?, ?)
+       ON CONFLICT(id) DO UPDATE SET metadata = excluded.metadata
+       RETURNING rowid`);
+        const row = upsertMeta.get(record.id, meta);
+        // vec0 virtual tables don't support ON CONFLICT upserts and reject JS
+        // floats for the rowid binding (node:sqlite passes plain numbers as
+        // REAL). Cast to BigInt and clear-then-reinsert.
+        const rowid = BigInt(row.rowid);
+        this.db.prepare(`DELETE FROM vec_items WHERE rowid = ?`).run(rowid);
+        this.db.prepare(`INSERT INTO vec_items(rowid, embedding) VALUES (?, ?)`).run(rowid, blob);
+    }
+    // k-nearest-neighbour search. Returns up to `topK` hits ordered by
+    // ascending L2 distance.
+    searchVector(query, topK = 10) {
+        if (query.length !== this.dim) {
+            throw new Error(`query length ${query.length} does not match configured dim ${this.dim}`);
+        }
+        const blob = Buffer.from(new Float32Array(query).buffer);
+        const stmt = this.db.prepare(`SELECT items.id AS id, vec_items.distance AS distance, items.metadata AS metadata
+       FROM vec_items
+       JOIN items ON items.rowid = vec_items.rowid
+       WHERE vec_items.embedding MATCH ? AND k = ?
+       ORDER BY vec_items.distance`);
+        const rows = stmt.all(blob, topK);
+        return rows.map((r) => ({
+            id: r.id,
+            distance: r.distance,
+            metadata: r.metadata ? JSON.parse(r.metadata) : undefined,
+        }));
+    }
+    close() {
+        this.db.close();
+    }
+}
+//# sourceMappingURL=vectors.js.map

package/dist/vectors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vectors.js","sourceRoot":"","sources":["../src/vectors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAA;AAC1C,OAAO,KAAK,SAAS,MAAM,YAAY,CAAA;AAIvC,MAAM,WAAW,GAAG,GAAG,CAAA;AAEvB,uEAAuE;AACvE,4EAA4E;AAC5E,MAAM,OAAO,OAAO;IACD,EAAE,CAAc;IAChB,GAAG,CAAQ;IAE5B,YAAY,IAAoB;QAC9B,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,WAAW,CAAA;QAElC,IAAI,CAAC,EAAE,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAA;QAC/D,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC,CAAA;QAElD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;0BAES,IAAI,CAAC,GAAG;;;;;;;KAO7B,CAAC,CAAA;IACJ,CAAC;IAED,oEAAoE;IACpE,gDAAgD;IAChD,SAAS,CAAC,MAAoB;QAC5B,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,IAAI,CAAC,GAAG,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CACb,oBAAoB,MAAM,CAAC,SAAS,CAAC,MAAM,kCAAkC,IAAI,CAAC,GAAG,EAAE,CACxF,CAAA;QACH,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAA;QACnE,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;QAErE,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAChC;;uBAEiB,CAClB,CAAA;QACD,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAsB,CAAA;QAEhE,sEAAsE;QACtE,oEAAoE;QACpE,iDAAiD;QACjD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QAC/B,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QACnE,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,uDAAuD,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAA;IAC3F,CAAC;IAED,mEAAmE;IACnE,yBAAyB;IACzB,YAAY,CAAC,KAA8B,EAAE,IAAI,GAAG,EAAE;QACpD,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CACb,gBAAgB,KAAK,CAAC,MAAM,kCAAkC,IAAI,CAAC,GAAG,EAAE,CACzE,CAAA;QACH,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAA;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B;;;;mCAI6B,CAC9B,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAI9B,CAAA;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtB,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;SAC1D,CAAC,CAAC,CAAA;IACL,CAAC;IAED,KAAK;QACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAA;IACjB,CAAC;CACF"}