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

chore: Update edge extraction prompt to paraphrase instead of quote

Browse source 
- Changed instruction 5 to request paraphrasing rather than verbatim quoting
- Updated string quotes to use double quotes for consistency

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

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Daniel Chalef 2025-10-01 08:58:51 -07:00
parent 1ebcda19c6
commit 36356f491b
1 changed files with 26 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
graphiti_core/prompts/extract_edges.py
View file

@ -23,17 +23,23 @@ from .prompt_helpers import to_prompt_json
class Edge(BaseModel):
relation_type: str = Field(..., description='FACT_PREDICATE_IN_SCREAMING_SNAKE_CASE')
source_entity_id: int = Field(..., description='The id of the source entity of the fact.')
target_entity_id: int = Field(..., description='The id of the target entity of the fact.')
fact: str = Field(..., description='')
relation_type: str = Field(
..., description="FACT_PREDICATE_IN_SCREAMING_SNAKE_CASE"
)
source_entity_id: int = Field(
..., description="The id of the source entity of the fact."
)
target_entity_id: int = Field(
..., description="The id of the target entity of the fact."
)
fact: str = Field(..., description="")
valid_at: str | None = Field(
None,
description='The date and time when the relationship described by the edge fact became true or was established. Use ISO 8601 format (YYYY-MM-DDTHH:MM:SS.SSSSSSZ)',
description="The date and time when the relationship described by the edge fact became true or was established. Use ISO 8601 format (YYYY-MM-DDTHH:MM:SS.SSSSSSZ)",
)
invalid_at: str | None = Field(
None,
description='The date and time when the relationship described by the edge fact stopped being true or ended. Use ISO 8601 format (YYYY-MM-DDTHH:MM:SS.SSSSSSZ)',
description="The date and time when the relationship described by the edge fact stopped being true or ended. Use ISO 8601 format (YYYY-MM-DDTHH:MM:SS.SSSSSSZ)",
)
@ -60,13 +66,13 @@ class Versions(TypedDict):
def edge(context: dict[str, Any]) -> list[Message]:
return [
Message(
role='system',
content='You are an expert fact extractor that extracts fact triples from text. '
'1. Extracted fact triples should also be extracted with relevant date information.'
'2. Treat the CURRENT TIME as the time the CURRENT MESSAGE was sent. All temporal information should be extracted relative to this time.',
role="system",
content="You are an expert fact extractor that extracts fact triples from text. "
"1. Extracted fact triples should also be extracted with relevant date information."
"2. Treat the CURRENT TIME as the time the CURRENT MESSAGE was sent. All temporal information should be extracted relative to this time.",
),
Message(
role='user',
role="user",
content=f"""
<FACT TYPES>
{context['edge_types']}
@ -111,7 +117,7 @@ You may use information from the PREVIOUS MESSAGES only to disambiguate referenc
2. Each fact must involve two **distinct** entities.
3. Use a SCREAMING_SNAKE_CASE string as the `relation_type` (e.g., FOUNDED, WORKS_AT).
4. Do not emit duplicate or semantically redundant facts.
5. The `fact_text` should quote or closely paraphrase the original source sentence(s).
5. The `fact_text` should closely paraphrase the original source sentence(s). Do not verbatim quote the original text.
6. Use `REFERENCE_TIME` to resolve vague or relative temporal expressions (e.g., "last week").
7. Do **not** hallucinate or infer temporal bounds from unrelated events.
@ -151,19 +157,19 @@ Given the above MESSAGES, list of EXTRACTED ENTITIES entities, and list of EXTRA
determine if any facts haven't been extracted.
"""
return [
Message(role='system', content=sys_prompt),
Message(role='user', content=user_prompt),
Message(role="system", content=sys_prompt),
Message(role="user", content=user_prompt),
]
def extract_attributes(context: dict[str, Any]) -> list[Message]:
return [
Message(
role='system',
content='You are a helpful assistant that extracts fact properties from the provided text.',
role="system",
content="You are a helpful assistant that extracts fact properties from the provided text.",
),
Message(
role='user',
role="user",
content=f"""
<MESSAGE>
@ -189,7 +195,7 @@ def extract_attributes(context: dict[str, Any]) -> list[Message]:
versions: Versions = {
'edge': edge,
'reflexion': reflexion,
'extract_attributes': extract_attributes,
"edge": edge,
"reflexion": reflexion,
"extract_attributes": extract_attributes,
}