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

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 b7ccf98b47
commit c9becdf5f4
1 changed files with 3 additions and 3 deletions
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", "") .replace("model", "")
.strip() .strip()
) )
result = "{" + result.split("{")[1].split("}")[0] + "}" result = "{" + result.split("{")[-1].split("}")[0] + "}"
keywords_data = json.loads(result) keywords_data = json.loads(result)
keywords = keywords_data.get("low_level_keywords", []) keywords = keywords_data.get("low_level_keywords", [])
@@ -691,7 +691,7 @@ async def global_query(
.replace("model", "") .replace("model", "")
.strip() .strip()
) )
result = "{" + result.split("{")[1].split("}")[0] + "}" result = "{" + result.split("{")[-1].split("}")[0] + "}"
keywords_data = json.loads(result) keywords_data = json.loads(result)
keywords = keywords_data.get("high_level_keywords", []) keywords = keywords_data.get("high_level_keywords", [])
@@ -940,7 +940,7 @@ async def hybrid_query(
.replace("model", "") .replace("model", "")
.strip() .strip()
) )
result = "{" + result.split("{")[1].split("}")[0] + "}" result = "{" + result.split("{")[-1].split("}")[0] + "}"
keywords_data = json.loads(result) keywords_data = json.loads(result)
hl_keywords = keywords_data.get("high_level_keywords", []) hl_keywords = keywords_data.get("high_level_keywords", [])
ll_keywords = keywords_data.get("low_level_keywords", []) ll_keywords = keywords_data.get("low_level_keywords", [])