feat: 添加回复情绪对好感度的影响,修改用户显示名字为群昵称

This commit is contained in:
tcmofashi
2025-03-02 08:24:00 +08:00
parent c24bf0cdcf
commit 97834e7a93
13 changed files with 77 additions and 14 deletions

View File

@@ -72,7 +72,12 @@ class LLMResponseGenerator:
if model_response:
print(f"'{model_response}' 获取到的情感标签为:{emotion}")
valuedict={
'happy':0.5,'angry':-1,'sad':-0.5,'surprised':0.5,'disgusted':-1.5,'fearful':-0.25,'neutral':0.25
}
await relationship_manager.update_relationship_value(message.user_id, relationship_value=valuedict[emotion[0]])
return model_response, emotion
async def _generate_base_response(
@@ -82,6 +87,8 @@ class LLMResponseGenerator:
model_params: Optional[Dict[str, Any]] = None
) -> Optional[str]:
sender_name = message.user_nickname or f"用户{message.user_id}"
if message.user_cardname:
sender_name=f"[({message.user_id}){message.user_nickname}]{message.user_cardname}"
# 获取关系值
if relationship_manager.get_relationship(message.user_id):
@@ -247,6 +254,8 @@ class LLMResponseGenerator:
for msg_dict in messages_list:
time_str = time.strftime("%m-%d %H:%M:%S", time.localtime(msg_dict['time']))
display_name = msg_dict.get('user_nickname', f"用户{msg_dict['user_id']}")
cardname = msg_dict.get('user_cardname', f"用户{msg_dict['user_id']}")
display_name = f"[({msg_dict['user_id']}){display_name}]{cardname}"
content = msg_dict.get('processed_plain_text', msg_dict['plain_text'])
group_chat += f"[{time_str}] {display_name}: {content}\n"