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

core: unify atexit to always call _finalize_process (covers both self-launched and adopted servers)

Browse Source
This commit is contained in:
Andy Lee
2025-08-13 22:42:25 -07:00
parent dfe60a152f
commit 0f110dc7b9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/leann-core/src/leann/embedding_server_manager.py
View File

@@ -352,8 +352,8 @@ class EmbeddingServerManager:
# Register atexit callback only when we actually start a process # Register atexit callback only when we actually start a process
if not self._atexit_registered: if not self._atexit_registered:
# Use a lambda to avoid issues with bound methods # Always attempt best-effort finalize at interpreter exit
atexit.register(lambda: self.stop_server() if self.server_process else None) atexit.register(self._finalize_process)
self._atexit_registered = True self._atexit_registered = True
# Touch finalizer so it knows there is a live process # Touch finalizer so it knows there is a live process
if getattr(self, "_finalizer", None) is not None and not self._finalizer.alive: if getattr(self, "_finalizer", None) is not None and not self._finalizer.alive: