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

feat: add metadata {"agent":xx} to assistant message

Browse source
This commit is contained in:
莘权 马 2024-04-30 12:32:03 +08:00
parent e37aa8e5bf
commit fc06cc3b1f
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/roles/role.py
View file

@ -407,10 +407,8 @@ class Role(SerializationMixin, ContextMixin, BaseModel):
)
elif isinstance(response, Message):
msg = response
elif isinstance(response, AIMessage):
msg = response.with_agent(self._setting)
else:
msg = AIMessage(content=response, cause_by=self.rc.todo, sent_from=self).with_agent(self._setting)
msg = AIMessage(content=response, cause_by=self.rc.todo, sent_from=self)
if self.enable_memory:
self.rc.memory.add(msg)
@ -534,6 +532,8 @@ class Role(SerializationMixin, ContextMixin, BaseModel):
else:
raise ValueError(f"Unsupported react mode: {self.rc.react_mode}")
self._set_state(state=-1) # current reaction is complete, reset state to -1 and todo back to None
if isinstance(rsp, AIMessage):
rsp.with_agent(self._setting)
return rsp
def get_memories(self, k=0) -> list[Message]: