File size: 226 Bytes
a360e3c
 
 
 
 
 
1
2
3
4
5
6
7
from pydantic import BaseModel, Field


class SQLQueryModel(BaseModel):
    sql_query: str = Field(..., description="SQL query to execute.")
    explanation: str = Field(..., description="Short explanation of the SQL query.")