fix: supress resources leak logs

This commit is contained in:
Andy Lee
2025-07-22 19:53:45 -07:00
parent d3f85678ec
commit 43155d2811
4 changed files with 59 additions and 28 deletions

View File

@@ -268,6 +268,17 @@ def create_hnsw_embedding_server(
if __name__ == "__main__":
import signal
import sys
def signal_handler(sig, frame):
logger.info(f"Received signal {sig}, shutting down gracefully...")
sys.exit(0)
# Register signal handlers for graceful shutdown
signal.signal(signal.SIGTERM, signal_handler)
signal.signal(signal.SIGINT, signal_handler)
parser = argparse.ArgumentParser(description="HNSW Embedding service")
parser.add_argument("--zmq-port", type=int, default=5555, help="ZMQ port to run on")
parser.add_argument(