Configurable SimpleKGPipeline Node label and Relationship type

neo4j / neo4j-graphrag-python

Neo4j GraphRAG for Python

https://neo4j.com/docs/neo4j-graphrag-python/current/

Other

231 stars 37 forks source link

Configurable SimpleKGPipeline Node label and Relationship type #197

Open martinohanlon opened 1 month ago

martinohanlon commented 1 month ago

It would be useful to fit into existing models to be able to configure the node labels and relationship types when using the SimpleKGPipeline.

e.g. being able to modify these constraints at run time.

https://github.com/neo4j/neo4j-graphrag-python/blob/main/src/neo4j_graphrag/experimental/components/entity_relation_extractor.py#L55

edit - update the link to the code

stellasia commented 3 weeks ago

Hi @martinohanlon , thank you for the request!

In v1.2.0 (released today), it is possible to configure the labels, but only in the low-level API (see this example). We're working on improving the SimpleKGPipeline interface, but this can be a solution for you in the meantime.