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(
65
65
llm = llm,
66
66
driver = driver,
67
67
embedder = OpenAIEmbeddings(),
68
- file_path = file_path,
69
68
entities = entities,
70
69
relations = relations,
71
70
)
Original file line number Diff line number Diff line change @@ -78,16 +78,24 @@ SinglePropertyExactMatchResolver
78
78
:members: run
79
79
80
80
81
-
82
81
.. _pipeline-section :
83
82
84
83
********
85
84
Pipeline
86
85
********
87
86
87
+ Pipeline
88
+ ========
89
+
88
90
.. autoclass :: neo4j_graphrag.experimental.pipeline.Pipeline
89
91
:members: run, add_component, connect, get_pygraphviz_graph
90
92
93
+ SimpleKGPipeline
94
+ ================
95
+
96
+ .. autoclass :: neo4j_graphrag.experimental.pipeline.kg_builder.SimpleKGPipeline
97
+ :members: run_async
98
+
91
99
92
100
.. _retrievers-section :
93
101
Original file line number Diff line number Diff line change @@ -81,9 +81,8 @@ class SimpleKGPipeline:
81
81
text_splitter (Optional[Any]): A text splitter component. Defaults to FixedSizeSplitter().
82
82
pdf_loader (Optional[Any]): A PDF loader component. Defaults to PdfLoader().
83
83
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 ".
85
85
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().
87
86
prompt_template (str): A custom prompt template to use for extraction.
88
87
"""
89
88
@@ -99,7 +98,7 @@ def __init__(
99
98
text_splitter : Optional [Any ] = None ,
100
99
pdf_loader : Optional [Any ] = None ,
101
100
kg_writer : Optional [Any ] = None ,
102
- on_error : str = "RAISE " ,
101
+ on_error : str = "CONTINUE " ,
103
102
prompt_template : Union [ERExtractionTemplate , str ] = ERExtractionTemplate (),
104
103
perform_entity_resolution : bool = True ,
105
104
):
You can’t perform that action at this time.
0 commit comments