diff --git a/pyproject.toml b/pyproject.toml index fabc88c1..992eae2c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -31,7 +31,7 @@ dependencies = [ "py-machineid", "protobuf==6.31.1", "dijkstar==2.6.0", - "lattica==1.0.7", + "lattica==1.0.8", ] [project.scripts] diff --git a/src/parallax/p2p/server.py b/src/parallax/p2p/server.py index 159ab580..cea2c62a 100644 --- a/src/parallax/p2p/server.py +++ b/src/parallax/p2p/server.py @@ -306,7 +306,7 @@ def run(self): node_info = self.get_node_info() if node_info == {}: logger.error("Failed to get node info, try again after 10 seconds") - del self.lattica + self.lattica.close() self.lattica = None time.sleep(10) return self.run() @@ -642,6 +642,8 @@ def shutdown(self): self.announcer.join() if self.routing_table_updater is not None: self.routing_table_updater.join() + if self.lattica is not None: + self.lattica.close() def launch_p2p_server( diff --git a/src/parallax/server/node_chat_http_server.py b/src/parallax/server/node_chat_http_server.py index e1762325..d6abd345 100644 --- a/src/parallax/server/node_chat_http_server.py +++ b/src/parallax/server/node_chat_http_server.py @@ -143,6 +143,7 @@ def build_lattica(self): logger.warning("Scheduler peer id not found, waiting for 1 second.") time.sleep(1) + self.lattica.close() return False def chat_completion(self, request_data, request_id: str, received_ts: int):