Function handle_ws_chat 

pub async fn handle_ws_chat(
    __arg0: State<AppState>,
    __arg1: Query<WsQuery>,
    headers: HeaderMap,
    ws: WebSocketUpgrade,
) -> impl IntoResponse

GET /ws/chat — WebSocket upgrade for agent chat