diff options
Diffstat (limited to 'autogpts/autogpt/tests/unit/test_api_manager.py')
-rw-r--r-- | autogpts/autogpt/tests/unit/test_api_manager.py | 12 |
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" |