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

Improve JSON entity extraction prompt (#949)

Browse source 
Add guideline to extract entities from all JSON properties, not just primary fields like name/user. This ensures comprehensive entity extraction while maintaining the existing exclusion of date properties.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
Jack Ryan 2025-09-30 10:00:14 -05:00 • committed by GitHub
parent f2c4c97362
commit f632a8ae9e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
graphiti_core/prompts/extract_nodes.py
View file

@ -152,7 +152,8 @@ Indicate the classified entity type by providing its entity_type_id.
Guidelines: Guidelines:
1. Always try to extract an entities that the JSON represents. This will often be something like a "name" or "user field 1. Always try to extract an entities that the JSON represents. This will often be something like a "name" or "user field
2. Do NOT extract any properties that contain dates 2. Extract all entities mentioned in all other properties throughout the JSON structure
3. Do NOT extract any properties that contain dates
""" """
return [ return [
Message(role='system', content=sys_prompt), Message(role='system', content=sys_prompt),