mcp_protocol.cpython-312.pyc•6.13 kB
�
���gr � � � d Z ddlZddlZddlmZ ddlmZmZmZmZ G d� d� Z G d� de � Z
G d � d
e � Z G d � d e � Z G d
� de � Z
G d� de � Zy)u
MCP协议实现
� N)�datetime)�Dict�Any�Optional�Listc � � e Zd ZdZ ddedededeeef dee f
d�Zd eeef fd
�Z d efd �Z
e d eeef d d fd
�� Z e ded d fd�� Z
y)�
MCPMessageu MCP协议消息基类N� message_type� sender_id� receiver_id�content�
message_idc �� � |xs t t j � � | _ || _ || _ || _ || _ t j � j � | _ y �N) �str�uuid�uuid4r r
r r r
r �now� isoformat� timestamp)�selfr
r r r
r s �.D:\AIProject\mcpTest\src\utils\mcp_protocol.py�__init__zMCPMessage.__init__ sO � � %�9��D�J�J�L�(9��� (���"���&��� �� �!� � � �1�1�3��� �returnc � � | j | j | j | j | j | j
dd�S )u 将消息转换为字典zMCP/1.0)r r
r r r
r �protocol)r r
r r r
r �r s r �to_dictzMCPMessage.to_dict s? � � �/�/� � -� -� � � � �+�+��|�|� � � �!�
�
r c �H � t j | j � � S )u 将消息转换为JSON字符串)�json�dumpsr r s r �to_jsonzMCPMessage.to_json% s � ��z�z�$�,�,�.�)�)r �datac �R � | |d |d |d |d |j d� �� S )u 从字典创建消息r
r r r
r �r
r r r
r )�get)�clsr$ s r � from_dictzMCPMessage.from_dict) s< � � � �n�-��;�'� �]�+���O��x�x� �-�
�
r �json_strc �N � t j |� }| j |� S )u 从JSON字符串创建消息)r! �loadsr) )r( r* r$ s r � from_jsonzMCPMessage.from_json4 s! � � �z�z�(�#���}�}�T�"�"r r )�__name__�
__module__� __qualname__�__doc__r r r r r r r# � classmethodr) r- � r r r r s� � � � .2� 4� "� 4� � 4� "� 4� �s�C�x�.� 4�
&�c�]� 4�
��c�3�h� �
�*�� *� �
�T�#�s�(�^�
� �
� �
� �#�� #� � #� �#r r c
�R � � e Zd ZdZd dedededee f� fd�
Zdeeef fd�Z � xZ
S )
� TextMessageu 文本消息r r �textr c �2 �� t �| � d||d|i|�� y )Nr6 r& ��superr )r r r r6 r � __class__s �r r zTextMessage.__init__> s( �� �
��� � �#��T�N�!� �
r r c �` � | j | j | j | j d�S )N�r r r
r
r< r s r r zTextMessage.to_dictF s, � � � � � �+�+� � -� -��|�|�
�
r r ) r. r/ r0 r1 r r r r r r �
__classcell__�r: s @r r5 r5 ; sB �� ��
�#�
�C�
�s�
�PX�Y\�P]�
�
��c�3�h� �
r r5 c �L � � e Zd ZdZ d dededededee dee f � fd�
Z� xZS )
� ImageMessageu 图像消息r r �
image_data�format� person_namer c �F �� ||d�}|�||d< t �| � d||||�� y )N)rA rB rC �imager& r8 ) r r r rA rB rC r r
r: s �r r zImageMessage.__init__R s> �� �!+�v�>�� � "�%0�G�M� "�
��� � �#��!� �
r )�base64NN�r. r/ r0 r1 r r r r= r> s @r r@ r@ O sN �� �� im�
�#�
�C�
�S�
� �
�6>�s�m�
�X`�ad�Xe�
�
r r@ c �B � � e Zd ZdZ ddededededee f
� fd�
Z� xZS ) � AudioMessageu 音频消息r r �
audio_datarB r c �4 �� t �| � d||||d�|�� y )N�audio)rJ rB r& r8 )r r r rJ rB r r: s �r r zAudioMessage.__init__d s) �� �
��� � �#�#-��@�!� �
r )rF NrG r> s @r rI rI a s@ �� �� FJ�
�#�
�C�
�S�
� �
�5=�c�]�
�
r rI c �L � � e Zd ZdZ ddedededeeef dee f
� fd�
Z� xZ S ) �CommandMessageu 命令消息r r �command�paramsr c �< �� t �| � d||||xs i d�|�� y )NrO )rO rP r& r8 )r r r rO rP r r: s �r r zCommandMessage.__init__r s- �� �
���"� �#� '�6�<�R�@�!� �
r �NN�
r. r/ r0 r1 r r r r r r= r> s @r rN rN o sH �� �� MQ�
�#�
�C�
�#�
� �c�3�h� �
�<D�S�M�
�
r rN c �L � � e Zd ZdZ ddedededeeef dee f
� fd�
Z� xZ S ) �
StatusMessageu 状态消息r r �status�detailsr c �< �� t �| � d||||xs i d�|�� y )NrV )rV rW r& r8 )r r r rV rW r r: s �r r zStatusMessage.__init__� s- �� �
���!� �#� %�'�-�R�@�!� �
r rR rS r> s @r rU rU } sH �� �� NR�
�#�
�C�
��
� �s�C�x�.�
�=E�c�]�
�
r rU )r1 r! r r �typingr r r r r r5 r@ rI rN rU r3 r r �<module>rZ sb ��� � � � ,� ,�/#� /#�d
�*�
�(
�:�
�$
�:�
�
�Z�
�
�J�
r