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

Merge pull request #953 from zhicwu/main

Browse Source 
Use consistent format as in examples
This commit is contained in:
zrguo
2025-02-26 23:23:44 +08:00
committed by GitHub
parent c785b4ba7d d79a9d7acc
commit 781ce7dfe2
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lightrag/operate.py
View File

@@ -362,7 +362,7 @@ async def extract_entities(
tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"], tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"],
record_delimiter=PROMPTS["DEFAULT_RECORD_DELIMITER"], record_delimiter=PROMPTS["DEFAULT_RECORD_DELIMITER"],
completion_delimiter=PROMPTS["DEFAULT_COMPLETION_DELIMITER"], completion_delimiter=PROMPTS["DEFAULT_COMPLETION_DELIMITER"],
entity_types=",".join(entity_types), entity_types=", ".join(entity_types),
language=language, language=language,
) )
# add example's format # add example's format

5
lightrag/prompt.py
View File

@@ -47,8 +47,9 @@ Format the content-level key words as ("content_keywords"{tuple_delimiter}<high_
############################# #############################
---Real Data--- ---Real Data---
###################### ######################
Entity_types: {entity_types} Entity_types: [{entity_types}]
Text: {input_text} Text:
{input_text}
###################### ######################
Output:""" Output:"""