gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove semaphore logic from EmbeddingFunc(cause num of instances is already control by limit_async_func_call)

Browse Source
This commit is contained in:
yangdx
2025-02-01 09:53:11 +08:00
parent 60b66b95e3
commit 1192727be7
1 changed files with 1 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lightrag/utils.py
View File

@@ -60,14 +60,7 @@ class EmbeddingFunc:
func: callable func: callable
concurrent_limit: int = 16 concurrent_limit: int = 16
def __post_init__(self):
if self.concurrent_limit != 0:
self._semaphore = asyncio.Semaphore(self.concurrent_limit)
else:
self._semaphore = UnlimitedSemaphore()
async def __call__(self, *args, **kwargs) -> np.ndarray: async def __call__(self, *args, **kwargs) -> np.ndarray:
async with self._semaphore:
return await self.func(*args, **kwargs) return await self.func(*args, **kwargs)