[gkorland]

We've done a very similar procedure just with FalkorDB as a Graph Database. Notice if you already have a schema/ontology it might be easier might you might miss some entities in the text you did realize exist.

So in our in GraphRAG-SDK we are running two phases, the first is sampling the data to suggest a schema and the second is using this schema to ground the LLM to this schema (as you suggested)

Check https://github.com/FalkorDB/GraphRAG-SDK