chore: make showin logs in chat optional

phil65 · Dec 16, 2024 · 96975ff · 96975ff
1 parent 908479b
commit 96975ff
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/llmling_agent/cli/chat_session/session.py b/src/llmling_agent/cli/chat_session/session.py
@@ -56,6 +56,7 @@ def __init__(
         agent: LLMlingAgent[str],
         *,
         log_level: int = logging.INFO,
+        show_log_in_chat: bool = False,
         stream: bool = False,
     ) -> None:
         """Initialize interactive session."""
@@ -71,10 +72,11 @@ def __init__(
         self.status_bar = StatusBar(self.console)
 
         # Setup logging
-        self._log_handler = SessionLogHandler(self._output_writer)
-        self._log_handler.setLevel(log_level)
-        logging.getLogger("llmling_agent").addHandler(self._log_handler)
-        logging.getLogger("llmling").addHandler(self._log_handler)
+        if show_log_in_chat:
+            self._log_handler = SessionLogHandler(self._output_writer)
+            self._log_handler.setLevel(log_level)
+            logging.getLogger("llmling_agent").addHandler(self._log_handler)
+            logging.getLogger("llmling").addHandler(self._log_handler)
 
         # Setup components
         self._setup_history()