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

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