加入头像信息

utils/wechat/contact_manager.py

@@ -51,7 +51,7 @@ class ContactManager:
             cls._instance = ContactManager()
         return cls._instance
 
-    def set_contacts(self, contacts: Dict[str, str]) -> None:
+    def set_contacts(self, contacts: Dict[str, str], head_imgs: Dict[str, str]) -> None:
         """设置联系人字典
         
         Args:
@@ -69,6 +69,7 @@ class ContactManager:
         """
         self._contacts = contacts
         self._friends = contacts
+        self._head_images = head_imgs
         self._logger.info(f"联系人信息已更新，共 {len(contacts)} 个联系人")
         # 分类联系人
         self._classify_contacts()