'commit'

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())