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

Save keywords to cache only when it's no empty

Browse Source
This commit is contained in:
yangdx
2025-02-01 22:54:23 +08:00
parent 3c3cdba499
commit 2d387fa6de
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/operate.py
View File

@@ -760,7 +760,7 @@ async def extract_keywords_only(
# 6. Parse out JSON from the LLM response # 6. Parse out JSON from the LLM response
match = re.search(r"\{.*\}", result, re.DOTALL) match = re.search(r"\{.*\}", result, re.DOTALL)
if not match: if not match:
logger.error("No JSON-like structure found in the result.") logger.error("No JSON-like structure found in the LLM respond.")
return [], [] return [], []
try: try:
keywords_data = json.loads(match.group(0)) keywords_data = json.loads(match.group(0))
@@ -772,6 +772,7 @@ async def extract_keywords_only(
ll_keywords = keywords_data.get("low_level_keywords", []) ll_keywords = keywords_data.get("low_level_keywords", [])
# 7. Cache only the processed keywords with cache type # 7. Cache only the processed keywords with cache type
if hl_keywords or ll_keywords:
cache_data = {"high_level_keywords": hl_keywords, "low_level_keywords": ll_keywords} cache_data = {"high_level_keywords": hl_keywords, "low_level_keywords": ll_keywords}
await save_to_cache( await save_to_cache(
hashing_kv, hashing_kv,