Skip to content

Commit 66d6c7d

Browse files
committed
Refactor main function to provide sync CLI entry point
1 parent 8777895 commit 66d6c7d

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

lightrag/tools/clean_llm_query_cache.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1129,11 +1129,16 @@ async def run(self):
11291129
pass
11301130

11311131

1132-
async def main():
1133-
"""Main entry point"""
1132+
async def async_main():
1133+
"""Async main entry point"""
11341134
tool = CleanupTool()
11351135
await tool.run()
11361136

11371137

1138+
def main():
1139+
"""Synchronous entry point for CLI command"""
1140+
asyncio.run(async_main())
1141+
1142+
11381143
if __name__ == "__main__":
1139-
asyncio.run(main())
1144+
main()

0 commit comments

Comments
 (0)