Build lightweight AI agent admin

2026-06-08 18:14:59 +08:00
commit e164840f43
2530 changed files with 435693 additions and 0 deletions
@@ -0,0 +1,108 @@
+# -*- coding: utf-8 -*-
+"""
+WebSocket 路由
+定义 WebSocket 端点
+"""
+from fastapi import APIRouter, WebSocket
+
+from core.websocket.consumers import (
+    TestWebSocketConsumer,
+    NotificationConsumer,
+    ServerMonitorConsumer,
+    RedisMonitorConsumer,
+    DatabaseMonitorConsumer,
+)
+from core.chat.ws import ChatConsumer
+
+router = APIRouter(redirect_slashes=False)
+
+
+@router.websocket("/ws/test")
+async def websocket_test(websocket: WebSocket):
+    """WebSocket测试连接"""
+    consumer = TestWebSocketConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/notifications")
+async def websocket_notifications(websocket: WebSocket):
+    """通知推送连接"""
+    consumer = NotificationConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/notification")
+async def websocket_notification(websocket: WebSocket):
+    """通知推送连接（兼容路径）"""
+    consumer = NotificationConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/notification/")
+async def websocket_notification_slash(websocket: WebSocket):
+    """通知推送连接（带斜杠兼容）"""
+    consumer = NotificationConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/notifications/")
+async def websocket_notifications_slash(websocket: WebSocket):
+    """通知推送连接（复数带斜杠兼容）"""
+    consumer = NotificationConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/server-monitor")
+async def websocket_server_monitor(websocket: WebSocket):
+    """服务器监控连接"""
+    consumer = ServerMonitorConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/server-monitor/")
+async def websocket_server_monitor_slash(websocket: WebSocket):
+    """服务器监控连接（带斜杠兼容）"""
+    consumer = ServerMonitorConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/redis-monitor")
+async def websocket_redis_monitor(websocket: WebSocket):
+    """Redis监控连接"""
+    consumer = RedisMonitorConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/redis-monitor/")
+async def websocket_redis_monitor_slash(websocket: WebSocket):
+    """Redis监控连接（带斜杠兼容）"""
+    consumer = RedisMonitorConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/database-monitor")
+async def websocket_database_monitor(websocket: WebSocket):
+    """数据库监控连接"""
+    consumer = DatabaseMonitorConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/database-monitor/")
+async def websocket_database_monitor_slash(websocket: WebSocket):
+    """数据库监控连接（带斜杠兼容）"""
+    consumer = DatabaseMonitorConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/chat")
+async def websocket_chat(websocket: WebSocket):
+    """聊天连接"""
+    consumer = ChatConsumer(websocket)
+    await consumer.run()
+
+
+@router.websocket("/ws/chat/")
+async def websocket_chat_slash(websocket: WebSocket):
+    """聊天连接（带斜杠兼容）"""
+    consumer = ChatConsumer(websocket)
+    await consumer.run()