mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-24 17:34:24 +01:00
8b8b3a2cdd
* Command name supports multiple names * Separate CommandRegistry.commands and .command_aliases * Update test_commands.py * Add __contains__ operator to CommandRegistry * Update error message for unknown commands --------- Co-authored-by: Reinier van der Leer <github@pwuts.nl>
24 lines
596 B
Python
24 lines
596 B
Python
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"""
|
|
command_name = "check_plan"
|
|
agent.ai_config.prompt_generator.add_command(
|
|
command_name,
|
|
"Read the plan.md with the next goals to achieve",
|
|
{},
|
|
check_plan,
|
|
)
|
|
command_result = execute_command(
|
|
command_name=command_name,
|
|
arguments={},
|
|
agent=agent,
|
|
)
|
|
assert command_result == "hi"
|