fix truncate.

This commit is contained in:
刘棒棒 2023-12-13 15:55:04 +08:00
parent ce9bd1cd02
commit 05ae935d8c

View file

@ -186,14 +186,13 @@ class ExecutePyCode(ExecuteCode, Action):
def truncate(result: str, keep_len: int = 2000) -> str:
desc = f"Truncated to show only the last {keep_len} characters\n"
if result.startswith(desc):
result = result[-len(desc) :]
result = result[len(desc) :]
if len(result) > keep_len:
result = result[-keep_len:]
if not result.startswith(desc):
return desc + result
return desc
return result
def remove_escape_and_color_codes(input_str):