
    Xj                         d dl mZmZmZ d dlmZ d dlmZ  ee          Z	 e            Z
e
                    d          dedefd            ZdS )	    )	APIRouter	WebSocketWebSocketDisconnect)manager)
get_loggerz/ws/training/{task_id}	websockettask_idc                   K   t          j        ||            d {V  t                              |                              d           	 	 |                                  d {V  # t          $ rM t                              |                              d           t          j        ||            d {V  Y d S w xY w)N)r	   zWebSocket connectedTzWebSocket disconnected)r   connectloggerbindinforeceive_textr   
disconnect)r   r	   s     F/lsinfo/ai/hellotax_ai/training_center/backend/app/api/v1/websocket.pywebsocket_endpointr      s      
/'9
-
--------
KKK  %%&;<<<5	+((*********	+ 5 5 5G$$))*BCCC )4444444444445s   A) )AC ?C N)fastapir   r   r   app.core.websocket_managerr   common_loggingr   __name__r   routerr   strr        r   <module>r      s    = = = = = = = = = = . . . . . . % % % % % %	H			*++5	 5C 5 5 5 ,+5 5 5r   