Skip to content

Commit 4538c6e

Browse files
committed
Add vector search example
1 parent c00ccb0 commit 4538c6e

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

examples/vector.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
import libsql_experimental as libsql
2+
3+
conn = libsql.connect("vector.db")
4+
5+
conn.execute(
6+
"CREATE TABLE movies (title TEXT, year INT, embedding F32_BLOB(3))")
7+
conn.execute("CREATE INDEX movies_idx ON movies (libsql_vector_idx(embedding))")
8+
9+
conn.execute(""""
10+
INSERT INTO movies (title, year, embedding) VALUES
11+
('Napoleon', 2023, vector('[1,2,3]')),
12+
('Black Hawk Down', 2001, vector('[10,11,12]')),
13+
('Gladiator', 2000, vector('[7,8,9]')),
14+
('Blade Runner', 1982, vector('[4,5,6]'));
15+
""")
16+
17+
cur = conn.execute("""
18+
SELECT title, year
19+
FROM vector_top_k('movies_idx', '[4,5,6]', 3)
20+
JOIN movies ON movies.rowid = id
21+
""")
22+
23+
print(cur.fetchal())

0 commit comments

Comments
 (0)