Rifqi Hafizuddin
[KM-512] add Pydantic model the LLM fills via function calling in sql_query, and add same signature for db and tabular
220f59e
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.")