eye_agent.cpython-312.pyc•11.8 kB
�
��g&. � � � d Z ddlZddlZddlZddlmZmZmZm Z m
Z
ddlZddlZddl
Z
ddlZddlZddlmZ ddlmZmZ ddlmZ ddlmZ G d� d e� Zy)
u
视觉智能体
� N)�Dict�Any�List�Tuple�Optional)� BaseAgent)�ImageMessage�TextMessage)�FaceRecognition)�PersonDatabasec
� � � e Zd Zdededef� fd�Z� fd�Z� fd�Zdedee fd �Z defd
�Z
dedefd�Zd
e
e de
e de
e deddf
d�Zd� Z� xZS )�EyeAgent�agent_id�host�portc �� �� t �| � |d||� d | _ d| _ d| _ d| _ t
� | _ t � | _ d | _
d| _ d| _ y )N�visionFr g $@g ��@)
�super�__init__�camera�is_capturing�last_analysis_time�analysis_intervalr �face_recognitionr �person_database�last_recognized_person_id�last_recognition_time�recognition_cooldown)�selfr r r � __class__s ��,D:\AIProject\mcpTest\src\agents\eye_agent.pyr zEyeAgent.__init__ si �� �
����8�T�4�8����!���"#���!%��� !0� 1��� .�/��� *.��&�%&��"�$)��!� c � �� �K � t �| � � � d{ ��� t j d� | _ | j j � s| j j d� yd| _ t j | j � � | j j d� | j j dt | j j � � � d�� y7 �ڭw)u 启动视觉智能体Nr zFailed to open cameraTu 视觉智能体已启动u 已加载u 个已知人物数据)r �start�cv2�VideoCapturer �isOpened�logger�errorr �asyncio�create_task�
_capture_loop�info�lenr �get_all_persons�r r s �r! r$ zEyeAgent.start&