корректная обработка ChatBusyError для WS эндпоинта. Docstring
This commit is contained in:
parent
69ec28037a
commit
58494ddea6
5 changed files with 52 additions and 15 deletions
|
|
@ -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())
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue