eye_agent.cpython-310.pyc•2.5 kB
o
R/�g
� @ sh d Z ddlZddlZddlZddlmZmZ ddlZddl Z ddl
m Z ddl m
Z
G dd� de �ZdS )u
视觉智能体
� N)�Dict�Any)� BaseAgent)� ImageMessagec sJ e Zd Zdededef� fdd� Z� fdd�Z� fdd �Zd
d � Z� Z S ) �EyeAgent�agent_id�host�portc s. t � �|d||� d | _d| _d| _d| _d S )N�visionFr g @)�super�__init__�camera� is_capturing�last_analysis_time�analysis_interval)�selfr r r �� __class__� �4D:\DeepSeek\MCPAgent\mcpTest\src\agents\eye_agent.pyr s
zEyeAgent.__init__c � s\ �t � �� I dH t�d�| _| j�� s | j�d� dS d| _t �
| � � � | j� d� dS )u 启动视觉智能体Nr zFailed to open cameraTu 视觉智能体已启动)
r �start�cv2Z VideoCapturer
ZisOpened�logger�errorr �asyncio� create_task�
_capture_loop�info�r r r r r s �
zEyeAgent.startc � s, �d| _ | jr | j�� t� �� I dH dS )u 停止视觉智能体FN)r r
�releaser �stopr r r r r " s
�
z
EyeAgent.stopc
� s �| j r�z_| j�� \}}|rZt�d|tjdg�\}}t�|��d�}ddl m
} |d|d��I dH }t � � }|| j | j
krZ|| _ t| jd |d
�} | j�d � | �d | �� �I dH t�d �I dH W n# ty� }
z| j�d
|
� �� t�d�I dH W Y d}
~
nd}
~
ww | j sdS dS )u 持续捕获视频的循环z.jpg�F zutf-8r )�broadcast_messager
)�type�contentN�brain)� sender_id� receiver_idZ
image_datau! 发送图像到大脑进行分析g�������?zError in capture loop: g �?)r r
�readr ZimencodeZIMWRITE_JPEG_QUALITY�base64� b64encode�decode�src.web.serverr" �timer r r r r r Z send_message�to_dictr �sleep� Exceptionr ) r �ret�frame�_�bufferZ image_base64r" �success� current_time�message�er r r r ) s: � �� ���zEyeAgent._capture_loop)
�__name__�
__module__� __qualname__�str�intr r r r �
__classcell__r r r r r s
r )�__doc__r r) ZnumpyZnp�typingr r r r- Zsrc.agents.base_agentr Zsrc.utils.mcp_protocolr r r r r r �<module> s