File size: 1,271 Bytes
942050b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | """Pipeline node factories. Each ``make_*_node`` returns a callable that
takes `PipelineState` and returns a partial state dict for LangGraph to merge.
Why factories: the graph topology is fixed but providers/index/registry are
runtime-injected. Keeping construction explicit (factory closures over deps)
makes tests trivial — pass fakes, run the node, assert the partial dict.
"""
from __future__ import annotations
from nl_sql.agent.nodes.context_builder import make_context_builder_node
from nl_sql.agent.nodes.execute import make_execute_node
from nl_sql.agent.nodes.explain_trace import make_explain_trace_node
from nl_sql.agent.nodes.format import make_format_node
from nl_sql.agent.nodes.generate_sql import make_generate_sql_node
from nl_sql.agent.nodes.grounded_critique import make_grounded_critique_node
from nl_sql.agent.nodes.plan_query import make_plan_node
from nl_sql.agent.nodes.repair_once import make_repair_once_node
from nl_sql.agent.nodes.validate import make_validate_node
__all__ = [
"make_context_builder_node",
"make_execute_node",
"make_explain_trace_node",
"make_format_node",
"make_generate_sql_node",
"make_grounded_critique_node",
"make_plan_node",
"make_repair_once_node",
"make_validate_node",
]
|