Spaces:
Sleeping
Sleeping
File size: 853 Bytes
eee3f75 570f7bd c1bc4eb 570f7bd 370553a 570f7bd c1bc4eb 570f7bd c1bc4eb 570f7bd a337fad 570f7bd eee3f75 570f7bd c1bc4eb 570f7bd c1bc4eb 570f7bd |
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 36 37 38 |
from pydantic import BaseModel, Field
from typing import List, Optional, Any, Dict, Mapping, Sequence
class NL2SQLRequest(BaseModel):
query: str
db_id: Optional[str] = None
schema_preview: Optional[str] = None
class Config:
extra = "ignore"
class TraceModel(BaseModel):
stage: str
duration_ms: float
token_in: int | None = 0
token_out: int | None = 0
cost_usd: float | None = 0
notes: Dict[str, Any] | None = None
class NL2SQLResponse(BaseModel):
ambiguous: bool = False
sql: Optional[str] = None
rationale: Optional[str] = None
traces: Sequence[TraceModel | Mapping[str, Any]] = Field(default_factory=list)
class ClarifyResponse(BaseModel):
ambiguous: bool = True
questions: List[str]
class ErrorResponse(BaseModel):
error: str
details: List[str] | None = None
|