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
Files changed (1) hide show
  1. 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
  # -------------------------------