gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #477 from theClassLight/GG/pr

Browse Source 
refactor(lightrag): 优化hashing_kv初始化逻辑
This commit is contained in:
zrguo
2024-12-17 15:09:26 +08:00
committed by GitHub
parent a3bf3a58f0 9893e93dd1
commit c840a053ac
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lightrag/lightrag.py
View File

@@ -252,7 +252,12 @@ class LightRAG:
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 @@ class LightRAG:
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 @@ class LightRAG:
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}")