diff --git a/evaluation/hypergraphrag/operate.py b/evaluation/hypergraphrag/operate.py index 58db667..92614fa 100644 --- a/evaluation/hypergraphrag/operate.py +++ b/evaluation/hypergraphrag/operate.py @@ -304,8 +304,8 @@ async def extract_entities( language=language, ) - continue_prompt = PROMPTS["entiti_continue_extraction"] - if_loop_prompt = PROMPTS["entiti_if_loop_extraction"] + continue_prompt = PROMPTS["entity_continue_extraction"] + if_loop_prompt = PROMPTS["entity_if_loop_extraction"] already_processed = 0 already_entities = 0 diff --git a/evaluation/hypergraphrag/prompt.py b/evaluation/hypergraphrag/prompt.py index c867d8b..56d9b38 100644 --- a/evaluation/hypergraphrag/prompt.py +++ b/evaluation/hypergraphrag/prompt.py @@ -176,13 +176,13 @@ """ PROMPTS[ - "entiti_continue_extraction" -] = """MANY knowdge fragements with entities were missed in the last extraction. Add them below using the same format: + "entity_continue_extraction" +] = """MANY knowledge fragments with entities were missed in the last extraction. Add them below using the same format: """ PROMPTS[ - "entiti_if_loop_extraction" -] = """Please check whether knowdge fragements cover all the given text. Answer YES | NO if there are knowdge fragements that need to be added. + "entity_if_loop_extraction" +] = """Please check whether knowledge fragments cover all the given text. Answer YES | NO if there are knowledge fragments that need to be added. """ PROMPTS["fail_response"] = "Sorry, I'm not able to provide an answer to that question." diff --git a/hypergraphrag/operate.py b/hypergraphrag/operate.py index 7269146..f0415ac 100644 --- a/hypergraphrag/operate.py +++ b/hypergraphrag/operate.py @@ -304,8 +304,8 @@ async def extract_entities( language=language, ) - continue_prompt = PROMPTS["entiti_continue_extraction"] - if_loop_prompt = PROMPTS["entiti_if_loop_extraction"] + continue_prompt = PROMPTS["entity_continue_extraction"] + if_loop_prompt = PROMPTS["entity_if_loop_extraction"] already_processed = 0 already_entities = 0 diff --git a/hypergraphrag/prompt.py b/hypergraphrag/prompt.py index c867d8b..56d9b38 100644 --- a/hypergraphrag/prompt.py +++ b/hypergraphrag/prompt.py @@ -176,13 +176,13 @@ """ PROMPTS[ - "entiti_continue_extraction" -] = """MANY knowdge fragements with entities were missed in the last extraction. Add them below using the same format: + "entity_continue_extraction" +] = """MANY knowledge fragments with entities were missed in the last extraction. Add them below using the same format: """ PROMPTS[ - "entiti_if_loop_extraction" -] = """Please check whether knowdge fragements cover all the given text. Answer YES | NO if there are knowdge fragements that need to be added. + "entity_if_loop_extraction" +] = """Please check whether knowledge fragments cover all the given text. Answer YES | NO if there are knowledge fragments that need to be added. """ PROMPTS["fail_response"] = "Sorry, I'm not able to provide an answer to that question."