fix:优化人格prompt提取

This commit is contained in:
SengokuCola
2025-04-06 14:22:34 +08:00
parent 915ad8c61a
commit 88a4db8894
7 changed files with 245 additions and 26 deletions

View File

@@ -26,23 +26,10 @@ class PromptBuilder:
) -> tuple[str, str]:
current_mind_info = heartflow.get_subheartflow(stream_id).current_mind
# 开始构建prompt
prompt_personality = ""
#person
individuality = Individuality.get_instance()
personality_core = individuality.personality.personality_core
prompt_personality += personality_core
personality_sides = individuality.personality.personality_sides
random.shuffle(personality_sides)
prompt_personality += f",{personality_sides[0]}"
identity_detail = individuality.identity.identity_detail
random.shuffle(identity_detail)
prompt_personality += f",{identity_detail[0]}"
prompt_personality = individuality.get_prompt(type = "personality",x_person = 2,level = 1)
prompt_identity = individuality.get_prompt(type = "identity",x_person = 2,level = 1)
# 关系
who_chat_in_group = [(chat_stream.user_info.platform,
chat_stream.user_info.user_id,
@@ -124,7 +111,7 @@ class PromptBuilder:
你刚刚脑子里在想:
{current_mind_info}
现在"{sender_name}"说的:{message_txt}。引起了你的注意,你想要在群里发言发言或者回复这条消息。\n
你的网名叫{global_config.BOT_NICKNAME},有人也叫你{"/".join(global_config.BOT_ALIAS_NAMES)}{prompt_personality}
你的网名叫{global_config.BOT_NICKNAME},有人也叫你{"/".join(global_config.BOT_ALIAS_NAMES)}{prompt_personality} {prompt_identity}
你正在{chat_target_2},现在请你读读之前的聊天记录,然后给出日常且口语化的回复,平淡一些,
尽量简短一些。{keywords_reaction_prompt}请注意把握聊天内容,不要回复的太有条理,可以有个性。{prompt_ger}
请回复的平淡一些,简短一些,说中文,不要刻意突出自身学科背景,尽量不要说你说过的话

View File

@@ -6,6 +6,7 @@ from dataclasses import dataclass
from ..config.config import global_config
from src.common.logger import get_module_logger, LogConfig, MOOD_STYLE_CONFIG
from ..person_info.relationship_manager import relationship_manager
from src.individuality.individuality import Individuality
mood_config = LogConfig(
# 使用海马体专用样式
@@ -129,16 +130,29 @@ class MoodManager:
current_time = time.time()
time_diff = current_time - self.last_update
# Valence 向中性0回归
valence_target = 0
# 获取人格特质
personality = Individuality.get_instance().personality
if personality:
# 神经质0.5为默认值0为极低1为极高
# 神经质越高,情绪变化越快(衰减率越高)
neuroticism_factor = 0.5 + (personality.neuroticism - 0.5) * 0.5 # 范围在0.25-0.75之间
# 宜人性0.5为默认值0为极低1为极高
# 宜人性越低,越容易走向负面情绪(向负值偏移)
agreeableness_bias = (0.5 - personality.agreeableness) * 0.2 # 范围在-0.1到0.1之间
else:
neuroticism_factor = 0.5 # 默认值
agreeableness_bias = 0.0
# Valence 向中性0回归考虑宜人性偏差
valence_target = agreeableness_bias
self.current_mood.valence = valence_target + (self.current_mood.valence - valence_target) * math.exp(
-self.decay_rate_valence * time_diff
-self.decay_rate_valence * time_diff * neuroticism_factor
)
# Arousal 向中性0.5)回归
arousal_target = 0.5
self.current_mood.arousal = arousal_target + (self.current_mood.arousal - arousal_target) * math.exp(
-self.decay_rate_arousal * time_diff
-self.decay_rate_arousal * time_diff * neuroticism_factor
)
# 确保值在合理范围内