From d79a9d7acc5ad8f62e6c45dacf180f9b323218c6 Mon Sep 17 00:00:00 2001
From: Zhichun Wu <zhicwu@gmail.com>
Date: Wed, 26 Feb 2025 23:04:21 +0800
Subject: [PATCH] consistent format

---
 lightrag/operate.py | 2 +-
 lightrag/prompt.py  | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/lightrag/operate.py b/lightrag/operate.py
index 89f9a777..d5dfc793 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -362,7 +362,7 @@ async def extract_entities(
         tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"],
         record_delimiter=PROMPTS["DEFAULT_RECORD_DELIMITER"],
         completion_delimiter=PROMPTS["DEFAULT_COMPLETION_DELIMITER"],
-        entity_types=",".join(entity_types),
+        entity_types=", ".join(entity_types),
         language=language,
     )
     # add example's format
diff --git a/lightrag/prompt.py b/lightrag/prompt.py
index c6cbf7db..d22244c9 100644
--- a/lightrag/prompt.py
+++ b/lightrag/prompt.py
@@ -47,8 +47,9 @@ Format the content-level key words as ("content_keywords"{tuple_delimiter}<high_
 #############################
 ---Real Data---
 ######################
-Entity_types: {entity_types}
-Text: {input_text}
+Entity_types: [{entity_types}]
+Text:
+{input_text}
 ######################
 Output:"""