From 2af2aff019df52082557eb5f090c6a9dc74b3e57 Mon Sep 17 00:00:00 2001 From: openhands Date: Thu, 16 Jan 2025 18:03:24 +0000 Subject: [PATCH] Fix pr #6274: Fix issue #6273: [Feature]: Disable LitLLM Print Message --- tests/unit/test_logger_litellm.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/unit/test_logger_litellm.py b/tests/unit/test_logger_litellm.py index 85bbd0adf179..365f6aa63f57 100644 --- a/tests/unit/test_logger_litellm.py +++ b/tests/unit/test_logger_litellm.py @@ -18,9 +18,9 @@ def reset_litellm(): del sys.modules['openhands.core.logger'] -def test_litellm_settings_debug_disabled(reset_litellm): - """Test that litellm settings are properly configured when DEBUG is disabled.""" - with mock.patch.dict(os.environ, {'DEBUG': 'false'}): +def test_litellm_settings_debug_llm_disabled(reset_litellm): + """Test that litellm settings are properly configured when DEBUG_LLM is disabled.""" + with mock.patch.dict(os.environ, {'DEBUG_LLM': 'false'}): import openhands.core.logger # noqa: F401 importlib.reload(openhands.core.logger) @@ -29,12 +29,12 @@ def test_litellm_settings_debug_disabled(reset_litellm): assert litellm.set_verbose is False -def test_litellm_settings_debug_enabled(reset_litellm): - """Test that litellm settings are properly configured when DEBUG is enabled.""" - with mock.patch.dict(os.environ, {'DEBUG': 'true'}): +def test_litellm_settings_debug_llm_enabled(reset_litellm): + """Test that litellm settings are properly configured when DEBUG_LLM is enabled.""" + with mock.patch.dict(os.environ, {'DEBUG_LLM': 'true'}): import openhands.core.logger # noqa: F401 importlib.reload(openhands.core.logger) - assert litellm.suppress_debug_info is True - assert litellm.set_verbose is False + assert litellm.suppress_debug_info is False + assert litellm.set_verbose is True