Spaces:
Sleeping
Sleeping
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- 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:
|