refactor: brain memory

This commit is contained in:
莘权 马 2023-09-07 23:41:31 +08:00
parent b49c7f2d70
commit d906bd1c81

View file

@ -83,12 +83,12 @@ class TalkAction(Action):
]
format_msgs = []
if self._knowledge:
format_msgs.append({"knowledge": self._knowledge})
format_msgs.append({"role": "knowledge", "content": self._knowledge})
if self._history_summary:
if CONFIG.LLM_TYPE == LLMType.METAGPT.value:
format_msgs.append(json.loads(self._history_summary))
else:
format_msgs.append({"knowledge": self._history_summary})
format_msgs.append({"role": "context", "content": self._history_summary})
rsp = await self.llm.aask(msg=self._talk, format_msgs=format_msgs, system_msgs=system_msgs)
self._rsp = ActionOutput(content=rsp)
return self._rsp