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

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

from agent_api import AgentApi, AgentException
from models import IM, OM, IncomingMessage, OutgoingMessage

__all__ = [
    "AgentApi",
    "AgentException",
    "IM",
    "OM",
    "IncomingMessage",
    "OutgoingMessage",
]