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", [])