ja.0ddlmZmZmZmZddlmZmZddlZddl Z ddl Z ddl m Z ddl mZe jeZeZej'ddedefd ZddlZdd lmZmZGd d ZeZej9d edfdededefdZej9ddedefdZej9ddedefdZej9dedededfdedededededef dZ y)) APIRouter WebSocketWebSocketDisconnectQuery)ListDictN)save_chat_message)insert_feedbackz /ws-test/{tenant_id}/{thread_id} tenant_id thread_idc"Kdd|d|ddSw)Nokz)WebSocket router is mounted. Testing for /T)statusmessage path_verified)r r s 0/var/www/html/strategist-ai/app/api/websocket.py ws_diagnosticr s(>yk9+V s )get_redis_client publish_eventc reZdZdZdedefdZdedefdZdedefdZddeded efd Z dded eded e fd Z y)ConnectionManagerc i|_i|_yN)active_connections pubsub_tasks)selfs r__init__zConnectionManager.__init__s>@;= websocket channel_idc K|jd{||jvrg|j|<|j|j|tj d|dt |j|t j|j||}||j|<y7w)NzNew connection to channel z. Total connections: ) acceptrappendloggerinfolenasyncio create_task_listen_to_redisr)rr!r"tasks rconnectzConnectionManager.connects   T44 424D # #J /  +229= 0 B<B&B>c Kd}d} t}|j4d{}|j|d{tj d|d}|j 23d{}|ddk(s|d} |j |d{tj d|tjd|b777a7;#t$r7} tjd |d | Yd} ~ dddd{7yd} ~ wwxYw6dddd{7n#1d{7swYnxYwn#tj$rtj d |Yyt$rV} tjd |d | d |dtj|d{7t|dz|}Yd} ~ nd} ~ wwxYwͭw)N<zSubscribed to Redis channel: typerdataz/Successfully sent message to client on channel zForwarded Redis message to z&Error sending message to websocket in : z(Redis listen task cancelled for channel zRedis subscription error for z. Retrying in zs...)rpubsub subscriber&r'listen send_textdebug Exceptionerrorr)CancelledErrorsleepmin) rr!r" retry_delay max_delay redis_clientr5rr2es rr+z"ConnectionManager._listen_to_redis*s   >/1 '..0''F **:666KK"? | LM"#K)/ ' 'g"6?i7#*6?D'&/&9&9$&? ? ? & .]^h]i,j k & /J:,-W X'6 '!@$-' & /UV`Uaacdecf-g h &''''*9 '''''))  FzlST > +sGECED/C.D//D3C4D7 D/D/CC4CD/ED/DC D"D=D/ E DEGDD/ E(D+)E/E5D8 6E=EG+G0G2G:7G 1F42G G GGcV||jvrK||j|vr|j|j||j|s |j|=||jvr*|j|j|j|=tj d|y)NzDisconnected from channel )rremovercancelr&r')rr!r"s r disconnectzConnectionManager.disconnectHs 00 0D33J??'' 3::9E**:6++J7 )) )   i ( / / 1!!), 0 =>r Nr sender_socketcK tj|}t||d{y#tj$rd|i}Y.wxYw7"w)zR Broadcasts a message to all participants in a channel via Redis. textN)jsonloadsJSONDecodeErrorr)rrr"rGpayloads r broadcastzConnectionManager.broadcastVsQ (jj)GJ000## (w'G ( 1s/A .A A A AA AA  event_typer2cNKd|||xsid}t||d{y7w)zS Broadcasts a system-level event to all participants in a channel. system)r1eventrr2N)r)rr"rOrr2rMs rbroadcast_system_eventz(ConnectionManager.broadcast_system_eventas1 JB   J000s %#%r) __name__ __module__ __qualname__rrstrr-r+rFrNdictrSrr rrrs}> ,y ,c ,> >s>< ?I ?3 ? 1s 1 1I 1 1s 1 1VY 1ae 1r rz/ws/{tenant_id}/dashboardadminr!rolecfK|dk7r|jdd{yddlm} |}|j5}|j d|f|j st jd|d ddd|jd |d }tj||d{t jd| |jd{7#1swYrxYw#t$r%}t jd |d |Yd}~d}~wwxYw7v7F#t$r1tj||t jd|Yyt$r=}t jd|d |dtj||Yd}~yd}~wwxYww)ze Dedicated WebSocket for the Admin Dashboard to receive real-time organization-wide updates. rYi)codeNrget_db_connectionJSELECT schema_name FROM information_schema.schemata WHERE schema_name = %sz)Dashboard WebSocket connected to tenant '' which has NO database schema.#Failed to verify schema for tenant r3tenant_z :dashboardz9Admin Dashboard connected to real-time stream for tenant Tz!Admin Dashboard disconnected for z,Unexpected error in Dashboard WebSocket for exc_info)closeapp.services.infra.databaser^cursorexecutefetchoner&warningr:managerr-r' receive_textrrFr;)r!r rZr^conncurrBr"s rdashboard_websocket_endpointroos woo4o(((>O " [[] wc KKdgpfr s<<>!J9+Utuv w 9+Z0J //)Z 000 KKKI;WX 2((* * *+ ) w w OC;F15D,6F1D0'D.(D0-F1/C84C;; D)D$F1$D))F1.D007F.'F1)F.13F)$F1)F..F1z!/ws/{tenant_id}/conversation_listcKddlm} |}|j5}|jd|f|j st j d|dddd|jd|d }tj||d{t jd | |jd{#1swYpxYw#t$r%}t j d|d|Yd}~d}~wwxYw7t7D#t$rtj||Yyt$r8}t jd |tj||Yd}~yd}~wwxYww) Nrr]r_z1Conversation list WebSocket connected to tenant 'r`rar3rbz:conversation_listz1Connected to conversation_list stream for tenant z&Error in conversation_list WebSocket: rfr^rgrhrir&rjrer:rkr-r'rlrrFr;r!r r^rmrnrBr"s r$conversation_list_websocket_endpointrss^>O " [[] c KKdgpfr s<<>!RS\R]]|}~  9+%78J //)Z 000 KKCI;OP2((* * *   OO " [[] Xc KKdgpfr s<<>!@ LV WX X 9+X.J //)Z 000 KK9)EF2((* * * X X OBfn":Rf0@$()-'. % (!% &$&"& ,! ##$5zQZ#[ [ [s 1 2'' $#  $   G rsDD  A7   8 $  ./330DS1S1j  -.g&2&2&2 &2/&2P562)2PS2722+,2 2c2-22/0 f ;'; N2N2N2N2  N2  N2  N21N2r