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))
@manager.websocket("/chats/<chat_id>/completions") # noqa: F821
@manager.websocket("/ws/chats/<chat_id>/completions") # noqa: F821
@ws_token_required
async def chat_completions_ws(tenant_id, chat_id):
"""
WebSocket endpoint for streaming chat completions.
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
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}")
@manager.websocket("/agents/<agent_id>/completions") # noqa: F821
@manager.websocket("/ws/agents/<agent_id>/completions") # noqa: F821
@ws_token_required
async def agent_completions_ws(tenant_id, agent_id):
"""
WebSocket endpoint for streaming agent completions.
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
if not UserCanvasService.query(user_id=tenant_id, id=agent_id):