aboutsummaryrefslogtreecommitdiff
path: root/autogpts/autogpt/tests/unit/test_api_manager.py
diff options
context:
space:
mode:
Diffstat (limited to 'autogpts/autogpt/tests/unit/test_api_manager.py')
-rw-r--r--autogpts/autogpt/tests/unit/test_api_manager.py12
1 files changed, 0 insertions, 12 deletions
diff --git a/autogpts/autogpt/tests/unit/test_api_manager.py b/autogpts/autogpt/tests/unit/test_api_manager.py
index da9b0c3d8..e669756cd 100644
--- a/autogpts/autogpt/tests/unit/test_api_manager.py
+++ b/autogpts/autogpt/tests/unit/test_api_manager.py
@@ -1,5 +1,3 @@
-from unittest.mock import patch
-
import pytest
from pytest_mock import MockerFixture
@@ -77,13 +75,3 @@ class TestApiManager:
assert api_manager.get_total_prompt_tokens() == prompt_tokens
assert api_manager.get_total_completion_tokens() == 0
assert api_manager.get_total_cost() == (prompt_tokens * 0.0004) / 1000
-
- @staticmethod
- def test_get_models():
- """Test if getting models works correctly."""
- with patch("openai.Model.list") as mock_list_models:
- mock_list_models.return_value = {"data": [{"id": "gpt-3.5-turbo"}]}
- result = api_manager.get_models()
-
- assert result[0]["id"] == "gpt-3.5-turbo"
- assert api_manager.models[0]["id"] == "gpt-3.5-turbo"