mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-26 02:14:27 +01:00
d923004e20
* Remove App Commands and Audio Text Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> * Remove self feedback Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> --------- Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> Co-authored-by: Erik Peterson <e@eriklp.com>
26 lines
901 B
Python
26 lines
901 B
Python
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
|