phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 10:54:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix score threshold in faiss

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-24 12:11:58 -07:00
parent 161aef0aae
commit 205bcfdd4e
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/impls/meta_reference/memory/faiss.py
View file

@ -47,7 +47,9 @@ class FaissIndex(EmbeddingIndex):
self.index.add(np.array(embeddings).astype(np.float32)) self.index.add(np.array(embeddings).astype(np.float32))
async def query(self, embedding: NDArray, k: int) -> QueryDocumentsResponse: async def query(
self, embedding: NDArray, k: int, score_threshold: float
) -> QueryDocumentsResponse:
distances, indices = self.index.search( distances, indices = self.index.search(
embedding.reshape(1, -1).astype(np.float32), k embedding.reshape(1, -1).astype(np.float32), k
) )