aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2026-02-04 13:54:31 +01:00
Code Issues Packages Projects Releases Wiki Activity

mock openai in test image gen (#3285)

Browse Source
This commit is contained in:
merwanehamadi
2023-04-26 09:11:31 -07:00
committed by GitHub
parent 3ae6c1b03f
commit 4241fbbbf0
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_agent_manager.py
View File

@@ -1,9 +1,7 @@
from typing import List
import pytest
from autogpt.agent.agent_manager import AgentManager
from tests.utils import requires_api_key
from autogpt.llm_utils import create_chat_completion
@pytest.fixture
@@ -29,7 +27,16 @@ def model():
return "gpt-3.5-turbo"
@requires_api_key("OPENAI_API_KEY")
@pytest.fixture(autouse=True)
def mock_create_chat_completion(mocker):
mock_create_chat_completion = mocker.patch(
"autogpt.agent.agent_manager.create_chat_completion",
wraps=create_chat_completion,
)
mock_create_chat_completion.return_value = "irrelevant"
return mock_create_chat_completion
def test_create_agent(agent_manager, task, prompt, model):
key, agent_reply = agent_manager.create_agent(task, prompt, model)
assert isinstance(key, int)
@@ -37,7 +44,6 @@ def test_create_agent(agent_manager, task, prompt, model):
assert key in agent_manager.agents
@requires_api_key("OPENAI_API_KEY")
def test_message_agent(agent_manager, task, prompt, model):
key, _ = agent_manager.create_agent(task, prompt, model)
user_message = "Please translate 'Good morning' to French."
@@ -45,7 +51,6 @@ def test_message_agent(agent_manager, task, prompt, model):
assert isinstance(agent_reply, str)
@requires_api_key("OPENAI_API_KEY")
def test_list_agents(agent_manager, task, prompt, model):
key, _ = agent_manager.create_agent(task, prompt, model)
agents_list = agent_manager.list_agents()
@@ -53,7 +58,6 @@ def test_list_agents(agent_manager, task, prompt, model):
assert (key, task) in agents_list
@requires_api_key("OPENAI_API_KEY")
def test_delete_agent(agent_manager, task, prompt, model):
key, _ = agent_manager.create_agent(task, prompt, model)
success = agent_manager.delete_agent(key)