add configuable funcname

This commit is contained in:
yichuan520030910320
2025-07-01 05:02:01 +00:00
parent b81b5d0f86
commit 371e3de04e
4 changed files with 973 additions and 1131 deletions

View File

@@ -246,7 +246,7 @@ class DiskannSearcher(LeannBackendSearcherInterface):
raise
def search(self, query: np.ndarray, top_k: int, **kwargs) -> Dict[str, any]:
complexity = kwargs.get("complexity", 100)
complexity = kwargs.get("complexity", 32)
beam_width = kwargs.get("beam_width", 4)
USE_DEFERRED_FETCH = kwargs.get("USE_DEFERRED_FETCH", False)