Skip to content

Commit d6d484d

Browse files
authored
Update the test case of pgvector store (#516)
Signed-off-by: SimFG <[email protected]>
1 parent 5aa7e1e commit d6d484d

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

gptcache/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
"""gptcache version"""
2-
__version__ = "0.1.39"
2+
__version__ = "0.1.39.1"
33

44
from gptcache.config import Config
55
from gptcache.core import Cache

tests/unit_tests/manager/test_pgvector.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,11 @@ def test_normal(self):
3232
db.mul_add([VectorData(id=size, data=data[0])])
3333
ret = db.search(data[0])
3434
print(ret)
35-
self.assertIn(ret[0][0], [0, size])
36-
self.assertIn(ret[1][0], [0, size])
35+
self.assertIn(ret[0][1], [0, size])
36+
self.assertIn(ret[1][1], [0, size])
3737
db.delete([0, 1, 2, 3, 4, 5, size])
3838
ret = db.search(data[0])
3939
print(ret)
40-
self.assertNotIn(ret[0][0], [0, size])
40+
self.assertNotIn(ret[0][1], [0, size])
4141
db.rebuild()
4242
db.close()

0 commit comments

Comments
 (0)