Merge pull request #477 from theClassLight/GG/pr

refactor(lightrag): 优化hashing_kv初始化逻辑
HKUDS · Dec 17, 2024 · c840a05 · c840a05
2 parents a3bf3a5 + 9893e93
commit c840a05
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
@@ -252,7 +252,12 @@ def __post_init__(self):
         self.llm_model_func = limit_async_func_call(self.llm_model_max_async)(
             partial(
                 self.llm_model_func,
-                hashing_kv=self.llm_response_cache,
+                hashing_kv=self.llm_response_cache
+                if self.llm_response_cache
+                and hasattr(self.llm_response_cache, "global_config")
+                else self.key_string_value_json_storage_cls(
+                    global_config=asdict(self),
+                ),
                 **self.llm_model_kwargs,
             )
         )
@@ -515,7 +520,12 @@ async def aquery(self, query: str, param: QueryParam = QueryParam()):
                 self.text_chunks,
                 param,
                 asdict(self),
-                hashing_kv=self.llm_response_cache,
+                hashing_kv=self.llm_response_cache
+                if self.llm_response_cache
+                and hasattr(self.llm_response_cache, "global_config")
+                else self.key_string_value_json_storage_cls(
+                    global_config=asdict(self),
+                ),
             )
         elif param.mode == "naive":
             response = await naive_query(
@@ -524,7 +534,12 @@ async def aquery(self, query: str, param: QueryParam = QueryParam()):
                 self.text_chunks,
                 param,
                 asdict(self),
-                hashing_kv=self.llm_response_cache,
+                hashing_kv=self.llm_response_cache
+                if self.llm_response_cache
+                and hasattr(self.llm_response_cache, "global_config")
+                else self.key_string_value_json_storage_cls(
+                    global_config=asdict(self),
+                ),
             )
         else:
             raise ValueError(f"Unknown mode {param.mode}")