From ccf6c8bfd7b7ed9b3bdb82925faa184aaaa6fd5c Mon Sep 17 00:00:00 2001 From: yichuan520030910320 Date: Mon, 21 Jul 2025 21:54:20 -0700 Subject: [PATCH] fix flush print --- .../leann_backend_hnsw/hnsw_embedding_server.py | 8 +++----- packages/leann-backend-hnsw/third_party/msgpack-c | 2 +- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_embedding_server.py b/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_embedding_server.py index 3d3c9fa..ad09b37 100644 --- a/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_embedding_server.py +++ b/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_embedding_server.py @@ -19,7 +19,7 @@ RED = "\033[91m" RESET = "\033[0m" # Set up logging based on environment variable -LOG_LEVEL = os.getenv("LEANN_LOG_LEVEL", "INFO").upper() +LOG_LEVEL = os.getenv("LEANN_LOG_LEVEL", "WARNING").upper() logging.basicConfig( level=getattr(logging, LOG_LEVEL, logging.INFO), format="%(asctime)s - %(levelname)s - %(message)s", @@ -100,7 +100,7 @@ def create_hnsw_embedding_server( e2e_start = time.time() request_payload = msgpack.unpackb(message_bytes) - # Handle direct text embedding request (for OpenAI and sentence-transformers) + # Handle direct text embedding request if isinstance(request_payload, list) and len(request_payload) > 0: # Check if this is a direct text request (list of strings) if all(isinstance(item, str) for item in request_payload): @@ -173,9 +173,7 @@ def create_hnsw_embedding_server( socket.send(response_bytes) e2e_end = time.time() - logger.info( - f"⏱️ Distance calculation E2E time: {e2e_end - e2e_start:.6f}s" - ) + print(f"⏱️ Distance calculation E2E time: {e2e_end - e2e_start:.6f}s") continue # Standard embedding request (passage ID lookup) diff --git a/packages/leann-backend-hnsw/third_party/msgpack-c b/packages/leann-backend-hnsw/third_party/msgpack-c index a0b2ec0..9b801f0 160000 --- a/packages/leann-backend-hnsw/third_party/msgpack-c +++ b/packages/leann-backend-hnsw/third_party/msgpack-c @@ -1 +1 @@ -Subproject commit a0b2ec09da4bd823e40fa591221713951d4ec995 +Subproject commit 9b801f087ab7434f2ab1ab3c0f48a966c19d3b70