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

Fix JSON parsing error

Browse Source
This commit is contained in:
Larfii
2024-12-05 18:26:55 +08:00
parent 567fde643f
commit 9af3676991
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/lightrag_lmdeploy_demo.py
View File

@@ -12,7 +12,11 @@ if not os.path.exists(WORKING_DIR):
async def lmdeploy_model_complete( 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: ) -> str:
model_name = kwargs["hashing_kv"].global_config["llm_model_name"] model_name = kwargs["hashing_kv"].global_config["llm_model_name"]
return await lmdeploy_model_if_cache( return await lmdeploy_model_if_cache(

2
lightrag/operate.py
View File

@@ -479,7 +479,7 @@ async def kg_query(
print(result) print(result)
try: try:
# json_text = locate_json_string_body_from_string(result) # handled in use_model_func # 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) 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", [])