fix bug of example prompt

2024-12-03 22:25:50 +08:00
parent b00f2d6027
commit 380a1d1bc3
2 changed files with 11 additions and 1 deletions
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -264,6 +264,16 @@ async def extract_entities(
    else:
        examples = "\n".join(PROMPTS["entity_extraction_examples"])

+    example_context_base = dict(
+        tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"],
+        record_delimiter=PROMPTS["DEFAULT_RECORD_DELIMITER"],
+        completion_delimiter=PROMPTS["DEFAULT_COMPLETION_DELIMITER"],
+        entity_types=",".join(PROMPTS["DEFAULT_ENTITY_TYPES"]),
+        language=language,
+    )
+    # add example's format
+    examples = examples.format(**example_context_base)
+
    entity_extract_prompt = PROMPTS["entity_extraction"]
    context_base = dict(
        tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"],