From 4bcb39dc14450bf30a2bb143211b723ecdb3e37f Mon Sep 17 00:00:00 2001
From: Reinier van der Leer <github@pwuts.nl>
Date: Thu, 24 Aug 2023 16:09:25 +0200
Subject: [PATCH] Fix logging setup for testing

---
 autogpt/llm/providers/openai.py | 2 +-
 autogpt/logs/formatters.py      | 2 +-
 tests/conftest.py               | 6 +++++-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/autogpt/llm/providers/openai.py b/autogpt/llm/providers/openai.py
index d4c2b15f..433e5b7b 100644
--- a/autogpt/llm/providers/openai.py
+++ b/autogpt/llm/providers/openai.py
@@ -174,7 +174,7 @@ def retry_api(
     }
     api_key_error_msg = (
         f"Please double check that you have setup a "
-        f"{Fore.CYAN + Style.BRIGHT}PAID{Style.RESET_ALL} OpenAI API Account. You can "
+        f"{Style.BRIGHT}PAID{Style.NORMAL} OpenAI API Account. You can "
         f"read more here: {Fore.CYAN}https://docs.agpt.co/setup/#getting-an-api-key{Fore.RESET}"
     )
     backoff_msg = "Waiting {backoff} seconds..."
diff --git a/autogpt/logs/formatters.py b/autogpt/logs/formatters.py
index 84f14d86..2c05f831 100644
--- a/autogpt/logs/formatters.py
+++ b/autogpt/logs/formatters.py
@@ -43,7 +43,7 @@ class AutoGptFormatter(logging.Formatter):
 
         # Determine color for title
         title = getattr(record, "title", "")
-        title_color = getattr(record, "title_color", level_color)
+        title_color = getattr(record, "title_color", "") or level_color
         if title and title_color:
             title = f"{title_color + Style.BRIGHT}{title}{Style.RESET_ALL}"
         # Make sure record.title is set, and padded with a space if not empty
diff --git a/tests/conftest.py b/tests/conftest.py
index 21dc900c..55a1a4e3 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -58,7 +58,6 @@ def config(
     config.plugins_config_file = temp_plugins_config_file
 
     config.plain_output = True
-    configure_logging(config, Path(__file__).parent / "logs")
 
     # avoid circular dependency
     from autogpt.plugins.plugins_config import PluginsConfig
@@ -78,6 +77,11 @@ def config(
     yield config
 
 
+@pytest.fixture(scope="session")
+def setup_logger(config: Config):
+    configure_logging(config, Path(__file__).parent / "logs")
+
+
 @pytest.fixture()
 def api_manager() -> ApiManager:
     if ApiManager in ApiManager._instances: