Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fix vectorstore tests #none

Browse Source
This commit is contained in:
taprosoft 2024-09-27 04:18:11 +00:00
parent 94cc3a96c2
commit 00df123309
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/kotaemon/tests/test_vectorstore.py
View File

@ -65,7 +65,7 @@ class TestChromaVectorStore:
db.add(embeddings=embeddings, metadatas=metadatas, ids=ids) db.add(embeddings=embeddings, metadatas=metadatas, ids=ids)
_, sim, out_ids = db.query(embedding=[0.1, 0.2, 0.3], top_k=1) _, sim, out_ids = db.query(embedding=[0.1, 0.2, 0.3], top_k=1)
assert sim == [1.0] assert sim[0] - 1.0 < 1e-6
assert out_ids == ["a"] assert out_ids == ["a"]
_, _, out_ids = db.query(embedding=[0.42, 0.52, 0.53], top_k=1) _, _, out_ids = db.query(embedding=[0.42, 0.52, 0.53], top_k=1)
@ -224,7 +224,7 @@ class TestMilvusVectorStore:
db.add(embeddings=normalized_embeddings, metadatas=metadatas, ids=ids) db.add(embeddings=normalized_embeddings, metadatas=metadatas, ids=ids)
_, sim, out_ids = db.query(embedding=normalized_embeddings[0], top_k=1) _, sim, out_ids = db.query(embedding=normalized_embeddings[0], top_k=1)
assert sim == [1.0] assert sim[0] - 1.0 < 1e-6
assert out_ids == ["a"] assert out_ids == ["a"]
query_embedding = [ query_embedding = [