Auto-GPT/tests/unit/test_make_agent.py

from unittest.mock import MagicMock

from pytest_mock import MockerFixture

from autogpt.agent.agent import Agent
from autogpt.app import list_agents, start_agent


def test_make_agent(agent: Agent, mocker: MockerFixture) -> None:
    """Test that an agent can be created"""
    mock = mocker.patch("openai.ChatCompletion.create")

    response = MagicMock()
    response.choices[0].message.content = "Test message"
    response.usage.prompt_tokens = 1
    response.usage.completion_tokens = 1
    del response.error

    mock.return_value = response
    start_agent("Test Agent", "chat", "Hello, how are you?", agent, "gpt-3.5-turbo")
    agents = list_agents(agent)
    assert "List of agents:\n0: chat" == agents
    start_agent("Test Agent 2", "write", "Hello, how are you?", agent, "gpt-3.5-turbo")
    agents = list_agents(agent.config)
    assert "List of agents:\n0: chat\n1: write" == agents