mouth_agent.cpython-312.pyc•8.76 kB
�
���g� � �v � d Z ddlZddlZddlmZmZ ddlZddlZddlZddl Z ddl
m Z ddl m
Z
G d� de � Zy)u&
嘴巴智能体 - 负责语音输出
� N)�Dict�Any)� BaseAgent)� TextMessagec �n � � e Zd Zdededef� fd� Z� fd�Zd� Zd� Zd� Z d � Z
d
e ee f fd �Z
� fd �Z� xZS )
�
MouthAgent�agent_id�host�portc � �� t �| � |d||� t j � | _ | j d| j � y )N�speech�text)�super�__init__�asyncio�Queue� tts_queue�register_handler�_handle_text_message)�selfr r
r � __class__s ��.D:\AIProject\mcpTest\src\agents\mouth_agent.pyr zMouthAgent.__init__ s; �� �
����8�T�4�8� � � � ���
� � �f�d�&?�&?�@� c � �( �K � t �| � � � d{ ��� | j � � d{ ��� }|s| j j d� t j | j � � | j j d� y7 �x7 �b�w)u 启动嘴巴智能体Nu- TTS测试失败，语音功能可能不可用u 嘴巴智能体已启动) r �start� _test_tts�logger�warningr � create_task�_process_tts_queue�info)r � test_resultr s �r r zMouthAgent.start st �� �� ��g�m�o� � � !�N�N� ,�,� �� �K�K� � � O� P� ���D� 3� 3� 5�6� � � ���3�4� �� -�s �B�B�B�B�A!B�Bc � �� K � t j | j � � d{ ��� }|S 7 �# t $ r(}| j j d|� �� Y d}~yd}~ww xY w�w)u 测试TTS引擎Nu TTS测试失败: F)r � to_thread�_init_and_test_tts� Exceptionr �error)r �result�es r r zMouthAgent._test_tts& s[ � �� � �"�,�,�T�-D�-D�E�E�F��M� F��� � �K�K� � � 1�!�� 5� 6��� �s6 �A#�"/ �-�/ �A#�/ � A � A�A#�A � A#c �� � | j j d� t j � }|j dd� |j dd� |j d� }| j j dt
|� � �� |D ]� }| j j d|j � d |j � �� d
|j j � v s d |j j � v s�q|j d |j � | j j d
|j � �� n d}| j j d|� �� |j |� |j � |j � | j j d� y# t $ r(}| j j d|� �� Y d}~yd}~ww xY w)u 初始化并测试TTS引擎u 初始化TTS引擎�rate� �volume��������?�voicesu 可用语音数量: u 语音: ID=z, Name=�chinese�zh�voiceu 设置中文语音: u 嘴巴智能体启动！u 测试TTS: u! TTS引擎初始化和测试完成Tu# TTS引擎初始化或测试失败: NF)r r! �pyttsx3�init� setProperty� getProperty�len�id�name�lower�say�
runAndWait�stopr& r' )r �enginer/ r2 � test_textr) s r r% z MouthAgent._init_and_test_tts0 s� � � � �K�K� � � 1� 2� �\�\�^�F�
� � �v�s� +� � � �x�� -� �'�'��1�F� �K�K� � � 3�C��K�=� A� B� �
��� � � � �;�u�x�x�j���
�
�|�!L�M� ��
�
� 0� 0� 2�2�d�e�h�h�n�n�>N�6N��&�&�w����9��K�K�$�$�';�E�H�H�:�%F�G��
� 3�I� �K�K� � �{�9�+� 6� 7� �J�J�y� !� � � � � �K�K�M� �K�K� � � @� A���� � �K�K� � � C�A�3� G� H��� �s �C=F4 � B3F4 �4 G%�= G � G%c � � K � | j j � � d{ ��� }t j | j |f�� }d|_ |j
� | j j � t j d� � d{ ��� ��7 �u# t $ r(}| j j d|� �� Y d}~�Ld}~ww xY w7 �;�w)u 处理TTS队列TN)�target�argsu 处理TTS队列时出错: g�������?)
r �get� threading�Thread� _speak_text�daemonr � task_doner& r r' r �sleep)r r �threadr) s r r z MouthAgent._process_tts_queueT s� � �� ��
D� !�^�^� /� /� 1�1�� #�)�)� �+�+� ���� !%��
�� � �����(�(�*� �-�-��$� $� $�! �1�� �
D�� � �!�!�$>�q�c�"B�C�C��
D�� $�sK �C� B �B�AB �:C�C�C�B � C �# C�C�C � Cc � � | j j d|� �� t j � }|j dd� |j dd� |j d� }|D ]X }d|j j � v s d|j j � v s�<|j d |j � n |j |� |j � |j � | j j d
� y # t $ r(}| j j d |� �� Y d }~y d }~ww xY w)
u 使用TTS引擎播放文本u 开始播放语音: r+ �� r- r. r/ r0 r1 r2 u 语音播放完成u 语音播放失败: N)r r! r3 r4 r5 r6 r9 r: r8 r; r< r= r& r' )r r r> r/ r2 r) s r rF zMouthAgent._speak_texth s � � :� �K�K� � � 3�D�6� :� ;� �\�\�^�F� � � �v�s� +� � � �x�� -� �'�'��1�F� �
�� ��
�
� 0� 0� 2�2�d�e�h�h�n�n�>N�6N��&�&�w����9��
�
�J�J�t� � � � � � �K�K�M� �K�K� � � 1� 2��� :� �K�K� � � 4�Q�C� 8� 9� 9�� :�s �B$D �'A+D � E� D?�?E�messagec � � K � | j j d|� �� d|v r)t |d t � rd|d v r |d d }n|d }n|j dd� }|j dd� }| j j d|� d|� �� d d
lm} |d |d �� � d
{ ��� | j j |� � d
{ ��� y
7 �(7 �# t $ r(}| j j d|� �� Y d
}~y
d
}~ww xY w�w)u$ 处理文本消息（语音输出）u 收到文本消息: �contentr � � sender_id�unknownu 收到来自u 的语音输出请求: r )�broadcast_messager
)�typerO Nu 处理文本消息时出错: ) r r! �
isinstance�dictrC �src.web.serverrS r �putr& r' )r rM r rQ rS r) s r r z MouthAgent._handle_text_message� s � �� � C� �K�K� � � 3�G�9� =� >��G�#� �g�i� 0�$�7�F�g�i�FX�<X�"�9�-�f�5�D�"�9�-�D� �{�{�6�2�.�� � � �K��;�I� �K�K� � �|�I�;�6M�d�V� T� U�
9�#� � �%� �
�
� �.�.�$�$�T�*� *� *�
��
+��� C� �K�K� � � =�a�S� A� B� B�� C�sS �D �B$C �(C�)"C � C� C �D �C �C � D� D�<D �D�D c � �� �K � t | d� r.| j r"| j j � � d{ ��� t �| � � � d{ ��� | j
j
d� y7 �67 �"�w)u 停止嘴巴智能体r Nu 嘴巴智能体已停止)�hasattrr �joinr r= r r! )r r s �r r= zMouthAgent.stop� sZ �� �� � �4�� %�$�.�.��.�.�%�%�'� '� '��g�l�n� � � � � ���3�4�
(�� �s! �6A4�A0�A4�A2�!A4�2A4)�__name__�
__module__� __qualname__�str�intr r r r% r rF r r r r= �
__classcell__)r s @r r r s[ �� �A�� A�C� A�s� A� 5� �"�H%�(:�2 C�$�s�C�x�.� C�<5� 5r r )�__doc__r �logging�typingr r r3 rD �time�os�src.agents.base_agentr �src.utils.mcp_protocolr r � r r �<module>rj s5 ��� � � � � � � � +� .�W5�� W5r