fix: prevent CLI commands from hanging on exit (#471)

Signed-off-by: phernandez <paul@basicmachines.co>
Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>

src/basic_memory/cli/commands/command_utils.py

@@ -1,12 +1,14 @@
 """utility functions for commands"""
 
-from typing import Optional
+import asyncio
+from typing import Optional, TypeVar, Coroutine, Any
 
 from mcp.server.fastmcp.exceptions import ToolError
 import typer
 
 from rich.console import Console
 
+from basic_memory import db
 from basic_memory.mcp.async_client import get_client
 
 from basic_memory.mcp.tools.utils import call_post, call_get
@@ -15,6 +17,30 @@ from basic_memory.schemas import ProjectInfoResponse
 
 console = Console()
 
+T = TypeVar("T")
+
+
+def run_with_cleanup(coro: Coroutine[Any, Any, T]) -> T:
+    """Run an async coroutine with proper database cleanup.
+
+    This helper ensures database connections are cleaned up before the event
+    loop closes, preventing process hangs in CLI commands.
+
+    Args:
+        coro: The coroutine to run
+
+    Returns:
+        The result of the coroutine
+    """
+
+    async def _with_cleanup() -> T:
+        try:
+            return await coro
+        finally:
+            await db.shutdown_db()
+
+    return asyncio.run(_with_cleanup())
+
 
 async def run_sync(project: Optional[str] = None, force_full: bool = False):
     """Run sync operation via API endpoint.