diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index bff2f3b5..69820d9a 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -256,7 +256,9 @@ def __post_init__(self): if self.llm_response_cache and hasattr(self.llm_response_cache, "global_config") else self.key_string_value_json_storage_cls( + namespace="llm_response_cache", global_config=asdict(self), + embedding_func=None, ), **self.llm_model_kwargs, ) @@ -524,7 +526,9 @@ async def aquery(self, query: str, param: QueryParam = QueryParam()): if self.llm_response_cache and hasattr(self.llm_response_cache, "global_config") else self.key_string_value_json_storage_cls( + namespace="llm_response_cache", global_config=asdict(self), + embedding_func=None, ), ) elif param.mode == "naive": @@ -538,7 +542,9 @@ async def aquery(self, query: str, param: QueryParam = QueryParam()): if self.llm_response_cache and hasattr(self.llm_response_cache, "global_config") else self.key_string_value_json_storage_cls( + namespace="llm_response_cache", global_config=asdict(self), + embedding_func=None, ), ) else: