|
|
@@ -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))
|
StrayWarrior