python/mcp/yazzh.py - master - ksilisk/spbtechrun_hack

python/mcp/yazzh.py
117 строк2 KB
Shaliko Salimov
documents for RAG and supported polyclinics + dou
10 дек 2025, 20:51
10 дек 2025, 20:517cd073f
from __future__ import annotations

from typing import Optional

from pydantic import BaseModel


# =========================
# Общие вспомогательные модели
# =========================


class GeoPoint(BaseModel):
    """Простая точка на карте."""
    lat: float
    lon: float


# =========================
# AfishaClient
# =========================


class AfishaEvent(BaseModel):
    """Краткая карточка события Афиши."""
    title: str
    start_time: Optional[str] = None
    end_time: Optional[str] = None
    address: Optional[str] = None
    location_title: Optional[str] = None


# =========================
# BeautifulPlacesClient
# =========================


class BeautifulPlace(BaseModel):
    """Маршрут или точка из раздела Красивых мест."""
    title: str
    description: Optional[str] = None
    address: Optional[str] = None


# =========================
# SportEventsClient
# =========================


class SportEvent(BaseModel):
    """Короткая карточка спортивного события."""
    title: str
    description: Optional[str] = None
    start_date: Optional[str] = None
    start_time: Optional[str] = None
    address: Optional[str] = None


# =========================
# SportgroundsClient
# =========================


class Sportground(BaseModel):
    """Объект спортплощадки."""
    name: str
    address: Optional[str] = None


# =========================
# MyPetsClient
# =========================


class PetVetClinic(BaseModel):
    title: str
    address: Optional[str] = None


class PetShelter(BaseModel):
    title: str
    address: Optional[str] = None


class PetWalkingArea(BaseModel):
    title: str
    address: Optional[str] = None


# =========================
# MfcClient
# =========================


class MfcOffice(BaseModel):
    name: str
    nearest_metro: Optional[str] = None
    address: Optional[str] = None


class Building(BaseModel):
    id: int
    district: str
    address: str

class Dou(BaseModel):
    name: str
    building_id: str

class DouFillInfo(BaseModel):
    name: str
    address: str

class Polyclinic(BaseModel):
    address: str
    url: str
    name: str