mcp_protocol.cpython-310.pyc•4.21 kB
o
$ �g� � @ s� d Z ddlZddlZddlmZ ddlmZmZmZmZ G dd� d�Z G dd� de �Z
G d d
� d
e �Z G d d � d e �Z G d
d� de �Z
G dd� de �ZdS )u
MCP协议实现
� N)�datetime)�Dict�Any�Optional�Listc @ s� e Zd ZdZ ddedededeeef dee f
dd �Zd
eeef fd d �Z d
efd
d�Z
e deeef d
d fdd��Z e ded
d fdd��Z
dS )�
MCPMessageu MCP协议消息基类N� message_type� sender_id� receiver_id�content�
message_idc C s< |pt t�� �| _|| _|| _|| _|| _t� � �
� | _ d S �N) �str�uuid�uuid4r r r r
r r �now� isoformat� timestamp)�selfr r r
r r � r �6D:\DeepSeek\MCPAgent\mcpTest\src\utils\mcp_protocol.py�__init__ s zMCPMessage.__init__�returnc C s | j | j| j| j| j| jdd�S )u 将消息转换为字典zMCP/1.0)r r r r
r r �protocol)r r r r
r r �r r r r �to_dict s �zMCPMessage.to_dictc C s t �| �� �S )u 将消息转换为JSON字符串)�json�dumpsr r r r r �to_json% s zMCPMessage.to_json�datac C s( | |d |d |d |d |� d�d�S )u 从字典创建消息r r r
r r �r r r
r r )�get)�clsr r r r � from_dict) s �zMCPMessage.from_dict�json_strc C s t �|�}| �|�S )u 从JSON字符串创建消息)r �loadsr# )r" r$ r r r r � from_json4 s
zMCPMessage.from_jsonr
)�__name__�
__module__� __qualname__�__doc__r r r r r r r � classmethodr# r& r r r r r s( ����
�
�
r c
sN e Zd ZdZd dedededee f� fdd�
Zd eeef fd
d �Z � Z
S )
� TextMessageu 文本消息Nr r
�textr c s t � jd||d|i|d� d S )Nr- r ��superr )r r r
r- r �� __class__r r r > s
�zTextMessage.__init__r c C s | j | j| j| jd�S )N�r r
r r r2 r r r r r F s
�zTextMessage.to_dictr
) r' r( r) r* r r r r r r �
__classcell__r r r0 r r, ; s $ r, c �> e Zd ZdZ d dededededee f
� fd d
�
Z� ZS ) � ImageMessageu 图像消息�base64Nr r
�
image_data�formatr c � t � jd||||d�|d� d S )N�image)r7 r8 r r. )r r r
r7 r8 r r0 r r r R �
�zImageMessage.__init__�r6 N�r' r( r) r* r r r r3 r r r0 r r5 O � ���r5 c r4 ) � AudioMessageu 音频消息r6 Nr r
�
audio_datar8 r c r9 )NZaudio)r@ r8 r r. )r r r
r@ r8 r r0 r r r ` r; zAudioMessage.__init__r< r= r r r0 r r? ] r> r? c �F e Zd ZdZ d
dedededeeef dee f
� fdd �
Z� Z S ) �CommandMessageu 命令消息Nr r
�command�paramsr c �$ t � jd||||p
i d�|d� d S )NrC )rC rD r r. )r r r
rC rD r r0 r r r n �
�zCommandMessage.__init__�NN�
r' r( r) r* r r r r r r3 r r r0 r rB k � �
��rB c rA ) �
StatusMessageu 状态消息Nr r
�status�detailsr c rE )NrK )rK rL r r. )r r r
rK rL r r0 r r r | rF zStatusMessage.__init__rG rH r r r0 r rJ y rI rJ )r* r r r �typingr r r r r r, r5 r? rB rJ r r r r �<module> s 2