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

use default value for return exceptions (#246)

Browse source
This commit is contained in:
Preston Rasmussen 2024-12-17 15:11:22 -05:00 • committed by GitHub
parent 7f94b0c648
commit fcff4ca67c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
graphiti_core/helpers.py
View file

@ -86,16 +86,11 @@ def normalize_l2(embedding: list[float]) -> list[float]:
# Use this instead of asyncio.gather() to bound coroutines # Use this instead of asyncio.gather() to bound coroutines
async def semaphore_gather( async def semaphore_gather(*coroutines: Coroutine, max_coroutines: int = SEMAPHORE_LIMIT):
*coroutines: Coroutine, max_coroutines: int = SEMAPHORE_LIMIT, return_exceptions=True
):
semaphore = asyncio.Semaphore(max_coroutines) semaphore = asyncio.Semaphore(max_coroutines)
async def _wrap_coroutine(coroutine): async def _wrap_coroutine(coroutine):
async with semaphore: async with semaphore:
return await coroutine return await coroutine
return await asyncio.gather( return await asyncio.gather(*(_wrap_coroutine(coroutine) for coroutine in coroutines))
*(_wrap_coroutine(coroutine) for coroutine in coroutines),
return_exceptions=return_exceptions,
)