feat: +role

This commit is contained in:
莘权 马 2023-09-02 15:39:17 +08:00
parent 578961ce2e
commit a7b933311e
2 changed files with 9 additions and 9 deletions

View file

@ -27,7 +27,14 @@ class TalkAction(Action):
@property
def prompt(self):
prompt = f"Background knowledge:\n{self._knowledge}\n\n" if self._knowledge else ""
prompt = ""
if CONFIG.agent_description:
prompt = (
f"You are {CONFIG.agent_description}. Your responses should align with the role-play agreement, "
f"maintaining the character's persona and habits. When faced with unrelated questions, playfully "
f"decline to answer without revealing your AI nature to preserve the character's image.\n\n"
)
prompt += f"Background knowledge:\n{self._knowledge}\n\n" if self._knowledge else ""
prompt += f"{self._history_summary}\n\n"
if self._history_summary != "":
prompt += "According to the historical conversation above, "

View file

@ -54,14 +54,7 @@ class Assistant(Role):
last_talk = await self.refine_memory()
if not last_talk:
return False
prompt = ""
if CONFIG.agent_description:
prompt = (
f"You are {CONFIG.agent_description}. Your responses should align with the role-play agreement, "
f"maintaining the character's persona and habits. When faced with unrelated questions, playfully "
f"decline to answer without revealing your AI nature to preserve the character's image.\n\n"
)
prompt += f"Refer to this sentence:\n {last_talk}\n"
prompt = f"Refer to this sentence:\n {last_talk}\n"
skills = self.skills.get_skill_list()
for desc, name in skills.items():
prompt += f"If want you to do {desc}, return `[SKILL]: {name}` brief and clear. For instance: [SKILL]: text_to_image\n"