Update entity retriever

SynaLinks · Sep 8, 2024 · 03d8a6c · 03d8a6c
1 parent ff90ebd
commit 03d8a6c
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 30 deletions.
diff --git a/hybridagi/modules/retrievers/integration/local/faiss_entity_retriever.py b/hybridagi/modules/retrievers/integration/local/faiss_entity_retriever.py
@@ -47,6 +47,7 @@ def __init__(
         self.max_distance = max_distance
         self.reranker = reranker
         self.k = k
+        self.reverse = reverse
         vector_dim = self.embeddings.dim
         if self.distance == "euclidean":
             self.index = faiss.IndexFlatL2(vector_dim)

diff --git a/notebooks/hotel_recommendation.ipynb b/notebooks/hotel_recommendation.ipynb
diff --git a/notebooks/knowledge_graph_rag.ipynb b/notebooks/knowledge_graph_rag.ipynb