sqlite-vec/site/using/js.md

145 lines
4 KiB
Markdown
Raw Normal View History

2024-06-11 23:58:00 -07:00
# Using `sqlite-vec` in Node.js, Deno, and Bun
2024-06-24 23:26:29 -07:00
[![npm](https://img.shields.io/npm/v/sqlite-vec.svg?color=green&logo=nodedotjs&logoColor=white)](https://www.npmjs.com/package/sqlite-vec)
2024-06-22 16:46:33 -07:00
To use `sqlite-vec` in Node.js, Deno or Bun, install the
[`sqlite-vec` NPM package](https://npmjs.com/package/sqlite-vec) using your
favorite package manager:
::: code-group
```bash [npm]
2024-06-11 23:58:00 -07:00
npm install sqlite-vec
```
2024-06-22 16:46:33 -07:00
```bash [Bun]
bun install sqlite-vec
```
```bash [Deno]
deno add npm:sqlite-vec
```
:::
Once installed, use the `sqliteVec.load()` function to load `sqlite-vec` SQL
functions into a SQLite connection.
```js
import * as sqliteVec from "sqlite-vec";
import Database from "better-sqlite3";
const db = new Database(":memory:");
sqliteVec.load(db);
const { vec_version } = db
.prepare("select vec_version() as vec_version;")
.get();
console.log(`vec_version=${vec_version}`);
```
The `load()` function is compatable with
[`better-sqlite3`](https://github.com/WiseLibs/better-sqlite3),
[`node-sqlite3`](https://github.com/TryGhost/node-sqlite3),
2024-06-24 23:26:29 -07:00
[`jsr:@db/sqlite`](https://jsr.io/@db/sqlite) (Deno), and
2024-06-22 16:46:33 -07:00
[`bun:sqlite`](https://bun.sh/docs/api/sqlite).
2024-06-11 23:58:00 -07:00
## Working with vectors in JavaScript
2024-06-24 23:26:29 -07:00
if your vectors are represented as an array of numbers, wrap it in a
[`Float32Array`](https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Float32Array),
and use
2024-06-22 16:46:33 -07:00
[`.buffer`](https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/TypedArray/buffer)
2024-06-24 23:26:29 -07:00
accessor to bind as a parameter to `sqlite-vec` SQL functions.
2024-06-22 16:46:33 -07:00
```js
2024-06-24 23:26:29 -07:00
const embedding = new Float32Array([0.1, 0.2, 0.3, 0.4]);
2024-07-31 12:55:03 -07:00
const stmt = db.prepare("select vec_length(?)");
2024-08-01 02:45:36 -07:00
console.log(stmt.run(embedding.buffer)); // 4
2024-06-24 23:26:29 -07:00
```
2024-06-22 16:46:33 -07:00
2024-06-11 23:58:00 -07:00
## Node.js
2024-07-31 12:55:03 -07:00
Here's a quick recipe of using `sqlite-vec` with
[`better-sqlite3`](https://github.com/WiseLibs/better-sqlite3) in Node.js.
2024-06-24 23:26:29 -07:00
```js
2024-08-01 02:45:36 -07:00
import * as sqliteVec from "sqlite-vec";
import Database from "better-sqlite3";
const db = new Database(":memory:");
sqliteVec.load(db);
const embedding = new Float32Array([0.1, 0.2, 0.3, 0.4]);
const { result } = db
.prepare("select vec_length(?)",)
.get(embedding);
console.log(result); // 4
2024-06-24 23:26:29 -07:00
```
2024-07-31 12:55:03 -07:00
See
[`simple-node/demo.mjs`](https://github.com/asg017/sqlite-vec/blob/main/examples/simple-node/demo.mjs)
2024-06-24 23:26:29 -07:00
for a more complete Node.js demo.
2024-06-11 23:58:00 -07:00
## Deno
2024-07-31 12:55:03 -07:00
Here's a quick recipe of using `sqlite-vec` with
[`jsr:@db/sqlite`](https://jsr.io/@db/sqlite) in Deno. It will only work on Deno
2024-08-01 02:45:36 -07:00
version `1.44` or greater, because of a bug in previous Deno versions.
2024-06-24 23:26:29 -07:00
2024-07-31 12:55:03 -07:00
Keep in mind, the `better-sqlite3` example above also works in Deno, you just
need to prefix the `better-sqlite3` import with `npm:`, like
`import * from "npm:better-sqlite3"`.
2024-06-24 23:26:29 -07:00
```ts
2024-08-01 02:45:36 -07:00
import { Database } from "jsr:@db/sqlite@0.11";
import * as sqliteVec from "npm:sqlite-vec@0.0.1-alpha.9";
const db = new Database(":memory:");
db.enableLoadExtension = true;
sqliteVec.load(db);
db.enableLoadExtension = false;
const embedding = new Float32Array([0.1, 0.2, 0.3, 0.4]);
const [result] = db
.prepare("select vec_length(?)")
.value<[string]>(new Uint8Array(embedding.buffer)!);
console.log(result); // 4
2024-06-24 23:26:29 -07:00
```
2024-07-31 12:55:03 -07:00
See
[`simple-deno/demo.ts`](https://github.com/asg017/sqlite-vec/blob/main/examples/simple-deno/demo.ts)
2024-06-24 23:26:29 -07:00
for a more complete Deno demo.
2024-06-11 23:58:00 -07:00
## Bun
2024-06-24 23:26:29 -07:00
2024-07-31 12:55:03 -07:00
Here's a quick recipe of using `sqlite-vec` with
[`bun:sqlite`](https://bun.sh/docs/api/sqlite) in Bun. The `better-sqlite3`
example above also works with Bun.
2024-06-24 23:26:29 -07:00
```ts
2024-08-01 02:45:36 -07:00
import { Database } from "bun:sqlite";
import * as sqliteVec from "sqlite-vec";
// MacOS *might* have to do this, as the builtin SQLite library on MacOS doesn't allow extensions
Database.setCustomSQLite("/usr/local/opt/sqlite3/lib/libsqlite3.dylib");
const db = new Database(":memory:");
sqliteVec.load(db);
const embedding = new Float32Array([0.1, 0.2, 0.3, 0.4]);
const { result } = db
.prepare("select vec_length(?)",)
.get(embedding);
console.log(result); // 4
2024-06-22 16:46:33 -07:00
```
2024-06-24 23:26:29 -07:00
2024-07-31 12:55:03 -07:00
See
[`simple-bun/demo.ts`](https://github.com/asg017/sqlite-vec/blob/main/examples/simple-bun/demo.ts)
2024-06-24 23:26:29 -07:00
for a more complete Bun demo.