From 1c00dbfa56037607f3e74c4576b5bb42d06c2a54 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 2fb57e76 --- lightrag/lightrag.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 67ec2308..47a6f7e6 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -527,6 +527,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,