diff --git a/lightrag/kg/milvus_impl.py b/lightrag/kg/milvus_impl.py
index 6d2520ce..c72498c5 100644
--- a/lightrag/kg/milvus_impl.py
+++ b/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",
diff --git a/lightrag/storage.py b/lightrag/storage.py
index 4c043893..534c6e2e 100644
--- a/lightrag/storage.py
+++ b/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",