gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
billvsme 2024-12-13 15:42:41 +08:00
parent b7552f35aa
commit cf0278ca43
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lightrag/kg/milvus_impl.py
View file

@ -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",

2
lightrag/storage.py
View file

@ -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",