from pydantic import BaseModel class QueryResponse(BaseModel): query_to_vector_database: str short_explanation: str