mouth_agent.cpython-310.pyc•4.74 kB
o
�+�g� � @ sp d Z ddlZddlZddlmZmZ ddlZddlZddlZddl Z ddl
m Z ddl m
Z
G dd� de �ZdS )u&
嘴巴智能体 - 负责语音输出
� N)�Dict�Any)� BaseAgent)� TextMessagec sx e Zd Zdededef� fdd� Z� fdd�Zdd � Zd
d � Zd d
� Z dd� Z
de ee f fdd�Z
� fdd�Z� ZS )�
MouthAgent�agent_id�host�portc s. t � �|d||� t�� | _| �d| j� d S )N�speech�text)�super�__init__�asyncio�Queue� tts_queue�register_handler�_handle_text_message)�selfr r r �� __class__� �6D:\DeepSeek\MCPAgent\mcpTest\src\agents\mouth_agent.pyr
s
zMouthAgent.__init__c � sN �t � �� I dH | �� I dH }|s| j�d� t�| �� � | j�d� dS )u 启动嘴巴智能体Nu- TTS测试失败，语音功能可能不可用u 嘴巴智能体已启动) r �start� _test_tts�logger�warningr � create_task�_process_tts_queue�info)r Z test_resultr r r r s � zMouthAgent.startc
� sT �z t �| j�I dH }|W S ty) } z| j�d|� �� W Y d}~dS d}~ww )u 测试TTS引擎Nu TTS测试失败: F)r � to_thread�_init_and_test_tts� Exceptionr �error)r �result�er r r r & s ���zMouthAgent._test_ttsc
C s0 z{| j �d� t�� }|�dd� |�dd� |�d�}| j �dt|�� �� |D ]1}| j �d|j� d |j� �� d
|j� � v sGd |j� � v rZ|�d |j� | j �d
|j� �� nq)d}| j �d|� �� |�
|� |� � |� � | j �d� W dS t
y� } z| j �d|� �� W Y d}~dS d}~ww )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) r, Z test_textr$ r r r r 0 s4
�
��z MouthAgent._init_and_test_ttsc
� s� � z | j �� I dH }tj| j|fd�}d|_|�� | j �� W n ty< } z| j �
d|� �� W Y d}~nd}~ww t � d�I dH q)u 处理TTS队列TN)�target�argsu 处理TTS队列时出错: g�������?)
r �get� threading�Thread� _speak_text�daemonr � task_doner! r r" r �sleep)r r �threadr$ r r r r T s" �� ���z MouthAgent._process_tts_queuec
C s� zP| j �d|� �� t�� }|�dd� |�dd� |�d�}|D ]}d|j�� v s1d|j�� v r:|�d |j� nq!|� |� |�
� |� � | j �d
� W d S t yl } z| j �
d |� �� W Y d }~d S d }~ww )
u 使用TTS引擎播放文本u 开始播放语音: r% r& r' r( r) r* r+ r, u 语音播放完成u 语音播放失败: N)r r r- r. r/ r0 r3 r4 r2 r5 r6 r7 r! r" )r r r8 r) r, r$ r r r r> h s&
�
��zMouthAgent._speak_text�messagec
� s� �zZ| j �d|� �� d|v r(t|d t�r#d|d v r#|d d }n |d }n|�dd�}|�dd�}| j �d|� d|� �� d d
lm} |d |d ��I d
H | j�|�I d
H W d
S t yw } z| j �
d|� �� W Y d
}~d
S d
}~ww )u$ 处理文本消息（语音输出）u 收到文本消息: �contentr � � sender_id�unknownu 收到来自u 的语音输出请求: r )�broadcast_messager
)�typerD Nu 处理文本消息时出错: ) r r �
isinstance�dictr; �src.web.serverrH r �putr! r" )r rC r rF rH r$ r r r r � s( �
� ��z MouthAgent._handle_text_messagec � sB �t | d�r| jr| j�� I dH t� �� I dH | j�d� dS )u 停止嘴巴智能体r Nu 嘴巴智能体已停止)�hasattrr �joinr r7 r r )r r r r r7 � s
�zMouthAgent.stop)�__name__�
__module__� __qualname__�str�intr
r r r r r> r r r r7 �
__classcell__r r r r r s
$ r )�__doc__r �logging�typingr r r- r< �time�os�src.agents.base_agentr �src.utils.mcp_protocolr r r r r r �<module> s