TBNilles/LEANN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(hnsw-server): be lenient to nested [[ids]] for both distance and embedding requests to match client expectations; prevents missing ID lookup when wrapper nests the list

Browse Source
This commit is contained in:
Andy Lee
2025-08-13 15:31:36 -07:00
parent 91d4b4fd94
commit f496621034
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_embedding_server.py
View File

@@ -301,6 +301,9 @@ def create_hnsw_embedding_server(
and isinstance(request[1], list) and isinstance(request[1], list)
): ):
node_ids = request[0] node_ids = request[0]
# Handle nested [[ids]] shape defensively
if len(node_ids) == 1 and isinstance(node_ids[0], list):
node_ids = node_ids[0]
query_vector = np.array(request[1], dtype=np.float32) query_vector = np.array(request[1], dtype=np.float32)
last_request_type = "distance" last_request_type = "distance"
last_request_length = len(node_ids) last_request_length = len(node_ids)
@@ -356,8 +359,17 @@ def create_hnsw_embedding_server(
logger.info(f"⏱️ Distance calculation E2E time: {e2e_end - e2e_start:.6f}s") logger.info(f"⏱️ Distance calculation E2E time: {e2e_end - e2e_start:.6f}s")
continue continue
# Fallback: treat as embedding-by-id request [[ids]] # Fallback: treat as embedding-by-id request
node_ids = request if isinstance(request, list) else [] if (
isinstance(request, list)
and len(request) == 1
and isinstance(request[0], list)
):
node_ids = request[0]
elif isinstance(request, list):
node_ids = request
else:
node_ids = []
last_request_type = "embedding" last_request_type = "embedding"
last_request_length = len(node_ids) last_request_length = len(node_ids)
logger.info(f"ZMQ received {len(node_ids)} node IDs for embedding fetch") logger.info(f"ZMQ received {len(node_ids)} node IDs for embedding fetch")