"""WebSocket streaming routes.

⚠️ AUTO-GENERATED BY PRISM - DO NOT EDIT
"""

from __future__ import annotations

from fastapi import APIRouter, Query, WebSocket

from .ws_handler import handle_websocket

router = APIRouter(tags=["streaming"])


@router.websocket("{{ ws_path }}/events")
async def ws_all_events(websocket: WebSocket) -> None:
    """WebSocket endpoint for all model events."""
    await handle_websocket(websocket)
{% for model_name in model_names %}


@router.websocket("{{ ws_path }}/{{ model_name | lower }}/events")
async def ws_{{ model_name | lower }}_events(websocket: WebSocket) -> None:
    """WebSocket endpoint for {{ model_name }} events."""
    await handle_websocket(websocket, model="{{ model_name }}")
{% endfor %}


__all__ = ["router"]
