You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
44 lines
1.2 KiB
import asyncio
|
|
import inspect
|
|
import logging
|
|
|
|
from lightrag import QueryParam
|
|
|
|
from Util.LightRagUtil import configure_logging, print_stream, initialize_pg_rag
|
|
|
|
# 详细地控制日志输出
|
|
logger = logging.getLogger('lightrag')
|
|
logger.setLevel(logging.INFO)
|
|
handler = logging.StreamHandler()
|
|
handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
|
|
logger.addHandler(handler)
|
|
|
|
WORKING_DIR = f"../dsWorking"
|
|
|
|
logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO)
|
|
|
|
|
|
async def main():
|
|
try:
|
|
rag = await initialize_pg_rag(WORKING_DIR=WORKING_DIR, workspace='Math')
|
|
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())
|