aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2026-02-15 11:14:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

add post_instruction hook

Browse Source
This commit is contained in:
BillSchumacher
2023-04-16 00:01:23 -05:00
parent 00225e01b3
commit 397627d1b9
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
autogpt/agent/agent_manager.py
View File

@@ -58,7 +58,10 @@ class AgentManager(metaclass=Singleton):
self.agents[key] = (task, messages, model)
return key, plugins_reply
for plugin in self.cfg.plugins:
agent_reply = plugin.post_instruction(agent_reply)
return key, agent_reply
def message_agent(self, key: Union[str, int], message: str) -> str:
"""Send a message to an agent and return its response
@@ -98,7 +101,10 @@ class AgentManager(metaclass=Singleton):
if plugins_reply and plugins_reply != "":
messages.append({"role": "assistant", "content": plugins_reply})
return plugins_reply
for plugin in self.cfg.plugins:
agent_reply = plugin.post_instruction(agent_reply)
return agent_reply
def list_agents(self) -> List[Tuple[Union[str, int], str]]:
"""Return a list of all agents