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

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

from lambda_agent_api.agent_api import AgentApi, AgentException
from lambda_agent_api.server import ServerMessage
from lambda_agent_api.client import ClientMessage

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