From 08975c3a0838cfda45fe69afe8bd804c9891e889 Mon Sep 17 00:00:00 2001 From: lidanyang Date: Wed, 17 Jul 2024 15:16:56 +0800 Subject: [PATCH] keep len to 20000 --- metagpt/actions/di/execute_nb_code.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/metagpt/actions/di/execute_nb_code.py b/metagpt/actions/di/execute_nb_code.py index b3bf4fcb8..dfeea7e67 100644 --- a/metagpt/actions/di/execute_nb_code.py +++ b/metagpt/actions/di/execute_nb_code.py @@ -158,7 +158,7 @@ class ExecuteNbCode(Action): else: cell["outputs"].append(new_output(output_type="stream", name="stdout", text=str(output))) - def parse_outputs(self, outputs: list[str], keep_len: int = 5000) -> Tuple[bool, str]: + def parse_outputs(self, outputs: list[str], keep_len: int = 20000) -> Tuple[bool, str]: """Parses the outputs received from notebook execution.""" assert isinstance(outputs, list) parsed_output, is_success = [], True @@ -191,10 +191,8 @@ class ExecuteNbCode(Action): 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. - if '' not in output_text: - output_text = output_text[:keep_len] if is_success else output_text[-keep_len:] - else: - output_text = output_text[:20000] if is_success else output_text[-keep_len:] + # if '' not in output_text: + output_text = output_text[:keep_len] if is_success else output_text[-keep_len:] parsed_output.append(output_text) return is_success, ",".join(parsed_output)