"""
Библиотека для взаимодействия с AI-агентом через WebSocket.

Экспортирует:
    - AgentApi: Основной класс клиента
    - AgentException: Исключение для ошибок агента
    - IM, OM, IncomingMessage, OutgoingMessage: Pydantic модели контракта
"""

from .agent_api import AgentApi, AgentException
from .models import IM, OM, ClientMessage, ServerMessage

__all__ = [
    "AgentApi",
    "AgentException",
    "IM",
    "OM",
    "ClientMessage",
    "ServerMessage",
]