From a2f809263a4232367ce03397788bf0dc89108953 Mon Sep 17 00:00:00 2001
From: lidanyang <lidanyang@fuzhi.ai>
Date: Thu, 27 Jun 2024 11:22:12 +0800
Subject: [PATCH] ignore warning info

---
 metagpt/actions/di/execute_nb_code.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/metagpt/actions/di/execute_nb_code.py b/metagpt/actions/di/execute_nb_code.py
index 64620d9cc..9a9f0483a 100644
--- a/metagpt/actions/di/execute_nb_code.py
+++ b/metagpt/actions/di/execute_nb_code.py
@@ -30,6 +30,12 @@ from metagpt.logs import logger
 from metagpt.utils.report import NotebookReporter
 
 INSTALL_KEEPLEN = 500
+INI_CODE = """import warnings
+import logging
+
+root_logger = logging.getLogger()
+root_logger.setLevel(logging.ERROR)
+warnings.filterwarnings('ignore')"""
 
 
 class RealtimeOutputNotebookClient(NotebookClient):
@@ -79,6 +85,10 @@ class ExecuteNbCode(Action):
         )
         self.reporter = NotebookReporter()
         self.set_nb_client()
+        asyncio.run(self._init_code())
+
+    async def _init_code(self):
+        await self.run(INI_CODE)
 
     def set_nb_client(self):
         self.nb_client = RealtimeOutputNotebookClient(
@@ -175,6 +185,8 @@ class ExecuteNbCode(Action):
                 is_success = False
 
             output_text = remove_escape_and_color_codes(output_text)
+            if is_success:
+                output_text = remove_log_and_warning_lines(output_text)
             # The useful information of the exception is at the end,
             # the useful information of normal output is at the begining.
             output_text = output_text[:keep_len] if is_success else output_text[-keep_len:]
@@ -268,6 +280,18 @@ class ExecuteNbCode(Action):
             return outputs, success
 
 
+def remove_log_and_warning_lines(input_str: str) -> str:
+    delete_lines = ["[warning]", "warning:", "[cv]", "[info]"]
+    result = "\n".join(
+        [
+            line
+            for line in input_str.split("\n")
+            if not any(dl in line.lower() for dl in delete_lines)
+        ]
+    ).strip()
+    return result
+
+
 def remove_escape_and_color_codes(input_str: str):
     # 使用正则表达式去除jupyter notebook输出结果中的转义字符和颜色代码
     # Use regular expressions to get rid of escape characters and color codes in jupyter notebook output.