"""
WebSocket module for AiCippy.

Provides real-time communication with the AiCippy backend
via API Gateway WebSocket.
"""

from __future__ import annotations

from aicippy.websocket.client import WebSocketClient
from aicippy.websocket.models import (
    WebSocketMessage,
    MessageType,
    AgentUpdate,
    ToolOutput,
)

__all__ = [
    "WebSocketClient",
    "WebSocketMessage",
    "MessageType",
    "AgentUpdate",
    "ToolOutput",
]
