Skip to content

Commit 4b39851

Browse files
fix LLMEntityRelationExtractor user guide (#438)
1 parent fc4d87a commit 4b39851

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

docs/source/user_guide_kg_builder.rst

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -903,6 +903,10 @@ It can be used in this way:
903903
from neo4j_graphrag.experimental.components.entity_relation_extractor import (
904904
LLMEntityRelationExtractor,
905905
)
906+
from neo4j_graphrag.experimental.components.types import (
907+
TextChunks,
908+
TextChunk
909+
)
906910
from neo4j_graphrag.llm import OpenAILLM
907911
908912
extractor = LLMEntityRelationExtractor(
@@ -914,7 +918,7 @@ It can be used in this way:
914918
},
915919
)
916920
)
917-
await extractor.run(chunks=TextChunks(chunks=[TextChunk(text="some text")]))
921+
await extractor.run(chunks=TextChunks(chunks=[TextChunk(text="some text", index=0)]))
918922
919923
920924
.. warning::

0 commit comments

Comments
 (0)