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

text knn NE

Browse source
This commit is contained in:
Alex Garcia 2024-11-18 12:15:25 -08:00
parent 31622209eb
commit 4ba167c315
3 changed files with 185 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -783,33 +783,163 @@
})
# ---
# name: test_long_text_knn[ne-bbbb]
dict({
'error': 'OperationalError',
'message': 'unrecognized token: "!"',
OrderedDict({
'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?",
'rows': list([
OrderedDict({
'rowid': 6,
'vector': b'\x11\x11\x11\x11',
'name': 'cccccccccccc_ccc',
}),
OrderedDict({
'rowid': 5,
'vector': b'\x11\x11\x11\x11',
'name': 'cccc',
}),
OrderedDict({
'rowid': 4,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbbbbbbbbbb_bbb',
}),
OrderedDict({
'rowid': 2,
'vector': b'\x11\x11\x11\x11',
'name': 'aaaaaaaaaaaa_aaa',
}),
OrderedDict({
'rowid': 1,
'vector': b'\x11\x11\x11\x11',
'name': 'aaaa',
}),
]),
})
# ---
# name: test_long_text_knn[ne-bbbbbbbbbbbb_aaa]
dict({
'error': 'OperationalError',
'message': 'unrecognized token: "!"',
OrderedDict({
'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?",
'rows': list([
OrderedDict({
'rowid': 6,
'vector': b'\x11\x11\x11\x11',
'name': 'cccccccccccc_ccc',
}),
OrderedDict({
'rowid': 5,
'vector': b'\x11\x11\x11\x11',
'name': 'cccc',
}),
OrderedDict({
'rowid': 4,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbbbbbbbbbb_bbb',
}),
OrderedDict({
'rowid': 3,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbb',
}),
OrderedDict({
'rowid': 2,
'vector': b'\x11\x11\x11\x11',
'name': 'aaaaaaaaaaaa_aaa',
}),
]),
})
# ---
# name: test_long_text_knn[ne-bbbbbbbbbbbb_bbb]
dict({
'error': 'OperationalError',
'message': 'unrecognized token: "!"',
OrderedDict({
'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?",
'rows': list([
OrderedDict({
'rowid': 6,
'vector': b'\x11\x11\x11\x11',
'name': 'cccccccccccc_ccc',
}),
OrderedDict({
'rowid': 5,
'vector': b'\x11\x11\x11\x11',
'name': 'cccc',
}),
OrderedDict({
'rowid': 3,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbb',
}),
OrderedDict({
'rowid': 2,
'vector': b'\x11\x11\x11\x11',
'name': 'aaaaaaaaaaaa_aaa',
}),
OrderedDict({
'rowid': 1,
'vector': b'\x11\x11\x11\x11',
'name': 'aaaa',
}),
]),
})
# ---
# name: test_long_text_knn[ne-bbbbbbbbbbbb_ccc]
dict({
'error': 'OperationalError',
'message': 'unrecognized token: "!"',
OrderedDict({
'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?",
'rows': list([
OrderedDict({
'rowid': 6,
'vector': b'\x11\x11\x11\x11',
'name': 'cccccccccccc_ccc',
}),
OrderedDict({
'rowid': 5,
'vector': b'\x11\x11\x11\x11',
'name': 'cccc',
}),
OrderedDict({
'rowid': 4,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbbbbbbbbbb_bbb',
}),
OrderedDict({
'rowid': 3,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbb',
}),
OrderedDict({
'rowid': 2,
'vector': b'\x11\x11\x11\x11',
'name': 'aaaaaaaaaaaa_aaa',
}),
]),
})
# ---
# name: test_long_text_knn[ne-longlonglonglonglonglonglong]
dict({
'error': 'OperationalError',
'message': 'unrecognized token: "!"',
OrderedDict({
'sql': "select * from v where vector match X'11111111' and k = 5 and name != ?",
'rows': list([
OrderedDict({
'rowid': 6,
'vector': b'\x11\x11\x11\x11',
'name': 'cccccccccccc_ccc',
}),
OrderedDict({
'rowid': 5,
'vector': b'\x11\x11\x11\x11',
'name': 'cccc',
}),
OrderedDict({
'rowid': 4,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbbbbbbbbbb_bbb',
}),
OrderedDict({
'rowid': 3,
'vector': b'\x11\x11\x11\x11',
'name': 'bbbb',
}),
OrderedDict({
'rowid': 2,
'vector': b'\x11\x11\x11\x11',
'name': 'aaaaaaaaaaaa_aaa',
}),
]),
})
# ---
# name: test_long_text_updates

2
tests/test-metadata.py
View file

@ -152,7 +152,7 @@ def test_long_text_knn(db, snapshot):
"bbbbbbbbbbbb_ccc",
"longlonglonglonglonglonglong",
]
ops = ["=", "!-", "<", "<=", ">", ">="]
ops = ["=", "!=", "<", "<=", ">", ">="]
op_names = ["eq", "ne", "lt", "le", "gt", "ge"]
for test in tests: