fix file path bug still compatiable bug in hnsw search
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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.")
|
||||
|
||||
|
||||
Submodule packages/leann-backend-hnsw/third_party/faiss updated: b906ceeb8f...2365db59a7
@@ -1,6 +1,6 @@
|
||||
{
|
||||
"version": "0.1.0",
|
||||
"backend_name": "diskann",
|
||||
"backend_name": "hnsw",
|
||||
"embedding_model": "facebook/contriever",
|
||||
"num_chunks": 826,
|
||||
"chunks": [
|
||||
|
||||
Reference in New Issue
Block a user