diff --git a/dsLightRag/Test/T1_WriteToPg.py b/dsLightRag/Test/T1_WriteToPg.py index 2598a18c..0a33cf2d 100644 --- a/dsLightRag/Test/T1_WriteToPg.py +++ b/dsLightRag/Test/T1_WriteToPg.py @@ -22,8 +22,7 @@ handler = logging.StreamHandler() handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) logger.addHandler(handler) -ROOT_DIR = '.' -WORKING_DIR = f"{ROOT_DIR}/dickens-pg" +WORKING_DIR = f"./dickens-pg" logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO) @@ -58,19 +57,15 @@ async def initialize_pg_rag(): vector_storage="PGVectorStorage", auto_manage_storages_states=False, ) - await rag.initialize_storages() await initialize_pipeline_status() - return rag async def main(): rag = await initialize_pg_rag() - with open(f"../Txt/sushi.txt", "r", encoding="utf-8") as f: await rag.ainsert(f.read()) - if __name__ == "__main__": asyncio.run(main())