sqlite-vec/examples/simple-sqlite/demo.sql

28 lines
529 B
MySQL
Raw Normal View History

2024-05-10 20:51:42 -07:00
.load ../../dist/vec0
.mode table
2024-05-10 20:51:42 -07:00
.header on
select sqlite_version(), vec_version();
2024-05-11 00:21:44 -07:00
CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[4]);
2024-05-10 20:51:42 -07:00
INSERT INTO vec_items(rowid, embedding)
select
value ->> 0,
value ->> 1
from json_each('[
2024-05-11 00:21:44 -07:00
[1, [0.1, 0.1, 0.1, 0.1]],
[2, [0.2, 0.2, 0.2, 0.2]],
[3, [0.3, 0.3, 0.3, 0.3]],
[4, [0.4, 0.4, 0.4, 0.4]],
[5, [0.5, 0.5, 0.5, 0.5]]
2024-05-10 20:51:42 -07:00
]');
SELECT
rowid,
distance
FROM vec_items
2024-05-11 00:21:44 -07:00
WHERE embedding MATCH '[0.3, 0.3, 0.3, 0.3]'
2024-05-10 20:51:42 -07:00
ORDER BY distance
2024-05-11 00:21:44 -07:00
LIMIT 3;