diff --git a/docs/source/user_guide_kg_builder.rst b/docs/source/user_guide_kg_builder.rst index 65ccbd82..71ad7bdc 100644 --- a/docs/source/user_guide_kg_builder.rst +++ b/docs/source/user_guide_kg_builder.rst @@ -897,6 +897,10 @@ It can be used in this way: from neo4j_graphrag.experimental.components.entity_relation_extractor import ( LLMEntityRelationExtractor, ) + from neo4j_graphrag.experimental.components.types import ( + TextChunks, + TextChunk + ) from neo4j_graphrag.llm import OpenAILLM extractor = LLMEntityRelationExtractor( @@ -908,7 +912,7 @@ It can be used in this way: }, ) ) - await extractor.run(chunks=TextChunks(chunks=[TextChunk(text="some text")])) + await extractor.run(chunks=TextChunks(chunks=[TextChunk(text="some text", index=0)])) .. warning::