File size: 991 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 33 34 35 | """LangGraph pipeline (stage 4 of v2 architecture).
Six-node graph (per docs/02_architecture_v2.md §3):
context_builder → generate_sql → validate
│
▼
┌─── repair_once ◄─── (validate or execute fail, once)
│
▼
execute → deterministic_format → explain_trace → END
`build_pipeline()` returns a compiled graph; pass `PipelineConfig` to inject
provider instances (lets tests swap real Mistral/Groq for fakes).
"""
from __future__ import annotations
from nl_sql.agent.graph import (
PipelineConfig,
PipelineRunResult,
build_pipeline,
run_pipeline,
)
from nl_sql.agent.state import GenerateSQLOutput, PipelineState
__all__ = [
"GenerateSQLOutput",
"PipelineConfig",
"PipelineRunResult",
"PipelineState",
"build_pipeline",
"run_pipeline",
]
|