Melika Kheirieh commited on
Commit
1615704
·
1 Parent(s): ed681b1

fix(pipeline): allow optional schema_preview in run() for better typing compatibility with benchmarks

Browse files
Files changed (1) hide show
  1. nl2sql/pipeline.py +2 -1
nl2sql/pipeline.py CHANGED
@@ -80,7 +80,7 @@ class Pipeline:
80
  self,
81
  *,
82
  user_query: str,
83
- schema_preview: str,
84
  clarify_answers: Optional[Dict[str, Any]] = None,
85
  ) -> FinalResult:
86
  traces: List[dict] = []
@@ -88,6 +88,7 @@ class Pipeline:
88
  sql: Optional[str] = None
89
  rationale: Optional[str] = None
90
  verified: Optional[bool] = None
 
91
 
92
  # --- 1) ambiguity detection ---
93
  try:
 
80
  self,
81
  *,
82
  user_query: str,
83
+ schema_preview: str | None = None,
84
  clarify_answers: Optional[Dict[str, Any]] = None,
85
  ) -> FinalResult:
86
  traces: List[dict] = []
 
88
  sql: Optional[str] = None
89
  rationale: Optional[str] = None
90
  verified: Optional[bool] = None
91
+ schema_preview = schema_preview or ""
92
 
93
  # --- 1) ambiguity detection ---
94
  try: