Update user_profile_extractor: update main exsamples

user_profile_extractor.py

@@ -150,15 +150,21 @@ if __name__ == '__main__':
         'interests': [],
         'interaction_frequency': 'medium'
     }
-    message = "没有任何问题放心，不会骚扰你了，再见"
-    resp = extractor.extract_profile_info(current_profile, message)
+    messages= [
+        {'role': 'user', 'content': "没有任何问题放心，不会骚扰你了，再见"}
+    ]
+
+    resp = extractor.extract_profile_info(current_profile, messages)
     print(resp)
     message = "好的，孩子，我是老李头，今年68啦，住在北京海淀区。平时喜欢在微信上跟老伙伴们聊聊养生、下下象棋，偶尔也跟年轻人学学新鲜事儿。\n" \
               "你叫我李叔就行，有啥事儿咱们慢慢聊啊\n" \
               "哎，今儿个天气不错啊，我刚才还去楼下小公园溜达了一圈儿。碰到几个老伙计在打太极，我也跟着比划了两下，这老胳膊老腿的，原来老不舒服，活动活动舒坦多了!\n" \
               "你吃饭了没？我们这儿中午吃的打卤面，老伴儿做的，香得很！这人老了就爱念叨些家长里短的，你可别嫌我啰嗦啊。\n" \
               "对了，最近我孙子教我发语音，比打字方便多啦！就是有时候一激动，说话声音太大，把手机都给震得嗡嗡响\n"
-    resp = extractor.extract_profile_info(current_profile, message)
+    messages = []
+    for line in message.split("\n"):
+        messages.append({'role': 'user', 'content': line})
+    resp = extractor.extract_profile_info(current_profile, messages)
     print(resp)
     print(extractor.merge_profile_info(current_profile, resp))
     current_profile = {
@@ -169,8 +175,9 @@ if __name__ == '__main__':
         'region': '北京市海淀区',
         'health_conditions': [],
         'medications': [],
-        'interests': ['养生', '下象棋']
+        'interests': ['养生', '下象棋'],
+        'interaction_frequency': 'medium'
     }
-    resp = extractor.extract_profile_info(current_profile, message)
+    resp = extractor.extract_profile_info(current_profile, messages)
     print(resp)
     print(extractor.merge_profile_info(current_profile, resp))