From c9becdf5f40e6f3b291b9a3d50a9624d677e5d65 Mon Sep 17 00:00:00 2001 From: luoyifan <1625370020@qq.com> Date: Tue, 19 Nov 2024 14:02:38 +0800 Subject: [PATCH] A more robust approach for result to json. --- lightrag/operate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lightrag/operate.py b/lightrag/operate.py index b11e14fe..eb600c4b 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -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", [])