From 66d6c7dd6fc78bc16e51566da8717001b5c930e0 Mon Sep 17 00:00:00 2001 From: yangdx Date: Fri, 21 Nov 2025 13:11:55 +0800 Subject: [PATCH] Refactor main function to provide sync CLI entry point --- lightrag/tools/clean_llm_query_cache.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/lightrag/tools/clean_llm_query_cache.py b/lightrag/tools/clean_llm_query_cache.py index dbe2e455..6fc7eea8 100644 --- a/lightrag/tools/clean_llm_query_cache.py +++ b/lightrag/tools/clean_llm_query_cache.py @@ -1129,11 +1129,16 @@ class CleanupTool: pass -async def main(): - """Main entry point""" +async def async_main(): + """Async main entry point""" tool = CleanupTool() await tool.run() +def main(): + """Synchronous entry point for CLI command""" + asyncio.run(async_main()) + + if __name__ == "__main__": - asyncio.run(main()) + main()