zrguo
commited on
Update lightrag.py
Browse files- lightrag/lightrag.py +5 -8
lightrag/lightrag.py
CHANGED
|
@@ -198,15 +198,12 @@ class LightRAG:
|
|
| 198 |
logger.info(f"Creating working directory {self.working_dir}")
|
| 199 |
os.makedirs(self.working_dir)
|
| 200 |
|
| 201 |
-
self.llm_response_cache = (
|
| 202 |
-
|
| 203 |
-
|
| 204 |
-
|
| 205 |
-
embedding_func=None,
|
| 206 |
-
)
|
| 207 |
-
if self.enable_llm_cache
|
| 208 |
-
else None
|
| 209 |
)
|
|
|
|
| 210 |
self.embedding_func = limit_async_func_call(self.embedding_func_max_async)(
|
| 211 |
self.embedding_func
|
| 212 |
)
|
|
|
|
| 198 |
logger.info(f"Creating working directory {self.working_dir}")
|
| 199 |
os.makedirs(self.working_dir)
|
| 200 |
|
| 201 |
+
self.llm_response_cache = self.key_string_value_json_storage_cls(
|
| 202 |
+
namespace="llm_response_cache",
|
| 203 |
+
global_config=asdict(self),
|
| 204 |
+
embedding_func=None,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 205 |
)
|
| 206 |
+
|
| 207 |
self.embedding_func = limit_async_func_call(self.embedding_func_max_async)(
|
| 208 |
self.embedding_func
|
| 209 |
)
|