mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-19 15:04:26 +01:00
c9adedf746
* Add links to github issues in the README and clarify run instructions * Rename agent subpackage to agents * Revert all unwanted changes * Use relative import in `agents/__init__.py` --------- Co-authored-by: Reinier van der Leer <github@pwuts.nl>
29 lines
852 B
Python
29 lines
852 B
Python
from autogpt.agents.agent import Agent, execute_command
|
|
|
|
|
|
def test_agent_initialization(agent: Agent):
|
|
assert agent.ai_name == "Base"
|
|
assert agent.history.messages == []
|
|
assert agent.next_action_count == 0
|
|
|
|
|
|
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",
|
|
{},
|
|
lambda: "hi",
|
|
)
|
|
command_result = execute_command(
|
|
command_name=command_name,
|
|
arguments={},
|
|
agent=agent,
|
|
)
|
|
assert command_result == "hi"
|
|
|
|
|
|
# More test methods can be added for specific agent interactions
|
|
# For example, mocking chat_with_ai and testing the agent's interaction loop
|