diff --git a/examples/main_cli_example.py b/examples/main_cli_example.py index be0cf96..e58b3f9 100644 --- a/examples/main_cli_example.py +++ b/examples/main_cli_example.py @@ -7,7 +7,7 @@ import asyncio import os import dotenv from leann.api import LeannBuilder, LeannSearcher, LeannChat -import leann_backend_diskann # Import to ensure backend registration +import leann_backend_hnsw # Import to ensure backend registration import shutil from pathlib import Path @@ -50,7 +50,7 @@ if INDEX_DIR.exists(): print(f"\n[PHASE 1] Building Leann index...") builder = LeannBuilder( - backend_name="diskann", + backend_name="hnsw", embedding_model="facebook/contriever", # Using a common sentence transformer model graph_degree=32, complexity=64 diff --git a/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py b/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py index 740678c..8901f00 100644 --- a/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py +++ b/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py @@ -151,7 +151,7 @@ class HNSWBackend(LeannBackendFactoryInterface): @staticmethod def searcher(index_path: str, **kwargs) -> LeannBackendSearcherInterface: path = Path(index_path) - meta_path = path.parent / f"{path.stem}.hnsw.meta.json" + meta_path = path.parent / f"{path.name}.meta.json" if not meta_path.exists(): raise FileNotFoundError(f"Leann metadata file not found at {meta_path}. Cannot infer vector dimension for searcher.") diff --git a/packages/leann-backend-hnsw/third_party/faiss b/packages/leann-backend-hnsw/third_party/faiss index b906cee..2365db5 160000 --- a/packages/leann-backend-hnsw/third_party/faiss +++ b/packages/leann-backend-hnsw/third_party/faiss @@ -1 +1 @@ -Subproject commit b906ceeb8f93c589545b47bef697a993ca9ef9a0 +Subproject commit 2365db59a7ba253e8b075fbfa43a5c0d15dbda84 diff --git a/test_pdf_index/pdf_documents.leann.meta.json b/test_pdf_index/pdf_documents.leann.meta.json index 38b478b..0fe5af0 100644 --- a/test_pdf_index/pdf_documents.leann.meta.json +++ b/test_pdf_index/pdf_documents.leann.meta.json @@ -1,6 +1,6 @@ { "version": "0.1.0", - "backend_name": "diskann", + "backend_name": "hnsw", "embedding_model": "facebook/contriever", "num_chunks": 826, "chunks": [