from pydantic import BaseModel

class RunQueryRequest(BaseModel):
    query: str

class ValidateQueryRequest(BaseModel):
    user_query: str
    expected_query: str