apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: brain memory

Browse source
This commit is contained in:
莘权 马 2023-09-07 23:29:08 +08:00
parent 1b267d34dc
commit 270b14e036
2 changed files with 26 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

33
metagpt/actions/talk_action.py
View file

@ -64,22 +64,35 @@ class TalkAction(Action):
logger.info(f"PROMPT: {prompt}")
return prompt
async def run(self, *args, **kwargs) -> ActionOutput:
if CONFIG.LLM_TYPE == LLMType.METAGPT.value:
rsp = await self.llm.aask(
msg=self._talk,
knowledge_msgs=[{"knowledge": self._knowledge}] if self._knowledge else None,
history_msgs=json.loads(self._history_summary) if self._history_summary else None,
)
self._rsp = ActionOutput(content=rsp)
return self._rsp
async def run_old(self, *args, **kwargs) -> ActionOutput:
prompt = self.prompt
rsp = await self.llm.aask(msg=prompt, system_msgs=[])
logger.debug(f"PROMPT:{prompt}\nRESULT:{rsp}\n")
self._rsp = ActionOutput(content=rsp)
return self._rsp
async def run(self, *args, **kwargs) -> ActionOutput:
language = CONFIG.language or DEFAULT_LANGUAGE
system_msgs = [
f"You are {CONFIG.agent_description}.",
"Your responses should align with the role-play agreement, "
"maintaining the character's persona and habits. When faced with unrelated questions, playfully "
"decline to answer without revealing your AI nature to preserve the character's image.",
"If the information is insufficient, you can search in the context or knowledge.",
f"Answer the following questions strictly in {language}, and the answers must follow the Markdown format.",
]
format_msgs = []
if self._knowledge:
format_msgs.append({"knowledge": 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({"context": 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
__FORMATION__ = """Formation: "Capacity and role" defines the role you are currently playing;
"[HISTORY_BEGIN]" and "[HISTORY_END]" tags enclose the historical conversation;
"[KNOWLEDGE_BEGIN]" and "[KNOWLEDGE_END]" tags enclose the knowledge may help for your responses;

9
metagpt/provider/base_gpt_api.py
View file

@ -42,17 +42,14 @@ class BaseGPTAPI(BaseChatbot):
self,
msg: str,
system_msgs: Optional[list[str]] = None,
history_msgs: Optional[list[dict[str, str]]] = None,
knowledge_msgs: Optional[list[dict[str, str]]] = None,
format_msgs: Optional[list[dict[str, str]]] = None,
generator: bool = False,
) -> str:
message = []
if system_msgs:
message = self._system_msgs(system_msgs)
if knowledge_msgs:
message.extend(knowledge_msgs)
if history_msgs:
message.extend(history_msgs)
if format_msgs:
message.extend(format_msgs)
message.append(self._user_msg(msg))
try:
rsp = await self.acompletion_text(message, stream=True, generator=generator)