From 31315655b56b447e1503720813e95d11144e2bb7 Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Wed, 4 Sep 2024 11:41:43 -0400
Subject: [PATCH] update test mode env

---
 src/controlflow/settings.py | 13 -------------
 tests/conftest.py           |  4 +++-
 2 files changed, 3 insertions(+), 14 deletions(-)

diff --git a/src/controlflow/settings.py b/src/controlflow/settings.py
index 3d62831f..3554262d 100644
--- a/src/controlflow/settings.py
+++ b/src/controlflow/settings.py
@@ -130,19 +130,6 @@ def set_log_level(self):
         setup_logging(level=self.log_level)
         return self
 
-    @model_validator(mode="after")
-    def test_mode(self):
-        if os.getenv("CONTROLFLOW_TEST_MODE"):
-            if self.log_level != "DEBUG":
-                self.log_level = "DEBUG"
-            if not self.log_all_messages:
-                self.log_all_messages = True
-            if not self.tools_verbose:
-                self.tools_verbose = True
-            if self.pretty_print_agent_events:
-                self.pretty_print_agent_events = False
-        return self
-
     @model_validator(mode="after")
     def _apply_prefect_settings(self):
         """
diff --git a/tests/conftest.py b/tests/conftest.py
index aeae52dd..65f85867 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -9,7 +9,9 @@
 @pytest.fixture(autouse=True, scope="session")
 def temp_controlflow_settings():
     with temporary_settings(
-        enable_print_handler=False,
+        pretty_print_agent_events=False,
+        log_all_messages=True,
+        log_level="DEBUG",
         orchestrator_max_turns=10,
         orchestrator_max_calls_per_turn=10,
     ):