Skip to content
Snippets Groups Projects
Unverified Commit 7c27b312 authored by Logan's avatar Logan Committed by GitHub
Browse files

filter out negative indexes from faiss query (#9907)

parent 3005dcb0
No related branches found
No related tags found
No related merge requests found
......@@ -189,6 +189,16 @@ class FaissVectorStore(BasePydanticVectorStore):
return VectorStoreQueryResult(similarities=[], ids=[])
# returned dimension is 1 x k
node_idxs = [str(i) for i in indices[0]]
return VectorStoreQueryResult(similarities=dists, ids=node_idxs)
node_idxs = indices[0]
filtered_dists = []
filtered_node_idxs = []
for dist, idx in zip(dists, node_idxs):
if idx < 0:
continue
filtered_dists.append(dist)
filtered_node_idxs.append(str(idx))
return VectorStoreQueryResult(
similarities=filtered_dists, ids=filtered_node_idxs
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment