From c446cc396301df9ab22e16230d0972ec494ecaf4 Mon Sep 17 00:00:00 2001 From: Alex Garcia Date: Sun, 12 May 2024 00:15:50 -0700 Subject: [PATCH] small demo fixes --- examples/simple-bun/demo.ts | 4 ++-- examples/simple-deno/demo.ts | 4 ++-- examples/simple-python/demo.py | 14 ++++---------- 3 files changed, 8 insertions(+), 14 deletions(-) diff --git a/examples/simple-bun/demo.ts b/examples/simple-bun/demo.ts index b7b1761..31595a7 100644 --- a/examples/simple-bun/demo.ts +++ b/examples/simple-bun/demo.ts @@ -22,7 +22,7 @@ const items = [ ]; const query = [0.3, 0.3, 0.3, 0.3]; -db.exec("CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[8])"); +db.exec("CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[4])"); const insertStmt = db.prepare( "INSERT INTO vec_items(rowid, embedding) VALUES (?, vec_f32(?))" @@ -48,6 +48,6 @@ const rows = db LIMIT 3 ` ) - .all(query); + .all(new Float32Array(query)); console.log(rows); diff --git a/examples/simple-deno/demo.ts b/examples/simple-deno/demo.ts index a50df47..376284b 100644 --- a/examples/simple-deno/demo.ts +++ b/examples/simple-deno/demo.ts @@ -1,5 +1,5 @@ import { Database } from "jsr:@db/sqlite@0.11"; -import * as sqliteVec from "npm:sqlite-vec@0.0.1-alpha.3"; +import * as sqliteVec from "npm:sqlite-vec@0.0.1-alpha.9"; const db = new Database(":memory:"); db.enableLoadExtension = true; @@ -8,7 +8,7 @@ db.enableLoadExtension = false; const [sqlite_version, vec_version] = db .prepare("select sqlite_version(), vec_version()") - .value<[string]>()!; + .value<[string, string]>()!; console.log(`sqlite_version=${sqlite_version}, vec_version=${vec_version}`); const items = [ diff --git a/examples/simple-python/demo.py b/examples/simple-python/demo.py index c2593b5..e089b35 100644 --- a/examples/simple-python/demo.py +++ b/examples/simple-python/demo.py @@ -10,25 +10,16 @@ def serialize_f32(vector: List[float]) -> bytes: return struct.pack("%sf" % len(vector), *vector) -def serialize_int8(vector: List[int]) -> bytes: - """serializes a list of int8 into a compact "raw bytes" format""" - return struct.pack("%sb" % len(vector), *vector) - - db = sqlite3.connect(":memory:") db.enable_load_extension(True) sqlite_vec.load(db) db.enable_load_extension(False) - sqlite_version, vec_version = db.execute( "select sqlite_version(), vec_version()" ).fetchone() print(f"sqlite_version={sqlite_version}, vec_version={vec_version}") - -db.execute("CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[4])") - items = [ (1, [0.1, 0.1, 0.1, 0.1]), (2, [0.2, 0.2, 0.2, 0.2]), @@ -36,6 +27,10 @@ items = [ (4, [0.4, 0.4, 0.4, 0.4]), (5, [0.5, 0.5, 0.5, 0.5]), ] +query = [0.3, 0.3, 0.3, 0.3] + +db.execute("CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[4])") + with db: for item in items: db.execute( @@ -43,7 +38,6 @@ with db: [item[0], serialize_f32(item[1])], ) -query = [0.3, 0.3, 0.3, 0.3] rows = db.execute( """ SELECT