change insert to be sync (#153)

aigc-apps · Aug 12, 2024 · b93d7ee · b93d7ee
1 parent 6a6a2e7
commit b93d7ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pai_rag/data/rag_dataloader.py b/src/pai_rag/data/rag_dataloader.py
@@ -222,7 +222,7 @@ async def aload(
 
         if enable_raptor:
             nodes_with_embeddings = await self.node_enhance.acall(nodes=nodes)
-            await self.index.vector_index.insert_nodes_async(nodes_with_embeddings)
+            self.index.vector_index.insert_nodes(nodes_with_embeddings)
 
             logger.info(
                 f"Async inserted {len(nodes)} and enhanced {len(nodes_with_embeddings)-len(nodes)} nodes successfully."