From bef82fde1c1e04878dbbe23401f6487176b6301a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20MANSUY?= Date: Thu, 4 Dec 2025 19:14:27 +0800 Subject: [PATCH] cherry-pick de4412dd --- lightrag/lightrag.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 72a4dc6d..518d347a 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -526,6 +526,16 @@ class LightRAG: logger.debug(f"LightRAG init with param:\n {_print_config}\n") # Init Embedding + # Step 1: Capture max_token_size before applying decorator (decorator strips dataclass attributes) + embedding_max_token_size = None + if self.embedding_func and hasattr(self.embedding_func, "max_token_size"): + embedding_max_token_size = self.embedding_func.max_token_size + logger.debug( + f"Captured embedding max_token_size: {embedding_max_token_size}" + ) + self.embedding_token_limit = embedding_max_token_size + + # Step 2: Apply priority wrapper decorator self.embedding_func = priority_limit_async_func_call( self.embedding_func_max_async, llm_timeout=self.default_embedding_timeout,