Fix execute_command coming from plugins (#4730)

This commit is contained in:
Erik Peterson
2023-06-17 14:45:34 -07:00
committed by GitHub
parent 7f6f18f642
commit 7bac56b57d
3 changed files with 30 additions and 7 deletions

View File

@@ -0,0 +1,24 @@
from autogpt.agent import Agent
from autogpt.app import execute_command
def check_plan():
return "hi"
def test_execute_command_plugin(agent: Agent):
"""Test that executing a command that came from a plugin works as expected"""
agent.ai_config.prompt_generator.add_command(
"check_plan",
"Read the plan.md with the next goals to achieve",
{},
check_plan,
)
command_name = "check_plan"
arguments = {}
command_result = execute_command(
command_name=command_name,
arguments=arguments,
agent=agent,
)
assert command_result == "hi"