ishaq101's picture
Merge dev_new to main (improve retriever and add querying) (#13)
52999bc
raw
history blame
313 Bytes
"""Structured output model for LLM-generated SQL queries."""
from pydantic import BaseModel, Field
class SQLQuery(BaseModel):
sql: str = Field(description="A single SQL SELECT statement. No markdown, no explanation inline.")
reasoning: str = Field(description="One sentence: what this query answers.")