Skip to content

Commit 6a0ba67

Browse files
committed
fix LLMEntityRelationExtractor user guide
1 parent 7b4e2d3 commit 6a0ba67

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
@@ -897,6 +897,10 @@ It can be used in this way:
897897
from neo4j_graphrag.experimental.components.entity_relation_extractor import (
898898
LLMEntityRelationExtractor,
899899
)
900+
from neo4j_graphrag.experimental.components.types import (
901+
TextChunks,
902+
TextChunk
903+
)
900904
from neo4j_graphrag.llm import OpenAILLM
901905
902906
extractor = LLMEntityRelationExtractor(
@@ -908,7 +912,7 @@ It can be used in this way:
908912
},
909913
)
910914
)
911-
await extractor.run(chunks=TextChunks(chunks=[TextChunk(text="some text")]))
915+
await extractor.run(chunks=TextChunks(chunks=[TextChunk(text="some text", index=0)]))
912916
913917
914918
.. warning::

0 commit comments

Comments
 (0)