From b561b2f98252c9174f885f4c82fc1c9eb4ee83df Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Tue, 5 Dec 2023 18:58:16 +0800 Subject: [PATCH] fix: change keep length of result from 1000 to 2000. --- metagpt/roles/ml_engineer.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py index 65583638e..e2203c4fb 100644 --- a/metagpt/roles/ml_engineer.py +++ b/metagpt/roles/ml_engineer.py @@ -143,10 +143,12 @@ class MLEngineer(Role): result, success = await self.execute_code.run(code) # truncated the result - print(truncate(result)) + _keep_result_len = 2000 + truncate_result = truncate(remove_escape_and_color_codes(result), keep_len=_keep_result_len) + print(truncate_result) # print(result) self.working_memory.add( - Message(content=truncate(remove_escape_and_color_codes(result)), role="user", cause_by=ExecutePyCode) + Message(content=truncate_result, keep_len=_keep_result_len), role="user", cause_by=ExecutePyCode) ) if "!pip" in code: