diff --git a/graphiti_core/prompts/extract_nodes.py b/graphiti_core/prompts/extract_nodes.py index 29e99978..f9d6eabd 100644 --- a/graphiti_core/prompts/extract_nodes.py +++ b/graphiti_core/prompts/extract_nodes.py @@ -260,10 +260,7 @@ def extract_attributes(context: dict[str, Any]) -> list[Message]: role='user', content=f""" - - {to_prompt_json(context['previous_episodes'], indent=2)} - {to_prompt_json(context['episode_content'], indent=2)} - + Given the above MESSAGES and the following ENTITY, update any of its attributes based on the information provided in MESSAGES. Use the provided attribute descriptions to better understand how each attribute should be determined. @@ -271,6 +268,11 @@ def extract_attributes(context: dict[str, Any]) -> list[Message]: Guidelines: 1. Do not hallucinate entity property values if they cannot be found in the current context. 2. Only use the provided MESSAGES and ENTITY to set attribute values. + + + {to_prompt_json(context['previous_episodes'], indent=2)} + {to_prompt_json(context['episode_content'], indent=2)} + {context['node']} @@ -289,17 +291,16 @@ def extract_summary(context: dict[str, Any]) -> list[Message]: Message( role='user', content=f""" + Given the MESSAGES and the ENTITY, update the summary that combines relevant information about the entity + from the messages and relevant information from the existing summary. + + {summary_instructions} {to_prompt_json(context['previous_episodes'], indent=2)} {to_prompt_json(context['episode_content'], indent=2)} - Given the above MESSAGES and the following ENTITY, update the summary that combines relevant information about the entity - from the messages and relevant information from the existing summary. - - {summary_instructions} - {context['node']}