gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

A more robust approach for result to json.

Browse source
This commit is contained in:
luoyifan 2024-11-19 14:02:38 +08:00
parent 633aa6c2ab
commit 92fe18b2ea
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lightrag/operate.py
View file

@ -418,7 +418,7 @@ async def local_query(
.replace("model", "")
.strip()
)
result = "{" + result.split("{")[1].split("}")[0] + "}"
result = "{" + result.split("{")[-1].split("}")[0] + "}"
keywords_data = json.loads(result)
keywords = keywords_data.get("low_level_keywords", [])
@ -691,7 +691,7 @@ async def global_query(
.replace("model", "")
.strip()
)
result = "{" + result.split("{")[1].split("}")[0] + "}"
result = "{" + result.split("{")[-1].split("}")[0] + "}"
keywords_data = json.loads(result)
keywords = keywords_data.get("high_level_keywords", [])
@ -940,7 +940,7 @@ async def hybrid_query(
.replace("model", "")
.strip()
)
result = "{" + result.split("{")[1].split("}")[0] + "}"
result = "{" + result.split("{")[-1].split("}")[0] + "}"
keywords_data = json.loads(result)
hl_keywords = keywords_data.get("high_level_keywords", [])
ll_keywords = keywords_data.get("low_level_keywords", [])