корректная обработка ChatBusyError для WS эндпоинта. Docstring

This commit is contained in:
Егор Кандрушин 2026-04-19 15:48:34 +03:00
parent 69ec28037a
commit 58494ddea6
5 changed files with 52 additions and 15 deletions

View file

@ -11,16 +11,17 @@ from lambda_agent_api.server import (
from lambda_agent_api.client import ClientMessage, MsgUserMessage
from src.agent import AgentChat
from src.api.dependencies import get_chat
from src.api.dependencies import get_chat_ws
router = APIRouter()
@router.websocket("/agent_ws/{chat_id}/")
@router.websocket("/v1/agent_ws/{chat_id}/")
async def websocket_endpoint(
ws: WebSocket,
chat: Annotated[AgentChat, Depends(get_chat)],
# важно использовать именно _ws вариант, чтобы корректно обрабатывались исключения
chat: Annotated[AgentChat, Depends(get_chat_ws)],
):
await ws.accept()
await ws.send_text(MsgStatus().model_dump_json())