aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/test_provider_openai.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/integration/test_provider_openai.py')
-rw-r--r--tests/integration/test_provider_openai.py54
1 files changed, 0 insertions, 54 deletions
diff --git a/tests/integration/test_provider_openai.py b/tests/integration/test_provider_openai.py
deleted file mode 100644
index f51ad9ac4..000000000
--- a/tests/integration/test_provider_openai.py
+++ /dev/null
@@ -1,54 +0,0 @@
-from unittest.mock import MagicMock, patch
-
-import pytest
-
-from autogpt.llm.api_manager import ApiManager
-from autogpt.llm.providers import openai
-
-api_manager = ApiManager()
-
-
-@pytest.fixture(autouse=True)
-def reset_api_manager():
- api_manager.reset()
- yield
-
-
-class TestProviderOpenAI:
- @staticmethod
- def test_create_chat_completion_debug_mode(caplog):
- """Test if debug mode logs response."""
- messages = [
- {"role": "system", "content": "You are a helpful assistant."},
- {"role": "user", "content": "Who won the world series in 2020?"},
- ]
- model = "gpt-3.5-turbo"
- with patch("openai.ChatCompletion.create") as mock_create:
- mock_response = MagicMock()
- del mock_response.error
- mock_response.usage.prompt_tokens = 10
- mock_response.usage.completion_tokens = 20
- mock_create.return_value = mock_response
-
- openai.create_chat_completion(messages, model=model)
-
- assert "Response" in caplog.text
-
- @staticmethod
- def test_create_chat_completion_empty_messages():
- """Test if empty messages result in zero tokens and cost."""
- messages = []
- model = "gpt-3.5-turbo"
-
- with patch("openai.ChatCompletion.create") as mock_create:
- mock_response = MagicMock()
- del mock_response.error
- mock_response.usage.prompt_tokens = 0
- mock_response.usage.completion_tokens = 0
- mock_create.return_value = mock_response
-
- openai.create_chat_completion(messages, model=model)
-
- assert api_manager.get_total_prompt_tokens() == 0
- assert api_manager.get_total_completion_tokens() == 0
- assert api_manager.get_total_cost() == 0