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 22:58:00 +08:00
parent 7c6b0325d8
commit f2da313548
3 changed files with 50 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

19
metagpt/provider/base_gpt_api.py
View file

@ -38,11 +38,22 @@ class BaseGPTAPI(BaseChatbot):
rsp = self.completion(message)
return self.get_choice_text(rsp)
async def aask(self, msg: str, system_msgs: Optional[list[str]] = None, generator: bool = False) -> str:
async def aask(
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,
generator: bool = False,
) -> str:
message = []
if system_msgs:
message = self._system_msgs(system_msgs) + [self._user_msg(msg)]
else:
message = [self._default_system_msg(), self._user_msg(msg)]
message = self._system_msgs(system_msgs)
if knowledge_msgs:
message.extend(knowledge_msgs)
if history_msgs:
message.extend(history_msgs)
message.append(self._user_msg(msg))
try:
rsp = await self.acompletion_text(message, stream=True, generator=generator)
except Exception as e: