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 1/2] 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: From 98b14bbcc38fd99d39731fe38342e6e2fac96961 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Wed, 6 Dec 2023 16:44:14 +0800 Subject: [PATCH 2/2] chore --- metagpt/roles/ml_engineer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py index e2203c4fb..34bd81110 100644 --- a/metagpt/roles/ml_engineer.py +++ b/metagpt/roles/ml_engineer.py @@ -148,7 +148,7 @@ class MLEngineer(Role): print(truncate_result) # print(result) self.working_memory.add( - Message(content=truncate_result, keep_len=_keep_result_len), role="user", cause_by=ExecutePyCode) + Message(content=truncate_result, role="user", cause_by=ExecutePyCode) ) if "!pip" in code: