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

9
lightrag/utils.py
View File

@@ -60,15 +60,8 @@ 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)
def locate_json_string_body_from_string(content: str) -> Union[str, None]: def locate_json_string_body_from_string(content: str) -> Union[str, None]: