diff --git a/packages/leann-core/src/leann/api.py b/packages/leann-core/src/leann/api.py index 6c773c4..bc8cbef 100644 --- a/packages/leann-core/src/leann/api.py +++ b/packages/leann-core/src/leann/api.py @@ -88,6 +88,9 @@ def compute_embeddings_via_server(chunks: list[str], model_name: str, port: int) context = zmq.Context() socket = context.socket(zmq.REQ) socket.setsockopt(zmq.LINGER, 0) # Don't block on close + socket.setsockopt(zmq.RCVTIMEO, 300000) # 300秒接收超时 + socket.setsockopt(zmq.SNDTIMEO, 300000) # 300秒发送超时 + socket.setsockopt(zmq.IMMEDIATE, 1) # 立即模式,避免队列阻塞 socket.connect(f"tcp://localhost:{port}") try: diff --git a/packages/leann-core/src/leann/searcher_base.py b/packages/leann-core/src/leann/searcher_base.py index fbe749f..d1dfa0c 100644 --- a/packages/leann-core/src/leann/searcher_base.py +++ b/packages/leann-core/src/leann/searcher_base.py @@ -138,6 +138,9 @@ class BaseSearcher(LeannBackendSearcherInterface, ABC): context = zmq.Context() socket = context.socket(zmq.REQ) socket.setsockopt(zmq.LINGER, 0) # Don't block on close + socket.setsockopt(zmq.RCVTIMEO, 300000) # 300秒接收超时 + socket.setsockopt(zmq.SNDTIMEO, 300000) # 300秒发送超时 + socket.setsockopt(zmq.IMMEDIATE, 1) # 立即模式,避免队列阻塞 socket.connect(f"tcp://localhost:{zmq_port}") # Send embedding request