fix tests

griptape-ai · Aug 14, 2024 · 0d5b0cd · 0d5b0cd
1 parent f72fe79
commit 0d5b0cd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/tests/unit/config/logging/test_newline_logging_filter.py b/tests/unit/config/logging/test_newline_logging_filter.py
@@ -1,15 +1,17 @@
 import io
+import logging
 from contextlib import redirect_stdout
 
 from griptape.config import config
-from griptape.config.logging import LoggingConfig, NewlineLoggingFilter
+from griptape.config.logging import NewlineLoggingFilter
 from griptape.structures import Agent
 
 
 class TestNewlineLoggingFilter:
     def test_filter(self):
         # use the filter in an Agent
-        config.logging = LoggingConfig(logger_filters=[NewlineLoggingFilter(replace_str="$$$")])
+        logger = logging.getLogger(config.logging.logger_name)
+        logger.addFilter(NewlineLoggingFilter(replace_str="$$$"))
         agent = Agent()
         # use a context manager to capture the stdout
         with io.StringIO() as buf, redirect_stdout(buf):

diff --git a/tests/unit/config/logging/test_truncate_logging_filter.py b/tests/unit/config/logging/test_truncate_logging_filter.py
@@ -1,15 +1,17 @@
 import io
+import logging
 from contextlib import redirect_stdout
 
 from griptape.config import config
-from griptape.config.logging import LoggingConfig, TruncateLoggingFilter
+from griptape.config.logging import TruncateLoggingFilter
 from griptape.structures import Agent
 
 
 class TestTruncateLoggingFilter:
     def test_filter(self):
         # use the filter in an Agent
-        config.logging = LoggingConfig(logger_filters=[TruncateLoggingFilter(max_log_length=0)])
+        logger = logging.getLogger(config.logging.logger_name)
+        logger.addFilter(TruncateLoggingFilter(max_log_length=100))
         agent = Agent()
         # use a context manager to capture the stdout
         with io.StringIO() as buf, redirect_stdout(buf):