From 3bfd0c8dadaf0e016e374d5aca28550d8f635f98 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= <liubangbang@fuzhi.ai>
Date: Mon, 22 Jan 2024 18:36:23 +0800
Subject: [PATCH] update truncate.

---
 metagpt/actions/execute_code.py | 21 ++++++++++-----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/metagpt/actions/execute_code.py b/metagpt/actions/execute_code.py
index 851794b91..a5a766ab2 100644
--- a/metagpt/actions/execute_code.py
+++ b/metagpt/actions/execute_code.py
@@ -237,8 +237,10 @@ class ExecutePyCode(ExecuteCode, Action):
 
 
 def truncate(result: str, keep_len: int = 2000, is_success: bool = True):
-    """执行失败的代码, 展示result后keep_len个字符; 执行成功的代码, 展示result前keep_len个字符。"""
+    """对于超出keep_len个字符的result: 执行失败的代码, 展示result后keep_len个字符; 执行成功的代码, 展示result前keep_len个字符。"""
     desc = f"Executed code {'successfully. ' if is_success else 'failed, please reflect the cause of bug and then debug. '}"
+    is_same_desc = False
+
     if is_success:
         desc += f"Truncated to show only first {keep_len} characters\n"
     else:
@@ -246,20 +248,17 @@ def truncate(result: str, keep_len: int = 2000, is_success: bool = True):
 
     if result.startswith(desc):
         result = result[len(desc) :]
+        is_same_desc = True
+
+    if result.strip().startswith("<coroutine object"):
+        result = "Executed code failed, you need use key word 'await' to run a async code."
+        return result, False
 
     if len(result) > keep_len:
         result = result[-keep_len:] if not is_success else result[:keep_len]
-        if not result:
-            result = "No output about your code. Only when importing packages it is normal case. Recap and go ahead."
-            return result, False
+        return desc + result, is_success
 
-        if result.strip().startswith("<coroutine object"):
-            result = "Executed code failed, you need use key word 'await' to run a async code."
-            return result, False
-
-        return desc + result[:keep_len], is_success
-
-    return result, is_success
+    return result if not is_same_desc else desc + result, is_success
 
 
 def remove_escape_and_color_codes(input_str):