Spaces:
Sleeping
Sleeping
Melika Kheirieh
commited on
Commit
·
787d215
1
Parent(s):
1f5cc07
fix(router): re-export Pipeline via alias to satisfy test monkeypatch; remove duplicate import
Browse files- app/routers/nl2sql.py +5 -1
app/routers/nl2sql.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
from dataclasses import asdict, is_dataclass
|
| 2 |
from fastapi import APIRouter, HTTPException, UploadFile, File
|
| 3 |
from app.schemas import NL2SQLRequest, NL2SQLResponse, ClarifyResponse
|
| 4 |
-
from nl2sql.pipeline import Pipeline, FinalResult
|
| 5 |
from nl2sql.ambiguity_detector import AmbiguityDetector
|
| 6 |
from nl2sql.safety import Safety
|
| 7 |
from nl2sql.planner import Planner
|
|
@@ -20,7 +20,11 @@ import json
|
|
| 20 |
import uuid
|
| 21 |
from typing import Union, Optional, Dict, TypedDict, Any, cast
|
| 22 |
|
|
|
|
|
|
|
|
|
|
| 23 |
__all__ = ["Pipeline", "FinalResult"]
|
|
|
|
| 24 |
router = APIRouter(prefix="/nl2sql")
|
| 25 |
|
| 26 |
# -------------------------------
|
|
|
|
| 1 |
from dataclasses import asdict, is_dataclass
|
| 2 |
from fastapi import APIRouter, HTTPException, UploadFile, File
|
| 3 |
from app.schemas import NL2SQLRequest, NL2SQLResponse, ClarifyResponse
|
| 4 |
+
from nl2sql.pipeline import Pipeline as _Pipeline, FinalResult as _FinalResult
|
| 5 |
from nl2sql.ambiguity_detector import AmbiguityDetector
|
| 6 |
from nl2sql.safety import Safety
|
| 7 |
from nl2sql.planner import Planner
|
|
|
|
| 20 |
import uuid
|
| 21 |
from typing import Union, Optional, Dict, TypedDict, Any, cast
|
| 22 |
|
| 23 |
+
# Re-export for tests & public API stability (pytest expects nl2sql.Pipeline)
|
| 24 |
+
Pipeline = _Pipeline
|
| 25 |
+
FinalResult = _FinalResult
|
| 26 |
__all__ = ["Pipeline", "FinalResult"]
|
| 27 |
+
|
| 28 |
router = APIRouter(prefix="/nl2sql")
|
| 29 |
|
| 30 |
# -------------------------------
|