o
    "i                     @   sF   d dl mZmZmZ d dlmZ e ZeddedefddZ	dS )	    )	APIRouter	WebSocketWebSocketDisconnect)managerz/ws/training/{task_id}	websockettask_idc                    sN   t || I d H  z		 |  I d H  q ty&   t || I d H  Y d S w )N)r   connectreceive_textr   
disconnect)r   r    r   F/lsinfo/ai/hellotax_ai/training_center/backend/app/api/v1/websocket.pywebsocket_endpoint   s   r   N)
fastapir   r   r   app.core.websocket_managerr   routerr   strr   r   r   r   r   <module>   s
    