|
|
@ -23,7 +23,7 @@ async def main():
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
rag = await initialize_pg_rag(WORKING_DIR=WORKING_DIR,workspace='dsideal')
|
|
|
|
rag = await initialize_pg_rag(WORKING_DIR=WORKING_DIR,workspace='dsideal')
|
|
|
|
with open(f"../Txt/sushi.txt", "r", encoding="utf-8") as f:
|
|
|
|
with open(f"../Txt/sushi.txt", "r", encoding="utf-8") as f:
|
|
|
|
await rag.ainsert(f.read())
|
|
|
|
await rag.ainsert(input=f.read(),file_paths=[f"sushi.txt"]) # 添加来源参数
|
|
|
|
finally:
|
|
|
|
finally:
|
|
|
|
if rag:
|
|
|
|
if rag:
|
|
|
|
await rag.finalize_storages()
|
|
|
|
await rag.finalize_storages()
|
|
|
|