gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Move WebSocket to SDK pattern with /ws/ prefix - Moved to api/apps/sdk/websocket.py following session.py pattern - Added ws_token_required decorator - WebSocket endpoints: /ws/chats/<id>/completions and /ws/agents/<id>/completions - Prevents routing conflicts with HTTP endpoints

Browse source
This commit is contained in:
SmartDever02 2025-12-05 01:32:32 -03:00
parent 9ce780fefd
commit 081f7f7b74
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
api/apps/sdk/websocket.py
View file

@ -55,12 +55,13 @@ async def send_ws_message(data, code=0, message=""):
await websocket.send(json.dumps(response, ensure_ascii=False)) await websocket.send(json.dumps(response, ensure_ascii=False))
@manager.websocket("/chats/<chat_id>/completions") # noqa: F821 @manager.websocket("/ws/chats/<chat_id>/completions") # noqa: F821
@ws_token_required @ws_token_required
async def chat_completions_ws(tenant_id, chat_id): async def chat_completions_ws(tenant_id, chat_id):
""" """
WebSocket endpoint for streaming chat completions. WebSocket endpoint for streaming chat completions.
Follows the same pattern as the HTTP POST /chats/<chat_id>/completions endpoint. Follows the same pattern as the HTTP POST /chats/<chat_id>/completions endpoint.
Uses /ws/ prefix to avoid routing conflicts with HTTP endpoints.
""" """
# Verify chat ownership # Verify chat ownership
if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value): if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value):
@ -141,12 +142,13 @@ async def chat_completions_ws(tenant_id, chat_id):
logging.info(f"WebSocket chat connection closed for chat_id: {chat_id}") logging.info(f"WebSocket chat connection closed for chat_id: {chat_id}")
@manager.websocket("/agents/<agent_id>/completions") # noqa: F821 @manager.websocket("/ws/agents/<agent_id>/completions") # noqa: F821
@ws_token_required @ws_token_required
async def agent_completions_ws(tenant_id, agent_id): async def agent_completions_ws(tenant_id, agent_id):
""" """
WebSocket endpoint for streaming agent completions. WebSocket endpoint for streaming agent completions.
Follows the same pattern as the HTTP POST /agents/<agent_id>/completions endpoint. Follows the same pattern as the HTTP POST /agents/<agent_id>/completions endpoint.
Uses /ws/ prefix to avoid routing conflicts with HTTP endpoints.
""" """
# Verify agent ownership # Verify agent ownership
if not UserCanvasService.query(user_id=tenant_id, id=agent_id): if not UserCanvasService.query(user_id=tenant_id, id=agent_id):