Fix pr #6274: Fix issue #6273: [Feature]: Disable LitLLM Print Message

All-Hands-AI · Jan 16, 2025 · f4b3445 · f4b3445
1 parent d76494a
commit f4b3445
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/openhands/core/logger.py b/openhands/core/logger.py
@@ -13,10 +13,12 @@
 
 LOG_LEVEL = os.getenv('LOG_LEVEL', 'INFO').upper()
 DEBUG = os.getenv('DEBUG', 'False').lower() in ['true', '1', 'yes']
+DEBUG_LLM = os.getenv('DEBUG_LLM', 'False').lower() in ['true', '1', 'yes']
+
+# Configure litellm logging based on DEBUG_LLM
+litellm.suppress_debug_info = not DEBUG_LLM
+litellm.set_verbose = DEBUG_LLM
 
-# Always disable litellm logging
-litellm.suppress_debug_info = True
-litellm.set_verbose = False
 if DEBUG:
     LOG_LEVEL = 'DEBUG'