From 457e683acd995ad8dce4872147860260e082d866 Mon Sep 17 00:00:00 2001 From: zrguo <49157727+LarFii@users.noreply.github.com> Date: Thu, 26 Dec 2024 22:14:04 +0800 Subject: [PATCH] Update lightrag.py --- lightrag/lightrag.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 81963328..51ac204d 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -198,15 +198,12 @@ class LightRAG: logger.info(f"Creating working directory {self.working_dir}") os.makedirs(self.working_dir) - self.llm_response_cache = ( - self.key_string_value_json_storage_cls( - namespace="llm_response_cache", - global_config=asdict(self), - embedding_func=None, - ) - if self.enable_llm_cache - else None + self.llm_response_cache = self.key_string_value_json_storage_cls( + namespace="llm_response_cache", + global_config=asdict(self), + embedding_func=None, ) + self.embedding_func = limit_async_func_call(self.embedding_func_max_async)( self.embedding_func )