apunkt/sqlite-vec
1
0
Fork 0
mirror of https://github.com/asg017/sqlite-vec.git synced 2026-04-25 00:36:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Temporarily block vector UPDATE for DiskANN and IVF indexes

Browse source 
vec0Update_UpdateVectorColumn writes to flat chunk blobs but does not
update DiskANN graph or IVF index structures, silently corrupting KNN
results. Now returns a clear error for these index types. Rescore
UPDATE is unaffected — it already has a full implementation that
updates both quantized chunks and float vectors.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Alex Garcia 2026-03-31 14:08:08 -07:00
parent 07f56e3cbe
commit 9df59b4c03
3 changed files with 35 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/test-diskann.py
View file

@ -891,7 +891,7 @@ def test_diskann_delete_preserves_graph_connectivity(db):
# ======================================================================
def test_diskann_update_vector(db):
"""UPDATE a vector on DiskANN table may not be supported; verify it either works or errors cleanly."""
"""UPDATE a vector on DiskANN table should error (will be implemented soon)."""
db.execute("""
CREATE VIRTUAL TABLE t USING vec0(
emb float[8] INDEXED BY diskann(neighbor_quantizer=binary, n_neighbors=8)
@ -901,17 +901,8 @@ def test_diskann_update_vector(db):
db.execute("INSERT INTO t(rowid, emb) VALUES (2, ?)", [_f32([0, 1, 0, 0, 0, 0, 0, 0])])
db.execute("INSERT INTO t(rowid, emb) VALUES (3, ?)", [_f32([0, 0, 1, 0, 0, 0, 0, 0])])
# UPDATE may not be fully supported for DiskANN yet; verify no crash
result = exec(db, "UPDATE t SET emb = ? WHERE rowid = 1", [_f32([0, 0.9, 0.1, 0, 0, 0, 0, 0])])
if "error" not in result:
# If UPDATE succeeded, verify KNN reflects the new value
rows = db.execute(
"SELECT rowid, distance FROM t WHERE emb MATCH ? AND k=3",
[_f32([0, 1, 0, 0, 0, 0, 0, 0])],
).fetchall()
assert len(rows) == 3
# rowid 2 should still be closest (exact match)
assert rows[0][0] == 2
with pytest.raises(sqlite3.OperationalError, match="UPDATE on vector column.*not supported for DiskANN"):
db.execute("UPDATE t SET emb = ? WHERE rowid = 1", [_f32([0, 0.9, 0.1, 0, 0, 0, 0, 0])])
# ======================================================================