Delete duplicate method declaration

Significant-Gravitas · Apr 17, 2023 · 6135526 · 6135526
1 parent 8670b30
commit 6135526
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 48 deletions.
diff --git a/tests/test_token_counter.py b/tests/test_token_counter.py
@@ -23,15 +23,6 @@ def test_count_message_tokens_empty_input(self):
         # Empty input should return 3 tokens
         self.assertEqual(count_message_tokens([]), 3)
 
-    def test_count_message_tokens_invalid_model(self):
-        # Invalid model should raise a KeyError
-        messages = [
-            {"role": "user", "content": "Hello"},
-            {"role": "assistant", "content": "Hi there!"},
-        ]
-        with self.assertRaises(KeyError):
-            count_message_tokens(messages, model="invalid_model")
-
     def test_count_message_tokens_gpt_4(self):
         messages = [
             {"role": "user", "content": "Hello"},

diff --git a/tests/unit/_test_json_parser.py b/tests/unit/_test_json_parser.py
@@ -33,45 +33,6 @@ def test_invalid_json_major_without_gpt(self):
         with self.assertRaises(Exception):
             fix_and_parse_json(json_str, try_to_fix_with_gpt=False)
 
-    def test_invalid_json_leading_sentence_with_gpt(self):
-        """Test that a REALLY invalid JSON string raises an error when try_to_fix_with_gpt is False."""
-
-        json_str = """I suggest we start by browsing the repository to find any issues that we can fix.
-
-{
-    "command": {
-        "name": "browse_website",
-        "args":{
-            "url": "https://github.com/Torantulino/Auto-GPT"
-        }
-    },
-    "thoughts":
-    {
-        "text": "I suggest we start browsing the repository to find any issues that we can fix.",
-        "reasoning": "Browsing the repository will give us an idea of the current state of the codebase and identify any issues that we can address to improve the repo.",
-        "plan": "- Look through the repository to find any issues.\n- Investigate any issues to determine what needs to be fixed\n- Identify possible solutions to fix the issues\n- Open Pull Requests with fixes",
-        "criticism": "I should be careful while browsing so as not to accidentally introduce any new bugs or issues.",
-        "speak": "I will start browsing the repository to find any issues we can fix."
-    }
-}"""
-        good_obj = {
-            "command": {
-                "name": "browse_website",
-                "args": {"url": "https://github.com/Torantulino/Auto-GPT"},
-            },
-            "thoughts": {
-                "text": "I suggest we start browsing the repository to find any issues that we can fix.",
-                "reasoning": "Browsing the repository will give us an idea of the current state of the codebase and identify any issues that we can address to improve the repo.",
-                "plan": "- Look through the repository to find any issues.\n- Investigate any issues to determine what needs to be fixed\n- Identify possible solutions to fix the issues\n- Open Pull Requests with fixes",
-                "criticism": "I should be careful while browsing so as not to accidentally introduce any new bugs or issues.",
-                "speak": "I will start browsing the repository to find any issues we can fix.",
-            },
-        }
-        # Assert that this raises an exception:
-        self.assertEqual(
-            fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj
-        )
-
     def test_invalid_json_leading_sentence_with_gpt(self):
         """Test that a REALLY invalid JSON string raises an error when try_to_fix_with_gpt is False."""
         json_str = """I will first need to browse the repository (https://github.com/Torantulino/Auto-GPT) and identify any potential bugs that need fixing. I will use the "browse_website" command for this.