From 4241fbbbf0333f6b28d46de766035302b4e16325 Mon Sep 17 00:00:00 2001 From: merwanehamadi Date: Wed, 26 Apr 2023 09:11:31 -0700 Subject: [PATCH] mock openai in test image gen (#3285) --- tests/test_agent_manager.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/tests/test_agent_manager.py b/tests/test_agent_manager.py index e02a6a59..d1bb3d8b 100644 --- a/tests/test_agent_manager.py +++ b/tests/test_agent_manager.py @@ -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)