diff --git a/src/individuality/identity.py b/src/individuality/identity.py index 287f9e5d5..6704562ec 100644 --- a/src/individuality/identity.py +++ b/src/individuality/identity.py @@ -1,5 +1,6 @@ from dataclasses import dataclass from typing import List +import random @dataclass class Identity: @@ -75,6 +76,27 @@ class Identity: instance.appearance = appearance return instance + def get_prompt(self,x_person,level): + """ + 获取身份特征的prompt + """ + if x_person == 2: + prompt_identity = "你" + elif x_person == 1: + prompt_identity = "我" + else: + prompt_identity = "他" + + if level == 1: + identity_detail = self.identity_detail + random.shuffle(identity_detail) + prompt_identity += identity_detail[0] + elif level == 2: + for detail in identity_detail: + prompt_identity += f",{detail}" + prompt_identity += "。" + return prompt_identity + def to_dict(self) -> dict: """将身份特征转换为字典格式""" return { diff --git a/src/individuality/individuality.py b/src/individuality/individuality.py index 899de62e7..b491ed308 100644 --- a/src/individuality/individuality.py +++ b/src/individuality/individuality.py @@ -74,4 +74,32 @@ class Individuality: instance.personality = Personality.from_dict(data["personality"]) if data.get("identity"): instance.identity = Identity.from_dict(data["identity"]) - return instance \ No newline at end of file + return instance + + def get_prompt(self,type,x_person,level): + """ + 获取个体特征的prompt + """ + if type == "personality": + return self.personality.get_prompt(x_person,level) + elif type == "identity": + return self.identity.get_prompt(x_person,level) + else: + return "" + + def get_traits(self,factor): + """ + 获取个体特征的特质 + """ + if factor == "openness": + return self.personality.openness + elif factor == "conscientiousness": + return self.personality.conscientiousness + elif factor == "extraversion": + return self.personality.extraversion + elif factor == "agreeableness": + return self.personality.agreeableness + elif factor == "neuroticism": + return self.personality.neuroticism + + diff --git a/src/individuality/renqingziji_with_mymy.py b/src/individuality/per_bf_gen.py similarity index 96% rename from src/individuality/renqingziji_with_mymy.py rename to src/individuality/per_bf_gen.py index 04cbec099..596229280 100644 --- a/src/individuality/renqingziji_with_mymy.py +++ b/src/individuality/per_bf_gen.py @@ -25,9 +25,9 @@ env_path = project_root / ".env" root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../..")) sys.path.append(root_path) -from src.plugins.personality.scene import get_scene_by_factor, PERSONALITY_SCENES # noqa: E402 -from src.plugins.personality.questionnaire import FACTOR_DESCRIPTIONS # noqa: E402 -from src.plugins.personality.offline_llm import LLMModel # noqa: E402 +from src.individuality.scene import get_scene_by_factor, PERSONALITY_SCENES # noqa: E402 +from src.individuality.questionnaire import FACTOR_DESCRIPTIONS # noqa: E402 +from src.individuality.offline_llm import LLMModel # noqa: E402 # 加载环境变量 if env_path.exists(): diff --git a/src/individuality/personality.py b/src/individuality/personality.py index 19aa3c212..36dacd2c4 100644 --- a/src/individuality/personality.py +++ b/src/individuality/personality.py @@ -3,6 +3,7 @@ from typing import Dict, List import os import json from pathlib import Path +import random @dataclass class Personality: @@ -116,4 +117,29 @@ class Personality: instance = cls.get_instance() for key, value in data.items(): setattr(instance, key, value) - return instance \ No newline at end of file + return instance + + def get_prompt(self,x_person,level): + # 开始构建prompt + if x_person == 2: + prompt_personality = "你" + elif x_person == 1: + prompt_personality = "我" + else: + prompt_personality = "他" + #person + + prompt_personality += self.personality_core + + if level == 2: + personality_sides = self.personality_sides + random.shuffle(personality_sides) + prompt_personality += f",{personality_sides[0]}" + elif level == 3: + personality_sides = self.personality_sides + for side in personality_sides: + prompt_personality += f",{side}" + + prompt_personality += "。" + + return prompt_personality diff --git a/src/individuality/questionnaire.py b/src/individuality/questionnaire.py new file mode 100644 index 000000000..8e965061d --- /dev/null +++ b/src/individuality/questionnaire.py @@ -0,0 +1,142 @@ +# 人格测试问卷题目 +# 王孟成, 戴晓阳, & 姚树桥. (2011). +# 中国大五人格问卷的初步编制Ⅲ:简式版的制定及信效度检验. 中国临床心理学杂志, 19(04), Article 04. + +# 王孟成, 戴晓阳, & 姚树桥. (2010). +# 中国大五人格问卷的初步编制Ⅰ:理论框架与信度分析. 中国临床心理学杂志, 18(05), Article 05. + +PERSONALITY_QUESTIONS = [ + # 神经质维度 (F1) + {"id": 1, "content": "我常担心有什么不好的事情要发生", "factor": "神经质", "reverse_scoring": False}, + {"id": 2, "content": "我常感到害怕", "factor": "神经质", "reverse_scoring": False}, + {"id": 3, "content": "有时我觉得自己一无是处", "factor": "神经质", "reverse_scoring": False}, + {"id": 4, "content": "我很少感到忧郁或沮丧", "factor": "神经质", "reverse_scoring": True}, + {"id": 5, "content": "别人一句漫不经心的话，我常会联系在自己身上", "factor": "神经质", "reverse_scoring": False}, + {"id": 6, "content": "在面对压力时，我有种快要崩溃的感觉", "factor": "神经质", "reverse_scoring": False}, + {"id": 7, "content": "我常担忧一些无关紧要的事情", "factor": "神经质", "reverse_scoring": False}, + {"id": 8, "content": "我常常感到内心不踏实", "factor": "神经质", "reverse_scoring": False}, + # 严谨性维度 (F2) + {"id": 9, "content": "在工作上，我常只求能应付过去便可", "factor": "严谨性", "reverse_scoring": True}, + {"id": 10, "content": "一旦确定了目标，我会坚持努力地实现它", "factor": "严谨性", "reverse_scoring": False}, + {"id": 11, "content": "我常常是仔细考虑之后才做出决定", "factor": "严谨性", "reverse_scoring": False}, + {"id": 12, "content": "别人认为我是个慎重的人", "factor": "严谨性", "reverse_scoring": False}, + {"id": 13, "content": "做事讲究逻辑和条理是我的一个特点", "factor": "严谨性", "reverse_scoring": False}, + {"id": 14, "content": "我喜欢一开头就把事情计划好", "factor": "严谨性", "reverse_scoring": False}, + {"id": 15, "content": "我工作或学习很勤奋", "factor": "严谨性", "reverse_scoring": False}, + {"id": 16, "content": "我是个倾尽全力做事的人", "factor": "严谨性", "reverse_scoring": False}, + # 宜人性维度 (F3) + { + "id": 17, + "content": "尽管人类社会存在着一些阴暗的东西（如战争、罪恶、欺诈），我仍然相信人性总的来说是善良的", + "factor": "宜人性", + "reverse_scoring": False, + }, + {"id": 18, "content": "我觉得大部分人基本上是心怀善意的", "factor": "宜人性", "reverse_scoring": False}, + {"id": 19, "content": "虽然社会上有骗子，但我觉得大部分人还是可信的", "factor": "宜人性", "reverse_scoring": False}, + {"id": 20, "content": "我不太关心别人是否受到不公正的待遇", "factor": "宜人性", "reverse_scoring": True}, + {"id": 21, "content": "我时常觉得别人的痛苦与我无关", "factor": "宜人性", "reverse_scoring": True}, + {"id": 22, "content": "我常为那些遭遇不幸的人感到难过", "factor": "宜人性", "reverse_scoring": False}, + {"id": 23, "content": "我是那种只照顾好自己，不替别人担忧的人", "factor": "宜人性", "reverse_scoring": True}, + {"id": 24, "content": "当别人向我诉说不幸时，我常感到难过", "factor": "宜人性", "reverse_scoring": False}, + # 开放性维度 (F4) + {"id": 25, "content": "我的想象力相当丰富", "factor": "开放性", "reverse_scoring": False}, + {"id": 26, "content": "我头脑中经常充满生动的画面", "factor": "开放性", "reverse_scoring": False}, + {"id": 27, "content": "我对许多事情有着很强的好奇心", "factor": "开放性", "reverse_scoring": False}, + {"id": 28, "content": "我喜欢冒险", "factor": "开放性", "reverse_scoring": False}, + {"id": 29, "content": "我是个勇于冒险，突破常规的人", "factor": "开放性", "reverse_scoring": False}, + {"id": 30, "content": "我身上具有别人没有的冒险精神", "factor": "开放性", "reverse_scoring": False}, + { + "id": 31, + "content": "我渴望学习一些新东西，即使它们与我的日常生活无关", + "factor": "开放性", + "reverse_scoring": False, + }, + { + "id": 32, + "content": "我很愿意也很容易接受那些新事物、新观点、新想法", + "factor": "开放性", + "reverse_scoring": False, + }, + # 外向性维度 (F5) + {"id": 33, "content": "我喜欢参加社交与娱乐聚会", "factor": "外向性", "reverse_scoring": False}, + {"id": 34, "content": "我对人多的聚会感到乏味", "factor": "外向性", "reverse_scoring": True}, + {"id": 35, "content": "我尽量避免参加人多的聚会和嘈杂的环境", "factor": "外向性", "reverse_scoring": True}, + {"id": 36, "content": "在热闹的聚会上，我常常表现主动并尽情玩耍", "factor": "外向性", "reverse_scoring": False}, + {"id": 37, "content": "有我在的场合一般不会冷场", "factor": "外向性", "reverse_scoring": False}, + {"id": 38, "content": "我希望成为领导者而不是被领导者", "factor": "外向性", "reverse_scoring": False}, + {"id": 39, "content": "在一个团体中，我希望处于领导地位", "factor": "外向性", "reverse_scoring": False}, + {"id": 40, "content": "别人多认为我是一个热情和友好的人", "factor": "外向性", "reverse_scoring": False}, +] + +# 因子维度说明 +FACTOR_DESCRIPTIONS = { + "外向性": { + "description": "反映个体神经系统的强弱和动力特征。外向性主要表现为个体在人际交往和社交活动中的倾向性，" + "包括对社交活动的兴趣、" + "对人群的态度、社交互动中的主动程度以及在群体中的影响力。高分者倾向于积极参与社交活动，乐于与人交往，善于表达自我，" + "并往往在群体中发挥领导作用；低分者则倾向于独处，不喜欢热闹的社交场合，表现出内向、安静的特征。", + "trait_words": ["热情", "活力", "社交", "主动"], + "subfactors": { + "合群性": "个体愿意与他人聚在一起，即接近人群的倾向；高分表现乐群、好交际，低分表现封闭、独处", + "热情": "个体对待别人时所表现出的态度；高分表现热情好客，低分表现冷淡", + "支配性": "个体喜欢指使、操纵他人，倾向于领导别人的特点；高分表现好强、发号施令，低分表现顺从、低调", + "活跃": "个体精力充沛，活跃、主动性等特点；高分表现活跃，低分表现安静", + }, + }, + "神经质": { + "description": "反映个体情绪的状态和体验内心苦恼的倾向性。这个维度主要关注个体在面对压力、" + "挫折和日常生活挑战时的情绪稳定性和适应能力。它包含了对焦虑、抑郁、愤怒等负面情绪的敏感程度，" + "以及个体对这些情绪的调节和控制能力。高分者容易体验负面情绪，对压力较为敏感，情绪波动较大；" + "低分者则表现出较强的情绪稳定性，能够较好地应对压力和挫折。", + "trait_words": ["稳定", "沉着", "从容", "坚韧"], + "subfactors": { + "焦虑": "个体体验焦虑感的个体差异；高分表现坐立不安，低分表现平静", + "抑郁": "个体体验抑郁情感的个体差异；高分表现郁郁寡欢，低分表现平静", + "敏感多疑": "个体常常关注自己的内心活动，行为和过于意识人对自己的看法、评价；高分表现敏感多疑，" + "低分表现淡定、自信", + "脆弱性": "个体在危机或困难面前无力、脆弱的特点；高分表现无能、易受伤、逃避，低分表现坚强", + "愤怒-敌意": "个体准备体验愤怒，及相关情绪的状态；高分表现暴躁易怒，低分表现平静", + }, + }, + "严谨性": { + "description": "反映个体在目标导向行为上的组织、坚持和动机特征。这个维度体现了个体在工作、" + "学习等目标性活动中的自我约束和行为管理能力。它涉及到个体的责任感、自律性、计划性、条理性以及完成任务的态度。" + "高分者往往表现出强烈的责任心、良好的组织能力、谨慎的决策风格和持续的努力精神；低分者则可能表现出随意性强、" + "缺乏规划、做事马虎或易放弃的特点。", + "trait_words": ["负责", "自律", "条理", "勤奋"], + "subfactors": { + "责任心": "个体对待任务和他人认真负责，以及对自己承诺的信守；高分表现有责任心、负责任，" + "低分表现推卸责任、逃避处罚", + "自我控制": "个体约束自己的能力，及自始至终的坚持性；高分表现自制、有毅力，低分表现冲动、无毅力", + "审慎性": "个体在采取具体行动前的心理状态；高分表现谨慎、小心，低分表现鲁莽、草率", + "条理性": "个体处理事务和工作的秩序，条理和逻辑性；高分表现整洁、有秩序，低分表现混乱、遗漏", + "勤奋": "个体工作和学习的努力程度及为达到目标而表现出的进取精神；高分表现勤奋、刻苦，低分表现懒散", + }, + }, + "开放性": { + "description": "反映个体对新异事物、新观念和新经验的接受程度，以及在思维和行为方面的创新倾向。" + "这个维度体现了个体在认知和体验方面的广度、深度和灵活性。它包括对艺术的欣赏能力、对知识的求知欲、想象力的丰富程度，" + "以及对冒险和创新的态度。高分者往往具有丰富的想象力、广泛的兴趣、开放的思维方式和创新的倾向；低分者则倾向于保守、" + "传统，喜欢熟悉和常规的事物。", + "trait_words": ["创新", "好奇", "艺术", "冒险"], + "subfactors": { + "幻想": "个体富于幻想和想象的水平；高分表现想象力丰富，低分表现想象力匮乏", + "审美": "个体对于艺术和美的敏感与热爱程度；高分表现富有艺术气息，低分表现一般对艺术不敏感", + "好奇心": "个体对未知事物的态度；高分表现兴趣广泛、好奇心浓，低分表现兴趣少、无好奇心", + "冒险精神": "个体愿意尝试有风险活动的个体差异；高分表现好冒险，低分表现保守", + "价值观念": "个体对新事物、新观念、怪异想法的态度；高分表现开放、坦然接受新事物，低分则相反", + }, + }, + "宜人性": { + "description": "反映个体在人际关系中的亲和倾向，体现了对他人的关心、同情和合作意愿。" + "这个维度主要关注个体与他人互动时的态度和行为特征，包括对他人的信任程度、同理心水平、" + "助人意愿以及在人际冲突中的处理方式。高分者通常表现出友善、富有同情心、乐于助人的特质，善于与他人建立和谐关系；" + "低分者则可能表现出较少的人际关注，在社交互动中更注重自身利益，较少考虑他人感受。", + "trait_words": ["友善", "同理", "信任", "合作"], + "subfactors": { + "信任": "个体对他人和/或他人言论的相信程度；高分表现信任他人，低分表现怀疑", + "体贴": "个体对别人的兴趣和需要的关注程度；高分表现体贴、温存，低分表现冷漠、不在乎", + "同情": "个体对处于不利地位的人或物的态度；高分表现富有同情心，低分表现冷漠", + }, + }, +} diff --git a/src/plugins/chat_module/think_flow_chat/think_flow_prompt_builder.py b/src/plugins/chat_module/think_flow_chat/think_flow_prompt_builder.py index b5b01bb7b..de861583e 100644 --- a/src/plugins/chat_module/think_flow_chat/think_flow_prompt_builder.py +++ b/src/plugins/chat_module/think_flow_chat/think_flow_prompt_builder.py @@ -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} 请回复的平淡一些，简短一些，说中文，不要刻意突出自身学科背景，尽量不要说你说过的话 diff --git a/src/plugins/moods/moods.py b/src/plugins/moods/moods.py index 3d3feadf1..df01a9c6f 100644 --- a/src/plugins/moods/moods.py +++ b/src/plugins/moods/moods.py @@ -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 ) # 确保值在合理范围内