add folder for faiss

autogluon · Jun 6, 2024 · a0a31c4 · a0a31c4
1 parent 86cb7c4
commit a0a31c4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/agrag/modules/vector_db/faiss.py → src/agrag/modules/vector_db/faiss/faiss.py b/src/agrag/modules/vector_db/faiss.py → src/agrag/modules/vector_db/faiss/faiss.py
diff --git a/src/agrag/modules/vector_db/vector_database.py b/src/agrag/modules/vector_db/vector_database.py
@@ -4,7 +4,7 @@
 import faiss
 import torch
 
-from agrag.modules.vector_db.faiss import construct_faiss_index
+from agrag.modules.vector_db.faiss.faiss import construct_faiss_index
 from agrag.modules.vector_db.utils import pad_embeddings, remove_duplicates
 
 logger = logging.getLogger("rag-logger")