Skip to content

Commit f9a7c8c

Browse files
Fix component connections
1 parent 9d05c76 commit f9a7c8c

File tree

1 file changed

+3
-10
lines changed

1 file changed

+3
-10
lines changed

src/neo4j_graphrag/experimental/pipeline/config/template_pipeline/simple_kg_builder.py

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -298,16 +298,6 @@ def _get_connections(self) -> list[ConnectionDefinition]:
298298
)
299299
)
300300
else:
301-
# handle automatic schema extraction for direct text input: ensure schema extraction uses the complete text
302-
if self.auto_schema_extraction and not self.has_user_provided_schema():
303-
connections.append(
304-
ConnectionDefinition(
305-
start="__input__", # connection to pipeline input
306-
end="schema",
307-
input_config={"text": "text"}, # use the original text input
308-
)
309-
)
310-
311301
connections.append(
312302
ConnectionDefinition(
313303
start="schema",
@@ -379,4 +369,7 @@ def get_run_params(self, user_input: dict[str, Any]) -> dict[str, Any]:
379369
"Expected 'text' argument when 'from_pdf' is False."
380370
)
381371
run_params["splitter"] = {"text": text}
372+
# Add full text to schema component for automatic schema extraction
373+
if self.auto_schema_extraction and not self.has_user_provided_schema():
374+
run_params["schema"] = {"text": text}
382375
return run_params

0 commit comments

Comments
 (0)