The asyncio.as_completed() function does not guarantee that the results are ordered

lightrag/kg/milvus_impl.py

@@ -59,7 +59,7 @@ class MilvusVectorDBStorge(BaseVectorStorage):
         embedding_tasks = [self.embedding_func(batch) for batch in batches]
         embeddings_list = []
         for f in tqdm_async(
-            asyncio.as_completed(embedding_tasks),
+            await asyncio.gather(*embedding_tasks),
             total=len(embedding_tasks),
             desc="Generating embeddings",
             unit="batch",

lightrag/storage.py

@@ -99,7 +99,7 @@ class NanoVectorDBStorage(BaseVectorStorage):
         embedding_tasks = [self.embedding_func(batch) for batch in batches]
         embeddings_list = []
         for f in tqdm_async(
-            asyncio.as_completed(embedding_tasks),
+            await asyncio.gather(*embedding_tasks),
             total=len(embedding_tasks),
             desc="Generating embeddings",
             unit="batch",