File size: 313 Bytes
52999bc
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
"""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.")