From 9af36769917942cc0552ee3c66cc87124517a397 Mon Sep 17 00:00:00 2001 From: Larfii <834462287@qq.com> Date: Thu, 5 Dec 2024 18:26:55 +0800 Subject: [PATCH] Fix JSON parsing error --- examples/lightrag_lmdeploy_demo.py | 6 +++++- lightrag/operate.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/examples/lightrag_lmdeploy_demo.py b/examples/lightrag_lmdeploy_demo.py index b44bfb1d..c0ee86cb 100644 --- a/examples/lightrag_lmdeploy_demo.py +++ b/examples/lightrag_lmdeploy_demo.py @@ -12,7 +12,11 @@ if not os.path.exists(WORKING_DIR): async def lmdeploy_model_complete( - prompt=None, system_prompt=None, history_messages=[], keyword_extraction=False, **kwargs + prompt=None, + system_prompt=None, + history_messages=[], + keyword_extraction=False, + **kwargs, ) -> str: model_name = kwargs["hashing_kv"].global_config["llm_model_name"] return await lmdeploy_model_if_cache( diff --git a/lightrag/operate.py b/lightrag/operate.py index d5e5f56e..947a8183 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -479,7 +479,7 @@ async def kg_query( print(result) try: # json_text = locate_json_string_body_from_string(result) # handled in use_model_func - result = re.search(r"{.*}", json_text, re.DOTALL) + result = re.search(r"{.*}", result, re.DOTALL) keywords_data = json.loads(result) hl_keywords = keywords_data.get("high_level_keywords", []) ll_keywords = keywords_data.get("low_level_keywords", [])