File tree Expand file tree Collapse file tree 3 files changed +11
-5
lines changed
src/neo4j_graphrag/experimental/pipeline Expand file tree Collapse file tree 3 files changed +11
-5
lines changed Original file line number Diff line number Diff line change @@ -65,7 +65,6 @@ kg_builder = SimpleKGPipeline(
6565 llm = llm,
6666 driver = driver,
6767 embedder = OpenAIEmbeddings(),
68- file_path = file_path,
6968 entities = entities,
7069 relations = relations,
7170)
Original file line number Diff line number Diff line change @@ -78,16 +78,24 @@ SinglePropertyExactMatchResolver
7878 :members: run
7979
8080
81-
8281.. _pipeline-section :
8382
8483********
8584Pipeline
8685********
8786
87+ Pipeline
88+ ========
89+
8890.. autoclass :: neo4j_graphrag.experimental.pipeline.Pipeline
8991 :members: run, add_component, connect, get_pygraphviz_graph
9092
93+ SimpleKGPipeline
94+ ================
95+
96+ .. autoclass :: neo4j_graphrag.experimental.pipeline.kg_builder.SimpleKGPipeline
97+ :members: run_async
98+
9199
92100.. _retrievers-section :
93101
Original file line number Diff line number Diff line change @@ -81,9 +81,8 @@ class SimpleKGPipeline:
8181 text_splitter (Optional[Any]): A text splitter component. Defaults to FixedSizeSplitter().
8282 pdf_loader (Optional[Any]): A PDF loader component. Defaults to PdfLoader().
8383 kg_writer (Optional[Any]): A knowledge graph writer component. Defaults to Neo4jWriter().
84- on_error (str): Error handling strategy. Defaults to "RAISE ". Possible values: "RAISE" or "IGNORE ".
84+ on_error (str): Error handling strategy. Defaults to "CONTINUE ". Possible values: "RAISE" or "CONTINUE ".
8585 perform_entity_resolution (bool): Merge entities with same label and name. Default: True
86- text_splitter (Optional[Any]): A text splitter component. Defaults to FixedSizeSplitter().
8786 prompt_template (str): A custom prompt template to use for extraction.
8887 """
8988
@@ -99,7 +98,7 @@ def __init__(
9998 text_splitter : Optional [Any ] = None ,
10099 pdf_loader : Optional [Any ] = None ,
101100 kg_writer : Optional [Any ] = None ,
102- on_error : str = "RAISE " ,
101+ on_error : str = "CONTINUE " ,
103102 prompt_template : Union [ERExtractionTemplate , str ] = ERExtractionTemplate (),
104103 perform_entity_resolution : bool = True ,
105104 ):
You can’t perform that action at this time.
0 commit comments