qi'dZddlZddlmZmZddlmZmZmZddl Z ddl m Z ddl m Z mZddlmZddlmZdd lmZGd d eZy) zFBase websocket service with automatic reconnection and error handling.N)ABCabstractmethod) AwaitableCallableOptional)logger)ConnectionClosedErrorConnectionClosedOK)State) ErrorFrame)exponential_backoff_timec eZdZdZdddefdZdefdZdedefd Z dd ed e e e ge d fdefd Z d e e ge d ffdZ dded e e ge d fde edefdZd e e ge d ffdZdZdZedZedZedZy )WebsocketServicea+Base class for websocket-based services with automatic reconnection. Provides websocket connection management, automatic reconnection with exponential backoff, connection verification, and error handling. Subclasses implement service-specific connection and message handling logic. T)reconnect_on_errorrc <d|_||_d|_d|_y)zInitialize the websocket service. Args: reconnect_on_error: Whether to automatically reconnect on connection errors. **kwargs: Additional arguments (unused, for compatibility). NF) _websocket_reconnect_on_error_reconnect_in_progress_disconnecting)selfrkwargss T/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/websocket_service.py__init__zWebsocketService.__init__s%IM#5 ,1#$)returncK |jr&|jjtjury|jj d{y7#t $r$}t j|d|Yd}~yd}~wwxYww)zVerify the websocket connection is active and responsive. Returns: True if connection is verified working, False otherwise. FNTz! connection verification failed: )rstater CLOSEDping Exceptionrerror)res r_verify_connectionz#WebsocketService._verify_connection*sr  ??doo&;&;u||&K//&&( ( ( )  LLD6!B1#F G sJB 2AB AAAB A B %B?B B  B attempt_numbercKtj|d|d|jd{|jd{|j d{S7477w)zReconnect the websocket with the current attempt number. Args: attempt_number: Current retry attempt number for logging. Returns: True if reconnection and verification successful, False otherwise. z reconnecting (attempt: )N)rwarning_disconnect_websocket_connect_websocketr#)rr$s r_reconnect_websocketz%WebsocketService._reconnect_websocket9sf $77GqIJ((***%%''',,... +'.s3/A,A&A, A( A,!A*"A,(A,*A,N max_retries report_errorc K|jrtj|dyd|_d} td|dzD]} tj|d||j |d{r#tj |d|d|_y t|}tj|d{|d |d }|r|d |z }tj||r|t|d d{ d|_y7#t $rM}|}tj|d|d||r"|t|d|d|d{7Yd}~d}~wwxYw77j#d|_wxYww) Nz/ reconnect attempt aborted: already in progressFTz reconnecting, attempt z% reconnected successfully on attempt z reconnection attempt z failed: z failed to reconnect after z attemptsz: )fatal) rrr'ranger*infor r!r r asynciosleep)rr+r,last_exceptionattemptr" wait_time fatal_msgs r_try_reconnectzWebsocketService._try_reconnectGs  & & NNdV#RS T&*#.2 0 K!O4 / NNdV+B7)#LM!66w??? tf,QRYQZ$[\#$+0D ')@5W= mmI... / & ;K= RIr.!122 LL #":it#DEEE*/D ')@!%&NLLD6)?y RSQT!UV#*&$/EgYiXYWZ'[\ / F+0D 's/E>E2.D3D4DE2E>$E2?E.AE2E0E2 E>D E+=E&EE&!E2&E++E20E22 E;;E>cK |jj|d{y7#t$r}tj|d|d|j |d{7}|r=tj |d|jj|d{7ntj|dYd}~yYd}~yd}~wwxYww)zAAttempt to send a message, retrying after reconnect if necessary.Nz send failed: z, will try to reconnectr,z6 reconnected successfully, will retry send the messagez! send failed; unable to reconnect)rsendr rr!r8r1)rmessager,r"successs rsend_with_retryz WebsocketService.send_with_retryms I//&&w/ / / I LLD6s2IJ K //\/JJJG tf$Z[\oo**7333 v%FGHH4 IsNC*(*C* C0C #A&$=C !B$"C  C CC error_messager!cDK|jr6|rtj|d|ytj|dytj||jr|j |d{}|S|t |d{y7 7w)aCheck if reconnection should be attempted and try if appropriate. Args: error_message: Human-readable error message for logging. report_error: Callback function to report connection errors. error: The exception that occurred (optional, may be None for graceful closes). Returns: True if should continue the receive loop, False if should break. z error during disconnect: z% receive loop ended during disconnectFr:N)rrr'debugrr8r )rr?r,r!r=s r_maybe_try_reconnectz%WebsocketService._maybe_try_reconnect|s"   $'A%IJ v%JKL }%  # # //\/JJGNz-89 9 9 K :s$A9B ;B<B BB B cK |jd{|d}|j||d{}|sy =7(7 #t$r$}tj|d|Yd}~yd}~wt $r4}|d|}|j|||d{7}|sYd}~yYd}~hd}~wt $r4}|d|}|j|||d{7}|sYd}~yYd}~d}~wwxYww)alHandle websocket message receiving with automatic retry logic. Continuously receives messages with automatic reconnection on errors. Uses exponential backoff between retry attempts and reports fatal errors after maximum retries are exhausted. Args: report_error: Callback function to report connection errors. Nz connection closed by serverz connection closed normally: z' connection closed, but with an error: z error receiving messages: )_receive_messagesrBr rrAr r )rr,r<should_continuer"s r_receive_task_handlerz&WebsocketService._receive_task_handlers ,,... "F">?(,(A(A'<(X"X&'. #Y&  v%B1#FG( "F"I!M(,(A(A'strr rBrFrJrMrr)r(rDrOrrrrsA6: *d * $  / / / JN$0$0x io(EFG$0  $0L I8ZLR[\`RaDa;b I&&* "" |Yt_<=" " "  "H%*yY]A^8_%N$#      rr)rXr2abcrrtypingrrr websocketslogururwebsockets.exceptionsr r websockets.protocolr pipecat.frames.framesr pipecat.utils.networkr rrOrrrds7M#00K%,:^ s^ r