8 lines
404 B
Python
8 lines
404 B
Python
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") |