From 87e4bc7bc4f786422744687997e4d4552ca3c342 Mon Sep 17 00:00:00 2001 From: Matt Zhou Date: Fri, 4 Oct 2024 15:28:22 -0700 Subject: [PATCH] Fix failing tests --- tests/helpers/endpoints_helper.py | 2 +- tests/test_endpoints.py | 5 ----- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/tests/helpers/endpoints_helper.py b/tests/helpers/endpoints_helper.py index 5149eec5..6ced28d8 100644 --- a/tests/helpers/endpoints_helper.py +++ b/tests/helpers/endpoints_helper.py @@ -371,7 +371,7 @@ def assert_inner_monologue_is_valid(message: Message) -> None: """ Helper function to check that the inner monologue is valid. """ - invalid_chars = '(){}[]"' + invalid_chars = "(){}[]" # Sometimes the syntax won't be correct and internal syntax will leak into message invalid_phrases = ["functions", "send_message"] diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py index 33be28df..c1938093 100644 --- a/tests/test_endpoints.py +++ b/tests/test_endpoints.py @@ -20,7 +20,6 @@ llm_config_dir = "configs/llm_model_configs" def test_openai_gpt_4_returns_valid_first_message(): filename = os.path.join(llm_config_dir, "gpt-4.json") response = check_first_response_is_valid_for_llm_endpoint(filename) - # Log out successful response print(f"Got successful response from client: \n\n{response}") @@ -29,7 +28,6 @@ def test_openai_gpt_4_returns_keyword(): keyword = "banana" filename = os.path.join(llm_config_dir, "gpt-4.json") response = check_response_contains_keyword(filename, keyword=keyword) - # Log out successful response print(f"Got successful response from client: \n\n{response}") @@ -37,7 +35,6 @@ def test_openai_gpt_4_returns_keyword(): def test_openai_gpt_4_uses_external_tool(): filename = os.path.join(llm_config_dir, "gpt-4.json") response = check_agent_uses_external_tool(filename) - # Log out successful response print(f"Got successful response from client: \n\n{response}") @@ -45,7 +42,6 @@ def test_openai_gpt_4_uses_external_tool(): def test_openai_gpt_4_recall_chat_memory(): filename = os.path.join(llm_config_dir, "gpt-4.json") response = check_agent_recall_chat_memory(filename) - # Log out successful response print(f"Got successful response from client: \n\n{response}") @@ -53,7 +49,6 @@ def test_openai_gpt_4_recall_chat_memory(): def test_openai_gpt_4_archival_memory_retrieval(): filename = os.path.join(llm_config_dir, "gpt-4.json") response = check_agent_archival_memory_retrieval(filename) - # Log out successful response print(f"Got successful response from client: \n\n{response}")