"""Pydantic-модели для входных параметров и выходных результатов инструментов.""" from datetime import date from typing import Literal from pydantic import BaseModel, Field class PaginationParams(BaseModel): """Базовая пагинация.""" limit: int = Field( default=20, ge=1, le=100, description=( "Количество результатов на странице. " "Если total > limit, используй offset+=limit для следующей страницы." ), ) offset: int = Field( default=0, ge=0, description="Смещение для пагинации. Следующая страница: offset += limit.", ) class MetadataFilters(BaseModel): """Фильтры по метаданным, общие для всех инструментов поиска. Все фильтры объединяются через AND. Если результатов мало — попробуй убрать 1–2 фильтра (обычно district или metro_station). """ deal_type: Literal["sale", "rent_long", "rent_short"] | None = Field( default=None, description=( "Тип сделки: sale=продажа, rent_long=долгосрочная аренда, " "rent_short=посуточная" ), ) city: str | None = Field( default=None, description=( "Город на украинском языке. Поиск по подстроке (ILIKE) — " "достаточно части названия, например 'Київ' или 'Львів'. Регистр не важен." ), ) district: str | None = Field( default=None, description=( "Район на украинском языке. Поиск по подстроке (ILIKE). " "Примеры: 'Печерський', 'Шевченківський', 'Галицький'." ), ) micro_district: str | None = Field( default=None, description="Микрорайон. Поиск по подстроке (ILIKE).", ) street: str | None = Field( default=None, description="Улица. Поиск по подстроке (ILIKE). Пример: 'вул. Шевченка'.", ) rooms_count: int | None = Field( default=None, ge=0, description="Количество комнат. 0 = студия, 1 = 1-комнатная и т.д.", ) bedrooms_count: int | None = Field( default=None, ge=0, description="Количество спален.", ) bathrooms_count: int | None = Field( default=None, ge=0, description="Количество ванных комнат.", ) min_price: float | None = Field( default=None, ge=0, description=( "Минимальная цена. Обязательно указывай currency вместе с min_price/max_price, " "иначе фильтр применится ко всем валютам одновременно." ), ) max_price: float | None = Field( default=None, ge=0, description="Максимальная цена. Указывай вместе с currency.", ) currency: Literal["USD", "EUR", "UAH"] | None = Field( default=None, description="Валюта цены. Обязательна при использовании min_price/max_price.", ) min_total_area: float | None = Field( default=None, ge=0, description="Минимальная общая площадь, м².", ) max_total_area: float | None = Field( default=None, ge=0, description="Максимальная общая площадь, м².", ) building_type: Literal["brick", "panel", "monolith", "gas_block", "wood"] | None = Field( default=None, description=( "Тип постройки: brick=кирпичный, panel=панельный, monolith=монолитный, " "gas_block=газоблок, wood=деревянный." ), ) floor: int | None = Field( default=None, ge=0, description="Этаж квартиры. 0 обычно означает цоколь/подвал (если есть в базе).", ) floors_total: int | None = Field( default=None, ge=0, description="Общее количество этажей в здании.", ) listing_status: Literal["active", "sold", "rented", "removed", "archived"] | None = Field( default=None, description="Статус объявления. По умолчанию сервер не фильтрует — включая все статусы.", ) metro_station: str | None = Field( default=None, description=( "Станция метро на украинском. Поиск по подстроке (ILIKE). " "Примеры: 'Арсенальна', 'Театральна', 'Площа Ринок'. " "Регистр не важен." ), ) metro_distance_type: Literal["walk", "transport"] | None = Field( default=None, description="Удалённость от метро: walk=пешком, transport=на транспорте.", ) metro_distance_min: int | None = Field( default=None, ge=0, description="Время до метро в минутах.", ) layout: Literal["studio", "separate", "adjacent"] | None = Field( default=None, description="Тип планировки: studio=студия, separate=раздельная, adjacent=смежная.", ) renovation_status: ( Literal["cosmetic", "euro", "designer", "none", "construction"] | None ) = Field( default=None, description="Состояние ремонта.", ) bathroom_type: Literal["combined", "separate", "multiple"] | None = Field( default=None, description="Тип санузла: combined=совмещённый, separate=раздельный, multiple=несколько.", ) parking_type: Literal["ground", "underground", "none", "garage"] | None = Field( default=None, description="Тип парковки.", ) heating_type: Literal["central", "autonomous", "floor", "none"] | None = Field( default=None, description=( "Тип отопления: central=центральное, autonomous=автономное, " "floor=подогрев полов, none=нет." ), ) window_view: Literal["yard", "street", "park", "water", "forest"] | None = Field( default=None, description="Вид из окон.", ) has_balcony: bool | None = Field( default=None, description="Наличие балкона.", ) has_loggia: bool | None = Field( default=None, description="Наличие лоджии.", ) class SearchSimilarInput(BaseModel): """Входные параметры для векторного поиска. Используй, когда пользователь описывает желания, атмосферу или качества («уютная квартира с ремонтом у метро»), а не конкретные ключевые слова. """ query: str = Field( ..., description=( "Текстовый запрос на украинском языке для поиска по смыслу. " "Формулируй конкретно, без разговорных стоп-слов. " "✓ Хорошо: '2-кімнатна квартира біля метро з ремонтом'. " "✗ Плохо: 'дай квартиру недорого' (стоп-слова)." ), ) filters: MetadataFilters = Field(default_factory=MetadataFilters) pagination: PaginationParams = Field(default_factory=PaginationParams) # min_similarity зашит в код всегда 0.7 (широкий поиск) class SearchMetadataInput(BaseModel): """Входные параметры для поиска по метаданным/FTS. Используй, когда пользователь называет конкретные ключевые слова: район, станция метро, улица, особенности ('Печерський район метро Арсенальна'). """ query: str = Field( ..., description=( "Текстовый запрос на украинском языке для полнотекстового поиска. " "Подходит для конкретных названий и ключевых слов. " "✓ Хорошо: 'Печерський район Арсенальна метро'. " "✗ Плохо: 'хочу квартиру' (слишком общее, используй search_similar)." ), ) filters: MetadataFilters = Field(default_factory=MetadataFilters) pagination: PaginationParams = Field(default_factory=PaginationParams) sort_by: Literal[ "relevance", "price_asc", "price_desc", "date_desc", "area_desc", ] = Field( default="relevance", description=( "Сортировка результатов: relevance=по релевантности FTS (умолч.), " "price_asc=цена по возрастанию, price_desc=цена по убыванию, " "date_desc=новые сверху, area_desc=большая площадь сверху." ), ) class GetListingInput(BaseModel): """Входные параметры для получения объявления по ID.""" listing_id: int = Field(..., ge=1, description="ID объявления (положительное целое число).") class ListingResult(BaseModel): """Результат поиска — одно объявление.""" id: int title: str description: str | None = None generated_description: str | None = None price: float | None = None currency: str | None = None original_price: float | None = None original_currency: str | None = None price_per_sqm: float | None = None deal_type: str | None = None city: str | None = None district: str | None = None micro_district: str | None = None street: str | None = None house_number: str | None = None address_raw: str | None = None rooms_count: int | None = None bedrooms_count: int | None = None bathrooms_count: int | None = None total_area: float | None = None living_area: float | None = None kitchen_area: float | None = None land_area: float | None = None floor: int | None = None floors_total: int | None = None building_type: str | None = None building_year: int | None = None renovation_status: str | None = None ceiling_height: float | None = None material: str | None = None balcony_count: int | None = None loggia_count: int | None = None has_balcony: bool | None = None has_loggia: bool | None = None bathroom_type: str | None = None elevator_count: int | None = None has_freight_elevator: bool | None = None parking_type: str | None = None heating_type: str | None = None internet: bool | None = None security: bool | None = None layout: str | None = None window_view: str | None = None windows_direction: str | None = None metro_station: str | None = None metro_distance_type: str | None = None metro_distance_min: int | None = None latitude: float | None = None longitude: float | None = None url_source: str | None = None publish_date: date | None = None images_count: int | None = None contact_phone: str | None = None contact_name: str | None = None contact_email: str | None = None is_agent: bool | None = None agency_name: str | None = None listing_status: str | None = None listing_quality_score: int | None = None reliability_rating: int | None = None sentiment_score: float | None = None archived_at: date | None = None created_at: date | None = None updated_at: date | None = None similarity_score: float | None = Field( default=None, description="Косинусная близость [0..1] для search_similar_listings. " "1.0 = максимально похоже, 0.0 = не похоже.", ) rank_score: float | None = Field( default=None, description="Релевантность FTS (ts_rank_cd) для search_by_metadata. " "Чем выше, тем релевантнее.", ) class SearchResult(BaseModel): """Результат поиска — список объявлений + мета. Если total > limit и offset + limit < total — есть ещё результаты. Для следующей страницы: offset += limit. """ total: int = Field(description="Общее количество найденных записей (без LIMIT).") limit: int offset: int listings: list[ListingResult]