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

text knn GT/GE fixes

Browse source
This commit is contained in:
Alex Garcia 2024-11-18 14:05:47 -08:00
parent 1ec1b89f60
commit df29e31ddc
3 changed files with 318 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

295
tests/__snapshots__/test-metadata.ambr
View file

@ -617,6 +617,13 @@
]),
})
# ---
# name: test_long_text_knn[eq-bb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?",
'rows': list([
]),
})
# ---
# name: test_long_text_knn[eq-bbbb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?",
@ -629,6 +636,13 @@
]),
})
# ---
# name: test_long_text_knn[eq-bbbbbb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?",
'rows': list([
]),
})
# ---
# name: test_long_text_knn[eq-bbbbbbbbbbbb_aaa]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name = ?",
@ -662,34 +676,175 @@
]),
})
# ---
# name: test_long_text_knn[ge-bb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name >= ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
OrderedDict({
'rowid': 3,
'name': 'bbbb',
'distance': 97.0,
}),
]),
})
# ---
# name: test_long_text_knn[ge-bbbb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name >= ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
OrderedDict({
'rowid': 3,
'name': 'bbbb',
'distance': 97.0,
}),
]),
})
# ---
# name: test_long_text_knn[ge-bbbbbb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name >= ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
]),
})
# ---
# name: test_long_text_knn[ge-bbbbbbbbbbbb_aaa]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name >= ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
]),
})
# ---
# name: test_long_text_knn[ge-bbbbbbbbbbbb_bbb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name >= ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
]),
})
# ---
# name: test_long_text_knn[ge-bbbbbbbbbbbb_ccc]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name >= ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
]),
})
# ---
# name: test_long_text_knn[ge-longlonglonglonglonglonglong]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name >= ?",
'rows': list([
]),
})
# ---
# name: test_long_text_knn[gt-bb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name > ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
OrderedDict({
'rowid': 3,
'name': 'bbbb',
'distance': 97.0,
}),
]),
})
# ---
# name: test_long_text_knn[gt-bbbb]
@ -714,6 +869,28 @@
]),
})
# ---
# name: test_long_text_knn[gt-bbbbbb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name > ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
]),
})
# ---
# name: test_long_text_knn[gt-bbbbbbbbbbbb_aaa]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name > ?",
@ -777,12 +954,24 @@
]),
})
# ---
# name: test_long_text_knn[le-bb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[le-bbbb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[le-bbbbbb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[le-bbbbbbbbbbbb_aaa]
dict({
'error': 'OperationalError',
@ -807,12 +996,24 @@
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[lt-bb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[lt-bbbb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[lt-bbbbbb]
dict({
'error': 'OperationalError',
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[lt-bbbbbbbbbbbb_aaa]
dict({
'error': 'OperationalError',
@ -837,6 +1038,38 @@
'message': 'Could not filter metadata fields',
})
# ---
# name: test_long_text_knn[ne-bb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
OrderedDict({
'rowid': 3,
'name': 'bbbb',
'distance': 97.0,
}),
OrderedDict({
'rowid': 2,
'name': 'aaaaaaaaaaaa_aaa',
'distance': 98.0,
}),
]),
})
# ---
# name: test_long_text_knn[ne-bbbb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?",
@ -869,6 +1102,38 @@
]),
})
# ---
# name: test_long_text_knn[ne-bbbbbb]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?",
'rows': list([
OrderedDict({
'rowid': 6,
'name': 'cccccccccccc_ccc',
'distance': 94.0,
}),
OrderedDict({
'rowid': 5,
'name': 'cccc',
'distance': 95.0,
}),
OrderedDict({
'rowid': 4,
'name': 'bbbbbbbbbbbb_bbb',
'distance': 96.0,
}),
OrderedDict({
'rowid': 3,
'name': 'bbbb',
'distance': 97.0,
}),
OrderedDict({
'rowid': 2,
'name': 'aaaaaaaaaaaa_aaa',
'distance': 98.0,
}),
]),
})
# ---
# name: test_long_text_knn[ne-bbbbbbbbbbbb_aaa]
OrderedDict({
'sql': "select rowid, name, distance from v where vector match '[100]' and k = 5 and name != ?",

2
tests/test-metadata.py
View file

@ -147,6 +147,8 @@ def test_long_text_knn(db, snapshot):
tests = [
"bbbb",
"bb",
"bbbbbb",
"bbbbbbbbbbbb_bbb",
"bbbbbbbbbbbb_aaa",
"bbbbbbbbbbbb_ccc",