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

Improve prompt clarity and structure

Browse source
This commit is contained in:
yangdx 2025-09-04 14:53:27 +08:00
parent 83b54975a2
commit 94114df995
1 changed files with 14 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
lightrag/prompt.py
View file

@ -11,7 +11,7 @@ PROMPTS["DEFAULT_COMPLETION_DELIMITER"] = "<|COMPLETE|>"
PROMPTS["DEFAULT_USER_PROMPT"] = "n/a" PROMPTS["DEFAULT_USER_PROMPT"] = "n/a"
PROMPTS["entity_extraction"] = """---Task--- PROMPTS["entity_extraction"] = """---Task---
Given a text document and a list of entity types, identify all entities of those types and all relationships among the identified entities. For a given text and a list of entity types, extract all entities and their relationships, then return them in the specified {language} and format described below.
---Instructions--- ---Instructions---
1. Recognizing definitively conceptualized entities in text. For each identified entity, extract the following information: 1. Recognizing definitively conceptualized entities in text. For each identified entity, extract the following information:
@ -49,6 +49,19 @@ Text:
---Output--- ---Output---
""" """
PROMPTS["entity_continue_extraction"] = """---Task---
Identify any missed entities or relationships in the last extraction task.
---Instructions---
1. Output the entities and realtionships in the same format as previous extraction task.
2. Do not include entities and relations that have been previously extracted.
3. If the entity doesn't clearly fit in any of`Entity_types` provided, classify it as "Other".
4. Return identified entities and relationships in {language}.
5. Output `{completion_delimiter}` when all the entities and relationships are extracted.
---Output---
"""
PROMPTS["entity_extraction_examples"] = [ PROMPTS["entity_extraction_examples"] = [
"""[Example 1] """[Example 1]
@ -180,28 +193,6 @@ Description List:
---Output--- ---Output---
""" """
PROMPTS["entity_continue_extraction"] = """---Task---
Identify any missed entities or relationships in the last extraction task.
---Instructions---
1. Output the entities and realtionships in the same format as previous extraction task.
2. Do not include entities and relations that have been previously extracted.
3. If the entity doesn't clearly fit in any of`Entity_types` provided, classify it as "Other".
4. Return identified entities and relationships in {language}.
5. Output `{completion_delimiter}` when all the entities and relationships are extracted.
---Output---
"""
# TODO: Deprecated
PROMPTS["entity_if_loop_extraction"] = """
---Goal---'
Check if it appears some entities may have still been missed. Output "Yes" if so, otherwise "No".
---Output---
Output:"""
PROMPTS["fail_response"] = ( PROMPTS["fail_response"] = (
"Sorry, I'm not able to provide an answer to that question.[no-context]" "Sorry, I'm not able to provide an answer to that question.[no-context]"
) )