fix:优化人格prompt提取
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user