Skip to content

Commit b3642fe

Browse files
committed
Update sqlitevec
Update iOS libs Fix script # Conflicts: # example/ios/Podfile.lock # scripts/turnOnLibsql.js
1 parent 4cc2c29 commit b3642fe

11 files changed

Lines changed: 73 additions & 2105 deletions

File tree

-1.9 KB
Binary file not shown.
-324 Bytes
Binary file not shown.

android/src/main/libsqlitevec/x86/libsqlite_vec.so

100644100755
528 Bytes
Binary file not shown.
336 Bytes
Binary file not shown.

example/ios/Podfile.lock

Lines changed: 0 additions & 2104 deletions
This file was deleted.

example/package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@
5151
"fts5": true,
5252
"rtree": true,
5353
"crsqlite": false,
54-
"sqliteVec": false,
54+
"sqliteVec": true,
5555
"performanceMode": true,
5656
"tokenizers": [
5757
"wordtokenizer",

example/src/tests/queries.ts

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import {
1414
describe,
1515
it,
1616
} from '@op-engineering/op-test';
17+
import pkg from '../../package.json'
1718

1819
describe('Queries tests', () => {
1920
let db: DB;
@@ -775,4 +776,50 @@ describe('Queries tests', () => {
775776
const res = db.executeSync('PRAGMA user_version');
776777
expect(res.rows).toDeepEqual([{user_version: 0}]);
777778
});
779+
780+
781+
const sqliteVecEnabled = pkg?.['op-sqlite']?.sqliteVec === true;
782+
if (sqliteVecEnabled) {
783+
it('sqlite-vec extension: vector similarity search', async () => {
784+
// Create a virtual table for storing vectors
785+
await db.execute(`
786+
CREATE VIRTUAL TABLE vec_items USING vec0(
787+
embedding FLOAT[8]
788+
)
789+
`);
790+
791+
// Insert some sample vectors
792+
await db.execute(`
793+
INSERT INTO vec_items(rowid, embedding)
794+
VALUES
795+
(1, '[-0.200, 0.250, 0.341, -0.211, 0.645, 0.935, -0.316, -0.924]'),
796+
(2, '[0.443, -0.501, 0.355, -0.771, 0.707, -0.708, -0.185, 0.362]'),
797+
(3, '[0.716, -0.927, 0.134, 0.052, -0.669, 0.793, -0.634, -0.162]'),
798+
(4, '[-0.710, 0.330, 0.656, 0.041, -0.990, 0.726, 0.385, -0.958]')
799+
`);
800+
801+
// Perform KNN query to find the 2 nearest neighbors
802+
const queryVector = '[0.890, 0.544, 0.825, 0.961, 0.358, 0.0196, 0.521, 0.175]';
803+
const result = await db.execute(`
804+
SELECT rowid, distance
805+
FROM vec_items
806+
WHERE embedding MATCH ?
807+
ORDER BY distance
808+
LIMIT 2
809+
`, [queryVector]);
810+
811+
// Verify results
812+
expect(result.rows.length).toEqual(2);
813+
expect(result.rows[0]!.rowid).toEqual(2);
814+
expect(result.rows[1]!.rowid).toEqual(1);
815+
816+
// Verify distances are positive numbers
817+
const distance0 = result.rows[0]!.distance as number;
818+
const distance1 = result.rows[1]!.distance as number;
819+
expect(typeof distance0).toEqual('number');
820+
expect(distance0 > 0).toBeTruthy();
821+
expect(distance1 > 0).toBeTruthy();
822+
});
823+
}
824+
778825
});
512 Bytes
Binary file not shown.
31.1 KB
Binary file not shown.

scripts/turnOnLibsql.js

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ packageJson['op-sqlite']['libsql'] = true;
88
packageJson['op-sqlite']['sqlcipher'] = false;
99
packageJson['op-sqlite']['ioSqlite'] = false;
1010
delete packageJson['op-sqlite']['tokenizers'];
11+
packageJson['op-sqlite']['sqliteVec'] = false;
1112

1213
// Save the updated package.json file
1314
fs.writeFileSync(

0 commit comments

Comments
 (0)