| | """Unit tests for the commands module""" |
| | from unittest.mock import MagicMock, patch |
| |
|
| | import pytest |
| |
|
| | import autogpt.agent.agent_manager as agent_manager |
| | from autogpt.app import execute_command, list_agents, start_agent |
| |
|
| |
|
| | @pytest.mark.integration_test |
| | def test_make_agent() -> None: |
| | """Test the make_agent command""" |
| | with patch("openai.ChatCompletion.create") as mock: |
| | obj = MagicMock() |
| | obj.response.choices[0].messages[0].content = "Test message" |
| | mock.return_value = obj |
| | start_agent("Test Agent", "chat", "Hello, how are you?", "gpt2") |
| | agents = list_agents() |
| | assert "List of agents:\n0: chat" == agents |
| | start_agent("Test Agent 2", "write", "Hello, how are you?", "gpt2") |
| | agents = list_agents() |
| | assert "List of agents:\n0: chat\n1: write" == agents |
| |
|
