server.cpython-310.pyc•3.13 kB
o
��g; � @ sF 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�e
j�e��Ze
j�e
j�e
j�e���Ze
j�edd�Ze
jed d
� e
j�edd �Ze
jed d
� ejd eed
�dd� eed
�Ze� Z d a e� d�dd� �Z e� d�dd� �Z e�!d�defdd��Z"de#fdd�Z$dS ) � )�FastAPI� WebSocket�WebSocketDisconnect)�Jinja2Templates)� StaticFiles)� HTMLResponse)�RequestNZmcpTest�staticT)�exist_ok� templatesz/static)� directory)�name�startupc � � �da td� d S )NTzWeb server started��server_running�print� r r �.D:\DeepSeek\MCPAgent\mcpTest\src\web\server.py�
startup_event! � � r �shutdownc � r )NFzWeb server shutting downr r r r r �shutdown_event' r r z/ws� websocketc
� s �| � � I d H t�| � tdtt�� �� z�z�| �t�ddd���I d H z�| �� I d H }t� |�}td|� �� |�
d�dkr�z7d d
l m } |red|d d i|d |d
dd�}|�
|�I d H ntd� ddddid�}t|�I d H W n- ty� } z!td|� �� ddddt|�� �id�}t|�I d H W Y d }~nd }~ww W n ty� } z
td|� �� W Y d }~nd }~ww q'W n ty� } z
td|� �� W Y d }~nd }~ww W | tv r�t�| � tdtt�� �� d S | tv r�t�| � tdtt�� �� w )Nu, WebSocket连接已建立，当前连接数: �status� connected)�type�contentTu 收到客户端消息: r �textr )�brain_instancer � sender_id� receiver_id)r r r! � message_typeu 未找到大脑智能体实例Zchat�systemu- 大脑智能体未启动，无法处理消息)r r r u 处理大脑消息时出错: u 处理消息时出错: u 处理WebSocket消息时出错: u WebSocket错误: u, WebSocket连接已关闭，剩余连接数: )�accept�websocket_connections�addr �len� send_text�json�dumps� receive_text�loads�getZsrc.brain.brain_agentr Z_handle_text_message�broadcast_message� Exception�str�remove)r �data�messager Z
brain_message�
error_message�er r r �websocket_endpoint. sz �
�
��������� ���2���
�
r6 r3 c � s� �t sdS d}tt �D ]3}z|�t�| ��I dH d}W q ty> } ztd|� �� |t v r4t �|� W Y d}~q d}~ww |S )u$ 广播消息到所有WebSocket连接FNTu 发送消息失败: )r% �listr( r) r* r/ r r1 )r3 Zsuccess�
connectionr5 r r r r. w s �
��r. )%�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 r6 �dictr. r r r r �<module> s4
H