Skip to content

Commit cb5e021

Browse files
Amir LayeghAmir Layegh
authored andcommitted
feat: add ConstraintType to GraphSchema for constraint extraction
1 parent e8da4cb commit cb5e021

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/neo4j_graphrag/generation/prompts.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -235,15 +235,15 @@ class SchemaExtractionTemplate(PromptTemplate):
235235
{{
236236
"name": "name",
237237
"type": "STRING"
238-
}},
238+
}}
239239
]
240-
}},
240+
}}
241241
...
242242
],
243243
"relationship_types": [
244244
{{
245245
"label": "WORKS_FOR"
246-
}},
246+
}}
247247
...
248248
],
249249
"patterns": [
@@ -255,7 +255,7 @@ class SchemaExtractionTemplate(PromptTemplate):
255255
"type": "UNIQUENESS",
256256
"node_type": "Person",
257257
"property_name": "person_id"
258-
}},
258+
}}
259259
...
260260
]
261261
}}

0 commit comments

Comments
 (0)