mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-29 03:44:28 +01:00
6e6e7fcc9a
* Extract open ai api calls and retry at lowest level * Forgot a test * Gotta fix my local docker config so I can let pre-commit hooks run, ugh * fix: merge artiface * Fix linting * Update memory.vector.utils * feat: make sure resp exists * fix: raise error message if created * feat: rename file * fix: partial test fix * fix: update comments * fix: linting * fix: remove broken test * fix: require a model to exist * fix: BaseError issue * fix: runtime error * Fix mock response in test_make_agent * add 429 as errors to retry --------- Co-authored-by: k-boikov <64261260+k-boikov@users.noreply.github.com> Co-authored-by: Nicholas Tindle <nick@ntindle.com> Co-authored-by: Reinier van der Leer <github@pwuts.nl> Co-authored-by: Nicholas Tindle <nicktindle@outlook.com> Co-authored-by: Luke K (pr-0f3t) <2609441+lc0rp@users.noreply.github.com> Co-authored-by: Merwane Hamadi <merwanehamadi@gmail.com>
25 lines
904 B
Python
25 lines
904 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()
|
|
del response.error
|
|
response.choices[0].messages[0].content = "Test message"
|
|
response.usage.prompt_tokens = 1
|
|
response.usage.completion_tokens = 1
|
|
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
|