import asyncio import inspect from Util.LightRagUtil import configure_logging, initialize_rag, print_stream from lightrag import QueryParam # 数学 #WORKING_DIR = "./Topic/Math" # 苏轼 #WORKING_DIR = "./Topic/Chinese" # 化学 WORKING_DIR = "./Topic/Chemistry" async def main(): try: rag = await initialize_rag(WORKING_DIR) resp = await rag.aquery( "氧化铁与硝酸的化学反应方程式是什么?", #"小学数学有哪些模型", param=QueryParam(mode="hybrid", stream=True), ) if inspect.isasyncgen(resp): await print_stream(resp) else: print(resp) except Exception as e: print(f"An error occurred: {e}") finally: if rag: await rag.finalize_storages() if __name__ == "__main__": configure_logging() asyncio.run(main())