�
��g; � �* � d dl mZmZmZ d dlmZ d dlmZ d dlm Z d dl m
Z
d dl Z d dl Z d dl
Z
e� Ze
j j! e
j j# e� � Ze
j j! e
j j! e
j j! e� � � Ze
j j+ edd� Z e
j. ed �
� e
j j+ edd � Z e
j. ed �
� ej3 d ee�
� d�� ee�
� Z e� Z d a ej= d� d� � Z ej= d� d� � Z ejC d� defd�� Z"de#fd�Z$y)� )�FastAPI� WebSocket�WebSocketDisconnect)�Jinja2Templates)� StaticFiles)� HTMLResponse)�RequestN�mcpTest�staticT)�exist_ok� templatesz/static)� directory)�name�startupc � �& K � da t d� y �w)NTzWeb server started��server_running�print� � �&D:\AIProject\mcpTest\src\web\server.py�
startup_eventr ! s � �� � �N� �
� �� ��shutdownc � �& K � da t d� y �w)NFzWeb server shutting downr r r r �shutdown_eventr ' s � �� � �N� �
$�%�r z/ws� websocketc � � K � | j � � d { ��� t j | � t dt t � � �� | j t
j ddd�� � � d { ��� | j � � d { ��� }t
j |� }t d|� �� |j d� dk( r^ dd l m } |r/d|d
d i|d |d dd
�}|j |� � d { ��� n&t d� ddddid�}t |� � d { ��� ��7 ��7 ��7 ��7 �37 �# t $ r@}t d|� �� ddddt! |� � �id�}t |� � d { ���7 Y d }~�Pd }~ww xY w# t $ r}t d|� �� Y d }~nd }~ww xY wn$# t $ r}t d|� �� Y d }~nd }~ww xY w| t v rt j# | � t dt t � � �� y # | t v rt j# | � t dt t � � �� w xY w�w)Nu, WebSocket连接已建立，当前连接数: �status� connected)�type�contentu 收到客户端消息: r! �textr )�brain_instancer" � sender_id� receiver_id)r" r% r& � message_typeu 未找到大脑智能体实例�chat�systemu- 大脑智能体未启动，无法处理消息)r! r% r" u 处理大脑消息时出错: u 处理消息时出错: u 处理WebSocket消息时出错: u WebSocket错误: u, WebSocket连接已关闭，剩余连接数: )�accept�websocket_connections�addr �len� send_text�json�dumps� receive_text�loads�get�src.brain.brain_agentr$ �_handle_text_message�broadcast_message� Exception�str�remove)r �data�messager$ �
brain_message�
error_message�es r �websocket_endpointr? . sg � �� �
�
�
�
� � �� � �i�(� � 8��=R�9S�8T�
U�V�A[��!�!�$�*�*� �"�.
� # � � � � �1
� &� 3� 3� 5�5�� �*�*�T�*���/��y�9�:� �;�;�v�&�&�0�%?�H�)� -3�G�I�4F�v�4N�+O�-4�[�-A�/6�}�/E�06� - �M� #1�"E�"E�m�"T� T� T� "�"B� C� )/�-5�$*�,[�,"�- �M� #4�M�"B� B� B�G � �� �� 6��( U�� C��$�
?� � =�a�S� A�B� %+�)1� &�*A�#�a�&��(J�( �)�
� 0�
� >�>�>��
?�� �
��8���<�=���
���� � '�
�!�!��%�&�&��'�� �-� -� !� (� (�� 3�
�<�S�AV�=W�<X�Y�Z�� �-� -� !� (� (�� 3�
�<�S�AV�=W�<X�Y�Z�s� �I�D+�4I� *F' �6D.�7F' �=F �D0�;F �
1D6 �>D2�?&D6 �%D4�&D6 �*F' �+I�.F' �0F �2D6 �4D6 �6 E?�?0E:�/E2�0E:�5F �:E?�?F � F#� F �F' � F#�#F' �&H �' G�0G�>H �G�H � 9I�:H>�>Ir; c � �, K � t syd}t t � D ]1 } |j t j | � � � d{ ��� d}�3 |S 7 �
# t
$ r5}t
d|� �� |t v rt j |� Y d}~�qd}~ww xY w�w)u$ 广播消息到所有WebSocket连接FNTu 发送消息失败: )r+ �listr. r/ r0 r7 r r9 )r; �success�
connectionr> s r r6 r6 w s� � �� � �� �G� �0�1� 9�
� 9� �&�&�t�z�z�'�':�;� ;� ;��G�9� �N�
<��� 9� �(���,� -�� 2�2�%�,�,�Z�8��� 9�s@ � B�'A�A�A�
B�A� B� +B �B� B�B)%�fastapir r r �fastapi.templatingr �fastapi.staticfilesr �fastapi.responsesr r �asyncior/ �os�app�path�dirname�abspath�__file__� current_dir�base_dir�join�
static_dir�makedirs�
templates_dir�mountr
�setr+ r �on_eventr r r r? �dictr6 r r r �<module>rY sc �� ;� ;� .� +� *� � � � �
�i�� �g�g�o�o�b�g�g�o�o�h� 7�8� �
�7�7�?�?�2�7�7�?�?�2�7�7�?�?�;�+G�H� I�� �W�W�\�\�(�I�x�
8�
� �� � �J�� &� ��� � �X�y�+�>�
� �� � �M�D� )� � � �)�[�:�6�X� � F� �m� 4� � �� ���� � �i�� � � �
� � �j��&� �&� � � �u��F[� � F[� �F[�P�T� r
