from pydantic import BaseModel
from typing import Optional

class RealEstateQuery(BaseModel):
    """부동산 검색 쿼리 모델"""
    text: str

class ParsedRealEstate(BaseModel):
    """파싱된 부동산 정보 모델"""
    price: Optional[str] = None
    location: Optional[str] = None
    area: Optional[str] = None
    rooms: Optional[int] = None
    transaction_type: Optional[str] = None  # 전세, 월세, 매매
    property_type: Optional[str] = None  # 아파트, 오피스텔, 주택, 빌라 등
    region_code: Optional[str] = None  # 시군구 코드 (5자리)
    region_name: Optional[str] = None  # 매칭된 정식 지역명
    raw_text: str