from pydantic import BaseModel, Field
from typing import Optional

class HouseFeatures(BaseModel):
    square_feet: float = Field(..., description="Total square feet of the house")
    bedrooms: int = Field(..., description="Number of bedrooms")
    bathrooms: float = Field(..., description="Number of bathrooms")
    number_of_floors: Optional[int] = Field(default=None, description="Number of floors")