gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit 781ce7dfe2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 3 deletions
Show stats 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:"""