models.py

from pydantic import BaseModel
from typing import List

class QueryInput(BaseModel):
    query: str

class QueryOutput(BaseModel):
    response: str
    sql_query: List[str]