Skip to content

Commit cfeda57

Browse files
authored
Changed OnError default to 'CONTINUE' (#167)
* Changed OnError default to 'CONTINUE' * Fix incorrect docstring description * Remove text_splitter
1 parent f63a375 commit cfeda57

File tree

3 files changed

+11
-5
lines changed

3 files changed

+11
-5
lines changed

README.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff 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
)

docs/source/api.rst

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,16 +78,24 @@ SinglePropertyExactMatchResolver
7878
:members: run
7979

8080

81-
8281
.. _pipeline-section:
8382

8483
********
8584
Pipeline
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

src/neo4j_graphrag/experimental/pipeline/kg_builder.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff 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
):

0 commit comments

Comments
 (0)